Spaces:
Sleeping
Sleeping
import os | |
from dotenv import load_dotenv | |
load_dotenv() | |
from sqlalchemy import create_engine, text | |
from fastapi import FastAPI, Depends | |
DATABASE_URL = os.getenv('DATABASE_URL') | |
def get_engine(): | |
if not DATABASE_URL: | |
raise ValueError("L'URL de la base de données (PG_URL) n'est pas définie dans l'environnement.") | |
engine = create_engine(DATABASE_URL) | |
return engine | |
def get_db_connection(engine=Depends(get_engine)): | |
with engine.connect() as connection: | |
yield connection | |
def load_job_offer(connection): | |
query_offre = text('SELECT entreprise, ville, poste, contrat, description_poste, publication, lien, id FROM "jobs_offers"') | |
resultats = connection.execute(query_offre).fetchall() | |
offres = [] | |
for row in resultats: | |
entreprise, ville, poste, contrat, description_poste, publication, lien, id = row | |
offres.append({ | |
"entreprise": entreprise, | |
"ville": ville, | |
"poste": poste, | |
"contrat": contrat, | |
"description_poste": description_poste, | |
"publication": publication, | |
"lien": lien, | |
"id": id | |
}) | |
return offres | |
def get_job_offer_by_id(connection, offre_id: str): | |
query = text('SELECT entreprise, ville, poste, contrat, description_poste, publication, lien, id FROM "jobs_offers" WHERE id = :id') | |
result = connection.execute(query, {"id": offre_id}).fetchone() | |
if result: | |
entreprise, ville, poste, contrat, description_poste, publication, lien, id = result | |
return { | |
"entreprise": entreprise, | |
"ville": ville, | |
"poste": poste, | |
"contrat": contrat, | |
"description_poste": description_poste, | |
"publication": publication, | |
"lien": lien, | |
"id": id | |
} | |
return None | |
app = FastAPI() | |
async def read_job_offer(connection = Depends(get_db_connection)): | |
offres = load_job_offer(connection) | |
if offres: | |
return offres | |
return {"message": "Aucune offre d'emploi trouvée."} | |
async def read_job_offer_by_id(offre_id: str, connection = Depends(get_db_connection)): | |
offre = get_job_offer_by_id(connection, offre_id) | |
if offre: | |
return offre | |
return {"message": f"Aucune offre trouvée avec l'id {offre_id}"} | |
if __name__ == "__main__": | |
import uvicorn | |
uvicorn.run(app, host="0.0.0.0", port=8010) |