1 #/usr/bin/env python
   2 # -*- coding:UTF-8 -*-
   3 
   4 
   5 """
   6 
   7 Este módulo fornece uma classe wrapper para ser usada com números de
   8 CPF, que além de oferecer um método simples de verificação, também
   9 conta com métodos para comparação e conversão.
  10 
  11 
  12 >>> a = CPF('56068332551')
  13 >>> b = CPF('560.683.325-51')
  14 >>> c = CPF((1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 0))
  15 >>> assert a.valido()
  16 >>> assert b.valido()
  17 >>> assert not c.valido()
  18 >>> assert a == b
  19 >>> assert not b == c
  20 >>> assert not a == c
  21 >>> assert eval(repr(a)) == a
  22 >>> assert eval(repr(b)) == b
  23 >>> assert eval(repr(c)) == c
  24 >>> assert str(a) == \"560.683.325-51\"
  25 >>> assert str(b) == str(a)
  26 >>> assert str(c) == \"123.456.789-00\"
  27 
  28 """
  29 
  30 
  31 
  32 
  33 class CPF(object):
  34 
  35     def __init__(self, cpf):
  36         """Classe representando um número de CPF
  37 
  38         >>> a = CPF('95524361503')
  39         >>> b = CPF('955.243.615-03')
  40         >>> c = CPF([9, 5, 5, 2, 4, 3, 6, 1, 5, 0, 3])
  41 
  42         """
  43         if isinstance(cpf, basestring):
  44             if not cpf.isdigit():
  45                 cpf = cpf.replace(".", "")
  46                 cpf = cpf.replace("-", "")
  47         
  48             if not cpf.isdigit:
  49                 raise ValueError("Valor não segue a forma xxx.xxx.xxx-xx")
  50 
  51         if len(cpf) < 11:
  52             raise ValueError("O número de CPF deve ter 11 digítos")
  53         
  54         self.cpf = map(int, cpf)
  55 
  56 
  57     def __getitem__(self, index):
  58         """Retorna o dígito em index como string
  59 
  60         >>> a = CPF('95524361503')
  61         >>> a[9] == '0'
  62         True
  63         >>> a[10] == '3'
  64         True
  65         >>> a[9] == 0
  66         False
  67         >>> a[10] == 3
  68         False
  69 
  70         """
  71         return str(self.cpf[index])
  72 
  73     def __repr__(self):
  74         """Retorna uma representação 'real', ou seja:
  75 
  76         eval(repr(cpf)) == cpf
  77 
  78         >>> a = CPF('95524361503')
  79         >>> print repr(a)
  80         CPF('95524361503')
  81         >>> eval(repr(a)) == a
  82         True
  83         
  84         """
  85         return "CPF('%s')" % ''.join([str(x) for x in self.cpf])
  86 
  87     def __eq__(self, other):
  88         """Provê teste de igualdade para números de CPF
  89 
  90         >>> a = CPF('95524361503')
  91         >>> b = CPF('955.243.615-03')
  92         >>> c = CPF('123.456.789-00')
  93         >>> a == b
  94         True
  95         >>> a != c
  96         True
  97         >>> b != c
  98         True
  99 
 100         """
 101         if isinstance(other, CPF):
 102             return self.cpf == other.cpf
 103         return False
 104     
 105     def __str__(self):
 106         """Retorna uma string do CPF na forma com pontos e traço
 107 
 108         >>> a = CPF('95524361503')
 109         >>> str(a)
 110         '955.243.615-03'
 111         
 112 
 113         """
 114         d = ((3, "."), (7, "."), (11, "-"))
 115         s = map(str, self.cpf)
 116         for i, v in d:
 117             s.insert(i, v)
 118         r = ''.join(s)
 119         return r
 120 
 121     def valido(self):
 122         """Valida o número de cpf
 123 
 124         >>> a = CPF('95524361503')
 125         >>> a.valido()
 126         True
 127         >>> b = CPF('12345678900')
 128         >>> b.valido()
 129         False
 130 
 131         """
 132         cpf = self.cpf[:9]
 133         # pegamos apenas os 9 primeiros dígitos do cpf e geramos os
 134         # dois dígitos que faltam
 135         while len(cpf) < 11:
 136 
 137             r = sum(map(lambda(i,v):(len(cpf)+1-i)*v,enumerate(cpf))) % 11
 138 
 139             if r > 1:
 140                 f = 11 - r
 141             else:
 142                 f = 0
 143             cpf.append(f)
 144 
 145         # se o número com os digítos faltantes coincidir com o número
 146         # original, então ele é válido
 147         return bool(cpf == self.cpf)
 148 
 149 
 150 
 151 if __name__ == "__main__":
 152 
 153     import doctest, sys
 154     doctest.testmod(sys.modules[__name__])