From e0f8c7b78b58f18d73bed4310f99d4fb1ebb5508 Mon Sep 17 00:00:00 2001 From: Lenczu Vex Date: Sat, 5 Apr 2025 12:22:31 +0200 Subject: Swapped db to postgresql --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'main.py') diff --git a/main.py b/main.py index 8ead667..1fa7768 100644 --- a/main.py +++ b/main.py @@ -1,5 +1,5 @@ from flask import Flask, request, render_template, redirect, make_response -import mariadb +import psycopg2 import argon2 import tomllib as toml import secrets @@ -68,18 +68,18 @@ def login(): page = request.args.get("page") username = request.form.get("username") password = request.form.get("password") - mariadb_connection = mariadb.connect( + psycopg2_connection = psycopg2.connect( user=app.config["DB_USER"], password=app.config["DB_PASSWORD"], host=app.config["DB_HOST"], port=app.config["DB_PORT"], database=app.config["DB_NAME"], ) - cur = mariadb_connection.cursor() + cur = psycopg2_connection.cursor() cur.execute("SELECT password FROM users WHERE username = ?", (username,)) result = cur.fetchone() cur.close() - mariadb_connection.close() + psycopg2_connection.close() if result is None: return render_template("login.html", error="Invalid username or password"), 401 -- cgit v1.2.3-70-g09d2