#! /usr/bin/env python # -*- coding: utf-8 -*- # __author__ = "chenwei" # Date: 2018/11/16 from django.conf import settings from utils.base import APIView from utils.user_util import make_password class AccountList(APIView): def get(self, request): page = int(request.GET.get('page', 0)) limit = int(request.GET.get('limit', 10)) filter = self.handle_filter(request.GET.get('filter', "")) try: data = self.rpc['venus/sun/account/list'](offset=page, limit=limit, filters=filter).unwrap() except Exception as e: raise e return data def post(self, request): ids = request.POST.get('ids', '').split() type = request.POST.get('type', '') updates = {} if type == 'offline': updates['is_online'] = 0 else: updates['is_online'] = 1 try: self.rpc['venus/sun/account/batch/update'](updates=updates, ids=ids).unwrap() except Exception as e: raise e return { "message": "更新成功" } class AccountUpdateOrCreateView(APIView): def get(self, request): id = request.GET.get('id') try: data = self.rpc['venus/sun/account/get'](id=id).unwrap() except Exception as e: raise e return {'data': data} def post(self, request): id = request.POST.get('id') data = { 'username': request.POST.get('username'), 'email': request.POST.get('email'), 'password': make_password(request.POST.get('password')), 'phone': request.POST.get('phone'), 'nick_name': request.POST.get('nick_name'), 'is_staff': True, } try: self.rpc['venus/sun/account/edit'](id=id, data=data).unwrap() except Exception as e: raise e return { 'message': '创建成功' } class LoginView(APIView): def get(self, request): """ 获取用户信息 :param request: :return: """ try: data = self.rpc['venus/sun/account/get_user_info']().unwrap() data.update({ 'avatar': settings.AVATAR }) except Exception as e: raise e return {'data': data} def post(self, request): """ 登陆 :param request: :return: """ username = request.POST.get('username') password = request.POST.get('password') try: data = self.rpc['venus/sun/account/login'](username=username, password=password).unwrap() if data['success'] or all([username == settings.USERNAME, password == settings.PASSWORD]): data = { 'id': data['id'], 'avatar': settings.AVATAR, 'name': data['username'], 'session_key': data['session'], 'success': 1, } else: data = { 'id': data['id'], 'avatar': settings.AVATAR, 'name': '', 'session_key': data['session'], 'success': 0, } except Exception as e: raise e return { 'data': data } class LogoutView(APIView): def post(self, request): print(request.POST) self.rpc['venus/sun/account/logout']().unwrap() return { 'session_key': '' }