1014 lines
38 KiB
Python
1014 lines
38 KiB
Python
from copy import copy
|
|
import datetime
|
|
import csv
|
|
import re
|
|
import json
|
|
import tomllib
|
|
import tomlkit
|
|
import io
|
|
import xml.etree.ElementTree as ET
|
|
# from fpdf import FPDF
|
|
|
|
|
|
# The event object stores all the event data.
|
|
# A .otime file is more or less just a json dump of the Event object.
|
|
class Event:
|
|
def __init__(self, eventid, name, start_time=None, end_time=None,
|
|
organiser=None, courses=[], o_classes=[], runners=[],
|
|
card_dumps=[], fees=[]):
|
|
self.id = eventid
|
|
self.name = name
|
|
self.start_time = start_time
|
|
self.end_time = end_time
|
|
self.organiser = organiser
|
|
self.courses = courses
|
|
self.o_classes = o_classes
|
|
self.runners = runners
|
|
self.card_dumps = card_dumps
|
|
self.fees = fees
|
|
|
|
def from_toml(f):
|
|
cfg = tomllib.load(f)
|
|
courses = []
|
|
for c in cfg['course']:
|
|
courses.append(Course(c['name'] , c['codes'], c['forked'], c['variations']))
|
|
o_classes = []
|
|
for c in cfg['classes']:
|
|
for n in courses:
|
|
if n.name == cfg['classes'][c]:
|
|
course = n
|
|
break
|
|
o_classes.append(OClass(c, cfg['classes'][c], course))
|
|
|
|
return Event(1, cfg['name'], cfg['start_time'], cfg['end_time'], cfg['organiser'], courses, o_classes, [], [], [])
|
|
|
|
def create_toml(self, dir='./'):
|
|
doc = tomlkit.document()
|
|
cdicts = []
|
|
for course in self.courses:
|
|
cdicts.append(course.asdict())
|
|
#Fjern None fra dictene
|
|
for c in cdicts:
|
|
if c['forked'] == False:
|
|
c['variations'] = []
|
|
|
|
|
|
ocdicts = []
|
|
for o_class in self.o_classes:
|
|
ocdicts.append(o_class.asdict())
|
|
|
|
fdicts = []
|
|
for fee in self.fees:
|
|
fdicts.append(fee.asdict())
|
|
|
|
data = {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'organiser': self.organiser,
|
|
'start_time': self.start_time.isoformat(),
|
|
'end_time': self.end_time.isoformat(),
|
|
'course': cdicts,
|
|
#'o_classes': ocdicts,
|
|
#'fees': fdicts
|
|
}
|
|
with open(dir + 'cfg.toml', "w") as f:
|
|
tomlkit.dump(data, f)
|
|
|
|
def add_course(self, *args):
|
|
for n in args:
|
|
self.courses.append(n)
|
|
|
|
def add_o_class(self, *args):
|
|
for n in args:
|
|
self.o_classes.append(n)
|
|
|
|
def add_runners(self, *args):
|
|
for n in args:
|
|
self.runners.append(n)
|
|
|
|
def add_fees(self, *args):
|
|
for n in args:
|
|
self.fees.append(n)
|
|
|
|
def import_xml_entries(self, xml_file):
|
|
self.add_runners(*runners_from_xml_entries(xml_file))
|
|
self.add_fees(*fees_from_xml_entries(xml_file))
|
|
|
|
def from_xml_entries(xml_file):
|
|
tree = ET.parse(xml_file)
|
|
root = tree.getroot()
|
|
|
|
url = '{http://www.orienteering.org/datastandard/3.0}'
|
|
event_el = root.find(f'./{url}Event')
|
|
event_id = int(event_el.find(f'./{url}Id').text)
|
|
name = event_el.find(f'./{url}Name').text
|
|
organiser = event_el.find(f'./{url}Organiser/{url}Name').text
|
|
|
|
start_ds = event_el.find(f'./{url}StartTime/{url}Date').text
|
|
start_ts = event_el.find(f'./{url}StartTime/{url}Time').text[:-1]
|
|
start_time = datetime.datetime.fromisoformat(f'{start_ds}T{start_ts}')
|
|
|
|
end_ds = event_el.find(f'./{url}EndTime/{url}Date').text
|
|
end_ts = event_el.find(f'./{url}EndTime/{url}Time').text[:-1]
|
|
end_time = datetime.datetime.fromisoformat(f'{end_ds}T{end_ts}')
|
|
|
|
runners = runners_from_xml_entries(xml_file)
|
|
fees = fees_from_xml_entries(xml_file)
|
|
|
|
return Event(event_id, name, organiser=organiser, runners=runners,
|
|
fees=fees, start_time=start_time, end_time=end_time)
|
|
def import_xml_courses(self, xml_file):
|
|
self.courses = courses_from_xml(xml_file)
|
|
|
|
def import_ttime_cnf(self, ttime_file):
|
|
self.add_course(*courses_from_ttime_conf(ttime_file))
|
|
if isinstance(ttime_file, io.TextIOBase):
|
|
ttime_file.seek(0)
|
|
self.add_o_class(*classes_from_ttime_conf(ttime_file, self.courses))
|
|
|
|
def import_ttime_db(self, ttime_file):
|
|
if type(ttime_file) == str:
|
|
f_list = ttime_file.splitlines()
|
|
elif isinstance(ttime_file, io.TextIOBase):
|
|
f_list = csv.reader(ttime_file, delimiter=';',)
|
|
runnerarray = []
|
|
for row in f_list:
|
|
if len(row) == 0 or row[1] == '':
|
|
continue
|
|
runnerarray.append(Runner.from_string(row, self.o_classes))
|
|
self.runners = runnerarray
|
|
|
|
def import_mtr_file(self, mtr_file):
|
|
self.card_dumps = CardDump.list_from_mtr_f(mtr_file)
|
|
|
|
def match_runners_cards(self):
|
|
for r in self.runners:
|
|
for d in self.card_dumps:
|
|
if r.card == d.card:
|
|
r.card_r = d
|
|
|
|
def match_runners_o_classes(self):
|
|
for r in self.runners:
|
|
for c in self.o_classes:
|
|
if r.o_class_str == c.name:
|
|
r.o_class = c
|
|
|
|
def match_o_classes_courses(self):
|
|
for oc in self.o_classes:
|
|
for c in self.courses:
|
|
if oc.course_str == c.name:
|
|
oc.course = c
|
|
|
|
def match_runners_fees(self):
|
|
for r in self.runners:
|
|
for f in self.fees:
|
|
if r.fee_id == f.id: tree = ET.parse(xml_file)
|
|
root = tree.getroot()
|
|
r.fee = f
|
|
|
|
def match_all(self):
|
|
self.match_runners_cards()
|
|
self.match_runners_o_classes()
|
|
self.match_o_classes_courses()
|
|
self.match_runners_fees()
|
|
|
|
def get_xml_res(self):
|
|
root = ET.Element('ResultList')
|
|
root.set('xmlns', 'http://www.orienteering.org/datastandard/3.0')
|
|
root.set('xmlns:xsi', 'http://www.w3.org/2001/XMLSchema-instance')
|
|
root.set('iofVersion', '3.0')
|
|
root.set('createTime', datetime.datetime.now().isoformat(timespec='seconds'))
|
|
root.set('creator', 'oTime')
|
|
root.set('status', 'Complete')
|
|
tree = ET.ElementTree(root)
|
|
event = ET.SubElement(root, 'Event')
|
|
xml_child(event, 'Id', self.id)
|
|
xml_child(event, 'Name', self.name)
|
|
for i in self.o_classes:
|
|
print('Hmmmmmm')
|
|
# <ClassResult>
|
|
class_result = ET.SubElement(root, 'ClassResult')
|
|
# <Class>
|
|
t = ET.SubElement(class_result, 'Class')
|
|
xml_child(t, 'Name', i.name)
|
|
# <PersonResult>
|
|
runners_same_c = get_runners_in_class(self.runners, i)
|
|
runners_ranked = rank_runners(runners_same_c, i)
|
|
# Put the OK runners first and Active last
|
|
runners_sorted = [i for i in runners_same_c if i not in runners_ranked]
|
|
runners_ranked.extend(runners_sorted)
|
|
|
|
for n in runners_ranked:
|
|
person_result = ET.SubElement(class_result, 'PersonResult')
|
|
# <Person>
|
|
person = ET.SubElement(person_result, 'Person')
|
|
xml_child(person, 'Id', n.id)
|
|
# <Name>
|
|
name = ET.SubElement(person, 'Name')
|
|
xml_child(name, 'Family', n.last)
|
|
xml_child(name, 'Given', n.first)
|
|
# </Name>
|
|
# </Person>
|
|
# <Organisation>
|
|
org = ET.SubElement(person_result, 'Organisation')
|
|
xml_child(org, 'Id', n.club_id)
|
|
xml_child(org, 'Name', n.club)
|
|
country = ET.SubElement(org, 'Country')
|
|
# TODO: hent land fra løperobjektet
|
|
country.text = 'Norway'
|
|
country.set('code', 'NOR')
|
|
# </Organisation>
|
|
# <Result>
|
|
result = ET.SubElement(person_result, 'Result')
|
|
|
|
# TODO: Dette bør skrives om til å bruke Runner metoder så mye som mulig.
|
|
if hasattr(n, 'card_r') and len(n.card_r.splits) > 2:
|
|
xml_child(result, 'StartTime', n.card_r.s_time.isoformat())
|
|
xml_child(result, 'FinishTime', n.card_r.f_time.isoformat())
|
|
xml_child(result, 'Time', n.totaltime())
|
|
if n.status() == 'OK':
|
|
# <TimeBehind>
|
|
xml_child(result, 'TimeBehind', n.totaltime() - runners_ranked[0].totaltime())
|
|
# </TimeBehind>
|
|
xml_child(result, 'Position', n.rank(self.runners))
|
|
xml_child(result, 'Status', n.status())
|
|
# <SplitTime>
|
|
# TODO: ta utgangspunkt i løypa, ikke det brikka har stempla
|
|
for code, split in zip(n.card_r.controls, n.card_r.splits):
|
|
st = ET.SubElement(result, 'SplitTime')
|
|
xml_child(st, 'ControlCode', code)
|
|
xml_child(st, 'Time', split)
|
|
if code == n.res_codes()[-1]:
|
|
break
|
|
# </SplitTime>
|
|
elif n.status() == 'Disqualified':
|
|
xml_child(result, 'Status', n.status())
|
|
for code, split in zip(n.res_codes(), n.res_splits()):
|
|
st = ET.SubElement(result, 'SplitTime')
|
|
xml_child(st, 'ControlCode', code)
|
|
if split is not None:
|
|
xml_child(st, 'Time', split)
|
|
else:
|
|
xml_child(result, 'Status', n.status())
|
|
else:
|
|
xml_child(result, 'Status', n.status())
|
|
# </Result>
|
|
# </PersonResult>
|
|
# </Class>
|
|
ET.indent(root, space=' ', level=0)
|
|
return tree
|
|
|
|
def create_json_file(self):
|
|
rdicts = []
|
|
for runner in self.runners:
|
|
rdicts.append(runner.asdict())
|
|
|
|
cdicts = []
|
|
for course in self.courses:
|
|
cdicts.append(course.asdict())
|
|
|
|
ocdicts = []
|
|
for o_class in self.o_classes:
|
|
ocdicts.append(o_class.asdict())
|
|
|
|
ddicts = []
|
|
for dump in self.card_dumps:
|
|
ddicts.append(dump.asdict())
|
|
|
|
fdicts = []
|
|
for fee in self.fees:
|
|
fdicts.append(fee.asdict())
|
|
|
|
json_data = {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'orginser': self.organiser,
|
|
'start_time': self.start_time.isoformat(),
|
|
'end_time': self.end_time.isoformat(),
|
|
'runners': rdicts,
|
|
'courses': cdicts,
|
|
'o_classes': ocdicts,
|
|
'card_dumps': ddicts,
|
|
'fees': fdicts
|
|
}
|
|
return json.dumps(json_data, sort_keys=True, indent=4)
|
|
|
|
# Get event object from .otime json file
|
|
def from_json(f):
|
|
data = json.load(f)
|
|
runners = []
|
|
for r in data['runners']:
|
|
runners.append(Runner(r['id'], r['first'], r['last'], club_id=r['club_id'],
|
|
club=r['club'], country=r['country'], card=r['card'],
|
|
o_class_str=r['o_class_str'], fork=r['fork'],
|
|
start_time=r['start_time'], fee_id=int(r['fee_id'])))
|
|
|
|
courses = []
|
|
for c in data['courses']:
|
|
courses.append(Course(c['name'], c['codes'], forked=c['forked'],
|
|
variations=c['variations']))
|
|
|
|
o_classes = []
|
|
for c in data['o_classes']:
|
|
o_classes.append(OClass(c['name'], c['course_str'], None))
|
|
|
|
card_dumps = []
|
|
for d in data['card_dumps']:
|
|
card_dumps.append(CardDump(d['card'], d['controls'], d['splits'],
|
|
datetime.datetime.fromisoformat(d['read_time']),
|
|
datetime.datetime.fromisoformat(d['s_time']),
|
|
datetime.datetime.fromisoformat(d['f_time'])))
|
|
|
|
fees = []
|
|
for f in data['fees']:
|
|
fees.append(Fee(f['id'], f['name'], f['currency'], f['amount'],
|
|
from_birth_date=f['from_birth_date'],
|
|
to_birth_date=f['to_birth_date']))
|
|
|
|
return Event(data['id'], data['name'], organiser=data['orginiser'],
|
|
start_time=datetime.datetime.fromisoformat(data['start_time']),
|
|
end_time=datetime.datetime.fromisoformat(data['end_time']),
|
|
runners=runners, courses=courses, o_classes=o_classes,
|
|
card_dumps=card_dumps, fees=fees)
|
|
|
|
def create_start_list_pdf(self, file_name):
|
|
pdf = FPDF()
|
|
pdf.add_page()
|
|
pdf.add_font("LiberationSans", fname="data/fonts/LiberationSans-Regular.ttf")
|
|
pdf.set_font("LiberationSans", size=10)
|
|
line_height = pdf.font_size * 2
|
|
col_width = pdf.epw / 4 # distribute content evenly
|
|
for runner in self.runners:
|
|
pdf.multi_cell(col_width, line_height, runner.fullname(), border=1, ln=3, max_line_height=pdf.font_size, align='L')
|
|
pdf.multi_cell(col_width, line_height, runner.o_class_str, border=1, ln=3, max_line_height=pdf.font_size)
|
|
pdf.multi_cell(col_width, line_height, str(runner.card), border=1, ln=3, max_line_height=pdf.font_size)
|
|
if runner.start_time is not None:
|
|
pdf.multi_cell(col_width, line_height, str(runner.start_time), border=1, ln=3, max_line_height=pdf.font_size)
|
|
else:
|
|
pdf.multi_cell(col_width, line_height, '', border=1, ln=3, max_line_height=pdf.font_size)
|
|
pdf.ln(line_height)
|
|
pdf.output(file_name)
|
|
|
|
def create_all_invoices(self, filename_prefix):
|
|
clubs = [x.club for x in self.runners if x.club != self.runners[self.runners.index(x)-1].club and x.club is not None]
|
|
for club in clubs:
|
|
self.create_club_invoice(club, filename_prefix + club + '.pdf')
|
|
|
|
def create_club_invoice(self, club, file_name):
|
|
# Get only runners in specified club
|
|
runners_ic = [x for x in self.runners if x.club == club]
|
|
payments = [x.fee.amount for x in runners_ic]
|
|
subtotal = sum(payments)
|
|
|
|
# Dict of runners in each fee
|
|
fee_dict = {}
|
|
for fee in self.fees:
|
|
fee_dict.update({fee.name: [x for x in runners_ic if x.fee.name == fee.name]})
|
|
if len(fee_dict[fee.name]) == 0:
|
|
fee_dict.pop(fee.name)
|
|
|
|
pdf = PDF()
|
|
|
|
pdf.set_title(f'Faktura {self.name} {club}')
|
|
pdf.set_producer('oTime 0.0.1')
|
|
pdf.set_creator('oTime 0.0.1')
|
|
pdf.set_creation_date()
|
|
|
|
pdf.add_page()
|
|
pdf.add_font("LiberationSans", fname="data/fonts/LiberationSans-Regular.ttf")
|
|
pdf.add_font("LiberationSans-Bold", fname="data/fonts/LiberationSans-Bold.ttf")
|
|
pdf.set_font("LiberationSans", size=10)
|
|
line_height = pdf.font_size * 1.5
|
|
|
|
# Topp venstre:
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(pdf.epw / 2, line_height, self.name, align='L')
|
|
pdf.ln(0.2)
|
|
pdf.set_font("LiberationSans", size=10)
|
|
pdf.multi_cell(pdf.epw / 2, line_height, self.organiser, align='L')
|
|
pdf.ln(14)
|
|
pdf.multi_cell(pdf.epw / 2, line_height, club, align='L')
|
|
pdf.ln()
|
|
|
|
# Topp høyre:
|
|
pdf.set_xy(-pdf.epw / 3, 10)
|
|
pdf.set_font("LiberationSans-Bold", size=20)
|
|
pdf.multi_cell(pdf.epw, line_height, 'Faktura', align='L')
|
|
pdf.set_xy(-pdf.epw / 3, 20)
|
|
|
|
# Dato
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(0, line_height, 'Dato:', align='L')
|
|
pdf.set_xy(-pdf.epw / 3, 20)
|
|
pdf.set_font("LiberationSans", size=10)
|
|
pdf.multi_cell(0, line_height, datetime.date.today().strftime('%d.%m.%Y'), align='R')
|
|
|
|
# Nummer
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height)
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(0, line_height, 'Nummer:', align='L')
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height)
|
|
pdf.multi_cell(0, line_height, '1', align='R')
|
|
|
|
# Kundeid
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height*2)
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(0, line_height, 'Kundeid:', align='L')
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height*2)
|
|
pdf.set_font("LiberationSans", size=10)
|
|
pdf.multi_cell(0, line_height, '123', align='R')
|
|
|
|
# Forfall
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height*3)
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(0, line_height, 'Forfall:', align='L')
|
|
pdf.set_xy(-pdf.epw / 3, 20 + line_height*3)
|
|
pdf.set_font("LiberationSans", size=10)
|
|
pdf.multi_cell(0, line_height, datetime.date.today().strftime('%d.08.%Y'), align='R')
|
|
|
|
pdf.set_xy(-pdf.epw, 20 + line_height*5)
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
date = self.start_time.strftime('%d.%m.%Y')
|
|
pdf.multi_cell(0, line_height, f'Kontingentliste {self.name} dato: {date}', align='R')
|
|
|
|
pdf.ln()
|
|
|
|
pdf.set_font("LiberationSans", size=10)
|
|
# Tabell
|
|
line_height = pdf.font_size * 2
|
|
col_width = pdf.epw / 8 # distribute content evenly
|
|
# Top row
|
|
pdf.set_fill_color(191, 191, 191)
|
|
pdf.set_draw_color(191, 191, 191)
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.multi_cell(col_width, line_height, 'Startnr', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width*2, line_height, 'Navn', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width, line_height, 'Klasse', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width, line_height, 'Brikke', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
# pdf.multi_cell(col_width, line_height, 'Starttid', border=1, ln=3,max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width, line_height, 'Resultat', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width, line_height, 'Plass', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.multi_cell(col_width, line_height, 'Kontigent', border=1, ln=3, max_line_height=pdf.font_size, align='L', fill=True)
|
|
pdf.ln()
|
|
pdf.set_draw_color(0, 0, 0)
|
|
for runners in fee_dict.values():
|
|
pdf.set_font("LiberationSans-Bold", size=11)
|
|
pdf.multi_cell(len(runners[0].fee.name)*2.3, pdf.font_size * 1.1, runners[0].fee.name, border='B', align='L')
|
|
pdf.set_font("LiberationSans", size=8)
|
|
line_height = pdf.font_size * 1.6
|
|
pdf.ln()
|
|
for runner in runners:
|
|
pdf.multi_cell(col_width, line_height, runner.id, ln=3, max_line_height=pdf.font_size, align='L') # Start Number
|
|
pdf.multi_cell(col_width*2, line_height, f'{runner.last}, {runner.first}', ln=3, max_line_height=pdf.font_size, align='L') # Name
|
|
pdf.multi_cell(col_width, line_height, runner.o_class_str, ln=3, max_line_height=pdf.font_size, align='L') # Class
|
|
pdf.multi_cell(col_width, line_height, str(runner.card), ln=3, max_line_height=pdf.font_size) # card
|
|
# Starttime:
|
|
#if runner.start_time != None:
|
|
# pdf.multi_cell(col_width, line_height, str(runner.start_time), ln=3, max_line_height=pdf.font_size)
|
|
#else:
|
|
# pdf.multi_cell(col_width, line_height, 'Fristart', ln=3, max_line_height=pdf.font_size)
|
|
# Time used:
|
|
if runner.status() == 'OK':
|
|
pdf.multi_cell(col_width, line_height, str(datetime.timedelta(seconds=runner.totaltime())), ln=3, max_line_height=pdf.font_size)
|
|
elif runner.status() == 'Disqualified':
|
|
pdf.multi_cell(col_width, line_height, 'DSQ', ln=3, max_line_height=pdf.font_size)
|
|
elif runner.status() == 'Active':
|
|
pdf.multi_cell(col_width, line_height, 'DNS', ln=3, max_line_height=pdf.font_size)
|
|
else:
|
|
pdf.multi_cell(col_width, line_height, runner.status(), ln=3, max_line_height=pdf.font_size)
|
|
|
|
# Rank:
|
|
if runner.status() == 'OK':
|
|
pdf.multi_cell(col_width, line_height, str(runner.rank(self.runners)) + '.', ln=3, max_line_height=pdf.font_size)
|
|
else:
|
|
pdf.multi_cell(col_width, line_height, '', ln=3, max_line_height=pdf.font_size)
|
|
|
|
pdf.multi_cell(col_width, line_height, str(runner.fee.amount), ln=3, max_line_height=pdf.font_size, align='R')
|
|
pdf.ln(line_height)
|
|
|
|
pdf.set_draw_color(0, 0, 0)
|
|
# sum
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.set_x(col_width*8)
|
|
pdf.cell(0, line_height, 'Sum ' + str(subtotal), border='TB', align='R')
|
|
|
|
pdf.set_font("LiberationSans", size=10)
|
|
line_height = pdf.font_size * 1.5
|
|
pdf.ln(20)
|
|
pdf.multi_cell(0, line_height, 'Antall løpere ' + str(len(runners_ic)), ln=3, max_line_height=pdf.font_size, align='L')
|
|
|
|
pdf.set_x(col_width*7)
|
|
pdf.multi_cell(col_width, line_height, 'Total sum', ln=3, max_line_height=pdf.font_size, align='L')
|
|
pdf.multi_cell(0, line_height, str(subtotal), ln=3, max_line_height=pdf.font_size, align='R')
|
|
pdf.ln()
|
|
pdf.set_x(col_width*7)
|
|
pdf.multi_cell(col_width, line_height, 'Betalt', ln=3, max_line_height=pdf.font_size, align='L')
|
|
pdf.multi_cell(0, line_height, '0', ln=3, max_line_height=pdf.font_size, align='R')
|
|
pdf.ln()
|
|
pdf.set_font("LiberationSans-Bold", size=10)
|
|
pdf.set_x(col_width*7)
|
|
pdf.multi_cell(col_width, line_height, 'Skyldig', border='B', ln=3, max_line_height=pdf.font_size, align='L')
|
|
pdf.multi_cell(0, line_height, str(subtotal), border='B',ln=3, max_line_height=pdf.font_size, align='R')
|
|
|
|
|
|
pdf.output(file_name)
|
|
"""
|
|
class PDF(FPDF):
|
|
def footer(self):
|
|
self.set_y(-15)
|
|
self.set_font("LiberationSans", size=10)
|
|
self.set_fill_color(191, 191, 191)
|
|
self.set_draw_color(191, 191, 191)
|
|
col_width = self.epw / 3
|
|
self.cell(col_width, 7, 'oTime', border=1, align='L', fill=True)
|
|
self.cell(col_width, 7, datetime.datetime.now().strftime('%d.%m.%Y %H:%M:%S'), border=1, align='C', fill=True)
|
|
self.cell(col_width, 7, f"Side {self.page_no()} av {{nb}}", border=1, align='R', fill=True)
|
|
"""
|
|
# The runner object stores all the data specific to a runner.
|
|
class Runner:
|
|
def __init__(self, runner_id, first, last, club=None, club_id=None,
|
|
country=None, card=None, o_class_str=None, o_class=None,
|
|
fork=0, start_time=None, fee_id=None, fee=None):
|
|
self.id = runner_id
|
|
self.first = first
|
|
self.last = last
|
|
self.club = club
|
|
self.club_id = club_id
|
|
self.country = country
|
|
self.card = card
|
|
self.o_class_str = o_class_str
|
|
self.o_class = o_class
|
|
self.fork = fork
|
|
self.start_time = start_time
|
|
self.fee_id = fee_id
|
|
self.fee = fee
|
|
|
|
def from_string(tt_line, o_classes):
|
|
#https://web.archive.org/web/20191229124046/http://wiki.ttime.no/index.php/Developer
|
|
eventorid = tt_line[0]
|
|
country = ''
|
|
name = tt_line[2].split(',')
|
|
try:
|
|
first = name[1].strip()
|
|
except:
|
|
first = ''
|
|
last = name[0]
|
|
try:
|
|
club = tt_line[4]
|
|
except:
|
|
club = "None"
|
|
try:
|
|
card = int(tt_line[6])
|
|
except:
|
|
card = 0
|
|
runner_o_class = None
|
|
try:
|
|
raw_class_str = tt_line[3]
|
|
except:
|
|
# VELDIG MIDLERTIDIG
|
|
runner_o_class = None
|
|
else:
|
|
if raw_class_str != '':
|
|
for i in o_classes:
|
|
if i.name == raw_class_str:
|
|
runner_o_class = i
|
|
break
|
|
else:
|
|
runner_o_class = None
|
|
# TODO: Gjør sånn at den lager nye o klasser om den ikke finnes fra før
|
|
options = tt_line[5].split(',')
|
|
try:
|
|
club_id = options[options.index('A')+3]
|
|
except:
|
|
club_id = 0
|
|
try:
|
|
start_time = options[options.index('U')+1]
|
|
except:
|
|
start_time = None
|
|
return Runner(eventorid, first, last, club=club, club_id=club_id,
|
|
country=country, card=card, o_class_str=raw_class_str,
|
|
o_class=runner_o_class, start_time=start_time)
|
|
|
|
def fullname(self):
|
|
return '{} {}'.format(self.first, self.last)
|
|
|
|
def check_codes(self):
|
|
return contains(self.res_codes(), self.card_r.controls)
|
|
|
|
def totaltime(self):
|
|
f_control = self.res_codes()[-1]
|
|
try:
|
|
index = self.card_r.controls.index(f_control)
|
|
return self.card_r.splits[index]
|
|
except:
|
|
return 0
|
|
|
|
def status(self):
|
|
if hasattr(self, 'card_r') is False or self.o_class is None:
|
|
return 'Active'
|
|
elif self.check_codes():
|
|
return 'OK'
|
|
elif self.check_codes() is False:
|
|
return 'Disqualified'
|
|
|
|
# TODO: må forbedres
|
|
def rank(self, allrunners):
|
|
c_ranked = rank_runners(allrunners, self.o_class)
|
|
try:
|
|
return c_ranked.index(self) + 1
|
|
except ValueError:
|
|
return None
|
|
|
|
def res_codes(self):
|
|
if self.o_class.course.forked is False:
|
|
return self.o_class.course.codes
|
|
else:
|
|
return self.o_class.course.variations[self.fork]
|
|
|
|
# TODO: Mange bugs med løyper som har samme post flere ganger
|
|
# Used for making result files and tables
|
|
def res_splits(self):
|
|
if self.status() == 'OK':
|
|
splits_cpy = self.card_r.splits.copy()
|
|
for control in self.card_r.controls:
|
|
if control not in self.res_codes():
|
|
index = self.card_r.controls.index(control)
|
|
split = self.card_r.splits[index]
|
|
splits_cpy.remove(split)
|
|
return splits_cpy
|
|
|
|
else:
|
|
splits_cpy = self.card_r.splits.copy()
|
|
for control in self.card_r.controls:
|
|
if control not in self.res_codes():
|
|
index = self.card_r.controls.index(control)
|
|
split = self.card_r.splits[index]
|
|
try:
|
|
splits_cpy.remove(split)
|
|
except Exception:
|
|
print('aaaaaa')
|
|
|
|
|
|
punches = self.card_r.controls.copy()
|
|
splits = []
|
|
for code in self.res_codes():
|
|
if punches[0] == code:
|
|
splits.append(splits_cpy[0])
|
|
splits_cpy.pop(0)
|
|
punches.pop(0)
|
|
continue
|
|
else:
|
|
splits.append(None)
|
|
return splits
|
|
|
|
def asdict(self):
|
|
return {
|
|
'id': self.id,
|
|
'first': self.first,
|
|
'last': self.last,
|
|
'club_id': self.club_id,
|
|
'club': self.club,
|
|
'country': self.country,
|
|
'card': self.card,
|
|
'o_class_str': self.o_class_str,
|
|
'fork' : self.fork,
|
|
'start_time': self.start_time,
|
|
'fee_id': self.fee_id
|
|
}
|
|
|
|
|
|
class CardDump:
|
|
def __init__(self, card, controls, splits, read_time, s_time, f_time):
|
|
self.card = card
|
|
self.controls = controls
|
|
self.splits = splits
|
|
self.read_time = read_time
|
|
self.s_time = s_time
|
|
self.f_time = f_time
|
|
|
|
def __repr__(self):
|
|
return f'card({self.card}) controls({self.controls}) splits({self.splits})'
|
|
|
|
def from_mtr_bytes(datamsg):
|
|
card = int.from_bytes(datamsg[20:23], 'little')
|
|
|
|
#Extract codes and splits:
|
|
controls = []
|
|
splits = []
|
|
splits_offset = 26
|
|
code_numbytes = 1
|
|
time_numbytes = 2
|
|
split_numbytes = code_numbytes + time_numbytes
|
|
for split_index in range(50):
|
|
code_offset = splits_offset + split_index * split_numbytes
|
|
time_offset = code_offset + code_numbytes
|
|
code = int.from_bytes(datamsg[code_offset:code_offset+code_numbytes], 'little')
|
|
time = int.from_bytes(datamsg[time_offset:time_offset+time_numbytes], 'little')
|
|
if code != 0:
|
|
controls.append(code)
|
|
splits.append(time)
|
|
|
|
# Extract start time:
|
|
year = int.from_bytes(datamsg[8:9], 'little')
|
|
month = int.from_bytes(datamsg[9:10], 'little')
|
|
day = int.from_bytes(datamsg[10:11], 'little')
|
|
hours = int.from_bytes(datamsg[11:12], 'little')
|
|
minutes = int.from_bytes(datamsg[12:13], 'little')
|
|
seconds = int.from_bytes(datamsg[13:14], 'little')
|
|
milliseconds = int.from_bytes(datamsg[14:16], 'little')
|
|
|
|
read_time = datetime.datetime(year, month, day, hours, minutes, seconds, milliseconds)
|
|
if len(controls) > 2:
|
|
s_time = read_time - datetime.timedelta(seconds=splits[-1])
|
|
f_time = read_time - (datetime.timedelta(seconds=splits[-1]) + datetime.timedelta(seconds=splits[-2]))
|
|
else:
|
|
s_time = read_time
|
|
f_time = read_time
|
|
|
|
return(CardDump(card, controls, splits, read_time, s_time, f_time))
|
|
|
|
def list_from_mtr_f(mtr_f):
|
|
if type(mtr_f) == str:
|
|
f_list = mtr_f.splitlines()
|
|
|
|
rows = []
|
|
cards = []
|
|
# hver rad er brikkenummer med tilhørende info
|
|
csvreader = csv.reader(f_list)
|
|
for row in csvreader:
|
|
if len(row) == 0:
|
|
continue
|
|
rows.append(row)
|
|
controls = []
|
|
splits = []
|
|
# postkodene kommer på oddetall fra og med den 11. De blir hevet inn i controls
|
|
for item in row[11::2]:
|
|
if item == '250':
|
|
controls.append(int(item))
|
|
break
|
|
elif item == '000':
|
|
break
|
|
else:
|
|
controls.append(int(item))
|
|
# strekktidene kommer på partall fra og med den 12. De blir hevet i splits.
|
|
for item in row[12::2]:
|
|
if item == '00000':
|
|
break
|
|
else:
|
|
splits.append(int(item))
|
|
# looper gjonnom løperobjektene og legger til poster og strekktider + start og sluttid
|
|
# usikker på om dette er riktig klokeslett
|
|
tl = row[5].split(' ')
|
|
tl[0] = tl[0].split('.')
|
|
tl[0][2] = '20' + tl[0][2]
|
|
tl[0] = list(map(int, tl[0]))
|
|
tl[1] = tl[1].split(':')
|
|
tl[1][2] = float(tl[1][2])
|
|
tl[1] = list(map(int, tl[1]))
|
|
read_time = datetime.datetime(tl[0][2], tl[0][1], tl[0][0], tl[1][0], tl[1][1], tl[1][2])
|
|
if len(controls) > 2 and len(splits) > 2:
|
|
s_time = read_time - datetime.timedelta(seconds=splits[-1])
|
|
f_time = read_time - (datetime.timedelta(seconds=splits[-1]) + datetime.timedelta(seconds=splits[-2]))
|
|
else:
|
|
s_time = read_time
|
|
f_time = read_time
|
|
cards.append(CardDump(int(row[6]), controls, splits, read_time, s_time, f_time))
|
|
return cards
|
|
|
|
def asdict(self):
|
|
return {
|
|
'card': self.card,
|
|
'controls': self.controls,
|
|
'splits': self.splits,
|
|
'read_time': self.read_time.isoformat(),
|
|
's_time': self.s_time.isoformat(),
|
|
'f_time': self.f_time.isoformat()
|
|
}
|
|
|
|
# Stored in Event.courses
|
|
class Course:
|
|
def __init__(self, name, codes, forked=False, variations=None):
|
|
self.name = name
|
|
self.codes = codes
|
|
self.forked = forked
|
|
# Variations is a list
|
|
self.variations = variations
|
|
|
|
def __repr__(self):
|
|
return f'name({self.name})'
|
|
|
|
def asdict(self):
|
|
return {
|
|
'name': self.name,
|
|
'codes': self.codes,
|
|
'forked': self.forked,
|
|
'variations': self.variations
|
|
}
|
|
|
|
# Stored in Event.o_classes
|
|
class OClass:
|
|
def __init__(self, name, course_str, course):
|
|
self.name = name
|
|
self.course_str = course_str
|
|
self.course = course
|
|
|
|
def __repr__(self):
|
|
return f'name({self.name})'
|
|
|
|
def asdict(self):
|
|
return {
|
|
'name': self.name,
|
|
'course_str': self.course_str
|
|
}
|
|
|
|
class Fee:
|
|
def __init__(self, fee_id, name, currency, amount, from_birth_date=None,
|
|
to_birth_date=None):
|
|
self.id = fee_id
|
|
self.name = name
|
|
self.currency = currency
|
|
self.amount = amount
|
|
self.from_birth_date = from_birth_date
|
|
self.to_birth_date = to_birth_date
|
|
|
|
def asdict(self):
|
|
return {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'currency': self.currency,
|
|
'amount': self.amount,
|
|
'from_birth_date': self.from_birth_date,
|
|
'to_birth_date': self.to_birth_date
|
|
}
|
|
|
|
# TODO: Take string instead of file.
|
|
def courses_from_ttime_conf(ttime_file):
|
|
if type(ttime_file) == str:
|
|
conf = ttime_file.splitlines()
|
|
elif isinstance(ttime_file, io.TextIOBase):
|
|
conf = ttime_file.readlines()
|
|
|
|
courses = []
|
|
for line in conf:
|
|
if '-codes' in line:
|
|
code_list = re.search(r'(?<=\")(.*?)(?=\")', line).group().split(';')
|
|
loops = 0
|
|
for n in code_list:
|
|
n = n.split(',')
|
|
loops += 1
|
|
n = list(map(int, n))
|
|
courses.append(Course('course_'+str(loops), n))
|
|
return courses
|
|
|
|
def classes_from_ttime_conf(ttime_file, courses):
|
|
if type(ttime_file) == str:
|
|
conf = ttime_file.splitlines()
|
|
elif isinstance(ttime_file, io.TextIOBase):
|
|
conf = ttime_file.readlines()
|
|
o_classes = []
|
|
for line in conf:
|
|
if '-courses' in line:
|
|
raw_courselist = re.search(r'(?<=\")(.*?)(?=\")', line).group().split(';')
|
|
loops = 0
|
|
for n in raw_courselist:
|
|
split = n.split(',')
|
|
for i in split:
|
|
o_classes.append(OClass(i, courses[loops].name, courses[loops]))
|
|
loops += 1
|
|
return o_classes
|
|
|
|
def runners_from_xml_entries(xml_file):
|
|
tree = ET.parse(xml_file)
|
|
root = tree.getroot()
|
|
|
|
url = '{http://www.orienteering.org/datastandard/3.0}'
|
|
runners = []
|
|
person_entries = root.findall(f'./{url}PersonEntry')
|
|
for p_entry in person_entries:
|
|
|
|
rid = p_entry[1][0].text
|
|
|
|
person = p_entry.find(f'./{url}Person')
|
|
name = person.find(f'./{url}Name')
|
|
first = name.find(f'./{url}Given').text
|
|
last = name.find(f'./{url}Family').text
|
|
|
|
organisation = p_entry.find(f'./{url}Organisation')
|
|
if organisation is not None:
|
|
club_id = organisation.find(f'./{url}Id').text
|
|
club_name = organisation.find(f'./{url}Name').text
|
|
club_name_short = organisation.find(f'./{url}ShortName').text
|
|
country = organisation.find(f'./{url}Country').attrib['code']
|
|
else:
|
|
club_id = club_name = club_name_short = country = None
|
|
|
|
class_el = p_entry.find(f'./{url}Class')
|
|
class_str = class_el.find(f'./{url}Name').text
|
|
|
|
fee_id = int(p_entry.find(f'./{url}AssignedFee/{url}Fee/{url}Id').text)
|
|
|
|
try:
|
|
card = int(p_entry.find(f'./{url}ControlCard').text)
|
|
except AttributeError:
|
|
card = None
|
|
|
|
start_time = None
|
|
runners.append(Runner(rid, first, last, club=club_name, club_id=club_id,
|
|
country=country,card=card, o_class_str=class_str,
|
|
start_time=start_time, fee_id=fee_id))
|
|
return runners
|
|
|
|
def fees_from_xml_entries(xml_file):
|
|
tree = ET.parse(xml_file)
|
|
root = tree.getroot()
|
|
|
|
url = '{http://www.orienteering.org/datastandard/3.0}'
|
|
allfees = root.findall(f'.//{url}Fee')
|
|
added_ids = []
|
|
fee_objs = []
|
|
for fee in allfees:
|
|
f_id = int(fee.find(f'./{url}Id').text)
|
|
if f_id not in added_ids:
|
|
added_ids.append(f_id)
|
|
|
|
fee_id = f_id
|
|
name = fee.find(f'./{url}Name').text
|
|
currency = fee.find(f'./{url}Amount').attrib['currency']
|
|
amount = int(fee.find(f'./{url}Amount').text)
|
|
try:
|
|
from_birth_date = fee.find(f'./{url}FromDateOfBirth').text
|
|
except AttributeError:
|
|
from_birth_date = None
|
|
try:
|
|
to_birth_date = fee.find(f'./{url}ToDateOfBirth').text
|
|
except AttributeError:
|
|
to_birth_date = None
|
|
|
|
fee_objs.append(Fee(fee_id, name, currency, amount,
|
|
from_birth_date=from_birth_date, to_birth_date=to_birth_date))
|
|
return fee_objs
|
|
|
|
def courses_from_xml(xml_file):
|
|
tree = ET.parse(xml_file)
|
|
root = tree.getroot()
|
|
|
|
url = '{http://www.orienteering.org/datastandard/3.0}'
|
|
allcourses = root.findall(f'.//{url}Course')
|
|
courseobjs = []
|
|
for c in allcourses:
|
|
name = c.find(f'./{url}Name').text
|
|
controls = []
|
|
allcontrols = c.findall(f'./{url}CourseControl')
|
|
for n in allcontrols:
|
|
controls.append(n.find(f'./{url}Control').text)
|
|
controls.remove('STA1')
|
|
controls.remove('FIN1')
|
|
controls = [int(l) for l in controls]
|
|
courseobjs.append(Course(name, controls))
|
|
return courseobjs
|
|
|
|
# Checks if small list is in big list
|
|
def contains(small, big):
|
|
valid = True
|
|
mark = 0
|
|
map_bl = []
|
|
for i in small:
|
|
for n, control in enumerate(big[mark:]):
|
|
if i == control:
|
|
mark += n
|
|
map_bl.append(mark)
|
|
break
|
|
else:
|
|
valid = False
|
|
if valid:
|
|
return map_bl
|
|
else:
|
|
return False
|
|
|
|
|
|
def get_runners_in_class(runners, o_class):
|
|
# Filters out runner objects that dont have the correct o_class
|
|
list_filtrd = []
|
|
for i in runners:
|
|
if i.o_class == o_class:
|
|
list_filtrd.append(i)
|
|
return list_filtrd
|
|
|
|
|
|
def rank_runners(allrunners, o_class):
|
|
runners = get_runners_in_class(allrunners, o_class)
|
|
runners_ranked = []
|
|
for i in runners:
|
|
if i.status() == 'OK':
|
|
runners_ranked.append(i)
|
|
runners_ranked.sort(key=lambda x: x.totaltime())
|
|
return runners_ranked
|
|
|
|
|
|
# Used to make creating xml files easier
|
|
def xml_child(parent, tag, content):
|
|
e = ET.SubElement(parent, tag)
|
|
e.text = str(content)
|
|
|
|
|