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') # class_result = ET.SubElement(root, 'ClassResult') # t = ET.SubElement(class_result, 'Class') xml_child(t, 'Name', i.name) # 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 = ET.SubElement(person_result, 'Person') xml_child(person, 'Id', n.id) # name = ET.SubElement(person, 'Name') xml_child(name, 'Family', n.last) xml_child(name, 'Given', n.first) # # # 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') # # 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': # xml_child(result, 'TimeBehind', n.totaltime() - runners_ranked[0].totaltime()) # xml_child(result, 'Position', n.rank(self.runners)) xml_child(result, 'Status', n.status()) # # 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 # 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()) # # # 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)