27 lines
858 B
Python
27 lines
858 B
Python
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)
|