diff --git a/app/database.py b/app/database.py index 1c18b07..b86b1bc 100644 --- a/app/database.py +++ b/app/database.py @@ -1,8 +1,8 @@ import os from dotenv import load_dotenv -from sqlalchemy import create_engine, Column, Integer, String, DateTime -from sqlalchemy.orm import sessionmaker, declarative_base -from datetime import datetime +from sqlalchemy import create_engine +from sqlalchemy.ext.declarative import declarative_base +from sqlalchemy.orm import sessionmaker # Cargar variables del archivo .env (para entorno local) load_dotenv() @@ -16,21 +16,6 @@ MYSQL_PORT = os.getenv("MYSQL_PORT", "3306") DATABASE_URL = f"mysql+pymysql://{MYSQL_USER}:{MYSQL_PASSWORD}@{MYSQL_HOST}:{MYSQL_PORT}/{MYSQL_DATABASE}" -# Modelo de datos de entrada -Base = declarative_base() - -class NewsItem(Base): - __tablename__ = 'news' - - id = Column(Integer, primary_key=True, autoincrement=True) - titulo = Column(String(255), unique=True, nullable=False) - contenido = Column(String, nullable=False) - autor = Column(String(100)) - fuente = Column(String(100)) - fecha = Column(DateTime, default=datetime.utcnow) - link = Column(String(255)) - keyword = Column(String(100)) - # Crear el motor de base de datos engine = create_engine(DATABASE_URL) @@ -38,6 +23,7 @@ engine = create_engine(DATABASE_URL) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) # Base para los modelos +Base = declarative_base() # Dependencia para obtener la sesión de la BD def get_db():