#!/usr/bin/env python3

import csv
import mxl_types
from argparse import ArgumentParser
from pathlib import Path
from sqlalchemy import create_engine, text
from sqlalchemy.orm import Session


def generate_tcs():
	# TODO
	# grab weapons, armor, misc? with spawnable=1 *and* who have itemtype with TreasureClass=1 (any itemtype? just direct?)
	# qlvl 1-3 in weap3, 4-6 in weap6, 7-9 in weap9 ... 91-93 in weap93
	# armo3 ... armo99
	# vanilla has bow3 ... bow87 as well, unknown but probably based on the itemtype having Shoots and Quiver
	# don't know if misc TCs are generated, neither median nor vanilla actually uses them if they are
	pass


def create_db(folder):
	engine = create_engine("sqlite+pysqlite:///:memory:", echo=False)
	mxl_types.BaseDAO.metadata.create_all(engine)

	try1 = (Path(folder) / "Data" / "global" / "excel")
	if try1.exists():
		data_folder = try1
	else:
		data_folder = folder

	types = [
		mxl_types.ItemRatio,
		mxl_types.ItemType,
		mxl_types.Weapon,
		mxl_types.Armor,
		mxl_types.Misc,
		mxl_types.UniqueItem,
		mxl_types.SetItem,
		mxl_types.TreasureClass,
	]
	for mytype in types:
		file = data_folder / mytype.filename
		with file.open(encoding='windows-1252') as csvfile, Session(engine) as session:
			reader = csv.DictReader(csvfile, dialect='excel-tab')
			for csv_row in reader:
				db_row = mytype(csv_row)
				if not db_row.is_valid():
					continue
				session.add(db_row)
			session.commit()
			count = session.scalar(text(f'select count(*) from {mytype.__tablename__}'))
			print(f'Counted {count} {mytype.__tablename__}')


def handle_args():
	parser = ArgumentParser()
	parser.add_argument('folder', help='Location of Median XL .txt files')
	return parser.parse_args()


def main():
	args = handle_args()
	create_db(args.folder)


if __name__ == '__main__':
	import sys

	sys.exit(main())