suricata/yt2mp3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

arg correct + verbose

Browse Source
This commit is contained in:
suricata
2026-01-19 13:23:16 +00:00
parent b3dcd3058a
commit f21d469280
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
server/py/fastapi_server/main.py
View File

@@ -61,11 +61,11 @@ async def fetch_user_info(arg: str, type: str):
if db.mode == "mysql": if db.mode == "mysql":
async with db.pool.acquire() as conn: async with db.pool.acquire() as conn:
async with conn.cursor(aiomysql.DictCursor) as cur: async with conn.cursor(aiomysql.DictCursor) as cur:
await cur.execute("SELECT * FROM users WHERE %s = %s", (arg,type)) await cur.execute("SELECT * FROM users WHERE %s = %s", (type, arg))
return await cur.fetchone() return await cur.fetchone()
elif db.mode == "sqlite": elif db.mode == "sqlite":
async with aiosqlite.connect(db.file) as conn: async with aiosqlite.connect(db.file) as conn:
async with conn.execute("SELECT * FROM users WHERE ? = ?", (arg,type)) as cursor: async with conn.execute("SELECT * FROM users WHERE ? = ?", (type, arg)) as cursor:
return await cursor.fetchone() return await cursor.fetchone()
def download_audio(video_url: str, output_folder: str) -> str: def download_audio(video_url: str, output_folder: str) -> str:
@@ -175,14 +175,16 @@ async def login_endpoint(req: Request):
body = await req.json() body = await req.json()
#print(body) #print(body)
try: try:
user_data = await fetch_user_info(body["email"]) user_data = await fetch_user_info(body["email"], "email")
if (user_data is None): if (user_data is None):
print("User doesn't exist")
raise HTTPException(status_code=401, detail="User doesn't exist") raise HTTPException(status_code=401, detail="User doesn't exist")
else: else:
pwd = user_data.get("password") pwd = user_data.get("password")
if (bcrypt.checkpw(body["password"].encode(), user_data.get("password").encode())): if (bcrypt.checkpw(body["password"].encode(), user_data.get("password").encode())):
return user_data return user_data
else: else:
print("Wrong password")
raise HTTPException(status_code=401, detail="Wrong password") raise HTTPException(status_code=401, detail="Wrong password")
except KeyError: except KeyError:
raise HTTPException(status_code=400) raise HTTPException(status_code=400)