otime-testing/otime.py

135 lines
4.2 KiB
Python

import datetime
import csv
import sqlite3
import re
from rich.console import Console
from rich.table import Table
from rich.traceback import install
import xml.etree.ElementTree as ET
install(show_locals=True)
class runner:
def __init__(self, first, last, club, country, card, o_class, controls, splits):
self.first = first
self.last = last
self.club = club
self.country = country
self.card = card
self.o_class = o_class
self.controls = controls
self.splits = splits
def fullname(self):
return '{} {}'.format(self.first, self.last)
# Returns False if not ok and touple if ok
def check_codes(self):
return contains(self.o_class.course.codes, self.controls)
def totaltime(self):
return self.splits[-2]
class course:
def __init__(self, name, codes):
self.name = name
self.codes = codes
class o_class:
def __init__(self, name, course):
self.name = name
self.course = course
def courses_from_ttime_conf(ttime_file = 'sc_2021_ttime/ttime.cnf.txt'):
courses = []
conf = open(ttime_file).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).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 xml_to_class(xml_file, o_class_list=[]):
tree = ET.parse(xml_file)
root = tree.getroot()
runnerarray = []
for person_root in root[1:]:
first = person_root[1][1][1].text
last = person_root[1][1][0].text
try:
club = person_root[2][1].text
except:
club = "None"
country = person_root[1][3].text
try:
card = int(person_root[3].text)
except:
card = 0
try:
xml_class_str = person_root[4][1].text
except:
# VELDIG MIDLERTIDIG
runner_o_class = o_class_list[0]
else:
for i in o_class_list:
if i.name == xml_class_str:
runner_o_class = i
break
# Gjør sånn at den lager nye o klasser om den ikke finnes fra før
runnerarray.append(runner(first, last, club, country, card, runner_o_class, [], []))
return runnerarray
def ttime_mtr_to_class(csv_file, runnerarray):
csvreader = csv.reader(open(csv_file))
fields = next(csvreader)
fields = []
rows = []
# hver rad er brikkenummer med tilhørende info
for row in csvreader:
rows.append(row)
for row in rows:
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 != '000' and item != '0000' and item != '00000':
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 != '000' and item != '0000' and item != '00000':
splits.append(int(item))
# looper gjonnom løperobjektene og legger til poster og strekktider
for runner in runnerarray:
if runner.card == int(row[6]):
runner.controls = controls
runner.splits = splits
def contains(small, big):
for i in range(len(big)-len(small)+1):
for j in range(len(small)):
if big[i+j] != small[j]:
break
else:
return i, i+len(small)
return True
return False