tools.py

import datetime, hashlib, re
from flask import session, current_app, request, flash
from services import svc_topic
def filter_sql(list):
	for x in range(0,len(list)):
		list[x] = list[x].replace("'", "''").replace('%', '__bfh_')
	return list

def page_turning_info(item_count, current_page = 1, page_count = 12):
	pgs = int(item_count / page_count)
	opg = item_count % page_count
	limit = '0, 1'
	if current_page <= 0 or item_count == 0:
		have = False
	else:
		have = True
		if current_page <= pgs:
			limit = '%d, %d' % (item_count - current_page * page_count, page_count)
		elif current_page is pgs + 1:
			limit = '0, %d' % opg
		else:
			have = False
	return {
		'pgs':pgs,
		'opg':opg,
		'have':have,
		'limit':limit
	}

def get_date_fashion(dt):
	cz  = datetime.datetime.now() - dt
	if cz.days > 0:
		mounths = cz.days/30
		if mounths >=1:
			return '%d月' % int(mounths)
		return '%d天' % cz.days
	else:
		secs = cz.seconds
		minutes = secs/60
		hours = minutes/60
		if hours >=1 :
			return '%d小时' % int(hours)
		elif minutes >= 1:
			return '%d分钟' % int(minutes)
		return '%d秒' % int(secs)

def get_post_device(user_agent):
	if user_agent is None:
		return None
	device = user_agent.split(')')[0].split('(')[-1]
	konwns_rel_devices = ['Letv X500', 'vivo X9', 'PRA-AL00X', 'Linux x86_64', 'iPhone', 'MI 5', 'MI 6', 'Redmi Note 4X', 'MI 4LTE', 'SM919', 
		'Windows NT 10', 'Windows NT 6', 'Windows NT 5', 'google.com', 'baidu.com', 'mj12bot.com',   'MHA-AL00',        'ZUK',     'm3 note', 'Redmi 5 Plus',
		'JMM-AL 10',       'MZ-PRO 6s', 'Le  X820']
	konwns_devices =         ['Le 1s',     'vivo X9', '荣耀8青春版', 'Linux64位',      'iPhone', 'MI 5', 'MI 6', '红米4X',    'MI 4', 'SmartisanT M1', 
		'Windows 10',           'Windows 7',    'Windows XP',             '匿名',           '匿名',         '匿名', 'HUAWEI Mate9', '联想手机', '魅蓝note3', '红米5 Plus', 
		'荣耀V9 Play', '魅族 Pro 6s', 'Le Max2']
	index = -1
	for d in konwns_rel_devices:
		index+=1
		if d in device:
			device = konwns_devices[index]
			break
	return device

def support_at(content):
	users = re.findall('@[^ @\r\n\t,，(;=<>]+ ', content)
	users = list(set(users))
	for user in users:
		content = content.replace(user, '<a href="/profile/%s">%s</a>' % (user[1:][:-1], user))
	content = content.replace(';\r\n', ';<br />').replace(';\n', ';<br />').replace('\n    ', '<br />&nbsp;&nbsp;&nbsp;&nbsp;').replace('    ', '&nbsp;&nbsp;&nbsp;&nbsp;').replace('</script>', '&lt;/script&gt;').replace('<script', '&lt;script')
	return content

def at_notify(content, conn, message):
	users = re.findall('@[^ @\r\n\t,，(;=<>]+ ', content)
	users = list(set(users))
	for user_raw in users:
		user = user_raw[1:-1]
		if user == session['ol_user']['name']:
			continue
		uid = conn.execute("select id from user where name='%s'" % user).first()[0]
		conn.execute("insert into message(content, from_id, to_id) values('%s', 1, %d)" % (message, uid))
def new_topic_notify(content, title, id, conn):
	username = session['ol_user']['name']
	at_notify(content, conn, '[%s](/profile/%s)在发布主题[%s](/t/%d)时提到了你.' % (username, username, title, id))

def new_answer_notify(content, title, id, conn):
	username = session['ol_user']['name']
	at_notify(content, conn, '[%s](/profile/%s)在主题[%s](/t/%d)的答案中提到了你.' % (username, username, title, id))

def update_online_users():
	online_usernames = current_app.online_usernames
	online_users = current_app.online_users
	online_user_updates_queue = current_app.online_user_updates_queue
	def create_user(name, last_time):
		return {
			'name':name,
			'last_time':last_time
		}
	if 'ol_user' in session:
		username = session['ol_user']['name']
	else:
		if hasattr(request, 'username'):
			username = request.username
		else:
			username = request.cookies.get(current_app.session_cookie_name)
		username = hashlib.md5(username.encode('utf-8')).hexdigest() +  ' • ' + get_post_device(request.headers.get('User_Agent')) + '用户'
			
	if username not in online_usernames:
		online_usernames.append(username)
		online_users.append(create_user(username, datetime.datetime.now()))
	else:
		online_user_updates_queue.put(username)

