thibautmodrin commited on
Commit
ef15422
·
1 Parent(s): 5ba7c5d

mod dockerfile et app.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -1,12 +1,12 @@
 
1
  from fastapi import FastAPI
2
  from pydantic import BaseModel
3
  from sqlalchemy import create_engine, Column, Integer, String, TIMESTAMP
4
  from sqlalchemy.ext.declarative import declarative_base
5
  from sqlalchemy.orm import sessionmaker
6
- import os
7
 
8
- # --- Remplace par l'URL de connexion Neon ---
9
- DATABASE_URL = "postgresql://username:password@host/dbname"
10
 
11
  # --- Initialisation SQLAlchemy ---
12
  engine = create_engine(DATABASE_URL)
@@ -43,4 +43,4 @@ def enregistrer_traitement(t: Traitement):
43
  db.add(nouveau)
44
  db.commit()
45
  db.close()
46
- return {"message": "Traitement enregistré"}
 
1
+ import os
2
  from fastapi import FastAPI
3
  from pydantic import BaseModel
4
  from sqlalchemy import create_engine, Column, Integer, String, TIMESTAMP
5
  from sqlalchemy.ext.declarative import declarative_base
6
  from sqlalchemy.orm import sessionmaker
 
7
 
8
+ # Lire l'URL de la base de données à partir d'une variable d'environnement
9
+ DATABASE_URL = os.getenv("DATABASE_URL")
10
 
11
  # --- Initialisation SQLAlchemy ---
12
  engine = create_engine(DATABASE_URL)
 
43
  db.add(nouveau)
44
  db.commit()
45
  db.close()
46
+ return {"message": "Traitement enregistré"}