51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import psycopg2
|
|
from werkzeug.security import generate_password_hash
|
|
|
|
DB_HOST = "db"
|
|
DB_PORT = "5432"
|
|
DB_NAME = "ecocharge"
|
|
DB_USER = "ecocharge_user"
|
|
DB_PASSWORD = "ecocharge_password"
|
|
|
|
USERNAME = "admin"
|
|
PLAIN_PASSWORD = "admin123"
|
|
|
|
def main():
|
|
password_hash = generate_password_hash(PLAIN_PASSWORD)
|
|
|
|
conn = psycopg2.connect(
|
|
host=DB_HOST,
|
|
port=DB_PORT,
|
|
database=DB_NAME,
|
|
user=DB_USER,
|
|
password=DB_PASSWORD
|
|
)
|
|
cur = conn.cursor()
|
|
|
|
cur.execute("SELECT id FROM users WHERE username = %s", (USERNAME,))
|
|
user = cur.fetchone()
|
|
|
|
if user:
|
|
cur.execute("""
|
|
UPDATE users
|
|
SET password = %s, is_admin = TRUE
|
|
WHERE username = %s
|
|
""", (password_hash, USERNAME))
|
|
print(f"Utilisateur '{USERNAME}' mis à jour avec un mot de passe haché.")
|
|
else:
|
|
cur.execute("""
|
|
INSERT INTO users (username, password, is_admin)
|
|
VALUES (%s, %s, TRUE)
|
|
""", (USERNAME, password_hash))
|
|
print(f"Utilisateur '{USERNAME}' créé avec un mot de passe haché.")
|
|
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
|
|
print("Terminé.")
|
|
print(f"Nom d'utilisateur : {USERNAME}")
|
|
print(f"Mot de passe : {PLAIN_PASSWORD}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |