def get(self, request, pk): user = User.get_by_pk(pk) form = UserForm(instance=user) return render(request, "my_admin/user_detail.html", context={ "form": form, "user": user })
def post(self, request, node_id): node = VmessNode.get_or_none_by_node_id(node_id) if not node: return HttpResponseNotFound() log_time = get_current_datetime() node_total_traffic = 0 need_clear_cache = False trafficlog_model_list = [] user_model_list = [] for log in request.json["user_traffics"]: user_id = log["user_id"] u = int(log["ut"] * node.enlarge_scale) d = int(log["dt"] * node.enlarge_scale) # 个人流量增量 user = User.get_by_pk(user_id) user.download_traffic += d user.upload_traffic += u user.last_use_time = log_time user_model_list.append(user) if user.overflow: need_clear_cache = True # 个人流量记录 trafficlog_model_list.append( UserTrafficLog( node_type=UserTrafficLog.NODE_TYPE_VMESS, node_id=node_id, user_id=user_id, download_traffic=u, upload_traffic=d, ) ) # 节点流量增量 node_total_traffic += u + d # 节点流量记录 VmessNode.increase_used_traffic(node_id, node_total_traffic) # 流量记录 UserTrafficLog.objects.bulk_create(trafficlog_model_list) # TODO 在线IP # 个人流量记录 User.objects.bulk_update( user_model_list, ["download_traffic", "upload_traffic", "last_use_time"], ) # 节点在线人数 NodeOnlineLog.add_log( NodeOnlineLog.NODE_TYPE_VMESS, node_id, len(request.json["user_traffics"]) ) # check node && user traffic if node.overflow: node.enable = False if need_clear_cache or node.overflow: node.save() return JsonResponse(data={})
def post(self, request, pk): user = User.get_by_pk(pk) form = UserForm(request.POST, instance=user) if form.is_valid(): form.save() messages.success(request, "数据更新成功", extra_tags="修改成功") return HttpResponseRedirect(reverse("sspanel:admin_user_list")) else: messages.error(request, "数据填写错误", extra_tags="错误") context = {"form": form, "user": user} return render(request, "my_admin/user_detail.html", context=context)
def get(self, request): token = request.GET.get("token") if not token: return HttpResponseNotFound() user = User.get_by_pk(encoder.string2int(token)) sub_links = user.get_sub_links() sub_links = base64.b64encode(bytes(sub_links, "utf8")).decode("ascii") resp = StreamingHttpResponse(sub_links) resp["Content-Type"] = "application/octet-stream; charset=utf-8" resp["Content-Disposition"] = "attachment; filename={}.txt".format(token) resp["Cache-Control"] = "no-store, no-cache, must-revalidate" resp["Content-Length"] = len(sub_links) return resp
def post(self, request, node_id): """ 这个接口操作比较重,所以为了避免发信号 所有写操作都需要用BULK的方式 1 更新节点流量 2 更新用户流量 3 记录节点在线IP 4 关闭超出流量的节点 """ ss_node = SSNode.get_or_none_by_node_id(node_id) if not ss_node: return HttpResponseNotFound() data = request.json["data"] node_total_traffic = 0 log_time = get_current_datetime() active_tcp_connections = 0 need_clear_cache = False user_model_list = [] trafficlog_model_list = [] online_ip_log_model_list = [] for user_data in data: user_id = user_data["user_id"] u = int(user_data["upload_traffic"] * ss_node.enlarge_scale) d = int(user_data["download_traffic"] * ss_node.enlarge_scale) # 个人流量增量 user = User.get_by_pk(user_id) user.download_traffic += d user.upload_traffic += u user.last_use_time = log_time user_model_list.append(user) if user.overflow: need_clear_cache = True # 个人流量记录 trafficlog_model_list.append( UserTrafficLog( node_type=UserTrafficLog.NODE_TYPE_SS, node_id=node_id, user_id=user_id, download_traffic=u, upload_traffic=d, ) ) # 节点流量增量 node_total_traffic += u + d # active_tcp_connections active_tcp_connections += user_data["tcp_conn_num"] # online ip log for ip in user_data.get("ip_list", []): online_ip_log_model_list.append( UserOnLineIpLog(user_id=user_id, node_id=node_id, ip=ip) ) # 用户流量 User.objects.bulk_update( user_model_list, ["download_traffic", "upload_traffic", "last_use_time"], ) # 节点流量记录 SSNode.increase_used_traffic(node_id, node_total_traffic) # 流量记录 UserTrafficLog.objects.bulk_create(trafficlog_model_list) # 在线IP UserOnLineIpLog.objects.bulk_create(online_ip_log_model_list) # 节点在线人数 NodeOnlineLog.add_log( NodeOnlineLog.NODE_TYPE_SS, node_id, len(data), active_tcp_connections ) # check node && user traffic if ss_node.overflow: ss_node.enable = False if need_clear_cache or ss_node.overflow: ss_node.save() return JsonResponse(data={})
def get(self, request, user_id): user = User.get_by_pk(user_id) user.delete() messages.success(request, "成功啦", extra_tags="删除用户") return HttpResponseRedirect( reverse("sspanel:admin_user_ss_config_list"))