1 from iri_tweet.models import setup_database, Message, UserMessage, User |
1 from iri_tweet.models import setup_database, Message, UserMessage, User |
2 from iri_tweet.utils import (get_oauth_token, get_user_query, set_logging_options, |
2 from iri_tweet.utils import (get_oauth_token, get_user_query, set_logging_options, |
3 set_logging, parse_date) |
3 set_logging, parse_date, logger) |
4 from optparse import OptionParser #@UnresolvedImport |
4 from optparse import OptionParser #@UnresolvedImport |
5 from sqlalchemy import BigInteger |
5 from sqlalchemy import BigInteger |
6 from sqlalchemy.orm import sessionmaker |
6 from sqlalchemy.orm import sessionmaker |
7 from sqlalchemy.schema import MetaData, Table, Column |
7 from sqlalchemy.schema import MetaData, Table, Column |
8 from sqlalchemy.sql import and_ |
8 from sqlalchemy.sql import and_ |
9 import datetime |
9 import datetime |
10 import logging #@UnresolvedImport |
|
11 import sys |
10 import sys |
12 import time |
11 import time |
13 import twitter |
12 import twitter |
14 |
13 |
15 APPLICATION_NAME = "Tweet recorder user" |
14 APPLICATION_NAME = "Tweet recorder user" |
52 |
51 |
53 (options, args) = get_options() |
52 (options, args) = get_options() |
54 |
53 |
55 set_logging(options) |
54 set_logging(options) |
56 |
55 |
57 logging.debug("OPTIONS : " + repr(options)) #@UndefinedVariable |
56 logger.debug("OPTIONS : " + repr(options)) #@UndefinedVariable |
58 |
57 |
59 if not options.message or len(options.message) == 0: |
58 if not options.message or len(options.message) == 0: |
60 sys.exit() |
59 sys.exit() |
61 |
60 |
62 engine, metadata = setup_database('sqlite:///'+options.database, echo=((options.verbose-options.quiet)>0), create_all = True) |
61 engine, metadata = setup_database('sqlite:///'+options.database, echo=((options.verbose-options.quiet)>0), create_all = True) |
106 |
105 |
107 for user in query_res: |
106 for user in query_res: |
108 screen_name = user.screen_name |
107 screen_name = user.screen_name |
109 |
108 |
110 message = u"@%s: %s" % (screen_name, base_message) |
109 message = u"@%s: %s" % (screen_name, base_message) |
111 logging.debug("new status : " + message) #@UndefinedVariable |
110 logger.debug("new status : " + message) #@UndefinedVariable |
112 if not options.simulate: |
111 if not options.simulate: |
113 t.statuses.update(status=message) |
112 t.statuses.update(status=message) |
114 user_message = UserMessage(user_id=user.id, message_id=message_obj.id) |
113 user_message = UserMessage(user_id=user.id, message_id=message_obj.id) |
115 session.add(user_message) |
114 session.add(user_message) |
116 session.flush() |
115 session.flush() |