forked from Trygve/otime
135 lines
4.2 KiB
Python
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
|
|
|
|
|