Thesis/web_service/manage.py

import argparse
import sqlite3
import uuid

DB_FILENAME = "db.sqlite"

SQL_CREATE_USERS = \
"""CREATE TABLE users (
username TEXT PRIMARY KEY,
api_key TEXT NOT NULL
);"""


SQL_CREATE_REQUESTS = \
"""CREATE TABLE requests (
id TEXT PRIMARY KEY,
username TEXT NOT NULL,
depth INTEGER NOT NULL,
width INTEGER NOT NULL,
int_cycles INTEGER NOT NULL,
ext_cycles INTEGER NOT NULL,
status TEXT NOT NULL,
FOREIGN KEY(username) REFERENCES users(username)
);"""

SQL_INSERT_USER = "INSERT INTO users VALUES('{username}', '{api_key}');"
SQL_DELETE_USER = "DELETE FROM users WHERE username='{username}';"

conn = sqlite3.connect(DB_FILENAME)
c = conn.cursor()


def init_db():
    c.execute(SQL_CREATE_USERS)
    c.execute(SQL_CREATE_REQUESTS)


def add_user(username):
    api_key = uuid.uuid4().hex
    sql = SQL_INSERT_USER.format(username=username, api_key=api_key)
    c.execute(sql)
    conn.commit()
    print("User '%s' added succesfully (api_key: %s)" % (username, api_key))


def remove_user(username):
    sql = SQL_DELETE_USER.format(username=username)
    c.execute(sql)
    conn.commit()
    print("User '%s' removed succesfully from DB" % username)


def parse_args():
    parser = argparse.ArgumentParser(description='Script to manage the DB related to the DEVStone comparative web service.')

    parser.add_argument('-i', '--init', action='store_true', help='Init the DB')
    parser.add_argument('-a', '--add_user', type=str, help='Add a user to the DB')
    parser.add_argument('-r', '--remove_user', type=str, help='Remove a user from the DB')

    return parser.parse_args()


if __name__ == '__main__':
    args = parse_args()

    if args.init:
        init_db()

    if args.add_user:
        add_user(args.add_user)

    if args.remove_user:
        remove_user(args.remove_user)