def purge(cur):
cur.execute(
'SELECT table_name FROM information_schema.tables WHERE table_type = \'BASE TABLE\' AND table_schema = \'public\';'
)
toDelete = cur.fetchall()
for table in toDelete:
rc.pcmd(' >>DROP TABLE ' + table[0] + ';')
cur.execute('DROP TABLE ' + table[0] + ';')
def genDat():
rc.pinfo('Reading directory setup files...')
dirfiles = []
for dirname, dirnames, filenames in os.walk('./setup/dir'):
for filename in filenames:
if not '.swp' in filename:
rc.pcmd(' >>found ' + dirname + '/' + filename)
dirfiles.append(dirname + '/' + filename)
for dirfile in dirfiles:
rc.pinfo(' in file ' + dirfile.strip())
with open(dirfile.strip()) as dirs:
for line in dirs:
line = line.strip()
rc.pinfo(' >>' + line)
os.makedirs(line, exist_ok=True)
def makeTable(tname, cur):
rc.pinfo('Parsing tables from ' + tname.strip())
with open(tname.strip()) as tables:
for table in tables:
if table.strip() == '':
continue
if (table.strip())[0] == '#':
continue
cur.execute(
'SELECT table_name FROM information_schema.tables WHERE table_type = \'BASE TABLE\' AND table_schema = \'public\' AND table_name = \''
+ table.strip().split('(')[0].strip().lower() + '\';')
if cur.fetchone() != None:
rc.pwarn(' >>Ignoring already-existing table ' + table.strip())
else:
rc.pcmd(' >>CREATE TABLE ' + table.strip() + ';')
cur.execute('CREATE TABLE ' + table.strip() + ';')
def updatePhrasebook(con):
cur = con.cursor()
try:
cur.execute('SELECT * FROM phrasebook;')
except psycopg2.ProgrammingError:
con.rollback()
rc.perr(
'It looks like the phrasebook table hasn\'t been created. Did you try initializing the database first?'
)
return
cur.execute('SELECT * FROM phrasebook;')
if not cur.fetchone() == None:
rc.pwarn(
'It looks like the phrasebook table is already populated. I will delete it and re-populate from sql/tables/phrasebook.tables.'
)
rc.pcmd(' >>DROP TABLE phrasebook;')
cur.execute('DROP TABLE phrasebook;')
makeTable('./sql/tables/phrasebook.tables', cur)
rc.pinfo('Table is initialized. Scanning for phrasefiles...')
phrasefiles = []
for dirname, dirnames, filenames in os.walk('./phrasebook/phrases'):
for filename in filenames:
if '.phrases' in filename and not '.swp' in filename:
rc.pcmd(' >>found ' + dirname + '/' + filename)
phrasefiles.append(dirname + '/' + filename)
rc.pinfo('Done. Reading phrasefiles and adding to database...')
for phrasefile in phrasefiles:
rc.pinfo('Parsing phrases from file ' + phrasefile.strip())
with open(phrasefile.strip()) as phrases:
jsonTranslation = json.load(phrases)
for module in jsonTranslation:
rc.pinfo(' In module ' + module["module"] + ':')
for context, cPList in module['phrases'].items():
rc.pinfo(' In context ' + context)
for phrase in cPList:
if len(phrase) > 25:
shortPhrase = phrase[:25] + '...'
else:
shortPhrase = phrase
rc.pcmd(' >>INSERT INTO phrasebook (' + context +
', ' + shortPhrase + ');')
cur.execute(
'INSERT INTO phrasebook (module, context, phrase) VALUES (%s, %s, %s);',
(module['module'], context, phrase))
cdb.commit()
rc.pinfo('Done.')
cdb = psycopg2.connect(dbname=rc.DBNAME,
user=rc.DBUSERNAME,
password=rc.DBPASSWORD,
host=rc.DBHOST,
port=rc.DBPORT)
module_tables = []
lightMode = False
rc.pinfo('Scanning table lists...')
for dirname, dirnames, filenames in os.walk('./sql/tables'):
for filename in filenames:
if '.tables' in filename and not '.swp' in filename:
rc.pcmd(' >>found ' + dirname + '/' + filename)
module_tables.append(dirname + '/' + filename)
rc.pinfo('Done.')
def initialize():
''' Initializes the database to have all of the proper tables '''
rc.pinfo('Preparing to initialize the database...')
cur = cdb.cursor()
cur.execute(
'SELECT table_name FROM information_schema.tables WHERE table_type = \'BASE TABLE\' AND table_schema = \'public\';'
)
if cur.fetchone() != None:
if lightMode:
rc.pwarn(
'There\'s already some data in the database, but we are running in LIGHT mode, so I won\'t touch it.'