src/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 import dirs
  19
  20 class Municipios:
  21     def __init__(self):
  22         self.l = []
  23
  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
  33     def get_municipio(self, code):
  34         for i in self.l:
  35             if i[0] == code:
  36                 return i
  37         return None
  38
  39     def carregar_estado(self, UF):
  40         irpf_dir = dirs.get_default_irpf_dir()
  41         self.l = []
  42         self.xml = xml.dom.minidom.parse(irpf_dir.get_resource_file("%s.xml" % (UF,)))
  43         self._list()
  44
  45     def verify_cep(self, m, cep):
  46         l = m[2][0:7]
  47         h = m[2][9:16]
  48         if cep >= l and cep <= h:
  49             return True
  50         return False
  51
  52 if __name__ == '__main__':
  53     municipios = Municipios()
  54
  55     municipios.carregar_estado('MG')
  56     m = municipios.get_municipio('4877')
  57     print m[1]
  58     print municipios.verify_cep(m, '36880000')
  59     print municipios.verify_cep(m, '05020000')
  60
  61     municipios.carregar_estado('SP')
  62     m = municipios.get_municipio('7107')
  63     print m[1]
  64     print municipios.verify_cep(m, '05020000')
  65     print municipios.verify_cep(m, '36880000')
  66
  67 # vim:tabstop=4:expandtab:smartindent