mlpccg/Tournament.py

   1 import csv
   2 import sqlite3
   3
   4 class Tournament:
   5     def __init__(self):
   6         #self.from_sqlite('data/db.sqlite3')
   7         self.from_csv('data/tournaments.csv')
   8
   9     def from_sqlite(self, sqlite_path):
  10         conn = sqlite3.connect(sqlite_path)
  11         conn.row_factory = sqlite3.Row
  12
  13         c = conn.cursor()
  14         for row in c.execute("SELECT strftime('%s', date) - 0 AS t, * FROM tournaments_record"):
  15             self.records += [dict(zip(row.keys(), row))]
  16         conn.close()
  17
  18     def from_csv(self, csv_path):
  19         with open(csv_path) as f:
  20             reader = csv.DictReader(f)
  21             for row in reader:
  22                 pass #row['date']
  23             self.records = [row for row in reader]
  24
  25     def add(self, record):
  26         self.records += [record]
  27
  28     def all(self):
  29         return self.records