def gtl_pdf_parser(docid, izena, data, lekua, import_to_db, filename, parse_to_csv):
structure = [(1, 'JN'), (2, 'JG'), (3, 'SN'), (4, 'SG'), (5, 'SG')]
e = None
if import_to_db:
import adaptors.couch as couchdb
try:
e = couchdb.getdoc(docid)
del e['sailkapena']
except KeyError as err:
e = {
'izena': izena,
'data': data,
'lekua': lekua
}
else:
e = {
'izena': izena,
'data': data,
'lekua': lekua
}
estropada = Estropada(**e)
kategoriak = []
if parse_to_csv:
for kategoria in structure:
logger.info(f'Reading page {kategoria[0]} for cat {kategoria[1]}')
try:
kategoria_emaitzak = parse_pdf_to_csv(filename, kategoria[0], kategoria[1], data)
except Exception as e:
print(f"Cannot parse {kategoria[0]} orrialdea")
logger.error(e)
else:
for kategoria in structure:
logger.info(f'Reading page {kategoria[0]} for cat {kategoria[1]}')
try:
kategoria_emaitzak = parse_data(kategoria[0], kategoria[1], data)
kategoriak.append(kategoria[1])
for t in kategoria_emaitzak:
estropada.taldeak_add(t)
except Exception as e:
print(f"Cannot parse {kategoria[0]} orrialdea")
logger.error(e)
estropada.kategoriak = kategoriak
puntuazioak_kalkulatu(estropada)
if import_to_db:
estropada_json = estropada.get_json()
estropada_obj = json.loads(estropada_json)
if hasattr(estropada, '_rev'):
estropada_obj['_rev'] = estropada._rev
couchdb.setdoc(docid, estropada_obj)
else:
estropada.dump_json()
kategoria_base = None
kategoriak_helper = []
for row in teamreader:
posizioa = None
try:
posizioa = int(row[1])
except ValueError as e:
continue
if posizioa == 1:
kategoria_base = kategoriak[tanda]
tanda = tanda + 1
kategoria = kategoria_base + ' ' + row[3]
kategoriak_helper.append(kategoria)
emaitza = {'tanda': tanda,
'tanda_postua': posizioa,
'kalea': 1,
'posizioa': posizioa,
'kategoria': kategoria,
'denbora': row[8],
'puntuazioa': row[10]}
t = TaldeEmaitza(row[2], **emaitza)
t.ziaboga_gehitu(row[5])
estropada.taldeak_add(t)
estropada.kategoriak = list(set(kategoriak_helper))
print(estropada.get_json())
estropada_json = estropada.get_json()
estropada_obj = json.loads(estropada_json)
estropada_obj['_rev'] = estropada._rev
db[estropada.id] = estropada_obj