データベースをSQLiteに変更

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,7 +20,23 @@ 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()