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
  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
  25     def _list(self):
  26         for i in self.xml.childNodes[0].childNodes:
  27             if "COL3" in i.attributes.keys():
  28                 self.l.append((i.attributes["COL1"].nodeValue, \
  29                         i.attributes["COL2"].nodeValue, \
  30                         i.attributes["COL3"].nodeValue))
  31                 def list(self):
  32                     return self.l
  33
  34     def get_municipio(self, code):
  35         for i in self.l:
  36             if i[0] == code:
  37                 return i
  38         return None
  39
  40     def verify_cep(self, m, cep):
  41         l = m[2][0:7]
  42         h = m[2][9:16]
  43         if cep >= l and cep <= h:
  44             return True
  45         return False
  46
  47 if __name__ == '__main__':
  48     municipios = Municipios('MG')
  49     m = municipios.get_municipio('4877')
  50     print m[1]
  51     print municipios.verify_cep(m, '36880000')
  52     print municipios.verify_cep(m, '05020000')
  53
  54     municipios = Municipios('SP')
  55     m = municipios.get_municipio('7107')
  56     print m[1]
  57     print municipios.verify_cep(m, '05020000')
  58     print municipios.verify_cep(m, '36880000')
  59
  60 # vim:tabstop=4:expandtab:smartindent