blob: de0eb7a5943f7e179105ff38fe0af09134d75ab2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
from f1elo.db import Session, find_driver
from f1elo.model import *
import sys
session = Session()
import csv
with open(sys.argv[1]) as f:
reader = csv.reader(f)
for row in reader:
driver = None
if len(row) == 6:
entry = Entry()
entry._race = row[0]
entry.result = row[1]
entry.car_no = row[2]
entry.result_group = row[5]
session.add(entry)
driver = find_driver(row[4].strip(), row[3].strip(), session)
entry.drivers.append(driver)
elif len(row) == 2:
driver = find_driver(row[1].strip(), row[0].strip(), session)
entry.drivers.append(driver)
else:
print row
session.commit()
|
