ASJServer/Controller/UserDevice/UserSubscriptionController.py

# -*- encoding: utf-8 -*-
"""
@File    : UserSubscriptionController.py
@Time    : 2024/5/21 16:31
@Author  : stephen
@Email   : zhangdongming@asj6.wecom.work
@Software: PyCharm
"""
import threading
import time

from django.http import QueryDict
from django.views import View

from Model.models import Device_User, CountryModel, UserEmailSubscriptions
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Object.YotpoCoreObject import YotpoCoreObject
from Ansjer.config import LOGGER


class UserSubscriptionControllerView(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 delete(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation')
        delete = QueryDict(request.body)
        if not delete:
            delete = request.GET
        return self.validation(delete, request, operation)

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

    def validation(self, request_dict, request, operation):
        response = ResponseObject('cn')
        tko = TokenObject(request.META.get('HTTP_AUTHORIZATION'))
        if tko.code != 0:
            return response.json(tko.code)
        response.lang = tko.lang
        userID = tko.userID
        if operation == 'checkSubscriptionStatus':
            return self.check_subscription_status(userID, response)
        elif operation == 'switchSubscription':
            return self.switch_subscription(userID, request_dict, response)
        else:
            return response.json(414)

    @staticmethod
    def check_subscription_status(user_id, response):
        """
        检查订阅状态
        @param user_id: str
        @param response: 响应
        @return: response
        """
        user_qs = Device_User.objects.filter(userID=user_id)
        if not user_qs.exists():
            return response.json(104)
        user_sub = UserEmailSubscriptions.objects.filter(user_id=user_id).values('status', 'push_sub_status').first()
        if not user_sub:
            UserEmailSubscriptions.objects.create(user_id=user_id, status=0, push_sub_status=1,
                                                  updated_time=int(time.time()), created_time=int(time.time()))
            return response.json(0, {"emailSubStatus": 0, 'pushSubStatus': 1})
        email_sub_status = 1 if user_sub["status"] == 1 else 0
        push_sub_status = 0 if user_sub["push_sub_status"] == 0 else 1
        user_sub = {"emailSubStatus": email_sub_status, 'pushSubStatus': push_sub_status}
        return response.json(0, user_sub)

    def switch_subscription(self, user_id, request_dict, response):
        """
        邮件订阅开关
        @param user_id: str
        @param request_dict: dict
        @param response
        """
        user_qs = Device_User.objects.filter(userID=user_id)
        email_sub_status = request_dict.get('emailSubStatus', None)
        push_sub_status = request_dict.get('pushSubStatus', None)
        if not email_sub_status and not push_sub_status:
            return response.json(444, "emailSubStatus or pushSubStatus")
        if not user_qs.exists():
            return response.json(104)

        # 用户推送订阅
        if push_sub_status:
            push_sub_status = int(push_sub_status)
            user_sub_qs = UserEmailSubscriptions.objects.filter(user_id=user_id)
            if user_sub_qs.exists():
                user_sub_qs.update(push_sub_status=push_sub_status, updated_time=int(time.time()))
            else:
                user = user_qs.values('userEmail', 'phone').first()
                push_sub = {
                    "user_id": user_id,
                    "push_sub_status": push_sub_status,
                    "created_time": int(time.time()),
                    "updated_time": int(time.time())
                }
                if user["userEmail"]:
                    push_sub["email"] = user["userEmail"]
                if user["phone"]:
                    push_sub["phone"] = user["phone"]
                UserEmailSubscriptions.objects.create(**push_sub)
        else:
            user_sub = UserEmailSubscriptions.objects.filter(user_id=user_id).values('status',
                                                                                     'push_sub_status').first()
            if not user_sub:
                push_sub_status = 1
            else:
                push_sub_status = user_sub["push_sub_status"]

        if email_sub_status:
            # 修改数据库中订阅状态
            email_sub_status = int(email_sub_status)
            user_sub_qs = UserEmailSubscriptions.objects.filter(user_id=user_id)
            # 邮件订阅
            if email_sub_status == 1:
                subscribers = user_qs.values('NickName', 'userEmail', 'region_country').first()
                if not subscribers["userEmail"]:
                    LOGGER.info(f'subscribers{user_id}邮箱为空，无法订阅')
                    return response.json(183)
                if user_sub_qs.exists():
                    user_sub_qs.update(email=subscribers["userEmail"], status=1, updated_time=int(time.time()))
                else:
                    UserEmailSubscriptions.objects.create(user_id=user_id, status=1, email=subscribers["userEmail"],
                                                          created_time=int(time.time()), updated_time=int(time.time()))
                subscription_thread = threading.Thread(target=self.subscription, args=(subscribers,))
                subscription_thread.start()
            # 取消订阅
            elif email_sub_status == 0:
                device_user = Device_User.objects.filter(userID=user_id).values('userEmail').first()
                if device_user:
                    customer = {
                        "email": device_user["userEmail"],
                        "first_name": device_user["userEmail"],
                        "last_name": "APP",
                    }
                    yotpo = YotpoCoreObject()
                    list_id = 8589406
                    subscription_thread = threading.Thread(target=yotpo.close_subscribers, args=(customer, list_id))
                    subscription_thread.start()
                    if user_sub_qs.exists():
                        customer["status"] = "unsubscription"
                        user_sub_qs.update(email=device_user["userEmail"], status=0, sub_result=customer,
                                           updated_time=int(time.time()))
        else:
            user_sub = UserEmailSubscriptions.objects.filter(user_id=user_id).values('status').first()
            if not user_sub:
                email_sub_status = 0
            else:
                email_sub_status = user_sub["status"]

        return response.json(0, {"emailSubStatus": email_sub_status, "pushSubStatus": push_sub_status})

    @staticmethod
    def subscription(subscribers):
        """
        订阅
        @param subscribers: dict
        @return: boolean
        """

        yotpo = YotpoCoreObject()

        try:
            # 查询顾客所在地区
            if subscribers["region_country"]:
                country = CountryModel.objects.filter(id=subscribers["region_country"]).values('country_code').first()
                if country:
                    country_code = country["country_code"]
                else:
                    country_code = ''
                # 构建顾客订阅格式
                customer = {
                    "email": subscribers["userEmail"],
                    "first_name": subscribers["userEmail"],
                    "last_name": "APP",
                    'address': {
                        "country_code": country_code,
                    },
                    "custom_properties": {
                        "subscription_office": "ZosiApp",
                    }
                }
            else:
                customer = {
                    "email": subscribers["userEmail"],
                    "first_name": subscribers["userEmail"],
                    "last_name": "APP",
                    "custom_properties": {
                        "subscription_office": "ZosiApp",
                    }
                }

            result = yotpo.creat_and_update_customers(customer)
            list_id = 8589406
            sub_status, sub_result = yotpo.create_subscribers(customer, list_id)

            if result and sub_status:
                # 创建结果写入数据库
                user_sub_qs = UserEmailSubscriptions.objects.filter(email=subscribers["userEmail"])
                if user_sub_qs.exists():
                    user_sub_qs.update(email=subscribers["userEmail"], status=1, sub_result=sub_result, list_id=list_id,
                                       updated_time=int(time.time()))
                    LOGGER.info(f'在yotpo创建客户并订阅成功，customer:{customer}')
                return True
            else:
                LOGGER.info(f'在yotpo创建客户并订阅失败，customer:{customer}')
            return False
        except Exception as e:
            LOGGER.error(f'{subscribers["userEmail"]}订阅失败：{e}')
            return False