premain_database_initialization.py

import os
import time

import mariadb

import secret


FLAGS = _ = None
DEBUG = False
STIME = time.time()


def main():
    if DEBUG:
        print(f'Parsed arguments {FLAGS}')
        print(f'Unparsed arguments {_}')

    try:
        conn = mariadb.connect(user=secret.db_username,
                               password=secret.db_password,
                               host=secret.db_host,
                               port=secret.db_port,
                               database=secret.db_databasename)
        cur = conn.cursor()
    except mariadb.Error as e:
        print(f'Error connecting to MariaDB: {e}')
        sys.exit(0)

    if DEBUG:
        print(f'[{int(time.time()-STIME)}] Connect to database')

    if FLAGS.reset:
        cur.execute('''DROP TABLE IF EXISTS blk;''')
        if DEBUG:
            print(f'[{int(time.time()-STIME)}] DROP all tables of {secret.db_databasename}')

    cur.execute('''CREATE TABLE blk (
                     id INT NOT NULL,
                     blkhash CHAR(64) NOT NULL,
                     miningtime TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
                     PRIMARY KEY (id),
                     UNIQUE (blkhash)
                   );''')
    cur.execute('''CREATE INDEX idx_miningtime ON blk (miningtime);''')
    conn.commit()

    cur.execute('''SHOW TABLES;''')
    res = cur.fetchall()
    tables = []
    if DEBUG:
        print(f'[{int(time.time()-STIME)}] CREATE TABLES:')
    for row in res:
        tables.append(row[0])
        if DEBUG:
            print(f'[{int(time.time()-STIME)}] > {table}')
    if DEBUG:
        for table in tables:
            print(f'[{int(time.time()-STIME)}] > DESCRIBE {table}')
            cur.execute(f'''DESCRIBE {table};''')
            res = cur.fetchall()
            for row in res:
                print(f'[{int(time.time()-STIME)}] >> {row}')


    conn.commit()
    conn.close()


if __name__ == '__main__':
    import argparse

    root_path = os.path.abspath(__file__)
    root_dir = os.path.dirname(root_path)
    os.chdir(root_dir)

    parser = argparse.ArgumentParser()
    parser.add_argument('--debug', action='store_true',
                        help='The present debug message')
    parser.add_argument('--reset', action='store_true',
                        help='Clear exists tables before creation')

    FLAGS, _ = parser.parse_known_args()
    DEBUG = FLAGS.debug

    main()