Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

WSAA Python3 UTF-8 Encoding #46

Open
wants to merge 5 commits into
base: py3k
Choose a base branch
from
Open

WSAA Python3 UTF-8 Encoding #46

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
2633426
Update wsaa.py
Jun 18, 2018
ab2a5d5
[FIX] TypeError: data must be bytes or None, not str
Jun 19, 2018
c4b9582
TypeError: can't concat bytes to str
Jun 19, 2018
8f13047
Remove whitespace from padron.py and utils.py
Jun 19, 2018
78222f6
[FIX] Obtener Tabla Parametros
Jun 19, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
64 changes: 32 additions & 32 deletions padron.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@

"Herramienta para procesar y consultar el Padrón Unico de Contribuyentes AFIP"

# Documentación e información adicional:
# Documentación e información adicional:
# http://www.sistemasagiles.com.ar/trac/wiki/PadronContribuyentesAFIP

__author__ = "Mariano Reingart <reingart@gmail.com>"
Expand Down Expand Up @@ -42,23 +42,23 @@
FORMATO = [
("nro_doc", 11, N, ""),
("denominacion", 30, A, ""),
("imp_ganancias", 2, A, "'NI', 'AC','EX', 'NC'"),
("imp_ganancias", 2, A, "'NI', 'AC','EX', 'NC'"),
("imp_iva", 2, A, "'NI' , 'AC','EX','NA','XN','AN'"),
("monotributo", 2, A, "'NI', 'Codigo categoria tributaria'"),
("integrante_soc", 1, A, "'N' , 'S'"),
("empleador", 1, A, "'N', 'S'"),
("actividad_monotributo", 2, A, ""),
("tipo_doc", 2, N, "80: CUIT, 96: DNI, etc."),
("cat_iva", 2, N, "1: RI, 4: EX, 5: CF, 6: MT, etc"),
("email", 250, A, ""),
]
("email", 250, A, ""),
]

# Mapeos constantes:

