diff -r e6b328970ee8 -r ffb0a6d08000 script/lib/iri_tweet/utils.py --- 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