--- a/script/lib/iri_tweet/utils.py	Wed Jul 27 12:24:43 2011 +0200
+++ b/script/lib/iri_tweet/utils.py	Wed Jul 27 12:25:45 2011 +0200
@@ -339,13 +339,17 @@
                       help="quiet", metavar="QUIET", default=0)
 
     
-def get_filter_query(session, start_date, end_date, hashtags, tweet_exclude_table):
+def get_filter_query(session, start_date, end_date, hashtags, tweet_exclude_table, user_whitelist):
     
     query = session.query(Tweet).join(EntityHashtag).join(Hashtag)
     if tweet_exclude_table is not None:
         query = query.filter(~Tweet.id.in_(select([tweet_exclude_table.c.id]))) #@UndefinedVariable
         
     query = query.filter(Tweet.created_at >=  start_date).filter(Tweet.created_at <=  end_date)
+
+    if user_whitelist:
+        query = query.join(User).filter(User.screen_name.in_(user_whitelist))
+
     
     if hashtags :
         def merge_hash(l,h):
@@ -354,6 +358,7 @@
         htags = reduce(merge_hash, hashtags, [])
         
         query = query.filter(or_(*map(lambda h: Hashtag.text.contains(h), htags))) #@UndefinedVariable
+        
     
     return query