# -*- coding: utf-8 -*-
# 推广系统对外接口，提供与客户端之间的广告信息接口及每分钟执行一次的缓存变更操作
# http://ad.kingsome.cn/webapp/index.php?c=Ops&a=getAdList&body={"gameid":1004,"locationid":1001}
# http://ad.kingsome.cn/webapp/index.php?c=Ops&a=upAdRecording&adid=1002
import tornado.ioloop
import tornado.web
import json
from myredis.myredis import my_redis
import datetime
from mysql.mmysql import MysqlBase
from config import mysql_promotion_config
from log.mylog import define_logger
import logging
from config import BEGIN, END, ad_list_interface_port
import pdb


define_logger("/data/logs/ad_interface_tornado.log")
log = logging.getLogger(__name__)

limit = 2


def send_cache_data():
	mydb = MysqlBase(**mysql_promotion_config)
	log.info("start update cache !")
	now = datetime.datetime.today().strftime("%Y-%m-%d %H:%M:%S")
	all = []
	# 添加无天数限定的记录
	get_full_data = f"select id,name,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort,companyid,locationid,gameid from ad where begin_time='{BEGIN}' or end_time='{END}'"
	full_data = mydb.query(get_full_data)

	if full_data:
		for line in full_data:
			if line:
				item = {}
				try:
					item['id'], item['name'], item['ad_num'], item['ad_title'], item['ad_body'], item['ad_image'], item['ad_url'], \
						item['ad_sort'], item['companyid'], item['locationid'], item['gameid'] = line
					all.append(item)
				except Exception:
					log.error("split data failed", exc_info=True)

	# 添加有天数限定的记录
	get_data_sql = f"select id,name,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort,companyid,locationid,gameid from ad where '{now}'>begin_time and '{now}'<end_time ;"

	data = mydb.query(get_data_sql)
	if data:
		for line in data:
			if line:
				item = {}
				try:
					item['id'], item['name'], item['ad_num'], item['ad_title'], item['ad_body'], item['ad_image'], item['ad_url'], \
						item['ad_sort'], item['companyid'], item['locationid'], item['gameid'] = line
					all.append(item)
				except Exception:
					log.error("split data failed", exc_info=True)
	# 检查ID是否存在播放列表中，以及播放次数是否完毕
	if all:

		# log.info(f"get data was {all}!\n")
		for line in all:
			key = f"ad::{line.get('gameid', 0)}::{line.get('locationid', 0)}"
			if line['ad_num'] > 0:
				num = my_redis.get(f"ad::{line['id']}::num")
				if not num:
					num = 0
				if line['ad_num'] > num:
					my_redis.sadd(key, line['id'])
					my_redis.expire(key, 120)
					log.info(f"add {line['id']} to {key} ,num was {line['ad_num']}!")
			elif line['ad_num'] == 0:
				my_redis.sadd(key, line['id'])
				my_redis.expire(key, 120)
				log.info(f"add {line['id']} to {key} ,num was unlimit !")
			else:
				log.error(f"get ad_num from mysql failed! ad_num={line['ad_num']}")

			if not my_redis.exists(f"ad::{line['id']}::info"):
				my_redis.hmset(f"ad::{line['id']}::info", line)
				my_redis.expire(f"ad::{line['id']}::info", 3600 * 24 * 7)
				log.info(f"add ad::{line['id']}::info to redis!")


class DispatchHandler(tornado.web.RequestHandler):

	def get(self):
		if self.get_query_argument('c') == 'Ops' and self.get_query_argument('a') == 'selfChecking':
			self._selfCheckingHandler()
		elif self.get_query_argument('c') == 'Ops' and self.get_query_argument('a') == 'getAdList':
			self._selfGetAdList()
		elif self.get_query_argument('c') == 'Ops' and self.get_query_argument('a') == 'upAdRecording':
			self._upAdRecording()
		else:
			self.write("pls check args!")

	def _upAdRecording(self):
		try:
			adid = self.get_query_argument('adid')
		except Exception:
			result = {'errcode': 2, "errmsg": f"get args failed`"}
			log.error(result, exc_info=True)
			self.write({'errcode': 1, "errmsg": 'get adid failed!'})

		if adid:
			key = f"ad::{adid}::num"
			my_redis.incr(key, amount=1)
			self.write({'errcode': 0, "errmsg": '', "message": f"{adid} incr success!"})


	def _selfCheckingHandler(self):
		self.write(json.dumps({
			'errcode': 0, 'errmsg': '', 'healthy': 1, 'max_rundelay': 10
			}, separators=(',', ':')))


	def _selfGetAdList(self):
		try:
			input = json.loads(self.get_query_argument('body'))
			gameid = input['gameid']
			locationid = input['locationid']
		except Exception as e:
			result = {'errcode': 2, "errmsg": f"get args failed,{str(e)}"}
			log.error(result)
			self.write_error(2)

		if gameid and locationid:
			key = f"ad::{gameid}::{locationid}"

			ids = my_redis.smembers(key)
			info = []
			if not ids:
				result = {'errcode': 0, "errmsg": '', "message": {"totoal": len(info), "result": info}}
			else:
				try:
					pdb.set_trace()
					# 如果取得的记录条数大于预设，扔掉多余的记录,当前采用的是随机选择，以后可能需要添加加权选择
					if limit < len(ids):
						ids = ids[0:int(limit)]
					for id in ids:
						temp = my_redis.hgetall(f"ad::{id}::info")
						info.append(temp)
					result = {'errcode': 0, "errmsg": '', "message": {"totoal": len(info), "result": info}}
				except Exception as e:
					result = {'errcode': 1, "errmsg": e}
			self.write(result)
		else:
			result = {'errcode': 2, "errmsg": f"get args failed!"}
			self.write(result)


def make_app():
	return tornado.web.Application([(r"/webapp/index[\.]php", DispatchHandler)])


if __name__ == "__main__":
	# send_cache_data()
	print('start!')
	app = make_app()
	app.listen(ad_list_interface_port)
	tornado.ioloop.PeriodicCallback(send_cache_data, 60000).start()
	tornado.ioloop.IOLoop.current().start()