migrations.py

import sqlite3
import os, json

config = json.load(open('src/config.json'))
magnetDatabase = config['magnetDatabase']
database = config['database']

input1 = input('Do you want to run migration for magnet link database? Y/[n]: ')
if input1 == 'y':
    if os.path.exists(magnetDatabase):
        os.remove(magnetDatabase)
        print('[-] Database already exists. Deleting it.')

    conn = sqlite3.connect(magnetDatabase)
    print('[+] Magnetlink Database opened successfully.')

    conn.execute('''CREATE TABLE data
            (hash       TEXT PRIMARY KEY COLLATE NOCASE,
            title      TEXT,
            magnetlink TEXT
            );''')

    print('[+] Table data created successfully.')

    conn.execute('''CREATE TABLE wishlist
            (wishlistId    INT PRIMARY KEY,
            ownerId        INT,
            hash      TEXT
            );''')

    print('[+] Table wishlist created successfully.')

    conn.close()

input1 = input('Do you want to run migration for Torrent Hunt database? Y/[n]: ')
if input1 == 'y':
    if os.path.exists(database):
        os.remove(database)
        print('[-] Database already exists. Deleting it.')

    conn = sqlite3.connect(database)
    print('[+] Database opened successfully.')

    conn.execute('''CREATE TABLE users
            (UserId       INTEGER PRIMARY KEY,
            date          STRING  NOT NULL,
            referrer      TEXT
            );''')

    print('[+] Table users created successfully.')

    conn.execute('''CREATE TABLE groups
            (UserId       INTEGER PRIMARY KEY,
            userName      TEXT,
            date          STRING  NOT NULL
            );''')

    print('[+] Table users created successfully.')

    conn.execute('''CREATE TABLE settings
            (ownerId       INTEGER PRIMARY KEY,
            language       TEXT DEFAULT "english",
            defaultSite    TEXT DEFAULT "piratebay",
            defaultMode    TEXT DEFAULT "link",
            restrictedMode INTEGER DEFAULT 1
            );''')

    print('[+] Table settings created successfully.')

    conn.execute('''CREATE TABLE flood
         (ownerId       INTEGER PRIMARY KEY,
         warned         INTEGER DEFAULT 0,
         lastMessage   INTEGER DEFAULT 0,
         blockTill     INTEGER DEFAULT 0
         );''')

    print('[+] Table flood created successfully.')

    conn.close()