script/lib/iri_tweet/models.py
changeset 250 6334869ab06d
parent 122 4c3a15877f80
child 253 e9335ee3cf71
equal deleted inserted replaced
249:b31aef9365a8 250:6334869ab06d
   122     profile_sidebar_border_color= Column(String)
   122     profile_sidebar_border_color= Column(String)
   123     profile_sidebar_fill_color= Column(String)
   123     profile_sidebar_fill_color= Column(String)
   124     profile_text_color= Column(String)
   124     profile_text_color= Column(String)
   125     profile_use_background_image= Column(Boolean)
   125     profile_use_background_image= Column(Boolean)
   126     protected= Column(Boolean)
   126     protected= Column(Boolean)
   127     screen_name= Column(String)
   127     screen_name= Column(String, index=True, unique=True)
   128     show_all_inline_media= Column(Boolean)
   128     show_all_inline_media= Column(Boolean)
   129     statuses_count = Column(Integer)
   129     statuses_count = Column(Integer)
   130     time_zone= Column(String)
   130     time_zone= Column(String)
   131     url= Column(String)
   131     url= Column(String)
   132     utc_offset = Column(Integer)
   132     utc_offset = Column(Integer)
   141     
   141     
   142 
   142 
   143 class Hashtag(Base):
   143 class Hashtag(Base):
   144     __tablename__ = "tweet_hashtag"
   144     __tablename__ = "tweet_hashtag"
   145     id = Column(Integer, primary_key=True)
   145     id = Column(Integer, primary_key=True)
   146     text = Column(String, unique = True)
   146     text = Column(String, unique = True, index = True)
   147     def __init__(self, **kwargs):
   147     def __init__(self, **kwargs):
   148         for key, value in kwargs.items():
   148         for key, value in kwargs.items():
   149             if hasattr(self,key):
   149             if hasattr(self,key):
   150                 setattr(self,key,value)
   150                 setattr(self,key,value)
   151 
   151 
   202                 setattr(self,key,value)
   202                 setattr(self,key,value)
   203                 
   203                 
   204                 
   204                 
   205 def setup_database(*args, **kwargs):
   205 def setup_database(*args, **kwargs):
   206         
   206         
   207     create_all = True        
   207     create_all = True
   208     if "create_all" in kwargs:
   208     if "create_all" in kwargs:
   209         create_all = kwargs["create_all"]
   209         create_all = kwargs["create_all"]
   210         del(kwargs["create_all"])
   210         del(kwargs["create_all"])
   211 
   211 
   212     engine = create_engine(*args, **kwargs)
   212     engine = create_engine(*args, **kwargs)