You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.5 KiB
55 lines
1.5 KiB
import tabula |
|
import config |
|
import csv |
|
from models import RegionalZone |
|
|
|
appName = config.APP_BASE_NAME |
|
usurpaPdf = config.USURPA_DOWNLOAD_PATH |
|
csvOut = config.USURPA_CSV |
|
|
|
zoneList = [] |
|
|
|
def parseCSV(input=csvOut): |
|
|
|
print("# Parsig", input) |
|
print ("* Parsing areas") |
|
with open(input, 'r') as csvFile: |
|
reader = csv.reader(csvFile) |
|
regionalZone = None |
|
for row in reader: |
|
zoneName = RegionalZone.checkIfIsZone(row) |
|
# If is zone append it to the list of zones |
|
if zoneName: |
|
regionalZone = RegionalZone(zoneName) |
|
zoneList.append(regionalZone) |
|
if regionalZone is not None and not zoneName: |
|
regionalZone.addRow(row) |
|
|
|
csvFile.close() |
|
|
|
|
|
print ("* Creating squat objects") |
|
for zone in zoneList: |
|
zone.generateSquats() |
|
|
|
print("# Printing all") |
|
x = 0 |
|
for zone in zoneList: |
|
print("--",zone.name, "squats:", zone.squatList.__len__()) |
|
for squat in zone.squatList: |
|
print("\t",squat.name) |
|
for row in squat.rows: |
|
print("\t -- ",row.__len__(), row) |
|
|
|
x += 1 |
|
|
|
print("Total squats found", x) |
|
print("Total areas found: ", zoneList.__len__()) |
|
|
|
|
|
|
|
def convertPdfToCsv(input=usurpaPdf, output=csvOut): |
|
print("# Converting", input, "to", output) |
|
df = tabula.convert_into(input, output, pages='all', multiple_tables=True, output_format="csv", stream=True) |
|
print("Done.") |
|
return output
|
|
|