Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
1 result
Blame History Permalink
TEIFile.py 1.72 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52 






from data_process.data_functions import read_tei

class TEIFile(object):
    def __init__(self, filename, textfilename):
        self.filename = filename
        self.soup = read_tei(filename)
        self._text = None
        self._Head = ''
        self._Objecttype = ''
        self._attribution = ''
        self._Class = ''
        self._normclass = ''
        self._englishclass = ''
        self._generatedclass = ''
        self._author = ''


        if self.soup.find('index', type='head'):
            self._Head = self.soup.find('index', type='head')['value']

        if self.soup.find('index', type='objecttype'):
            self._Objecttype = self.soup.find('index', type='objecttype')['value']


        if self.soup.find('index', type='attribution'):
            self._attribution = self.soup.find('index', type='attribution')['value']

        if self.soup.find('index', type='class') and self.soup.find('index', type='class').has_attr('value') :

            self._Class = self.soup.find('index', type='class')['value']

        if self.soup.find('index', type='normclass'):
            self._normclass =  self.soup.find('index', type='normclass')['value']

        if self.soup.find('index', type='englishclass'):
            self._englishclass = self.soup.find('index', type='englishclass')['value']

        if self.soup.find('index', type='generatedclass'):
            self._generatedclass = self.soup.find('index', type='generatedclass')['value']

        if self.soup.find('index', type = 'author'):
            self._author = self.soup.find('index', type='author')['value']



        ps = self.soup.find_all('p')
        Texts = []
        for p in ps[1:]:
            Texts.append(p.getText())

        self._text = ' '.join(Texts)