def see_top_tags(cls):
keys = cache.keys(pattern='%s*' %BASIC_CACHE_PREFIX)
set_name = 'TEMP:SET'
for key in keys:
value = float(cache.get(key))
key = key[9:]
cache.zadd(set_name, value, key)
results = cache.zrevrange(name=set_name, start=0, num=LIMIT, withscores=True)
results = [{'name': r[0].decode('utf-8'), 'value': r[1]} for r in results]
from pymongo import Connection
c = Connection()
db = c.names
if results:
db.results.insert(results)
cache.delete(set_name)
return results
def create_action(self):
cache_key = "WEIBO:HOT:%s" %self.user.sns_id
cache.delete(cache_key)
tmp_cache_key = "TEMP:WEIBO:HISTORY:%s:::" %self.user.sns_id
weibo_history = self.user.weibo_history
for text in weibo_history:
terms = seg_txt(text.encode('utf-8'))
for term in terms:
index_key = '%s%s' %(BASIC_TAG_PREFIX, term)
if cache.exists(index_key):
key = tmp_cache_key + term.decode('utf-8')
cache.incr(name=key, amount=1)
keys = cache.keys(pattern="%s*" %tmp_cache_key)
for key in keys:
name = key.split(":::")[1]
value = float(cache.get(key))
cache.zadd(cache_key, value, name)
cache.delete(key)
tag = BasicTag.get_by_name(name=name)
if not tag:
continue
relations = tag.friends
score = tag.score
for f in relations:
items = f.split(':::')
obj_name = items[0]
obj_value = float(items[1])
result = obj_value/50*value
cache.zadd(cache_key, result, obj_name)
results = cache.zrevrange(name=cache_key, start=0, num=30, withscores=True)
tags = [result[0].decode('utf-8') +'__' + str(result[1]) for result in results]
self.user.update(set__tags=tags)
def calculate_relations(cls):
items = cls.objects(friends__size=0)
for item in items:
try:
name = item.name
tag = db.tags.find_one({'name': name})
except:
continue
if not tag:
continue
item_tags = tag['tags']
for item_tag in item_tags:
try:
cls.__calculate_relation(obj_tag=item_tag, item_name=name)
except:
continue
keys = cache.keys('TEMP:FRIENDS:%s*' %item_tag)
for key in keys:
name = key.split(':::')[1]
cache.zadd('TEMP:%s' %item_tag, float(cache.get(key)), name)
cache.delete(key)
results = cache.zrevrange(name='TEMP:%s' %item_tag, start=0, num=DEFAULT_LIMIT, withscores=True)
cache.delete('TEMP:%s' %item_tag)
for r in results:
name = r[0].decode('utf-8')
score = r[1]/1000
if name == item_tag:
continue
try:
obj = cls.objects.get(name=item_tag)
obj.update(add_to_set__friends='%s:::%f' %(name, score))
print("push %s:::%f to tag %s" %(name, score, item_tag))
except :
cls.create_tag(name=item_tag, friends=['%s:::%f' %(name, score),])
print("push %s:::%f to create tag %s" %(name, score, item_tag))
def add(self, word, key, score):
try:
cache.zadd(word, score, key)
except:
logger.debug("cannot add search index key: %s score: %s" %(key, score))
