try out django

[mlpccg-meta.git] / mlpccg / Tournament.py

diff --git a/mlpccg/Tournament.py b/mlpccg/Tournament.py
new file mode 100644 (file)
index 0000000..94723eb
--- /dev/null
+++ b/mlpccg/Tournament.py
@@ -0,0 +1,29 @@
+import csv
+import sqlite3
+
+class Tournament:
+    def __init__(self):
+        #self.from_sqlite('data/db.sqlite3')
+        self.from_csv('data/tournaments.csv')
+
+    def from_sqlite(self, sqlite_path):
+        conn = sqlite3.connect(sqlite_path)
+        conn.row_factory = sqlite3.Row
+
+        c = conn.cursor()
+        for row in c.execute("SELECT strftime('%s', date) - 0 AS t, * FROM tournaments_record"):
+            self.records += [dict(zip(row.keys(), row))]
+        conn.close()
+
+    def from_csv(self, csv_path):
+        with open(csv_path) as f:
+            reader = csv.DictReader(f)
+            for row in reader:
+                pass #row['date']
+            self.records = [row for row in reader]
+
+    def add(self, record):
+        self.records += [record]
+
+    def all(self):
+        return self.records