DIFAL_ES/xml_file.py at main · z9sb/DIFAL_ES · GitHub

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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
from bs4 import BeautifulSoup as bs

#Retorna a aliquota interestadual para o estado do ES
def aliquota(uf: str) -> float:
    dicionario = {
        'AC': 12.00, 'AL': 12.00, 'AM': 12.00,
        'AP': 12.00, 'BA': 12.00, 'CE': 12.00,
        'DF': 12.00, 'GO': 12.00, 'MA': 12.00,
        'MT': 12.00, 'MS': 12.00, 'MG': 7.00,
        'PA': 12.00, 'PB': 12.00, 'PR': 12.00,
        'PE': 7.00, 'PI': 12.00, 'RN': 12.00,
        'RS': 7.00, 'RJ': 7.00, 'RO': 12.00,
        'RR': 12.00, 'SC': 7.00, 'SP': 7.00,
        'SE': 12.00, 'TO': 12.00, 'IM': 4.00,
        'ES': 17.00
        }
    return dicionario[uf]

class xmlns:
    def __init__(self, file: str):
        self._file = bs(open(file, encoding='UTF-8'), 'xml')


class NFe(xmlns):
    def __init__(self, file):
        super().__init__(file)

    #Busca o primeiro item informado no xml
    def xml_find(self, item):
        xml_file = self._file.find(item)
        return xml_file if xml_file else None


    #Busca o todos itens informados no xml
    def xml_find_all(self, item):
        xml_file = self._file.find_all(item)
        return [i.text for i in xml_file] if xml_file else None


    #Busca todos os subitens informados e retorna uma lista zerada caso não o tenha
    def xml_find_all_subitem_int(
        self, item, subitem, add = 0, substituto = None
        ):
        lista = []
        xml_file = self._file.find_all(item)
        for i in xml_file:
            if i.find(subitem):
                lista.append(i.find(subitem).text)

            elif substituto:
                lista.append(i.find(substituto).text)

            else:
                lista.append(add)
        return lista if lista != [] else [0 for i in  self.ncm()]


    #Retorna o itens das NF's sendo suportados os modelos (NF: 55, NFCE: 65)
    def model(self) -> str:
        return self.xml_find('mod').text

    def serie(self) -> str:
        return self.xml_find('serie').text

    def number_nf(self) -> str:
        return self.xml_find('nNF').text

    def date_emition(self) -> str:
        return self.xml_find('dhEmi').text

    def cnpj_emit(self) -> int:
        return self.xml_find('emit').find('CNPJ').text

    def uf_emit(self) -> str:
        return self.xml_find('emit').find('UF').text

    def cnpj_dest(self) -> int:
        return self.xml_find('dest').find('CNPJ').text

    def name_for(self) -> str:
        return self.xml_find('emit').find('xNome').text

    def estado_for(self) -> str:
        return self.xml_find('emit').find('UF').text

    def name_cli(self) -> str:
        return self.xml_find('dest').find('xNome').text

    def estado_cli(self) -> str:
        return self.xml_find('dest').find('UF').text

    def acess_key(self) -> int:
        return self.xml_find('chNFe').text

    def ncm(self) -> int:
        return self.xml_find_all('NCM')

    def ali_icms(self) -> int:
        return self.xml_find_all_subitem_int(
            'ICMS', 'pICMS', aliquota(self.uf_emit()))

    def csosn(self) -> int:
        return self.xml_find_all('CSOSN')

    def name_prod(self) -> str:
        return self.xml_find_all('xProd')

    def cest(self) -> int:
        return self.xml_find_all_subitem_int('prod','CEST')

    def v_ipi(self) -> int:
        return self.xml_find_all_subitem_int('prod', 'vIPI')

    def v_prod(self) -> float:
        return self.xml_find_all('vProd')

    def v_frete(self) -> float:
        return self.xml_find_all_subitem_int('prod','vFrete')

    def v_desc(self) -> float:
        return self.xml_find_all_subitem_int('prod','vDesc')

    def v_outros(self) -> float:
        return self.xml_find_all_subitem_int('prod','vOutro')

    def u_com(self) -> float:
        return self.xml_find_all_subitem_int('prod','uCom')

    def q_com(self) -> float:
        return self.xml_find_all_subitem_int('prod','qCom')

    def v_uncom(self) -> float:
        return self.xml_find_all_subitem_int('prod','vUnCom')

    def icms_orig(self) -> float:
        return self.xml_find_all_subitem_int('ICMS','orig')

    def icms_cst(self) -> float:
        return self.xml_find_all_subitem_int('ICMS','CST', substituto= 'CSOSN')

    def cfop(self) -> float:
        return self.xml_find_all_subitem_int('prod','CFOP')

    def v_icms(self):
        return self.xml_find_all_subitem_int('ICMS','vICMS')

    def Valor_total(self) -> float:
        return self.xml_find('ICMSTot').find('vNF').text