otime/event_mgr.py

116 lines
3.5 KiB
Python
Raw Normal View History

2022-03-13 20:19:07 +00:00
#!/usr/bin/env python
2022-03-11 14:33:09 +00:00
import otime
2022-03-23 16:54:48 +00:00
#import mtr_log_extractor as mle
import json
2022-03-23 16:54:48 +00:00
import pickle
2022-03-13 16:33:26 +00:00
import asyncio
2022-03-23 16:54:48 +00:00
import argparse
2022-03-13 20:19:07 +00:00
from difflib import Differ
2022-03-13 16:33:26 +00:00
from watchgod import awatch
2022-03-11 14:33:09 +00:00
from config import config
from rich import print
from rich import inspect
from rich.console import Console
from rich.columns import Columns
from rich.table import Table
2022-03-23 16:54:48 +00:00
def save(obj):
f = open(config['otime_file'], "wb")
pickle.dump(obj,f)
f.close()
def start_parse():
parser = argparse.ArgumentParser(description='Event manager')
subparsers = parser.add_subparsers(dest='command')
open_event = subparsers.add_parser('continue', help='Open pickle')
open_event.add_argument('--file', help='.otime file')
args = parser.parse_args()
if args.command == 'continue':
print('Later ' + args.file)
else:
print('Starter fra scratch')
start_event()
def start_event():
global event
2022-03-11 14:33:09 +00:00
event = otime.event(0, config['event_name'])
2022-03-23 16:54:48 +00:00
event.import_ttime_cnf('/home/trygve/Dokumenter/sprintcup040330/sprintcup040330.cnf')
2022-03-11 14:33:09 +00:00
event.import_ttime_db(config['db_file'])
2022-03-23 16:54:48 +00:00
event.import_mtr_file('/home/trygve/Dokumenter/sprintcup040330/sprintcup040330.log')
2022-03-11 14:33:09 +00:00
event.match_runners_cards()
2022-03-23 16:54:48 +00:00
save(event)
global db_file
global db_file_u
db_file = open(config['db_file'], 'r', encoding='latin_1').read().splitlines()
def load_event(ot_file):
global event
f = open(ot_file, "r")
event = pickle.load(f)
f.close()
global db_file
global db_file_u
db_file = open(config['db_file'], 'r', encoding='latin_1').read().splitlines()
2022-03-11 14:33:09 +00:00
2022-03-13 16:33:26 +00:00
async def main():
async for changes in awatch(config['db_file']):
global event
global db_file
global db_file_u
print(len(event.runners))
db_file_u = open(config['db_file'], 'r', encoding='latin_1').read().splitlines()
2022-03-13 20:19:07 +00:00
d = Differ()
result = list(d.compare(db_file, db_file_u))
db_file = db_file_u
2022-03-13 20:19:07 +00:00
added_raw = []
removed_raw = []
added = []
removed = []
changed = []
2022-03-13 20:19:07 +00:00
for line in result:
if line[:1] == '+':
#print(line)
clean = str(line).replace('+ ', '').split(';')
runner = otime.runner.from_string(clean, event.o_classes)
added_raw.append(runner)
elif line[:1] == '-':
#print(line)
clean = str(line).replace('- ', '').split(';')
runner = otime.runner.from_string(clean, event.o_classes)
removed_raw.append(runner)
for plus in added_raw:
for index, minus in enumerate(removed_raw):
if plus.id == minus.id:
changed.append(plus)
removed_raw[index] = plus
break
added = [x for x in added_raw if x not in changed]
removed = [x for x in removed_raw if x not in changed]
#print(len(added_raw),len(removed_raw))
#print(len(added),len(removed),len(changed))
event.runners.extend(added)
for r in removed:
for n in event.runners:
if n.id == r.id:
event.runners.remove(n)
for change in changed:
for i, n in enumerate(event.runners):
if n.id == runner.id:
event.runners[i] = runner
event.match_runners_cards()
print(len(event.runners))
event.get_xml_res().write(config['xml_res_file'])
2022-03-13 20:19:07 +00:00
db_file_u = None
2022-03-11 14:33:09 +00:00
if __name__ == "__main__":
2022-03-23 16:54:48 +00:00
start_parse()
2022-03-13 16:33:26 +00:00
asyncio.run(main())