import os
import glob
from abstractboard import get_gameinfo_from_file, apply_gameinfo_to_sgfmodel
from sgfmodels import Collection, Sgf, CollectionSgf, db
default_cols = ['Pro10','IMeijin','Kisei','ITengen','Gosei','Honinbo','Shusaku','Meijin','CJSuperGo','Tengen','Judan','Oza','NihonKiin','Ing']
#collectionslist = CollectionsList()
if not Sgf.table_exists():
Sgf.create_table()
if not Collection.table_exists():
Collection.create_table()
if not CollectionSgf.table_exists():
CollectionSgf.create_table()
for entry in default_cols:
sgfs = glob.glob('./games/' + entry + '/*.sgf')
sgfmodels = []
collections = []
colsgfmodels = []
for sgf in sgfs:
with open(sgf) as fileh:
sgfstr = fileh.read()
model = Sgf()
model.filename = sgf
model.sgf = sgfstr
colmod = list(Collection.select().where(Collection.name == entry))
if len(colmod) == 0:
