database.py

from sqlalchemy import Integer, String, Float, DateTime, create_engine, Column
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
from datetime import datetime
import os
from dotenv import load_dotenv

load_dotenv()

DB_URL = os.getenv("DATABASE_URL")
engine = create_engine(DB_URL)
Session = sessionmaker(bind=engine)
Base = declarative_base()

class Tasks(Base):
    __tablename__ = "tasks"

    id = Column(Integer, primary_key=True)
    task = Column(String, nullable=True)
    status = Column(String, nullable=True)
    result = Column(String, nullable=True)
    time = Column(DateTime, default=datetime.utcnow)

Base.metadata.create_all(engine)

def get_db():
    db = Session()
    try:
        yield db
    finally:
        db.close()