def update(self, request, *args, **kwargs):
code = 0
msg = "修改成功"
id = self.kwargs.get("id", None)
user = User.objects.filter(id=id).first()
username = request.data.get("username", None)
realname = request.data.get("realname", None)
phone = request.data.get("phone", None)
email = request.data.get("email", None)
remark = request.data.get("remark", None)
if user:
try:
user.username = username
user.realname = realname
user.email = email
user.phone = phone
user.info = remark
user.save()
except IntegrityError as e:
code = 1
msg = str(e)
return JsonResponse(code=code, msg=msg)
msg = "未找到!"
return JsonResponse(code=1, msg=msg)
def sign_up(request):
phone = request.POST.get("phone", "")
password = request.POST.get("password", "")
if not is_phone(phone):
return JsonResponse(4001, "请输入正确的手机号")
user = User.get_by_phone(phone)
if user:
return JsonResponse(4002, "手机号已被注册")
try:
User.create(phone, password)
except:
return JsonResponse(4003, "注册失败，请稍后再试")
return JsonResponse(200, "注册成功")
def sign_in(request):
phone = request.POST.get("phone", "")
password = request.POST.get("password", "")
user = User.get_by_phone(phone)
if not user:
return JsonResponse(4001, "用户不存在")
if not user.check_password(password):
return JsonResponse(4002, "账户或密码错误")
login(request, user)
# save sessionid
r_db1 = get_redis(**{'db': 1})
session_key = request.session._get_session_key()
r_db1.set('SESSION::%s' % user.id, session_key)
return JsonResponse(200, "登录成功")
def get(self, request, **kwargs):
msg = "错误的参数"
code = 1
host = request.GET.get("host", None)
password = request.GET.get("pwd", None)
user = request.GET.get("user", "root")
port = request.GET.get("port", 22)
if host and user and password and port:
ssh = SSHClient()
ssh.set_missing_host_key_policy(AutoAddPolicy)
try:
ssh.connect(hostname=host,
username=user,
password=password,
port=port,
timeout=5)
except Exception as e:
print(e)
msg = str(e)
code = 1
else:
code = 0
msg = None
ssh.close()
return JsonResponse(code=code, msg=msg)
def create(self, request, *args, **kwargs):
code = 0
msg = None
username = request.data.get("username", None)
realname = request.data.get("realname", None)
many = request.data.get("many", 'false')
password = request.data.get("password", None)
phone = request.data.get("phone", None)
email = request.data.get("email", None)
remark = request.data.get("remark", None)
print(request.data)
if many == "true":
msg_list = []
usernames = username.split("|")
realnames = realname.split("|")
if len(usernames) == len(realnames):
for i in range(len(usernames)):
user = User()
try:
user.username = usernames[i]
user.realname = realnames[i]
user.password = make_password(password)
user.email = email
user.phone = phone
user.save()
except IntegrityError as e:
msg_list.append("用户" + realnames[i] + "创建失败 " + str(e))
else:
msg = "创建成功 ! ," + "|".join(msg_list)
return JsonResponse(code=code, msg=msg)
else:
return JsonResponse(code=1, msg="账户名与姓名不匹配!")
else:
user = User()
try:
user.username = username
user.realname = realname
user.password = make_password(password)
user.email = email
user.phone = phone
user.info = remark
user.save()
except IntegrityError as e:
code = 1
msg = str(e)
return JsonResponse(code=code, msg=msg)
def list(self, request, *args, **kwargs):
queryset = self.filter_queryset(self.get_queryset())
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
serializer = self.get_serializer(queryset, many=True)
return JsonResponse(code=0, msg="success", data=serializer.data)
def get(self, request, **kwargs):
logout(request)
return JsonResponse({"code": 0, "msg": "Success !"})
def get(self, request, **kwargs):
id = request.GET.get("key")
role = get_object_or_404(Role, id=id)
role.status = not role.status
role.save()
return JsonResponse(code=0, msg="success")
def get(self, request, **kwargs):
id = request.GET.get("key")
p = get_object_or_404(UrlPermission, id=id)
p.status = not p.status
p.save()
return JsonResponse(code=0, msg="success")