from sqlalchemy import *
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Text(Base):
    __tablename__ = 'texts'
    id = Column(Integer, primary_key=True)
    text = Column(String(140))
    category = Column(String(50))

    def __repr__(self):
        return "<Tweet (category='%s', | text='%s')>" % (
                                self.category, self.text)

def get_engine(database):
    engine = create_engine('sqlite:///'+database, echo=False)
    Base.metadata.create_all(engine)
    return engine

if __name__ == '__main__':
    get_engine('test.db')
    get_engine('train.db')