alschameri/helping-source

1from sqlalchemy import create_engine, Column, Integer2from sqlalchemy.ext.declarative import declarative_base3from sqlalchemy.orm import sessionmaker4 5# Base class for models6Base = declarative_base()7 8# Visitor count model9class VisitorCount(Base):10    __tablename__ = 'visitor_count'11    id = Column(Integer, primary_key=True)12    count = Column(Integer, default=0)13 14def connect_to_postgres(url):15    """Connect to PostgreSQL and return a session and engine."""16    engine = create_engine(url, echo=True)  # echo=True for SQL logs17    Base.metadata.create_all(engine)  # create tables18    Session = sessionmaker(bind=engine)19    return Session(), engine