class SearchInterface:
threads = []
def __init__(self):
self.db = dbFacade()
self.db.connect()
self.db.create_keyspace_and_schema()
def initialize_scorer(self, search_packet):
self.scorer = Scorer(search_packet)
'''
Starts search crawling threads with inputed query string.
'''
def search(self, query, args):
self.threads = []
if args['GooglePlus']:
self.threads.append(
GooglePlusThread(self.db, self.scorer, query, args))
if args['Twitter']:
self.threads.append(
TwitterThread(self.db, self.scorer, query, args))
for thread in self.threads:
thread.start()
for thread in self.threads:
thread.join()
'''
Ends search crawling threads;
waits for them to terminate before continuing.
'''
def stop_search(self):
print "Closing threads.."
for thread in self.threads:
try:
thread.raiseExc(KeyboardInterrupt)
except threading.ThreadError:
pass
for thread in self.threads:
while thread.isAlive():
time.sleep(1)
thread.join()
'''
Retrieves users, calculates user scores, and
updates score in database
'''
def score(self, progress_que):
print "Scoring..\n"
users = self.db.get_users_dict()
progress_que.put(33)
self.populate_user_scores(users, progress_que)
def populate_user_scores(self, users, progress_que):
completion = 0.0
if len(users) == 0:
return
step_progress = ((1.0 / len(users)) * 0.67) * 100
sums = self.fit_scorer(users)
for user, sum in zip(users, sums):
self.update_user_score(user, sum)
completion = completion + step_progress
if completion >= 1:
progress_que.put(1)
completion = completion - 1
if completion < 0:
completion = 0.0
def update_user_score(self, user, sum):
posts = self.db.get_posts(user['username'])
score = self.scorer.get_prob(sum)
user['score'] = score
self.db.insert_user_score(user['username'], user['score'],
user['website'])
def fit_scorer(self, users):
sums = []
for user in users:
posts = self.db.get_posts(user['username'])
sum = self.scorer.sum_posts(posts)
self.scorer.add_point(sum)
sums.append(sum)
self.scorer.fit_graphs()
return sums
