--- a/script/rest/search_twitter.py Fri Aug 12 18:17:27 2011 +0200
+++ b/script/rest/search_twitter.py Wed Aug 17 18:32:07 2011 +0200
@@ -36,7 +36,13 @@
(options, args) = get_option()
twitter = twitter.Twitter(domain="search.twitter.com")
- engine, metadata = models.setup_database('sqlite:///'+args[0], echo=((options.verbose-options.quiet)>0))
+
+ conn_str = args[0].strip()
+ if not re.match("^\w+://.+", conn_str):
+ conn_str = 'sqlite:///' + conn_str
+
+
+ engine, metadata = models.setup_database(conn_str, echo=((options.verbose-options.quiet)>0), create_all=True)
Session = sessionmaker(bind=engine)
session = Session()
try:
@@ -54,7 +60,7 @@
print tweet
tweet_str = anyjson.serialize(tweet)
#invalidate user id
- processor = utils.TwitterProcessor(tweet, tweet_str, None, session, options.token_filename)
+ processor = utils.TwitterProcessor(tweet, tweet_str, None, session, None, options.token_filename)
processor.process()
session.flush()
session.commit()