Bring over refactoring and start of unit tests

sumpfork · Nov 21, 2015 · 0060cf3 · 0060cf3
1 parent 6aa9c28
commit 0060cf3
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/convert_csv.py b/convert_csv.py
@@ -0,0 +1,33 @@
+# coding=utf-8
+import csv
+import sys
+import json
+import chardet
+
+converted = []
+
+typemap = {'Aktion': 'Action',
+           'Geld': 'Treasure',
+           'Fluch': 'Curse',
+           'Punkte': 'Victory',
+           'Reaktion': 'Reaction',
+           'Angriff': 'Attack',
+           'Dauer': 'Duration'}
+
+with open(sys.argv[1], 'U') as csvfile:
+    reader = csv.DictReader(csvfile)
+    for row in reader:
+        if not ''.join(row.itervalues()):
+            continue
+        # row = {k: v.decode('ISO-8859-2').encode('utf-8') if type(v) == str else v for k, v in row.iteritems()}
+        print chardet.detect(row['Kartentext'])
+        print row['Kartentext']
+        converted_row = {'name': row['Kartenname'],
+                         'cost': row['Kosten'],
+                         'cardset': row['Edition'],
+                         'description': row['Kartentext'],
+                         'extra': row['Lange Erklärung'],
+                         'types': [typemap[t.strip()] for t in row['Typ'].split('/')],
+                         'potcost': row.get('potcost', 0)}
+        converted.append(converted_row)
+json.dump(converted, open(sys.argv[2], 'wb'), indent=True)