diff -r ee6305b4a7dc -r cdd7d3c0549c script/lib/iri_tweet/models.py --- a/script/lib/iri_tweet/models.py Tue Jul 26 23:57:09 2011 +0200 +++ b/script/lib/iri_tweet/models.py Wed Jul 27 00:04:55 2011 +0200 @@ -42,7 +42,34 @@ if hasattr(self,key): setattr(self,key,value) +class TweetSource(Base): + __tablename__ = 'tweet_tweet_source' + id = Column(Integer, primary_key = True, autoincrement=True) + original_json = Column(String) + received_at = Column(DateTime, default=datetime.datetime.now()) + + def __init__(self, **kwargs): + for key, value in kwargs.items(): + if hasattr(self,key): + setattr(self,key,value) + +class TweetLog(Base): + + TWEET_STATUS = { + 'OK' : 1, + 'ERROR' : 2, + } + + __tablename__ = 'tweet_tweet_log' + id = Column(Integer, primary_key = True, autoincrement=True) + tweet_source_id = Column(Integer, ForeignKey('tweet_tweet_source.id')) + tweet_source = relationship("TweetSource", backref="logs") + status = Column(Integer) + error = Column(String) + error_stack = Column(String) + + class Tweet(Base): __tablename__ = 'tweet_tweet' @@ -65,7 +92,10 @@ text = Column(String) truncated = Column(Boolean) user_id = Column(Integer, ForeignKey('tweet_user.id')) - original_json = Column(String) + user = relationship("TweetUser", backref="tweets") +# original_json = Column(String) + tweet_source_id = Column(Integer, ForeignKey('tweet_tweet_source.id')) + tweet_source = relationship("TweetSource", backref="tweet") entity_list = relationship(Entity, backref='tweet') received_at = Column(DateTime, default=datetime.datetime.now()) @@ -81,11 +111,11 @@ id = Column(Integer, primary_key = True) user_id = Column(Integer, ForeignKey('tweet_user.id')) + user = relationship("TweetUser", backref="messages") created_at = Column(DateTime, default=datetime.datetime.now()) message_id = Column(Integer, ForeignKey('tweet_message.id')) class Message(Base): - __tablename__ = "tweet_message" id = Column(Integer, primary_key = True)