cli.py

@@ -1,11 +1,7 @@
-#!/usr/bin/env python
 import otime
 
-import argparse
 import datetime
-from rich import print
 from rich.console import Console
-from rich.columns import Columns
 from rich.table import Table
 def print_runners(Runners):
     table = Table(title="Runners")
@@ -34,48 +30,21 @@ def print_time(Runners):
         tottime = 0
       table.add_row(i.fullname(), i.o_class.name, str(i.splits), str(i.check_codes()), str(tottime))
 
-    console = Console()
-    console.print(table)
-def print_class_result(runners, o_class):
-    table = Table(title=o_class.name)
-    table.add_column("Pos", style="red", no_wrap=True)
-    table.add_column("Name", justify="right", style="cyan", no_wrap=True)
-    table.add_column("Club", style="magenta")
-    table.add_column("Time", justify="right", style="blue")
-    pos = 0
-    if o_class:
-        runners = otime.get_runners_in_class(runners, o_class)
-        runners_ok = []
-        for n in runners:
-            if n.status() == 'OK':
-                runners_ok.append(n)
-        runners_ok.sort(key=lambda x: x.totaltime())
-    for i in runners_ok:
-        #if i.status() == 'OK':
-        	pos += 1
-        	table.add_row(str(pos)+'.',i.fullname(), i.status(), str(datetime.timedelta(seconds=i.totaltime())))
-
     console = Console()
     console.print(table)
 def print_class_splits(runners, o_class):
     table = Table(title=o_class.name)
-    table.add_column("Pos", style="red", no_wrap=True)
     table.add_column("Name", justify="right", style="cyan", no_wrap=True)
     table.add_column("club", style="magenta")
-    if o_class:
+    table.add_column("card", style="red")
+    table.add_column("class", justify="right", style="green")
+    table.add_column("controls", justify="right", style="blue")
     for i in o_class.course.codes:
         table.add_column(str(i))
-        runners = otime.get_runners_in_class(runners, o_class)
+    
-        runners_ok = []
+    for i in runners:
-        for n in runners:
+      if i.o_class == o_class:
-            if n.status() == 'OK':
+          table.add_row(i.fullname(), i.club ,str(i.card))
-                runners_ok.append(n)
-        runners_ok.sort(key=lambda x: x.totaltime())
-    pos = 0
-    for i in runners_ok:
-        pos += 1
-        list_string = map(lambda x:str(datetime.timedelta(seconds=x)), i.res_splits())
-        table.add_row(str(pos)+'.', i.fullname(), i.club, *list_string)
 
     console = Console()
     console.print(table)
@@ -91,72 +60,15 @@ def print_o_classes(class_list):
     console = Console()
     console.print(table)
     
-def main():
-    parser = argparse.ArgumentParser()
-    parser.add_argument(
-        '--debug',
-        action='store_true',
-        help='Print debug info'
-    )
-    subparsers = parser.add_subparsers(dest='command')
-    show_runners = subparsers.add_parser('show_runners', help='show a table of runners')
-    show_runners.add_argument('--ttcnf', required=True, help='ttime configuration file')
-    show_runners.add_argument('--ttdb', required=True, help='ttime database file')
-    show_runners.add_argument('--mtr', required=True, help='mtr csv file')
-
-    show_runners = subparsers.add_parser('show_result', help='show a table of runners')
-    show_runners.add_argument('--ttcnf', required=True, help='ttime configuration file')
-    show_runners.add_argument('--ttdb', required=True, help='ttime database file')
-    show_runners.add_argument('--mtr', required=True, help='mtr csv file')
-    show_runners.add_argument('--class', dest='o_class_str', help='Which class to show. Prints all classes if not set')
-    show_runners.add_argument('--splits', action='store_true', help='Shows split times')
-
-    show_classes = subparsers.add_parser('show_classes', help='show a table of classes')
-    show_classes.add_argument('--ttcnf', required=True, help='ttime configuration file')
-    show_classes.add_argument('--ttdb', required=True, help='ttime db files')
-    show_classes.add_argument('--mtr', required=True, help='mtr csv file')
-
-    create_xml = subparsers.add_parser('create_xml', help='Create xml result file')
-    create_xml.add_argument('--ttcnf', required=True, help='ttime configuration file')
-    create_xml.add_argument('--ttdb', required=True, help='ttime database file')
-    create_xml.add_argument('--mtr', required=True, help='mtr csv file')
-    create_xml.add_argument('--file', required=True, help='Filename for result file')
-
-    args = parser.parse_args()
-    if args.command == 'show_runners':
-        courses = otime.courses_from_ttime_conf(args.ttcnf)
-        o_classes = otime.classes_from_ttime_conf(args.ttcnf, courses)
-        runner_list = otime.ttime_db_to_class(args.ttdb, o_classes)
-        otime.ttime_mtr_to_class(args.mtr, runner_list)
-        print_runners(runner_list)
-    if args.command == 'show_result':
-        courses = otime.courses_from_ttime_conf(args.ttcnf)
-        o_classes = otime.classes_from_ttime_conf(args.ttcnf, courses)
-        runner_list = otime.ttime_db_to_class(args.ttdb, o_classes)
-        otime.ttime_mtr_to_class(args.mtr, runner_list)
-        for n in o_classes:
-            if args.o_class_str == n.name:
-                o_class = n
-                break
-        else:
-            print('Class not found')
-        if args.splits:
-            print_class_splits(runner_list, o_class)
-        else:
-            print_class_result(runner_list, o_class)
-    elif args.command == 'show_classes':
-        courses = otime.courses_from_ttime_conf(args.ttcnf)
-        o_classes = otime.classes_from_ttime_conf(args.ttcnf, courses)
-        runner_list = otime.ttime_db_to_class(args.ttdb, o_classes)
-        otime.ttime_mtr_to_class(args.mtr, runner_list)
-        print_o_classes(o_classes)
-
-    if args.command == 'create_xml':
-        courses = otime.courses_from_ttime_conf(args.ttcnf)
-        o_classes = otime.classes_from_ttime_conf(args.ttcnf, courses)
-        runner_list = otime.ttime_db_to_class(args.ttdb, o_classes)
-        otime.ttime_mtr_to_class(args.mtr, runner_list)
-        otime.gen_xml_result(runner_list, o_classes).write(args.file)
-
 if __name__ == "__main__":
