Importa recursos do irpf-livre.
[cascardo/irpf-gui.git] / municipios.py
1 #
2 #   Copyright 2013 Thadeu Lima de Souza Cascardo <cascardo@cascardo.info>
3 #
4 #   This program is free software: you can redistribute it and/or modify
5 #   it under the terms of the GNU General Public License as published by
6 #   the Free Software Foundation, either version 3 of the License, or
7 #   (at your option) any later version.
8 #
9 #   This program is distributed in the hope that it will be useful,
10 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
11 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 #   GNU General Public License for more details.
13 #
14 #   You should have received a copy of the GNU General Public License
15 #   along with this program.  If not, see <http://www.gnu.org/licenses/>.
16 # -*- mode: python; encoding: utf-8; -*-
17 import xml.dom.minidom
18
19 class Municipios:
20         def __init__(self, UF):
21                 self.xml = xml.dom.minidom.parse("res/%s.xml" % (UF,))
22                 self.l = []
23                 self._list()
24         def _list(self):
25                 for i in self.xml.childNodes[0].childNodes:
26                         if "COL3" in i.attributes.keys():
27                                 self.l.append((i.attributes["COL1"].nodeValue, \
28                                         i.attributes["COL2"].nodeValue, \
29                                         i.attributes["COL3"].nodeValue))
30         def list(self):
31                 return self.l
32         def get_municipio(self, code):
33                 for i in self.l:
34                         if i[0] == code:
35                                 return i
36                 return None
37         def verify_cep(self, m, cep):
38                 l = m[2][0:7]
39                 h = m[2][9:16]
40                 if cep >= l and cep <= h:
41                         return True
42                 return False
43
44 if __name__ == '__main__':
45         municipios = Municipios('MG')
46         m = municipios.get_municipio('4877')
47         print m[1]
48         print municipios.verify_cep(m, '36880000')
49         print municipios.verify_cep(m, '05020000')
50
51         municipios = Municipios('SP')
52         m = municipios.get_municipio('7107')
53         print m[1]
54         print municipios.verify_cep(m, '05020000')
55         print municipios.verify_cep(m, '36880000')