main.py

from db.QuestList import *
from db.NpcList import *
from db.ObjList import *
from db.CoordList import *
from db.ItemList import *
from preExtract.CoordPreExtract import printCoordFiles

import sys
import pymysql
import config

import time

version = config.version
debug = config.debug

if version not in ['classic', 'tbc', 'wotlk']:
    print(f'Unknown version {version}')
    sys.exit(1)


def getClassInstances(recache=False):
    """Get new instances of the list classes"""
    quests = QuestList(cursor, dictCursor, version, recache=recache)
    npcs = NpcList(cursor, dictCursor, version, recache=recache, debug=debug)
    obj = ObjList(cursor, dictCursor, version, recache=recache)
    items = ItemList(dictCursor, version, recache=recache)
    return quests, npcs, obj, items

def recache():
    _, _, _, _ = getClassInstances(True)

def main(recache):
    """Extracts and prints quest related data"""
    quests, npcs, objects, items = getClassInstances(recache)
    print("Printing files...")
    quests.printQuestFile(f'output/{version}/{version}QuestDB.lua')
    npcs.printNpcFile(f'output/{version}/{version}NpcDB.lua')
    objects.printObjFile(f'output/{version}/{version}ObjectDB.lua')
    items.writeFile(f'output/{version}/{version}ItemDB.lua')
    print("Done.")
    return 0

def getCursors(v):
    connection = pymysql.connect(
        host=config.dbInfo['host'],
        user=config.dbInfo['user'],
        password=config.dbInfo['password'],
        database=config.dbInfo[v],
        port=config.dbInfo["port"],
        charset='utf8'
    )
    c = connection.cursor()
    dc = connection.cursor(pymysql.cursors.DictCursor)

    return c, dc

def preExtract(v):
    c, dc = getCursors(v)
    printCoordFiles(c, v)

# DB connection needs to be set up first and globally, but after CLI 
# arguments are checked, in case the argument differs from the config
# file, so the main function being run is delayed by use of this variable
runMain = False

reCache = False

if __name__ == "__main__":
    """Executes only if run as a script"""
    if len(sys.argv) > 1:
        for arg in sys.argv[1:]:
            if arg == '-r':
                reCache = True
            elif arg in ['classic', 'tbc', 'wotlk']:
                version = arg
            else:
                print(f'Unknown argument "{arg}"')
    print(f'Using version {version}')
    runMain = True

cursor, dictCursor = getCursors(version)

if runMain:
    start_time = time.time()
    main(reCache)
    print("--- %s seconds ---" % (time.time() - start_time))
else:
    print(f'Using version {version}')