From 6d4717e4711536e4dd3140ba13d4cc67e81577df Mon Sep 17 00:00:00 2001 From: suti7yk5032 Date: Tue, 4 Jun 2024 13:26:28 +0900 Subject: [PATCH] =?UTF-8?q?=E3=83=87=E3=83=BC=E3=82=BF=E3=83=99=E3=83=BC?= =?UTF-8?q?=E3=82=B9=E3=82=92SQLite=E3=81=AB=E5=A4=89=E6=9B=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dislocker.py | 22 +++++++++++++++++++--- 1 file changed, 19 insertions(+), 3 deletions(-) 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);"