PROVINCIAS = {0: 'CIUDAD AUTONOMA BUENOS AIRES', 1: 'BUENOS AIRES',
PROVINCIAS = {0: 'CIUDAD AUTONOMA BUENOS AIRES', 1: 'BUENOS AIRES',
2: 'CATAMARCA', 3: 'CORDOBA', 4: 'CORRIENTES', 5: 'ENTRE RIOS', 6: 'JUJUY',
7: 'MENDOZA', 8: 'LA RIOJA', 9: 'SALTA', 10: 'SAN JUAN', 11: 'SAN LUIS',
12: 'SANTA FE', 13: 'SANTIAGO DEL ESTERO', 14: 'TUCUMAN', 16: 'CHACO',
7: 'MENDOZA', 8: 'LA RIOJA', 9: 'SALTA', 10: 'SAN JUAN', 11: 'SAN LUIS',
12: 'SANTA FE', 13: 'SANTIAGO DEL ESTERO', 14: 'TUCUMAN', 16: 'CHACO',
17: 'CHUBUT', 18: 'FORMOSA', 19: 'MISIONES', 20: 'NEUQUEN', 21: 'LA PAMPA',
22: 'RIO NEGRO', 23: 'SANTA CRUZ', 24: 'TIERRA DEL FUEGO'}

Expand All @@ -75,12 +75,12 @@ class PadronAFIP():

_public_methods_ = ['Buscar', 'Descargar', 'Procesar', 'Guardar',
'ConsultarDomicilios', 'Consultar', 'Conectar',
'DescargarConstancia', 'MostrarPDF',
'DescargarConstancia', 'MostrarPDF',
"ObtenerTablaParametros",
]
_public_attrs_ = ['InstallDir', 'Traceback', 'Excepcion', 'Version',
'cuit', 'dni', 'denominacion', 'imp_ganancias', 'imp_iva',
'monotributo', 'integrante_soc', 'empleador',
'cuit', 'dni', 'denominacion', 'imp_ganancias', 'imp_iva',
'monotributo', 'integrante_soc', 'empleador',
'actividad_monotributo', 'cat_iva', 'domicilios',
'tipo_doc', 'nro_doc', 'LanzarExcepciones',
'tipo_persona', 'estado', 'impuestos', 'actividades',
Expand All @@ -101,7 +101,7 @@ def __init__(self):
self.LanzarExcepciones = False
self.inicializar()
self.client = None

def inicializar(self):
self.Excepcion = self.Traceback = ""
self.cuit = self.dni = 0
Expand All @@ -114,14 +114,14 @@ def inicializar(self):
self.impuestos = []
self.actividades = []
self.imp_iva = self.empleador = self.integrante_soc = self.cat_iva = ""
self.monotributo = self.actividad_monotributo = ""
self.monotributo = self.actividad_monotributo = ""
self.data = {}
self.response = ""

@inicializar_y_capturar_excepciones_simple
def Conectar(self, url=URL_API, proxy="", wrapper=None, cacert=None, trace=False):
self.client = WebClient(location=url, trace=trace, cacert=cacert)
self.client.method = "GET" # metodo RESTful predeterminado
self.client.method = "GET" # metodo RESTful predeterminado
self.client.enctype = None # no enviar body
return True

Expand All @@ -136,8 +136,8 @@ def Descargar(self, url=URL, filename="padron.txt", proxy=None):
print("Abriendo URL %s ..." % url)
req = urllib.request.Request(url)
if os.path.exists(filename):
http_date = formatdate(timeval=os.path.getmtime(filename),
localtime=False, usegmt=True)
http_date = formatdate(timeval=os.path.getmtime(filename),
localtime=False, usegmt=True)
req.add_header('If-Modified-Since', http_date)
try:
web = urllib.request.urlopen(req)
Expand Down Expand Up @@ -177,20 +177,20 @@ def Descargar(self, url=URL, filename="padron.txt", proxy=None):
tf.write(zf.read(fn))
tf.close()
return 200

@inicializar_y_capturar_excepciones_simple
def Procesar(self, filename="padron.txt", borrar=False):
"Analiza y crea la base de datos interna sqlite para consultas"
"Analiza y crea la base de datos interna sqlite para consultas"
f = open(filename, "r")
keys = [k for k, l, t, d in FORMATO]
# conversion a planilla csv (no usado)
if False and not os.path.exists("padron.csv"):
csvfile = open('padron.csv', 'wb')
import csv
wr = csv.writer(csvfile, delimiter=',',
quotechar='"', quoting=csv.QUOTE_MINIMAL)
quotechar='"', quoting=csv.QUOTE_MINIMAL)
for i, l in enumerate(f):
if i % 100000 == 0:
if i % 100000 == 0:
print("Progreso: %d registros" % i)
r = leer(l, FORMATO)
row = [r[k] for k in keys]
Expand Down Expand Up @@ -237,7 +237,7 @@ def Procesar(self, filename="padron.txt", borrar=False):
db.commit()
c.close()
db.close()

@inicializar_y_capturar_excepciones_simple
def Buscar(self, nro_doc, tipo_doc=80):
"Devuelve True si fue encontrado y establece atributos con datos"
Expand Down Expand Up @@ -279,7 +279,7 @@ def Buscar(self, nro_doc, tipo_doc=80):
def ConsultarDomicilios(self, nro_doc, tipo_doc=80, cat_iva=None):
"Busca los domicilios, devuelve la cantidad y establece la lista"
self.cursor.execute("SELECT direccion FROM domicilio WHERE "
" tipo_doc=? AND nro_doc=? ORDER BY id ",
" tipo_doc=? AND nro_doc=? ORDER BY id ",
[tipo_doc, nro_doc])
filas = self.cursor.fetchall()
self.domicilios = [fila['direccion'] for fila in filas]
Expand All @@ -300,7 +300,7 @@ def Guardar(self, tipo_doc, nro_doc, denominacion, cat_iva, direccion, email):
# agregar el domicilio solo si no existe:
if direccion:
self.cursor.execute("SELECT * FROM domicilio WHERE direccion=? "
"AND tipo_doc=? AND nro_doc=?",
"AND tipo_doc=? AND nro_doc=?",
[direccion, tipo_doc, nro_doc])
if self.cursor.rowcount < 0:
sql = ("INSERT INTO domicilio (nro_doc, tipo_doc, direccion)"
Expand Down Expand Up @@ -357,7 +357,7 @@ def Consultar(self, nro_doc):
self.cod_postal = ""
# retrocompatibilidad:
self.domicilios = ["%s - %s (%s) - %s" % (
self.direccion, self.localidad,
self.direccion, self.localidad,
self.cod_postal, self.provincia,) ]
# analizo impuestos:
self.impuestos = data.get("impuestos", [])
Expand Down Expand Up @@ -410,11 +410,11 @@ def MostrarPDF(self, archivo, imprimir=False):

@inicializar_y_capturar_excepciones_simple
def ObtenerTablaParametros(self, tipo_recurso, sep="||"):
"Devuelve un array de elementos que tienen id y descripción"
"Devuelve un array de elementos que tienen id y descripción"
if not self.client:
self.Conectar()
self.response = self.client("parametros", "v1", tipo_recurso)
result = json.loads(self.response)
result = json.loads(self.response.decode('utf8'))
ret = {}
if result['success']:
data = result['data']
Expand All @@ -431,7 +431,7 @@ def ObtenerTablaParametros(self, tipo_recurso, sep="||"):
return ["%s%%s%s%%s%s" % (sep, sep, sep) % it for it in sorted(ret.items())]
else:
return ret



# busco el directorio de instalación (global para que no cambie si usan otra dll)
Expand Down Expand Up @@ -472,18 +472,18 @@ def ObtenerTablaParametros(self, tipo_recurso, sep="||"):
print('\n'.join(padron.ObtenerTablaParametros("categoriasAutonomo")))

if '--csv' in sys.argv:
csv_reader = csv.reader(open("entrada.csv", "rU"),
csv_reader = csv.reader(open("entrada.csv", "rU"),
dialect='excel', delimiter=",")
csv_writer = csv.writer(open("salida.csv", "w"),
csv_writer = csv.writer(open("salida.csv", "w"),
dialect='excel', delimiter=",")
encabezado = next(csv_reader)
columnas = ["cuit", "denominacion", "estado", "direccion",
"localidad", "provincia", "cod_postal",
"impuestos", "actividades", "imp_iva",
"monotributo", "actividad_monotributo",
"impuestos", "actividades", "imp_iva",
"monotributo", "actividad_monotributo",
"empleador", "imp_ganancias", "integrante_soc"]
csv_writer.writerow(columnas)

for fila in csv_reader:
cuit = (fila[0] if fila else "").replace("-", "")
if cuit.isdigit():
Expand Down Expand Up @@ -515,7 +515,7 @@ def ObtenerTablaParametros(self, tipo_recurso, sep="||"):
ok = padron.Consultar(cuit)
print('ok' if ok else "error", padron.Excepcion)
print("Denominacion:", padron.denominacion)
print("CUIT:", padron.cuit)
print("CUIT:", padron.cuit)
print("Tipo:", padron.tipo_persona, padron.tipo_doc, padron.dni)
print("Estado:", padron.estado)
print("Direccion:", padron.direccion)
Expand Down
Loading