def create_recent_topic_item(row, conn):
	user = conn.execute('select id,`name`, avatar from user where id=%d' % row.author_id).first()
	forum = conn.execute('select id, `name` from forum where id=%d' % row.forum_id).first()
	answer_count = conn.execute('select count(id) from answer where topic_id=%d' % row.id).first()[0]
	answer_info = conn.execute('select post_date, author_id from answer where topic_id=%d order by post_date desc limit 1' % row.id).first()
	if answer_info:
		 answer_author = conn.execute('select id, `name`, avatar from user where id=%d' % answer_info.author_id).first()
	return {
		'title':row.title,
		'id':row.id,
		'post_date':get_date_fashion(row.post_date),
		'author':{
			'id':user.id,
			'name':user.name,
			'avatar':user.avatar
		},
		'forum':{
			'id':forum.id,
			'name':forum.name
		},
		'view_count':row.view_count,
		'answer_count':answer_count,
		'last_reply_date': get_date_fashion(answer_info.post_date) if answer_info else 'None',
		'last_reply_author': {
			'id':answer_author.id,
			'name':answer_author.name,
			'avatar':answer_author.avatar
		} if answer_info else 'None'
	}

def check_auth_topic(topic_id, forum_id, conn):
	if 'ol_user' not in session:
		return ''
	the_forum_id = conn.execute('select forum_id from administrator where author_id=%d' % session['ol_user']['id']).first()
	if the_forum_id == None:
		return ''
	else:
		the_forum_id = the_forum_id[0]
	count = conn.execute('select count(id) from topic where author_id=%d and id=%d' % (session['ol_user']['id'], topic_id)).first()[0]
	if the_forum_id == 0 or forum_id == the_forum_id  or count == 1:
		return '<a href="javascript:deltopic();">删除</a>'
	else:
		return ''

def check_topic_append(topic_id, conn):
	if 'ol_user' not in session:
		return ''
	user = session['ol_user']
	author_id = conn.execute('select author_id from topic where id=%d' % topic_id).first()
	if not author_id:
		return ''
	else:
		author_id = author_id[0]
	auth_code = conn.execute('select forum_id from administrator where author_id=%d' % user['id']).first()
	if user['id'] == author_id or (auth_code and auth_code[0] == 0):
		return '<a href="/append/%d">续贴</a>' % topic_id
	else:
		return ''

def ensure_recent_queue_when_remove(topic_id):
	if topic_id in current_app.all_recent_topic_ids:
		current_app.all_recent_topic_ids.remove(topic_id)
	if topic_id in current_app.art_recent_topic_ids:
		current_app.art_recent_topic_ids.remove(topic_id)
	if topic_id in current_app.trade_recent_topic_ids:
		current_app.trade_recent_topic_ids.remove(topic_id)
	if topic_id in current_app.programming_recent_topic_ids:
		current_app.programming_recent_topic_ids.remove(topic_id)
	if topic_id in current_app.paid_recent_topic_ids:
		current_app.paid_recent_topic_ids.remove(topic_id)
def ensure_recent_queue_when_post(topic_id, forumid_str):
	if len(current_app.all_recent_topic_ids) == 36:
		current_app.all_recent_topic_ids.pop()
	current_app.all_recent_topic_ids.insert(0, topic_id)
	forumid = forumid_str
	if forumid in current_app.programming_forum_ids.split(','):
		if len(current_app.programming_recent_topic_ids) == 36:
			current_app.programming_recent_topic_ids.pop()
		current_app.programming_recent_topic_ids.insert(0, topic_id)
	if forumid in current_app.art_forum_ids.split(','):
		if len(current_app.art_recent_topic_ids) == 36:
			current_app.art_recent_topic_ids.pop()
		current_app.art_recent_topic_ids.insert(0, topic_id)
	if forumid in current_app.trade_forum_ids.split(','):
		if len(current_app.trade_recent_topic_ids) == 36:
			current_app.trade_recent_topic_ids.pop()
		current_app.trade_recent_topic_ids.insert(0, topic_id)
	if svc_topic.price(topic_id) > 0:
		if len(current_app.paid_recent_topic_ids) == 36:
			current_app.paid_recent_topic_ids.pop()
		current_app.paid_recent_topic_ids.insert(0, topic_id)
def ensure_recent_queue_when_reply(topic_id, forumid_str):
	all_ids = current_app.all_recent_topic_ids
	programming_forumids = current_app.programming_forum_ids.split(',')
	programming_ids = current_app.programming_recent_topic_ids
	art_forumids = current_app.art_forum_ids.split(',')
	art_ids = current_app.art_recent_topic_ids
	trade_forumids = current_app.trade_forum_ids.split(',')
	trade_ids = current_app.trade_recent_topic_ids
	paid_ids = current_app.paid_recent_topic_ids
	if topic_id in all_ids:
		all_ids.remove(topic_id)
	all_ids.insert(0, topic_id)
	if forumid_str in programming_forumids:
		if topic_id in programming_ids:
			programming_ids.remove(topic_id)
		programming_ids.insert(0, topic_id)
	if forumid_str in art_forumids:
		if topic_id in art_ids:
			art_ids.remove(topic_id)
		art_ids.insert(0, topic_id)
	if forumid_str in trade_forumids:
		if topic_id in trade_ids:
			trade_ids.remove(topic_id)
		trade_ids.insert(0, topic_id)
	if svc_topic.price(topic_id) > 0:
		if topic_id in paid_ids:
			paid_ids.remove(topic_id)
		paid_ids.insert(0, topic_id)

def transactional(func):
	def t_wrapper(*args, **kwargs):
		conn = current_app.connect()
		conn.begin()
		try:
			result = func(*args, **kwargs)
			conn.commit()
		except Exception as e:
			conn.rollback()
			flash(e)
			return render_template('404.html'), 404
		return result
	return t_wrapper