from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Boolean, Integer, String, DateTime, Float, create_engine
from sqlalchemy.orm import sessionmaker, scoped_session


Base = declarative_base()

class TrainedItem(Base):
	__tablename__ = 'trained_items'
	name = Column(String, primary_key=True)
	state = Column(Boolean)

class FoundItem(Base):
        __tablename__ = 'found_items'
        id = Column(Integer, primary_key=True)
        hnid = Column(String)
        comment_count = Column(Integer)
        name = Column(String)
        url = Column(String, unique=True, index=True)
        date = Column(DateTime, index=True)
        rating = Column(Float)

engine = create_engine('sqlite:///hnlearn.db')
Base.metadata.create_all(engine)
session_factory = sessionmaker(bind=engine)
Session = scoped_session(session_factory)