script/rest/search_twitter.py
changeset 255 500cd0405c7a
parent 242 cdd7d3c0549c
child 289 a5eff8f2b81d
--- 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()