ASJServer/Controller/SensorGateway/GatewayFamilyMemberController.py

# -*- encoding: utf-8 -*-
"""
@File    : GatewayFamilyMemberController.py
@Time    : 2022/5/25 10:21
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""

import json
import time

from django.db import transaction
from django.views.generic.base import View

from Model.models import UserFamily, FamilyMemberJoin, FamilyMember, Device_User, SysMsgModel
from Object.RedisObject import RedisObject
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService


# 家庭房间管理
class GatewayFamilyMemberView(View):

    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.GET, request, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        return self.validation(request.POST, request, operation)

    def validation(self, request_dict, request, operation):
        token = request.META.get('HTTP_AUTHORIZATION')
        token = TokenObject(token)
        lang = request_dict.get('lang', None)
        response = ResponseObject(lang) if lang else ResponseObject(token.lang)
        if token.code != 0:
            return response.json(token.code)
        app_user_id = token.userID
        # 添加设备关联房间
        if operation == 'join':
            return self.member_join(app_user_id, request_dict, response)
        elif operation == 'join/page':
            return self.member_join_page(request_dict, response)
        elif operation == 'qrcode/generate':
            return self.family_generate_token(app_user_id, request_dict, response)
        elif operation == 'qrcode/check':
            return self.check_qrcode(request_dict, response)

    @classmethod
    def member_join(cls, app_user_id, request_dict, response):
        """
        家庭成员加入
        @param app_user_id:
        @param request_dict:
        @param response:
        @return:
        """
        family_id = request_dict.get('familyId', None)
        invite = request_dict.get('QRCodeInvite', None)
        user_id = request_dict.get('app_user_id', None)
        lang = request_dict.get('lang', 'en')
        if not all([family_id, invite]):
            return response.json(444)
        invite = int(invite)
        family_id = int(family_id)
        now_time = int(time.time())
        # 判断当前家庭是否已有该成员
        if invite == 1:
            family_qs = FamilyMember.objects.filter(family_id=family_id, user_id=app_user_id)
        elif user_id:
            family_qs = FamilyMember.objects.filter(family_id=family_id, user_id=user_id)
        else:
            return response.json(173)
        if family_qs.exists():
            return response.json(174)
        try:
            user_family_qs = UserFamily.objects.filter(id=family_id)
            if not user_family_qs:
                return response.json(173)
            user_family = user_family_qs.first()
            family_name = user_family.name
            data = {
                'addTime': now_time,
                'updTime': now_time,
                'eventType': 705
            }
            # 请求加入家庭，用户信息
            user_qs = Device_User.objects.filter(userID=app_user_id)
            nick_name = user_qs.first().NickName

            if invite == 1:
                msg_text = cls.get_msg_text(nick_name, family_name, lang, True)
                # 通知对象 家庭所有者
                join_user_id = app_user_id
                data['userID_id'] = user_family.user_id
            elif app_user_id:
                msg_text = cls.get_msg_text(nick_name, family_name, lang, False)
                # 通知对象 被邀请人
                join_user_id = user_id
                data['userID_id'] = user_id
            else:
                return response.json(173)
            data['msg'] = msg_text
            with transaction.atomic():
                family_user_join = FamilyMemberJoin.objects.create(family_id=family_id, user_id=join_user_id,
                                                                   status=0,
                                                                   updated_time=now_time,
                                                                   created_time=now_time)
                data['uid'] = family_user_join.id
                SysMsgModel.objects.create(**data)
                return response.json(0)
        except Exception as e:
            print(e)
            return response.json(177, repr(e))

    @classmethod
    def get_msg_text(cls, nick_name, family_name, lang, is_invite=False):
        """
        系统消息通知内容，中英文判断
        @param nick_name:
        @param family_name:
        @param lang:
        @param is_invite:
        @return:
        """
        if lang and lang == 'cn':
            if is_invite:
                msg_text = '{}请求加入你的{}'.format(nick_name, family_name)
            else:
                msg_text = '{}邀请你加入{}'.format(nick_name, family_name)
        else:
            if is_invite:
                msg_text = '{} Request to join your Homegroup {}'.format(nick_name, family_name)
            else:
                msg_text = '{} invite you to join the Homegroup {}'.format(nick_name, family_name)

        return msg_text

    @classmethod
    def member_join_page(cls, request_dict, response):
        """
        家庭成员邀请记录
        @param request_dict:
        @param response:
        @return:
        """
        page_no = request_dict.get('pageNo', None)
        page_size = request_dict.get('pageSize', None)
        family_id = request_dict.get('familyId', None)
        if not all([page_no, page_size, family_id]):
            return response.json(444)
        member_join = FamilyMemberJoin.objects.filter(family_id=family_id).values('status', 'user__username',
                                                                                  'user__userIconUrl', 'user__phone',
                                                                                  'user__userEmail', 'updated_time',
                                                                                  'created_time')

        page_no = int(page_no)
        page_size = int(page_size)
        member_join = member_join.order_by('-created_time')[(page_no - 1) * page_size: page_no * page_size]
        if not member_join.exists():
            return response.json(0, [])
        result = []
        for item in member_join:
            result.append({
                'status': item['status'],
                'userName': item['user__username'],
                'userIconUrl': item['user__userIconUrl'],
                'phone': item['user__phone'],
                'nickName': item['user__NickName'],
                'userEmail': item['user__userEmail'],
                'updatedTime': item['updated_time'],
                'createdTime': item['created_time']
            })
        return response.json(0, result)

    @classmethod
    def family_generate_token(cls, app_user_id, request_dict, response):
        """
        获取家庭令牌信息
        @param app_user_id:
        @param request_dict:
        @param response:
        @return:
        """
        family_id = request_dict.get('familyId', None)
        if not family_id:
            return response.json(444)
        family_id = int(family_id)
        family_qs = UserFamily.objects.filter(id=family_id, user_id=app_user_id) \
            .values('name', 'user__userEmail',
                    'user__userIconUrl',
                    'user__phone', 'user__NickName',
                    'user__username')
        if not family_qs.exists():
            return response.json(173)
        family_qs = family_qs.first()
        data = {'sharer': app_user_id, 'familyId': family_id}
        # -----------------
        # share_token = jwt.encode(data, '.#^^&', algorithm='HS256').decode('utf-8')
        # +++++++++++++++++
        redisObj = RedisObject()
        share_token = str(int(time.time())) + CommonService.RandomStr(8)
        if not redisObj.set_data(key=share_token, val=json.dumps(data), expire=300):
            return response.json(10, '生成失败')
        return response.json(0, {
            'shareToken': share_token,
            'userPhone': family_qs['user__phone'],
            'nickName': family_qs['user__NickName'],
            'userIconUrl': family_qs['user__userIconUrl'],
            'userName': family_qs['user__username'],
            'familyName': family_qs['name'],
            'userEmail': family_qs['user__userEmail'],
        })

    @classmethod
    def check_qrcode(cls, request_dict, response):
        """
        检查二维码是否过期
        @param request_dict:
        @param response:
        @return:
        """
        share_token = request_dict.get('shareToken', None)
        if not share_token:
            return response.json(444)
        redisObj = RedisObject()
        share_data = redisObj.get_data(key=share_token)
        if not share_data:
            return response.json(119)
        data = json.loads(share_data)
        sharer = data.get('sharer', None)
        family_id = data.get('familyId', None)
        if not all([sharer, family_id]):
            return response.json(173)
        family_id = int(family_id)
        user_family_qs = UserFamily.objects.filter(id=family_id)
        if not user_family_qs.exists():
            return response.json(173)
        user_family_qs = user_family_qs.values('name', 'user__userEmail',
                                               'user__userIconUrl',
                                               'user__phone', 'user__NickName',
                                               'user__username')
        return response.json(0, {
            'userPhone': user_family_qs['user__phone'],
            'nickName': user_family_qs['user__NickName'],
            'userIconUrl': user_family_qs['user__userIconUrl'],
            'userName': user_family_qs['user__username'],
            'familyName': user_family_qs['name'],
            'userEmail': user_family_qs['user__userEmail'],
        })