def callback():
code = request.args.get('code', '')
client = get_client()
try:
client.set_code(code)
r = client.token_info
uid = r['uid']
access_token = r['access_token']
expires_in = r['expires_at']
except:
flash(u'微博登录没有成功')
return redirect(url_for('simple.login'))
try:
userinfo = client.get('users/show', uid=uid)
screen_name = userinfo['screen_name']
profile_image_url = userinfo['profile_image_url']
mongo.db.users.update({'uid': uid},
{'$set': {'uid': uid, 'access_token': access_token,
'expires_in': expires_in, 'screen_name': screen_name,
'profile_image_url': profile_image_url}}, upsert=True, safe=True)
session['uid'] = uid
session['screen_name'] = screen_name
session['profile_image_url'] = profile_image_url
return redirect(url_for('simple.index'))
except Exception:
flash(u'获取用户微博信息没有成功')
return redirect(url_for('simple.login'))
def index(mid, page=None):
user = login_user(session)
client = get_client(user['access_token'], user['expires_in'])
per_page = 200
total_page = 0
reposts_count = 0
source_weibo = None
if page is None:
source_weibo = client.get('statuses/show', id=mid)
mongo.db.all_source_weibos.update({'id': source_weibo['id']}, source_weibo, upsert=True)
items2mongo(resp2item_v2(source_weibo))
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
page = total_page
else:
source_weibo = mongo.db.all_source_weibos.find_one({'id': mid})
if source_weibo is None:
return ''
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
try:
reposts = client.get('statuses/repost_timeline', id=mid,
count=200, page=page)['reposts']
# 如果reposts为空，且是最开始访问的一页，有可能是页数多算了一页,直接将页数减一页跳转
if reposts == [] and total_page > 1 and page == total_page:
return redirect(url_for('graph.index', mid=mid, page=page - 1))
items = []
for repost in reposts:
items.extend(resp2item_v2(repost))
items2mongo(items)
for item in items:
if isinstance(item, WeiboItem) and item['id'] != source_weibo['id']:
item = item.to_dict()
item['source_weibo'] = source_weibo['id']
mongo.db.all_repost_weibos.update({'id': item['id']}, item, upsert=True)
except RuntimeError:
pass
reposts = list(mongo.db.all_repost_weibos.find({'source_weibo': source_weibo['id']}))
if reposts == []:
return ''
page_count = total_page - page + 1 if total_page >= page else 0
tree, tree_stats = reposts2tree(source_weibo, reposts, per_page, page_count)
graph, max_depth, max_width = tree2graph(tree)
tree_stats['max_depth'] = max_depth
tree_stats['max_width'] = max_width
# 存储转发状态
tree_stats['id'] = mid
tree_stats['page'] = page
mongo.db.tree_stats.update({'id': mid, 'page': page}, tree_stats, upsert=True, w=1)
return graph
def index():
user = login_user(session)
query = request.args.get('query', '')
if user:
client = get_client(user['access_token'], user['expires_in'])
results = []
for s in client.get('search/suggestions/at_users', q=query, type=0, count=10):
results.append(s['nickname'])
return jsonify(query=query, suggestions=results)
return jsonify(query=query, suggestions=[])
def index():
user = login_user(session)
q = request.args.get('q', '')
q = q.strip('@ \r\n\t')
client = get_client(user['access_token'], user['expires_in'])
try:
target_user = client.get('users/show', screen_name=q)
mongo.db.all_visited_users.update({'id': target_user['id']}, target_user, upsert=True)
items2mongo(resp2item_v2(target_user))
return redirect(url_for('search.weibos_by_uid_and_page', uid=target_user['id']))
except RuntimeError:
flash(u'您输入的昵称不存在,请重新输入')
return redirect(url_for('simple.index'))
def index(mid, page=None):
if page is None:
per_page = 200
user = login_user(session)
client = get_client(user['access_token'], user['expires_in'])
source_weibo = client.get('statuses/show', id=mid)
mongo.db.all_source_weibos.update({'id': source_weibo['id']}, source_weibo, upsert=True)
items2mongo(resp2item_v2(source_weibo))
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
page = total_page
return redirect(url_for('show_graph.index', mid=mid, page=page))
screen_name = session['screen_name']
profile_image_url = session['profile_image_url']
return render_template('graph.html', btnuserpicvisible='inline',
btnloginvisible='none',
screen_name=screen_name, profile_image_url=profile_image_url,
mid=mid,
page=page)
def index(mid, page=None):
client = get_client()
per_page = 200
total_page = 0
reposts_count = 0
source_weibo = None
if page is None:
source_weibo = client.get('/showBatch', ids=mid)['statuses'][0]
items = resp2item_search(source_weibo, 8)
items2mongo(items, weibo_mode='REPOST')
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
page = total_page
else:
source_weibo = mongo.db.master_timeline_weibo_repost.find_one({'id': mid})
if source_weibo is None:
return ''
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
if page == 0:
return ''
# try:
reposts = client.get('/queryReList', tweetId=mid, count=per_page, page=page)['reposts']
# 如果reposts为空，且是最开始访问的一页，有可能是页数多算了一页,直接将页数减一页跳转
if reposts == [] and total_page > 1 and page == total_page:
return redirect(url_for('graph.index', mid=mid, page=page - 1))
items = []
for repost in reposts:
items.extend(resp2item_search(repost, 4))
items2mongo(items, weibo_mode='REPOST')
for item in items:
if isinstance(item, WeiboItem_search) and item['id'] != source_weibo['id']:
item = item.to_dict()
item['source_weibo'] = source_weibo['id']
mongo.db.master_timeline_weibo_repost.update({'id': item['id']}, item, upsert=True)
#except RuntimeError:
# pass
reposts = list(mongo.db.master_timeline_weibo_repost.find({'source_weibo': source_weibo['id']}))
if reposts == []:
return ''
page_count = total_page - page + 1 if total_page >= page else 0
tree, tree_stats = reposts2tree(source_weibo, reposts, per_page, page_count)
sort_results, max_depth, max_width = tree2graph(tree)
tree_stats['max_depth'] = max_depth
tree_stats['max_width'] = max_width
# 存储转发状态
tree_stats['id'] = mid
tree_stats['page'] = page
mongo.db.tree_stats.update({'id': mid, 'page': page}, tree_stats, upsert=True, w=1)
return sort_results
def login():
client = get_client()
url = client.authorize_url
flash(u'请用新浪微博登录')
return render_template('simple.html', btnuserpicvisible='none', btnloginvisible='inline', url=url)
def weibos_by_uid_and_page(uid, page=1):
user = login_user(session)
q = request.args.get('q', '')
q = q.strip('@ \r\n\t')
auto_redirect = request.args.get('auto_redirect')
if auto_redirect:
auto_redirect = int(auto_redirect)
target_user = mongo.db.all_visited_users.find_one({'id': uid})
tar_screen_name = target_user['screen_name']
tar_profile_image_url = target_user['profile_image_url']
tar_location = target_user['location']
if q and auto_redirect and auto_redirect > 3:
statuses = []
else:
try:
client = get_client(user['access_token'], user['expires_in'])
statuses = client.get('statuses/user_timeline', uid=uid, count=50, page=page)['statuses']
items = []
for status in statuses:
items.extend(resp2item_v2(status))
items2mongo(items)
if q:
temp_statuses = []
for status in statuses:
if ('text' in status and q in status['text']) or ('retweeted_status' in status and q in status['retweeted_status']['text']):
temp_statuses.append(status)
statuses = temp_statuses
if statuses == []:
page += 1
if auto_redirect:
auto_redirect += 1
else:
auto_redirect = 1
return redirect(url_for('search.weibos_by_uid_and_page', uid=uid,
page=page) + '?q=%s&auto_redirect=%s' % (q, auto_redirect))
except RuntimeError:
flash(u'获取微博信息失败,请刷新')
statuses = []
if statuses == []:
flash(u'没有搜索到相关微博,请尝试下一页或者采用其他关键词')
for i in xrange(len(statuses)):
weibo_url = base62.weiboinfo2url(statuses[i]['user']['id'], statuses[i]['mid'])
statuses[i]['weibo_url'] = weibo_url
screen_name = session['screen_name']
profile_image_url = session['profile_image_url']
has_prev = True if page > 1 else False
has_next = True # 默认始终有下一页
page_url = lambda page: url_for('search.weibos_by_uid_and_page', uid=uid,
page=page)
return render_template('weibolist.html', btnuserpicvisible='inline',
btnloginvisible='none',
screen_name=screen_name, profile_image_url=profile_image_url,
tar_screen_name=tar_screen_name,
tar_profile_image_url=tar_profile_image_url,
tar_location=tar_location,
statuses=statuses,
page=page,
has_prev=has_prev,
has_next=has_next,
page_url=page_url
)
def index(mid, page=None):
client = get_client()
per_page = 200
total_page = 0
reposts_count = 0
source_weibo = None
if page is None:
source_weibo = client.get('/showBatch', ids=mid)['statuses'][0]
items = resp2item_search(source_weibo, 8)
items2mongo(items, weibo_mode='REPOST')
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
page = total_page
else:
source_weibo = mongo.db.master_timeline_weibo_repost.find_one(
{'id': mid})
if source_weibo is None:
return ''
reposts_count = source_weibo['reposts_count']
total_page = int(math.ceil(reposts_count * 1.0 / per_page))
if page == 0:
return ''
# try:
reposts = client.get('/queryReList',
tweetId=mid,
count=per_page,
page=page)['reposts']
# 如果reposts为空，且是最开始访问的一页，有可能是页数多算了一页,直接将页数减一页跳转
if reposts == [] and total_page > 1 and page == total_page:
return redirect(url_for('graph.index', mid=mid, page=page - 1))
items = []
for repost in reposts:
items.extend(resp2item_search(repost, 4))
items2mongo(items, weibo_mode='REPOST')
for item in items:
if isinstance(item,
WeiboItem_search) and item['id'] != source_weibo['id']:
item = item.to_dict()
item['source_weibo'] = source_weibo['id']
mongo.db.master_timeline_weibo_repost.update({'id': item['id']},
item,
upsert=True)
#except RuntimeError:
# pass
reposts = list(
mongo.db.master_timeline_weibo_repost.find(
{'source_weibo': source_weibo['id']}))
if reposts == []:
return ''
page_count = total_page - page + 1 if total_page >= page else 0
tree, tree_stats = reposts2tree(source_weibo, reposts, per_page,
page_count)
sort_results, max_depth, max_width = tree2graph(tree)
tree_stats['max_depth'] = max_depth
tree_stats['max_width'] = max_width
# 存储转发状态
tree_stats['id'] = mid
tree_stats['page'] = page
mongo.db.tree_stats.update({
'id': mid,
'page': page
},
tree_stats,
upsert=True,
w=1)
return sort_results
