1.0

flask/add_user.py

@@ -0,0 +1,51 @@
+import pymysql
+import getpass
+import bcrypt
+from dotenv import load_dotenv
+from datetime import datetime
+import os
+
+load_dotenv()
+
+def init():
+    try:
+        conn = pymysql.connect(
+            host=os.getenv("DB_HOST", "127.0.0.1"),
+            port=int(os.getenv("DB_PORT", 3306)),
+            user=os.getenv("DB_USER"),
+            password=os.getenv("DB_PASSWORD"),
+            database=os.getenv("DB_NAME"),
+            charset=os.getenv("DB_CHARSET", "utf8mb4")
+        )
+        return conn
+
+    except pymysql.err.OperationalError as e:
+        print(f"❌ Erreur de connexion : {e}")
+        return None
+
+def add():
+    conn = init()
+    if conn is None:
+        return False
+
+    username = input("Quel est l'username du nouvel utilisateur : ")
+    password = getpass.getpass("Quel est le password : ")
+
+    # Hashage du mot de passe avec bcrypt
+    hashed = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+
+    try:
+        cursor = conn.cursor()
+        requete = "INSERT INTO Auth (username, password, created_at) VALUES (%s, %s, %s)"
+        cursor.execute(requete, (username, hashed, datetime.now()))
+        conn.commit()
+        print(f"✅ Utilisateur '{username}' ajouté avec succès !")
+
+    except pymysql.err.IntegrityError:
+        print("❌ Cet utilisateur existe déjà")
+
+    finally:
+        cursor.close()
+        conn.close()
+
+add()