main.py

from generator.join_gen import *
from generator.ddl_gen import *
from generator.data_gen import *
from config.config import *

import signal
import sys

def _signal_handler(sig, frame):
    sys.exit(0)

def _interactiveMode(ddls, data):
    print(ddls)
    print(data)

    while True:
        res = input('Press x to exit or Enter to generate more')
        if res.lower() == 'x':
            print('Exit from query generation mode!')
            sys.exit(0)

        query = newQuery()
        print(query)

        resetConfig()

def _fileMode(ddls, data):
    ddlFileName = 'out/' + getConfig().ddlOutFile
    with open(ddlFileName, 'w') as ddlFile:
        ddlFile.writelines([ddls, data])

    queryCount = getConfig().queryCount
    fileName = 'out/' + getConfig().queryOutFile
    with open(fileName, 'w') as f:
        queryLines = []
        for _ in range(queryCount):
            query = newQuery()
            queryLine = query + '\n\n'
            queryLines.append(queryLine)
        f.writelines(queryLines)

if __name__ == '__main__':
    signal.signal(signal.SIGINT, _signal_handler)

    resetConfig()

    ddls = getTableDDLs()
    data = getTableData()

    if getConfig().interactiveMode:
        _interactiveMode(ddls, data)
    else:
        _fileMode(ddls, data)