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