s.py

#!/usr/bin/env python
# encoding: utf-8
"""
s.py

Created by Hilary Mason on 2010-08-15.
Copyright (c) 2010 Hilary Mason. All rights reserved.
"""


import sys, os
import re
import datetime
from optparse import OptionParser

import pymongo

import settings
from lib import mongodb
from lib import display


class Search(object):
    def __init__(self, options, args):
        self.debug = options.debug
        self.db = mongodb.connect('tweets')
        d = display.Display()        

        if options.user_search:
            twitterers = self.user_search(options.user_search, int(options.num))
            d.display_users(twitterers)
        else:
            tweets = self.tweet_search(args, int(options.num))
            d.display_tweets(tweets)
        
        
    def tweet_search(self, query_terms, num=10):
        r = re.compile(' '.join(query_terms), re.I)
        tweets = []
        for t in self.db['tweets'].find(spec={'text': r }).sort('created_at',direction=pymongo.DESCENDING):
            t['_display'] = True
            t['_datetime'] = datetime.datetime.strftime(t['created_at'], "%I:%M%p, %b %d, %Y %Z")
            tweets.append(t)
        
        if self.debug:
            print "%s total results" % (len(tweets))
            
        return tweets[:num]
        
    def user_search(self, user_terms, num=10):
        r = re.compile(user_terms, re.I)
        tweets = []
        
        # search username
        for t in self.db['users'].find(spec={'_id': r}).sort('_updated', direction=pymongo.DESCENDING):
            t['_display'] = True
            t['_datetime'] = datetime.datetime.strftime(t['_updated'], "%I:%M%p, %b %d, %Y %Z")
            tweets.append(t)

        usernames = [t['_id'] for t in tweets]

        # search name
        for t in self.db['users'].find(spec={'name': r}).sort('_updated', direction=pymongo.DESCENDING):
            if t['_id'] not in usernames:
                t['_display'] = True
                t['_datetime'] = datetime.datetime.strftime(t['_updated'], "%I:%M%p, %b %d, %Y %Z")
                tweets.append(t)
            
        



        if self.debug:
            print "%s total results" % (len(tweets))
            
        return tweets[:num]
    
    
if __name__ == "__main__":
    parser = OptionParser("usage: %prog [options] query terms")
    parser.add_option("-d", "--debug", dest="debug", action="store_true", default=False, help="set debug mode = True")
    parser.add_option("-n", "--num", dest="num", action="store", default=10, help="number of tweets to retrieve")
    parser.add_option("-u", "--user", dest="user_search", action="store", default=None, help="search users")
    (options, args) = parser.parse_args()

    t = Search(options, args)