from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, ForeignKey, Boolean
from sqlalchemy.orm import declarative_base, sessionmaker

Base = declarative_base()

class User(Base):
    __tablename__ = "users"
    line_id = Column(String, primary_key=True, index=True)
    current_state = Column(String, default="NORMAL")

class Restaurant(Base):
    __tablename__ = "restaurants"
    id = Column(Integer, primary_key=True, autoincrement=True)
    line_id = Column(String, ForeignKey("users.line_id"))
    original_name = Column(String) 
    zh_name = Column(String)       
    en_name = Column(String)       
    address = Column(String)
    rating = Column(Float, default=0.0)
    is_in_google_list = Column(Boolean, default=True)
    last_visited = Column(DateTime, nullable=True)

# 換成 v4 版本，強制重新建立乾淨無污染的資料表
engine = create_engine("sqlite:///./foodie_bot_v4.db", connect_args={"check_same_thread": False})
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base.metadata.create_all(bind=engine)