from flask import request
from functools import wraps
import tokenlib as TK
import json
import sys

# 我不知道还有没有更好的方法，如果有的话麻烦把下面这几行改掉
# 不要吐槽这堆奇奇怪怪的变量名了。。全部改掉太麻烦了
# 在其他文件中请使用：json_data(),tkData()
# ！！！注意上面这两个括号！！！
'''
postdata = {}
def json_data():
	return postdata

tkst = TK.BAD
tkdata = {}
def tkStatus():
	return tkst

def tkData():
	return tkdata
'''
# 以后把调试的代码写在这边，把一些公用的功能也可以移到这边
# 在所有函数名前面加上@Deco
# 这样路由的函数直接返回一个字典就好了
def Deco(func):
	@wraps(func)
	def wrapper(*args,**kwargs):
		# tmp,sys.stdout=sys.stdout,open("logs/debug.log","w+") # 重定向到文件输出 #
		# print("Entering Function->%s:"%func.__name__)
		postdata = {}
		tkst = TK.BAD
		tkdata = {} # 重要！！
		try: # 为了防止空POST出锅
			postdata=json.loads(request.get_data().decode("utf-8"))
			# print("Postdata:",postdata) # 加载到的POST数据
		except:
			postdata={}
			# print("No Postdata loaded.")

		if not "NoToken" in func.__name__:
		# 为了判断是否需要Token验证
		# 我知道这很不好，但是带参数的修饰器和Flask冲突了（估计是）
		# 所以请在不用Token的函数名后面加上"_NoToken"
			try: # 获取Token
				tkst, tkdata=TK.readToken(request.headers.get("Authorization"))
				# print("Loading Token:",tkst, tkdata)
				if tkst==TK.EXPIRED:
					return json.dumps({'type':'ERROR', 'message':"token过期，请重新登陆"})
				elif tkst==TK.BAD:
					return json.dumps({'type':'ERROR', 'message':"token失效，请重新登陆"})
			except:
				tksk=TK.ERROR
				tkdata={}
				return json.dumps({'type':'ERROR', 'message':"未获取到Token，请重新登陆"})
		# 上面这一段是在上面的if里面的啊，要加上缩进的

		# try:
		r = func(*args, **kwargs, json_data = postdata, token_data = tkdata)
		# print("result->",r) # 函数返回的JSON
		# 如果想做错误输出的话加在这里
		# sys.stdout=tmp # 重新回到控制台输出 #
		return json.dumps(r)
		# except:
		# 	# 理论上不应该有这个
		# 	# 出现这种情况说明代码锅了或者前端接口错了
		# 	# sys.stdout=tmp # 重新回到控制台输出 #
		# 	return json.dumps({'type':'ERROR','message':'未知错误'})
	return wrapper