diff --git a/dislocker.py b/dislocker.py index 01595be..50afa26 100644 --- a/dislocker.py +++ b/dislocker.py @@ -1,7 +1,7 @@ import json import discord import os -import psycopg2 +import sqlite3 class Bot(discord.Client): async def on_ready(self): @@ -20,8 +20,24 @@ class Bot(discord.Client): class Cert(): def __init__(self): - self.db = psycopg2.connect("host=localhost dbname=dislocker user=suti7 password=testing") - + if not os.path.isfile("./db/dislocker.db"): + initial = 1 + else: + initial = 0 + self.db = sqlite3.connect("./db/dislocker.db") + if initial == 1: + self.initial() + + def initial(self): + cursor = self.db.cursor() + initial_sql_web_auth = "CREATE TABLE web_auth (id INTEGER PRIMARY KEY AUTOINCREMENT, username VARCHAR(10) NOT NULL, password VARCHAR(32) NOT NULL);" + cursor.execute(initial_sql_web_auth) + initial_sql_discord_user = "CREATE TABLE discord_user (id INTEGER PRIMARY KEY AUTOINCREMENT, discord_username VARCHAR(32) NOT NULL, discord_userid VARCHAR(18) NOT NULL);" + cursor.execute(initial_sql_discord_user) + initial_sql_insert_web_admin = "INSERT INTO discord_user (username, password) VALUES ('admin', 'admin');" + cursor.execute(initial_sql_insert_web_admin) + self.db.commit() + def register(self, username, userid): cursor = self.db.cursor() insert_sql = "INSERT INTO user_list (discord_username, discord_userid) VALUES (%s, %s);"