-    main()
+  courses = otime.courses_from_ttime_conf('sc_2021_ttime/ttime.cnf.txt')
+  o_classes = otime.classes_from_ttime_conf('sc_2021_ttime/ttime.cnf.txt', courses)
+  runner_list = otime.ttime_db_to_class('sc_2021_ttime/db_eventor.csv', o_classes)
+  #runner_list = otime.xml_to_class('entries_KOK_Sommercup,_løp_2.xml', o_classes)
+  otime.ttime_mtr_to_class('sc_2021_ttime/mtr.csv', runner_list)
+  #print_runners(runner_list)
+  #print_time(runner_list)
+  #print_o_classes(o_classes)
+  #print_class_splits(runner_list, o_classes[0])
+  #print(otime.check_codes(runner_list[1]))
+  otime.gen_xml_result(runner_list, o_classes)

otime.py

@@ -1,10 +1,13 @@
 import datetime
 import csv
 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, eventorid, first, last, club, country, card, o_class, controls, splits):
+    def __init__(self, first, last, club, country, card, o_class, controls, splits):
-      self.id = eventorid
       self.first = first
       self.last = last
       self.club = club
@@ -13,6 +16,7 @@ class runner:
       self.o_class = o_class
       self.controls = controls
       self.splits = splits 
+      self.id = 0
       self.s_time = 0
       self.f_time = 0
 
@@ -61,7 +65,7 @@ class o_class:
 
 def courses_from_ttime_conf(ttime_file = 'sc_2021_ttime/ttime.cnf.txt'):
     courses = []
-    conf = open(ttime_file, 'r', encoding='latin_1').readlines()
+    conf = open(ttime_file).readlines()
     for line in conf:
         if '-codes' in line:
             code_list = re.search(r'(?<=\")(.*?)(?=\")', line).group().split(';')
@@ -74,7 +78,7 @@ def courses_from_ttime_conf(ttime_file = 'sc_2021_ttime/ttime.cnf.txt'):
     return courses
 def classes_from_ttime_conf(ttime_file, courses):
     o_classes = []
-    conf = open(ttime_file, 'r', encoding='latin_1').readlines()
+    conf = open(ttime_file).readlines()
     for line in conf:
         if '-courses' in line:
             raw_courselist = re.search(r'(?<=\")(.*?)(?=\")', line).group().split(';')
@@ -122,7 +126,6 @@ def ttime_db_to_class(ttime_file, o_class_list=[]):
     csvreader = csv.reader(open(ttime_file, 'r', encoding='latin_1'), delimiter=';',)
     runnerarray = []
     for row in csvreader:
-        eventorid = row[0]
         country = ''
         name = row[2].split(',')
         first = name[1]
@@ -146,7 +149,7 @@ def ttime_db_to_class(ttime_file, o_class_list=[]):
                     runner_o_class = i
                     break
                 # 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, [], []))
+        runnerarray.append(runner(first, last, club, country, card, runner_o_class, [], []))
     return runnerarray
 
 def ttime_mtr_to_class(csv_file, runnerarray): 
@@ -229,10 +232,6 @@ def xml_child(parent, tag, content):
 
 def gen_xml_result(runners, o_classes):
     root = ET.Element('ResultList')
-    root.set('iofVersion', '3.0')
-    root.set('createTime', datetime.datetime.now().isoformat())
-    root.set('creator', 'oTime')
-    root.set('status', 'Complete')
     tree = ET.ElementTree(root)
     event = ET.SubElement(root, 'Event')
     xml_child(event, 'Name', 'PLACEHOLDER')
@@ -249,7 +248,7 @@ def gen_xml_result(runners, o_classes):
             person_result = ET.SubElement(class_result, 'PersonResult')
             #<Person>
             person = ET.SubElement(person_result, 'Person')
-            xml_child(person, 'Id', n.id)
+            xml_child(person, 'Id', '0')
             #<Name>
             name = ET.SubElement(person, 'Name')
             xml_child(name, 'Family', n.last)
@@ -278,19 +277,11 @@ def gen_xml_result(runners, o_classes):
                     xml_child(result, 'Position', n.rank(runners))
                     xml_child(result, 'Status', n.status())
                     #<SplitTime>
-                    for code,split in zip(n.o_class.course.codes, n.res_splits(), strict=True):
+                    for code,split in zip(n.o_class.course.codes, n.res_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.controls, n.splits):
-                                if code == control:
-                                    xml_child(st, 'Time', split)
                 else:
                     xml_child(result, 'Status', n.status())
             else:
@@ -298,6 +289,6 @@ def gen_xml_result(runners, o_classes):
             #</Result>
         #</PersonResult>
         #</Class>
-    ET.indent(root, space=' ', level=0)
+
     tree.write('Resultater.xml') 
-    return tree
+    return ET.dump(root)