def __init__(self):
register_default_functions()
StreamListener.__init__(self)
try:
self.username = settings.TWITTER_USERNAME
self.password = settings.TWITTER_PASSWORD
except AttributeError:
print "TWITTER_USERNAME and TWITTER_PASSWORD not defined in settings.py"
self.username = raw_input('Twitter username: '******'Twitter password: ')
self.status_lock = RLock()
self.statuses = []
self.query_builder = gen_query_builder()
self.stream = None
# initializing database engine - could have done this anywhere. do it here for now
StatusSource.set_engine()
def __get_source(self, parsed):
source = parsed.sources[0]
if source == QueryTokens.TWITTER:
return StatusSource.TWITTER_FILTER
elif source.startswith(QueryTokens.TWITTER_SAMPLE):
return StatusSource.TWITTER_SAMPLE
elif source.startswith(QueryTokens.TABLE):
if StatusSource.table_exists(source):
return source
else:
raise DbException("table does not exist")
elif source.startswith(QueryTokens.STREAM):
if not StatusSource.has_saved_stream(source):
raise DbException("stream does not exist")
else:
return source
else:
raise QueryException("Unknown query source: %s" % (source))
def __init__(self, listener, dump_name):
self.listener = listener
self.status_dump = StatusSource.get_saved_stream(dump_name)
