script/iri_tweet/models.py
changeset 11 54d7f1486ac4
parent 9 bb44692e09ee
equal deleted inserted replaced
10:eb885a117aa0 11:54d7f1486ac4
     1 from sqlalchemy import Boolean, Table, Column, BigInteger, \
     1 from sqlalchemy import Boolean, Table, Column, BigInteger, Integer, String, \
     2     Integer, String, MetaData, ForeignKey, DateTime
     2     MetaData, ForeignKey, DateTime, create_engine
     3 from sqlalchemy.ext.declarative import declarative_base
     3 from sqlalchemy.ext.declarative import declarative_base
     4 from sqlalchemy.orm import relationship, backref, sessionmaker
     4 from sqlalchemy.orm import relationship, backref
     5 import datetime
     5 import datetime
     6 import email.utils
     6 import email.utils
     7 import simplejson
     7 import simplejson
     8 
     8 
     9 
     9 
    10 Base = declarative_base()
    10 Base = declarative_base()
    11 
    11 
       
    12 APPLICATION_NAME = "IRI_TWITTER" 
    12 CONSUMER_KEY = "54ThDZhpEjokcMgHJOMnQA"
    13 CONSUMER_KEY = "54ThDZhpEjokcMgHJOMnQA"
    13 CONSUMER_SECRET = "wUoL9UL2T87tfc97R0Dff2EaqRzpJ5XGdmaN2XK3udA"
    14 CONSUMER_SECRET = "wUoL9UL2T87tfc97R0Dff2EaqRzpJ5XGdmaN2XK3udA"
    14 ACCESS_TOKEN_KEY= "47312923-LiNTtz0I18YXMVIrFeTuhmH7bOvYsK6p3Ln2Dc"
    15 #ACCESS_TOKEN_KEY= "47312923-LiNTtz0I18YXMVIrFeTuhmH7bOvYsK6p3Ln2Dc"
    15 ACCESS_TOKEN_SECRET = "r3LoXVcjImNAElUpWqTu2SG2xCdWFHkva7xeQoncA"
    16 #ACCESS_TOKEN_SECRET = "r3LoXVcjImNAElUpWqTu2SG2xCdWFHkva7xeQoncA"
    16 
    17 
    17 def adapt_date(date_str):
    18 def adapt_date(date_str):
    18     ts = email.utils.parsedate_tz(date_str)
    19     ts = email.utils.parsedate_tz(date_str)
    19     return datetime.datetime(*ts[0:7])
    20     return datetime.datetime(*ts[0:7])
    20 
    21 
   177     def __init__(self, **kwargs):
   178     def __init__(self, **kwargs):
   178         super(EntityUser, self).__init__(**kwargs)
   179         super(EntityUser, self).__init__(**kwargs)
   179         for key, value in kwargs.items():
   180         for key, value in kwargs.items():
   180             if hasattr(self,key):
   181             if hasattr(self,key):
   181                 setattr(self,key,value)
   182                 setattr(self,key,value)
       
   183                 
       
   184                 
       
   185 def setup_database(*args, **kwargs):
       
   186         
       
   187     create_all = True        
       
   188     if "create_all" in kwargs:
       
   189         create_all = kwargs["create_all"]
       
   190         del(kwargs["create_all"])
       
   191 
       
   192     engine = create_engine(*args, **kwargs)
       
   193     metadata = Base.metadata
       
   194             
       
   195     if create_all:
       
   196         metadata.create_all(engine)
       
   197 
       
   198     return (engine, metadata)
   182 
   199 
   183 rest_tweet_tweet = {
   200 rest_tweet_tweet = {
   184     u'iso_language_code': 'unicode',
   201     u'iso_language_code': 'unicode',
   185     u'text': 'unicode',
   202     u'text': 'unicode',
   186     u'from_user_id_str': 'unicode',
   203     u'from_user_id_str': 'unicode',