forked from Trygve/otime
340 lines
12 KiB
Python
340 lines
12 KiB
Python
import copy
|
|
import datetime
|
|
import csv
|
|
import re
|
|
import xml.etree.ElementTree as ET
|
|
|
|
class event:
|
|
def __init__(self, eventid, name, **kwargs):
|
|
self.id = eventid
|
|
self.name = name
|
|
|
|
self.courses = []
|
|
self.o_classes = []
|
|
self.runners = []
|
|
self.card_punches= []
|
|
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 import_ttime_cnf(self, ttime_file):
|
|
self.add_course(*courses_from_ttime_conf(ttime_file))
|
|
self.add_o_class(*classes_from_ttime_conf(ttime_file, self.courses))
|
|
|
|
def import_ttime_db(self, ttime_file):
|
|
csvreader = csv.reader(open(ttime_file, 'r', encoding='latin_1'), delimiter=';',)
|
|
runnerarray = []
|
|
for row in csvreader:
|
|
if len(row) == 0 or row [1] == '':
|
|
continue
|
|
eventorid = row[0]
|
|
country = ''
|
|
name = row[2].split(',')
|
|
try:
|
|
first = name[1].strip()
|
|
except:
|
|
first = ''
|
|
last = name[0]
|
|
try:
|
|
club = row[4]
|
|
except:
|
|
club = "None"
|
|
try:
|
|
card = int(row[6])
|
|
except:
|
|
card = 0
|
|
try:
|
|
raw_class_str = row[3]
|
|
except:
|
|
# VELDIG MIDLERTIDIG
|
|
runner_o_class = None
|
|
else:
|
|
if raw_class_str != '':
|
|
for i in self.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
|
|
runnerarray.append(runner(eventorid, first, last, club, country, card, runner_o_class, [], []))
|
|
self.runners = runnerarray
|
|
def import_mtr_file(self, mtr_file):
|
|
self.card_punches = card_punch.list_from_mtr_f(mtr_file)
|
|
def match_runners_cards(self):
|
|
for n in self.runners:
|
|
for i in self.card_punches:
|
|
if n.card == i.card:
|
|
n.card_r = i
|
|
|
|
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:
|
|
#<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', '0')
|
|
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')
|
|
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>
|
|
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)
|
|
#</SplitTime>
|
|
elif n.status() == 'Disqualified':
|
|
xml_child(result, 'Status', n.status())
|
|
for code in n.o_class.course.codes:
|
|
st = ET.SubElement(result, 'SplitTime')
|
|
xml_child(st, 'ControlCode', code)
|
|
for control,split in zip(n.card_r.controls,n.card_r.splits):
|
|
if code == control:
|
|
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
|
|
class runner:
|
|
def __init__(self, eventorid, first, last, club, country, card, o_class, controls, splits):
|
|
self.id = eventorid
|
|
self.first = first
|
|
self.last = last
|
|
self.club = club
|
|
self.country = country
|
|
self.card = card
|
|
self.o_class = o_class
|
|
|
|
def fullname(self):
|
|
return '{} {}'.format(self.first, self.last)
|
|
def check_codes(self):
|
|
# Returns False if not ok and touple if ok
|
|
return contains(self.o_class.course.codes, list(self.card_r.controls))
|
|
def totaltime(self):
|
|
f_control = self.o_class.course.codes[-1]
|
|
try:
|
|
index = self.card_r.controls.index(f_control)
|
|
return self.card_r.splits[index]
|
|
except:
|
|
return self.card_r.splits[-1]
|
|
|
|
def status(self):
|
|
if hasattr(self, 'card_r') == False or self.o_class == None:
|
|
return 'Active'
|
|
elif self.check_codes():
|
|
return 'OK'
|
|
elif self.check_codes() == False:
|
|
return 'Disqualified'
|
|
def rank(self, allrunners):
|
|
c_ranked = rank_runners(allrunners, self.o_class)
|
|
return c_ranked.index(self) + 1
|
|
def res_splits(self):
|
|
splits_cpy = self.card_r.splits.copy()
|
|
for control in self.card_r.controls:
|
|
if control not in self.o_class.course.codes:
|
|
index = self.card_r.controls.index(control)
|
|
split = self.card_r.splits[index]
|
|
splits_cpy.remove(split)
|
|
return splits_cpy
|
|
|
|
class card_punch:
|
|
def __init__(self, card, controls, splits, s_time, f_time):
|
|
self.card = card
|
|
self.controls = controls
|
|
self.splits = splits
|
|
self.s_time = s_time
|
|
self.f_time = f_time
|
|
def __repr__(self):
|
|
return f'card({self.card}) punches({self.punches})'
|
|
|
|
def list_from_mtr_f(mtr_f):
|
|
csvreader = csv.reader(open(mtr_f))
|
|
fields = next(csvreader)
|
|
|
|
fields = []
|
|
rows = []
|
|
|
|
cards = []
|
|
# hver rad er brikkenummer med tilhørende info
|
|
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]))
|
|
f_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:
|
|
s_time = f_time - datetime.timedelta(seconds = splits[-2])
|
|
else:
|
|
s_time = f_time
|
|
cards.append(card_punch(int(row[6]), controls, splits, s_time, f_time))
|
|
return cards
|
|
class course:
|
|
def __init__(self, name, codes):
|
|
self.name = name
|
|
self.codes = codes
|
|
def __repr__(self):
|
|
return f'name({self.name})'
|
|
|
|
class o_class:
|
|
def __init__(self, name, course):
|
|
self.name = name
|
|
self.course = course
|
|
def __repr__(self):
|
|
return f'name({self.name})'
|
|
|
|
def courses_from_ttime_conf(ttime_file = 'sc_2021_ttime/ttime.cnf.txt'):
|
|
courses = []
|
|
conf = open(ttime_file, 'r', encoding='latin_1').readlines()
|
|
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):
|
|
o_classes = []
|
|
conf = open(ttime_file, 'r', encoding='latin_1').readlines()
|
|
for line in conf:
|
|
if '-courses' in line:
|
|
raw_courselist = re.search(r'(?<=\")(.*?)(?=\")', line).group().split(';')
|
|
loops = 0
|
|
for n in raw_courselist:
|
|
n = n.split(',')
|
|
for i in n:
|
|
o_classes.append(o_class(i,courses[loops]))
|
|
loops += 1
|
|
|
|
|
|
return o_classes
|
|
|
|
|
|
|
|
def contains(small, big):
|
|
# Checks if small list is in big list
|
|
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:
|
|
marker = 0
|
|
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
|
|
|
|
def xml_child(parent, tag, content):
|
|
# Used to make creating xml files easier
|
|
e = ET.SubElement(parent, tag)
|
|
e.text = str(content)
|