File size: 917 Bytes
58e450d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
# Database URL (e.g., SQLite, PostgreSQL, MySQL)
DATABASE_URL = "sqlite:///./instagram_ai.db"
# Create engine and session
engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
# Base class for models
Base = declarative_base()
# Define Post model
class Post(Base):
__tablename__ = "posts"
id = Column(Integer, primary_key=True, index=True)
caption = Column(String, nullable=False)
hashtags = Column(String)
image_url = Column(String)
posting_time = Column(DateTime)
engagement_rate = Column(Float)
viral_score = Column(Float)
promote = Column(Integer) # 1 for promote, 0 for don't promote
# Create tables
Base.metadata.create_all(bind=engine) |