Ejemplo n.º 1
0
 def post(self, request):
     # return JsonResponse(ref.set(400, "用户或密码错误"))
     userInfo = json.loads(request.POST.get("data"))
     print(userInfo, "uuuuuu")
     isUser = authenticate(username=userInfo["username"],
                           password=userInfo["password"])
     if isUser:
         login(request, isUser)
         tk = Token().encodejwt(isUser.username)
         cache.set(str(tk), isUser.username, 60 * 60)
         data = {"token": tk}
         data.update(ref.code(200))
         return JsonResponse(data)
     else:
         return JsonResponse(ref.set(400, "用户或密码错误"))
Ejemplo n.º 2
0
 def dataStorage(cls, newUser, data, num):
     msgList = []
     #"1"DNS多对一 "2" DNS多对多 "5" SQL多对一" 6" SQL多对多
     regExJump = {
         1: {
             "jumptarget": newUser.target,
             "is_jump": False
         },
         2: {
             "jumptarget": data["url"],
             "is_jump": False
         },
         5: {
             "jumptarget": newUser.target,
             "is_jump": True
         },
         6: {
             "jumptarget": data["url"],
             "is_jump": True
         },
     }
     for one in data["textarea"]:
         try:
             temp = re.split("\.", one)[-2:]
             newOne = "www." + temp[0].strip() + "." + temp[1].strip()
         except Exception as f:
             msg = "{}无效域名,添加失败".format(one)
             msgList.append(msg)
             continue
         newJ = Jump.objects.filter(name=newOne)
         if newJ:
             newId = newJ.first().relationship_id
             if newId != data["id"]:
                 other = newJ.first().relationship.username
                 otherMsg = str(other)[:2] + "***" + str(other)[-2:]
                 msg = "警告:{},用户:{}已添加,操作失败".format(one, otherMsg)
                 msgList.append(msg)
                 continue
             _num = int(data["mode"])
             if _num in [5, 6]:
                 isJump = True
             else:
                 isJump = False
             newJ.update(name=newOne,
                         jumptarget=regExJump[num]["jumptarget"],
                         is_jump=isJump,
                         relationship_id=data["id"])
         else:
             oneUserObj = User.objects.filter(id=data["id"]).first()
             JumpObjNum = len(oneUserObj.jump_target.all())
             print("oneUserObj.number", oneUserObj.number,
                   type(oneUserObj.number))
             if (oneUserObj.number - JumpObjNum) <= 0:
                 raise MyException(ref.set(403, "数量不足,无权添加"))
             Jump.objects.create(name=newOne,
                                 jumptarget=regExJump[num]["jumptarget"],
                                 is_jump=regExJump[num]["is_jump"],
                                 relationship_id=data["id"])
     if msgList:
         raise MyException(ref.set(403, msgList))
     return True