# 引用企業(yè)微信JDK
from WXcrypt.WXBizMsgCrypt import WXBizMsgCrypt

def work_weixin_api(request):
 # 獲取微信Post參數(shù)
 msg_signature = request.GET.get('msg_signature', '')
 timestamp = request.GET.get('timestamp', '')
 nonce = request.GET.get('nonce', '')
 echostr = request.GET.get('echostr', '')

 # 構(gòu)造微信信息解析方法
 wxcpt = WXBizMsgCrypt(WXTOKEN, WXENCODINGAESKEY, WXCROPID)
 if request.method == 'POST':
  eagle_branch = request.POST.get('eagle_branch', 'master')

  if eagle_branch == "master":
   request_data = request.body
   # 解析接收到的文本
   ret, msg = wxcpt.DecryptMsg(request_data, msg_signature, timestamp,
          nonce)
   request_xml = ET.fromstring(msg)

   # 獲取信息內(nèi)容
   content = request_xml.find("Content").text

   # 獲取信息類型
   msg_type = request_xml.find("MsgType").text

   # 獲取發(fā)送人
   from_user = request_xml.find("FromUserName").text
  else:
   content = request.POST.get('content', '')
   from_user = request.POST.get('from_user', '')

# 加密
def enc_dict(d):

 # 加密
 s = URLSafeSerializer('1234')
 st = s.dumps(d)

 # 加密后再生成基于時(shí)間戳的令牌
 t = TimestampSigner('4567')
 ts = t.sign(st)
 return ts

# 菜單功能的基類
class Function:
 def __init__(self, data):
  self._data = data
  self._functions = []

 # 傳入的方法的描述
 def __str__(self):
  return str(self._data())

 # 返回當(dāng)前對(duì)象類型
 def f_type(self):
  return self._data.f_type

 # 返回當(dāng)前對(duì)象
 def getData(self):
  return self._data

 # 返回所有子菜單
 def getFunctions(self):
  return self._functions

 # 新增子菜單
 def add(self, function):
  self._functions.append(function)

 # 遞歸搜索
 def go(self, num):
  for _, i in enumerate(self._functions):
   if int(num) == _ :
    return i
  return None

 # 新增一個(gè)類型屬性
 def f_type(self):
  return self._data.f_type

 # 遞歸搜索
 def go(self, num):
  for _, i in enumerate(self._functions):
   f_type = i._data().f_type
   # 如果類型是默認(rèn)且存在列表中，或動(dòng)態(tài)生成類型的，直接返回
   if f_type == "default" and int(num) == _ or f_type == "dynamic":
    return i
  return None

class Menu:
 def __init__(self):
  self._head = Function(FunctionNodeBase())
  self.input_text = None

 # 鏈接
 def linkToHead(self, function):
  self._head.add(function)

 # 搜索
 def search(self, text):
  cur = self._head
  for i in text.split('-'):
   if cur.go(i) == None:
    return None
   else:
    self.input_text = i
    cur = cur.go(i)
  return cur

# 空的功能Node
class FunctionNodeBase:
 __metaclass__ = ABCMeta

 def __init__(self,
     user=None,
     f_type="default",
     input_text=None,
     sub_text=None):
  self.user = user
  self.sub_text = sub_text
  self.input_text = input_text
  self.f_type = f_type
  self.f_mark = []

 # 菜單通過(guò)run方法執(zhí)行與生成文本
 @abstractmethod
 def run(self):
  return self.__str__()

 # 描述
 @abstractmethod
 def __str__(self):
  return "菜單樹(shù)頂層"

# 動(dòng)態(tài)生成
class SelectDeploymentTop(FunctionNodeBase):
 # 動(dòng)態(tài)生成的菜單需要聲明f_type
 def __init__(self):
  super().__init__()
  self.f_type = "dynamic"

 def run(self):

  text       = "請(qǐng)選擇事業(yè)部\n\n"
  deployment_list = [i for i in FunctionList.keys()]

  for _, i in enumerate(deployment_list):
   self.f_mark.append(_)
   text += "%s %s\n" % (_, i)

  return text

 # 微信顯示的文本信息
 def __str__(self):
  return "選擇事業(yè)部"

# 靜態(tài)
class MySQLFunctionTop(FunctionNodeBase):
 def __init__(self):
  super().__init__()

 def run(self):
  text = "您選擇的是%s,請(qǐng)選擇您想要操作:\n" % str(self.__str__())
  text += "%s\n" % self.sub_text
  return text

 def __str__(self):
  return "MySQL操作"

def api(tid,user):

 # 實(shí)例化
 menu    = Menu()
 top     = Function(SelectDeploymentTop)
 function_top = Function(FunctionTop)
 mysql_top  = Function(MySQLFunctionTop)

 # 鏈接
 top.add(function_top
 function_top.add(mysql_top)

 # 關(guān)聯(lián)菜單樹(shù)
 menu.linkToHead(top)

 # 遞歸搜索
 function = menu.search(tid)

class redis_db:
 def __init__(self):
  # 按符號(hào)隔開(kāi)
  self.mark = '-'
  self.redis_db = redis.StrictRedis(
   host = host, port=6379, db=1, decode_responses=True)

 # 默認(rèn)回話過(guò)期600秒,每次存儲(chǔ) '-'隔開(kāi)
 def add(self,key,text,Timeout=600):
  if key not in self.keys():
   self.redis_db.set(key,'',ex=Timeout)
  if self.get(key):
   self.redis_db.append(key,self.mark)
  self.redis_db.append(key,text)