Bring over refactoring and start of unit tests

This commit is contained in:
Sumpfork 2015-11-20 17:00:36 -08:00
parent 6aa9c28c80
commit 0060cf311d

33
convert_csv.py Normal file
View File

@ -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)