logger.setLevel(logging.DEBUG)
# create file handler which logs even debug messages
fh = logging.FileHandler(u'../logs/exetask.log')
fh.setLevel(logging.DEBUG)
# create console handler with warn log level
ch = logging.StreamHandler()
ch.setLevel(logging.WARN)
# create logger output formater
formatter = logging.Formatter(u'%(asctime)s %(name)s %(levelname)s %(message)s')
fh.setFormatter(formatter)
ch.setFormatter(formatter)
logger.addHandler(fh)
logger.addHandler(ch)
# setup weibo
[access_token, expires_in] = djangodb.get_weibo_auth_info(3041970403)
if time.time() > float(expires_in):
raise Exception(u'授权过期了，with expires_in:%s' % expires_in)
weibo = weiboAPI.weiboAPI(access_token=access_token, expires_in=expires_in, u_id=3041970403)
reqInterval(31)
logger.info(u'Sina Weibo 登录信息:\t' + weibo.getUserInfo()[u'name'])
# setup google reader
from libgreader import readerAPI
[access_token, refresh_token, access_expires] = djangodb.get_google_auth_info(u_id=1)
reader = readerAPI.readerAPI(u_id=1, access_token=access_token, \
refresh_token=refresh_token, expires_access=access_expires)
reqInterval(31)
logger.info(u'Google Reader 登录信息:\t' + reader.getUserInfo()[u'userName'])
def remindUserTopicUpdates(topicTitle):
try:
logger.debug(u'Start remind user for topic:%s' % topicTitle)
topic = djangodb.Topic.objects.get(title=topicTitle)
if topic.alive():
topic_news = topic.news_set.all()[0]
else:
logger.warn(u'topic:%s is already dead, add unsubscribe task!' % topicTitle)
djangodb.add_task(topic=topic, type=u'unsubscribe')
return False
except:
logger.exception(u'topic or news may not exist in database')
return False
topicWatchers = topic.watcher.all()
topicWatcherWeibo = topic.watcher_weibo.all()
'''
得到订阅该话题的所有用户，分两类，已经授权的和没有授权的
'''
watcherWithAuth = set()
watcherWithoutAuth = set()
for watcher in topicWatchers:
if watcher.has_oauth():
watcherWithAuth.add(watcher)
else:
watcherWithoutAuth.add(watcher)
'''
筛选出其中需要提醒的用户，分为四类：
watcherWithoutStatusAndAuth
watcherWithStatusAndAuth
watcherWithStatusWithoutAuth
watcherWithoutStatusWithAuth
'''
watcherWithStatusAndAuth = set()
watcherWithStatusWithoutAuth = set()
for watcherWeibo in topicWatcherWeibo:
watcher = watcherWeibo.user
if not watcher.to_remind():
# 去掉不需要提醒的用户
continue
watcher.original_weibo = watcherWeibo # 人工添加的字段
if watcher in watcherWithAuth:
watcherWithStatusAndAuth.add(watcher)
elif watcher in watcherWithoutAuth:
watcherWithStatusWithoutAuth.add(watcher)
# 去掉不需要提醒的用户
watcherWithoutStatusAndAuth = set([watcher for watcher in (watcherWithoutAuth - watcherWithStatusWithoutAuth) if watcher.to_remind()])
watcherWithoutStatusWithAuth = set([watcher for watcher in (watcherWithAuth - watcherWithStatusAndAuth) if watcher.to_remind()])
_shorturl = weibo.getShortUrl(Conf.get_timeline_url(topic.id))
_msg = u'#%s#有新进展：%s 『%s』' % (topicTitle, topic_news.title, _shorturl)
reqInterval(61)
logger.debug(u'topicWatcherWeibo:%s' % topicWatcherWeibo)
logger.debug(u'topicWatchers:%s' % topicWatchers)
logger.debug(u'watcherWithAuth:%s' % watcherWithAuth)
logger.debug(u'watcherWithoutAuth:%s' % watcherWithoutAuth)
logger.debug(u'posgMsg:%s' % _msg)
_user_reminded = []
'''
先更新有授权同时也发过微博的用户的状态
'''
for watcher in watcherWithStatusAndAuth:
'''
自己转发或评论自己的原始微博提醒自己，如果允许提醒他人的话，顺带提醒～
'''
[access_token, expires_in] = djangodb.get_weibo_auth_info(watcher.weiboId)
_weibo = weiboAPI.weiboAPI(access_token=access_token, \
expires_in=expires_in, \
u_id=watcher.weiboId)
watcher_btw = None
if watcher.allow_remind_others and watcherWithStatusWithoutAuth:
watcher_btw = watcherWithStatusWithoutAuth.pop()
postMsg = _msg + u' @' + watcher_btw.weiboName + u' 顺便提醒你一下～'
else:
postMsg = _msg + u' @' + watcher.weiboName
logger.info(u'remind user:%s \ntopic:%s \nupdate with msg:%s\noriginalWeibo:%s' % \
(watcher.weiboName, topicTitle, postMsg, watcher.original_weibo))
res = {}
try:
if watcher.repost_remind:
res[u'type'] = u'repost status'
res[u'status'] = _weibo.repostStatus(weibo_id=watcher.original_weibo.weibo_id, content=postMsg)
else:
res[u'type'] = u'comment status'
res[u'status'] = _weibo.postComment(weibo_id=watcher.original_weibo.weibo_id, content=postMsg)
except APIError, err:
logger.warn(u'%s failed:\t%s' % (res[u'type'], err.error))
if err.error == u'target weibo does not exist!':
topic.watcher_weibo.remove(watcher.original_weibo)
topic.watcher.remove(watcher)
logger.info(u'remove watcher:%s and delete watcherWeibo:%s' % (watcher, watcher.original_weibo))
watcher.original_weibo.delete()
else:
logger.exception(u'')
else:
logger.info(u'%s Succeed!' % res[u'type'])
if watcher_btw:
watcher_btw.add_remind()
watcher.add_remind()
logger.info(u'added remind history for user: [%s, %s]' % (watcher.weiboName, watcher_btw))
finally:
logger.setLevel(logging.DEBUG)
# create file handler which logs even debug messages
fh = logging.FileHandler(u"../logs/checkweibo.log")
fh.setLevel(logging.DEBUG)
# create console handler with warn log level
ch = logging.StreamHandler()
ch.setLevel(logging.WARN)
# create logger output formater
formatter = logging.Formatter(u"%(asctime)s %(name)s %(levelname)s %(message)s")
fh.setFormatter(formatter)
ch.setFormatter(formatter)
logger.addHandler(fh)
logger.addHandler(ch)
# setup weibo
[access_token, expires_in] = djangodb.get_weibo_auth_info(3041970403)
if time.time() > float(expires_in):
raise Exception(u"授权过期了，with expires_in:%s" % str(expires_in))
weibo = weiboAPI.weiboAPI(access_token=access_token, expires_in=expires_in, u_id=3041970403)
reqInterval(31)
logger.info(u"Sina Weibo 登录信息:%s" % weibo.getUserInfo()[u"name"])
def fetchUserMention():
"""
获取微博上用户的订阅话题,之后更新订阅
用户订阅话题的方式可以发布包含:@狗狗追踪 *订阅或取消订阅* #话题内容#的微博；或者转发上述微博
TODO:
1. mentions只获取了前50条
"""