suti7yk5032/Dislocker
1
0
Fork 0
Code Issues 11 Pull requests Projects 3 Releases Packages Wiki Activity Actions

sqlite用にちょっと変更

Browse source
This commit is contained in:
suti7yk5032 2024-06-04 14:17:35 +09:00
parent 83e44da330
commit ab92ecfb4a
2 changed files with 6 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
dislocker.py
View file

@ -40,13 +40,13 @@ class Cert():
def register(self, username, userid):
cursor = self.db.cursor()
insert_sql = "INSERT INTO user_list (discord_username, discord_userid) VALUES (%s, %s);"
insert_sql = "INSERT INTO user_list (discord_username, discord_userid) VALUES (?, ?);"
cursor.execute(insert_sql, (username, userid))
self.db.commit()
def get_username(self, username, password):
cursor = self.db.cursor()
get_sql = "SELECT * FROM web_auth WHERE username = %s AND password = %s "
get_sql = "SELECT * FROM web_auth WHERE username = ? AND password = ? "
cursor.execute(get_sql, (username, password))
result = cursor.fetchall()
print(result)
@ -55,14 +55,8 @@ class Cert():
elif result[0][1] == username and result[0][2] == password:
return 0
class Web(flask.Flask):
async def temp(self):
print("")
cert = Cert()
print(cert.get_username("suti7", "0126"))
bot_config_path = "./config/bot.json"
if not os.path.isfile(bot_config_path):

8
webui.py
View file

@ -1,6 +1,6 @@
from flask import Flask, request, render_template, redirect, url_for
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user
import psycopg2
import sqlite3
app = Flask(__name__, static_folder="./resource/")
app.secret_key = 'your_secret_key'
@ -9,17 +9,17 @@ login_manager.init_app(app)
class Cert():
def __init__(self):
self.db = psycopg2.connect("host=localhost dbname=dislocker user=suti7 password=testing")
self.db = sqlite3.connect("host=localhost dbname=dislocker user=suti7 password=testing")
def register(self, username, userid):
cursor = self.db.cursor()
insert_sql = "INSERT INTO user_list (discord_username, discord_userid) VALUES (%s, %s);"
insert_sql = "INSERT INTO user_list (discord_username, discord_userid) VALUES (?, ?)"
cursor.execute(insert_sql, (username, userid))
self.db.commit()
def login(self, username, password):
cursor = self.db.cursor()
get_sql = "SELECT * FROM web_auth WHERE username = %s AND password = %s "
get_sql = "SELECT * FROM web_auth WHERE username = ? AND password = ? "
cursor.execute(get_sql, (username, password))
result = cursor.fetchall()
print(result)