ASJServer/Controller/AppCampaignController.py

import time
import pytz
import json
from datetime import datetime
from datetime import time as Time

from Model.models import AppAdvertiseCampaign, DeviceTypeModel, CountryModel, Device_User, OpenScreenCampaign
from Ansjer.config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, SERVER_TYPE

from django.core.paginator import Paginator
from django.views import View
from django.db.models import Prefetch
from django.db.models import Q

from Object.AWS.AmazonS3Util import AmazonS3Util
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject


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

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

    def validation(self, request_dict, request, operation):
        language = request_dict.get('language', 'en')
        response = ResponseObject(language, 'pc')
        if operation == 'getCampaignList':  # 获取广告活动列表
            return self.get_campaign_list(request_dict, response)
        elif operation == 'addCampaign':  # 添加广告活动
            return self.add_campaign(request, request_dict, response)
        elif operation == 'updateCampaign':  # 更新广告活动
            return self.update_campaign(request, request_dict, response)
        elif operation == 'deleteCampaign':  # 删除广告活动
            return self.delete_campaign(request_dict, response)
        elif operation == 'switchCampaign':  # 广告活动开关
            return self.switch_campaign(request_dict, response)
        elif operation == 'appGetCampaigns':
            return self.app_get_campaigns(request_dict, response)
        elif operation == 'recordUserBehavior':
            return self.record_user_behavior(request_dict, response)
        elif operation == 'getUserBehaviorLog':
            return self.get_user_behavior_log(request_dict, response)
        else:
            return response.json(414)

    def get_campaign_list(self, request_dict, response):
        """
        查询广告活动列表
        @param request_dict: 请求参数
        @param response: 响应对象
        @return: 响应对象包含广告活动列表
        """
        campaign_name = request_dict.get('campaign_name', None)
        campaign_country = request_dict.get('campaign_country', None)
        status = request_dict.get('status', None)
        pageNo = request_dict.get('pageNo', 1)
        pageSize = request_dict.get('pageSize', 20)
        unknown_country = 0

        # 连接并获取国家和设备类型
        country_prefetch = Prefetch('country', queryset=CountryModel.objects.only('country_name'),
                                    to_attr='country_list')
        device_type_prefetch = Prefetch('device_type', queryset=DeviceTypeModel.objects.only('name'),
                                        to_attr='device_type_list')
        app_advertise_campaign_qs = AppAdvertiseCampaign.objects.prefetch_related(country_prefetch,
                                                                                  device_type_prefetch)

        # 过滤
        if campaign_name:
            app_advertise_campaign_qs = app_advertise_campaign_qs.filter(campaign_name=campaign_name)
        if status:
            app_advertise_campaign_qs = app_advertise_campaign_qs.filter(status=status)
        if campaign_country:
            campaign_country_list = campaign_country.split(',')
            if "未知地区" in campaign_country_list:
                unknown_country = 1
            app_advertise_campaign_qs = app_advertise_campaign_qs.filter(
                Q(country__country_name__in=campaign_country_list) | Q(unknown_country=unknown_country)).distinct()
        app_advertise_campaign_qs = app_advertise_campaign_qs.filter(~Q(status=2))
        # 分页
        paginator = Paginator(app_advertise_campaign_qs.order_by('id'), pageSize)
        campaigns = paginator.page(pageNo)

        if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings':
            s3_url = "https://ansjerfilemanager.s3.cn-northwest-1.amazonaws.com.cn/app/campaign/"
        else:
            s3_url = "https://ansjerfilemanager.s3.amazonaws.com/app/campaign/"

        # 添加设备名和地区返回
        campaign_list = []
        for campaign in campaigns.object_list:
            if campaign.unknown_country == 0:
                countries = ",".join([country.country_name for country in campaign.country_list])
            else:
                country_list = campaign.country_list
                country_names = []
                for country in country_list:
                    country_names.append(country.country_name)
                country_names.append("未知地区")
                countries = ",".join(country_names)
            campaign_data = {
                'id': campaign.id,
                'image_url': s3_url + campaign.image_url,
                'campaign_name': campaign.campaign_name,
                'campaign_url': campaign.campaign_url,
                'campaign_type': campaign.campaign_type,
                'status': campaign.status,
                'campaign_start_date': campaign.campaign_start_date,
                'campaign_end_date': campaign.campaign_end_date,
                'campaign_show_stime': campaign.campaign_show_stime,
                'campaign_show_etime': campaign.campaign_show_etime,
                'countries': countries,
                'device_types': ",".join([device.name for device in campaign.device_type_list]),
            }
            campaign_list.append(campaign_data)

        data = {
            'list': campaign_list,
            'total': paginator.count,
        }

        return response.json(0, data)

    def add_campaign(self, request, request_dict, response):
        """
        添加新的广告活动
        @param request_dict: 包含所有请求参数的字典
        @param response: 响应对象
        @return: 响应对象
        """
        file = request.FILES.get('posterFile', None)
        campaign_name = request_dict.get('campaign_name', None)
        campaign_url = request_dict.get('campaign_url', None)
        campaign_type = request_dict.get('campaign_type', None)
        status = request_dict.get('status', None)
        device_type_names = json.loads(request_dict.get('device_type_list', None))  # 设备类型名称列表
        country_name_list = json.loads(request_dict.get('country_name_list', None))  # 地区列表
        campaign_start_time = request_dict.get('campaign_start_time', None)
        campaign_end_time = request_dict.get('campaign_end_time', None)
        campaign_show_stime = request_dict.get('campaign_show_stime', 0)
        campaign_show_etime = request_dict.get('campaign_show_etime', 86399)

        if not all([campaign_name, campaign_type, device_type_names, campaign_url,
                    country_name_list, campaign_start_time, campaign_end_time, file]):
            return response.json(444)

        # 针对特殊地区的处理，表没设计好用这个处理挽救一下
        unknown_country = 0
        if "未知地区" in country_name_list:
            unknown_country = 1
            country_name_list.remove("未知地区")
        if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings':
            regin = 0
            AWS_SES_ACCESS_REGION = "cn-northwest-1"
        else:
            regin = 1
            AWS_SES_ACCESS_REGION = 'us-east-1'
        fileName = file.name
        try:
            create_time = int(time.time())
            update_time = int(time.time())
            # 保存图片到存储桶
            bucket_name = 'ansjerfilemanager'
            file_key = f'app/campaign/OpenScreenAdvertise/{update_time}_{fileName}'
            s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[0], AWS_SECRET_ACCESS_KEY[0], "cn-northwest-1")
            # 地址：https://ansjerfilemanager.s3.amazonaws.com/app/campaign/OpenScreenAdvertise/XXX.jpg
            s3.upload_file_obj(
                bucket_name,
                file_key,
                file,
                {'ContentType': file.content_type, 'ACL': 'public-read'})
            # 创建 AppAdvertiseCampaign 实例
            new_campaign = AppAdvertiseCampaign.objects.create(
                image_url=f"OpenScreenAdvertise/{update_time}_{fileName}",
                campaign_name=campaign_name,
                campaign_url=campaign_url,
                campaign_type=campaign_type,
                status=status,
                unknown_country=unknown_country,
                campaign_start_date=campaign_start_time,
                campaign_end_date=campaign_end_time,
                campaign_show_stime=campaign_show_stime,
                campaign_show_etime=campaign_show_etime,
                create_time=create_time,
                update_time=update_time,
            )

            # 根据名称查找 DeviceTypeModel 的实例并建立关系
            device_type_instances = DeviceTypeModel.objects.filter(name__in=device_type_names)
            for device_type_instance in device_type_instances:
                new_campaign.device_type.add(device_type_instance)

            # 根据 ID 关联 CountryModel 实例
            country_instances = CountryModel.objects.filter(country_name__in=country_name_list)
            for country_instance in country_instances:
                new_campaign.country.add(country_instance)

        except Exception as e:
            return response.json(178)

        return response.json(0)

    def update_campaign(self, request, request_dict, response):
        campaign_id = request_dict.get('id', None)
        campaign_name = request_dict.get('campaign_name', None)
        campaign_url = request_dict.get('campaign_url', None)
        campaign_type = request_dict.get('campaign_type', None)
        device_type_names = json.loads(request_dict.get('device_type_list', '[]'))  # 设备类型名称列表
        country_name_list = json.loads(request_dict.get('country_name_list', '[]'))  # 地区列表
        campaign_start_time = request_dict.get('campaign_start_time', None)
        campaign_end_time = request_dict.get('campaign_end_time', None)
        campaign_show_stime = request_dict.get('campaign_show_stime', None)
        campaign_show_etime = request_dict.get('campaign_show_etime', None)
        file = request.FILES.get('posterFile', None)
        if not campaign_id:
            return response.json(444)

        if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings':
            regin = 0
            AWS_SES_ACCESS_REGION = "cn-northwest-1"
        else:
            regin = 1
            AWS_SES_ACCESS_REGION = 'us-east-1'

        try:
            update_time = int(time.time())
            campaign = AppAdvertiseCampaign.objects.get(id=campaign_id)
            if country_name_list is not None:
                if "未知地区" in country_name_list:
                    campaign.unknown_country = 1
                    country_name_list.remove("未知地区")
                else:
                    campaign.unknown_country = 0
            if file is not None:
                # 删除存储桶原来的图片
                s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[regin], AWS_SECRET_ACCESS_KEY[regin], AWS_SES_ACCESS_REGION)
                bucket_name = 'ansjerfilemanager'
                if campaign.image_url:
                    s3.delete_obj(bucket_name, f"app/campaign/{campaign.image_url}")
                # 添加新的图片
                file_key = f'app/campaign/OpenScreenAdvertise/{update_time}_{file.name}'
                s3.upload_file_obj(
                    bucket_name,
                    file_key,
                    file,
                    {'ContentType': file.content_type, 'ACL': 'public-read'}
                )
                campaign.image_url = f'OpenScreenAdvertise/{update_time}_{file.name}'
            if campaign_name is not None:
                campaign.campaign_name = campaign_name
            if campaign_url is not None:
                campaign.campaign_url = campaign_url
            if campaign_type is not None:
                campaign.campaign_type = campaign_type
            if campaign_start_time is not None:
                campaign.campaign_start_date = campaign_start_time
            if campaign_end_time is not None:
                campaign.campaign_end_date = campaign_end_time
            if campaign_show_stime is not None:
                campaign.campaign_show_stime = campaign_show_stime
            if campaign_show_etime is not None:
                campaign.campaign_show_etime = campaign_show_etime

            # 更新多对多字段 - 设备类型
            if device_type_names:
                device_types = DeviceTypeModel.objects.filter(name__in=device_type_names)
                campaign.device_type.set(device_types)

            # 更新多对多字段 - 国家/地区
            if country_name_list:
                countries = CountryModel.objects.filter(country_name__in=country_name_list)
                campaign.country.set(countries)

            campaign.update_time = update_time

            # 保存更新
            campaign.save()
        except Exception as e:
            return response.json(177)

        return response.json(0)

    def switch_campaign(self, request_dict, response):
        campaign_id = request_dict.get('id')
        status = request_dict.get('status')
        if not campaign_id:
            return response.json(444)
        try:
            AppAdvertiseCampaign.objects.filter(pk=campaign_id).update(status=status, update_time=int(time.time()))
            return response.json(0)
        except Exception as e:
            return response.json(444)

    def delete_campaign(self, request_dict, response):
        campaign_id = request_dict.get('id')
        if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings':
            regin = 0
            AWS_SES_ACCESS_REGION = "cn-northwest-1"
        else:
            regin = 1
            AWS_SES_ACCESS_REGION = 'us-east-1'
        if not campaign_id:
            return response.json(444)
        try:
            campaign = AppAdvertiseCampaign.objects.get(pk=campaign_id)
            s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[regin], AWS_SECRET_ACCESS_KEY[regin], AWS_SES_ACCESS_REGION)
            bucket_name = 'ansjerfilemanager'
            if campaign.image_url:
                s3.delete_obj(bucket_name, f"app/campaign/{campaign.image_url}")
            # 清除多对多关系
            campaign.device_type.clear()
            campaign.country.clear()
            # 保留在广告表中
            campaign.status = 2
            campaign.update_time = int(time.time())
            campaign.save()
            return response.json(0)

        except Exception as e:
            return response.json(176)

    def get_timezone_offset(self, tz):
        """
        UTC-8格式转化
        """
        sign = tz[3]  # "+" 或 "-"
        hours = int(tz[4:])  # 时差小时数
        timezone_name = f"Etc/GMT{sign}{hours}" if sign == "-" else f"Etc/GMT+{hours}"
        return pytz.timezone(timezone_name)

    def app_get_campaigns(cls, request_dict, response):
        """
        APP获取广告活动列表
        @param request_dict: 请求参数
        @param response: 响应对象
        @return: 响应对象
        """
        tz = request_dict.get('tz', 'UTC+0')
        token = request_dict.get('token', None)
        if not token:
            return response.json(444)
        token = TokenObject(token)
        if token.code != 0:
            return response.json(token.code)
        user_id = token.userID

        if SERVER_TYPE == 'Ansjer.cn_config.formal_settings' or SERVER_TYPE == 'Ansjer.cn_config.test_settings':
            s3_url = "https://ansjerfilemanager.s3.cn-northwest-1.amazonaws.com.cn/app/campaign/"
        else:
            s3_url = "https://ansjerfilemanager.s3.amazonaws.com/app/campaign/"

        # 当日时间戳区间获取
        timezone = cls.get_timezone_offset(tz)
        current_date = datetime.now().date()
        start_of_day = datetime.combine(current_date, Time.min)
        end_of_day = datetime.combine(current_date, Time.max)
        start_timestamp = int(start_of_day.timestamp())
        end_timestamp = int(end_of_day.timestamp())

        open_screen_campaign_qs = OpenScreenCampaign.objects.filter(
            user_id=user_id,
            create_time__gt=start_timestamp,
            create_time__lt=end_timestamp
        )
        if not open_screen_campaign_qs.exists():
            OpenScreenCampaign.objects.create(user_id=user_id,
                                              create_time=int(time.time()))
        try:
            country_prefetch = Prefetch('country', queryset=CountryModel.objects.only('id'), to_attr='country_list')
            device_type_prefetch = Prefetch('device_type', queryset=DeviceTypeModel.objects.only('type'),
                                            to_attr='device_type_list')
            app_advertise_campaign_qs = AppAdvertiseCampaign.objects.prefetch_related(country_prefetch,
                                                                                      device_type_prefetch)

            device_info_qs = Device_User.objects.filter(userID=user_id).values("region_country").first()
            country_id = device_info_qs.get('region_country')
            if country_id != 0:
                app_advertise_campaign_qs = app_advertise_campaign_qs.filter(country__id=country_id, status=1)
            else:
                app_advertise_campaign_qs = app_advertise_campaign_qs.filter(status=1, unknown_country=1)

            # 返回 广告名称、广告类型、开始时间、结束时间、广告图片、活动链接
            campaigns_list = []

            for campaign in app_advertise_campaign_qs:
                # 时区处理
                campaigns_start_date = (datetime.utcfromtimestamp(campaign.campaign_start_date)
                                        .replace(tzinfo=pytz.utc).astimezone(timezone).strftime('%Y-%m-%d'))
                campaign_end_date = (datetime.utcfromtimestamp(campaign.campaign_end_date)
                                     .replace(tzinfo=pytz.utc).astimezone(timezone).strftime('%Y-%m-%d'))
                campaign_start_firstday = (
                    datetime.utcfromtimestamp(campaign.campaign_start_date + campaign.campaign_show_stime)
                    .replace(tzinfo=pytz.utc).astimezone(timezone).strftime('%Y-%m-%d %H:%M'))
                campaign_end_firstday = (
                    datetime.utcfromtimestamp(campaign.campaign_start_date + campaign.campaign_show_etime)
                    .replace(tzinfo=pytz.utc).astimezone(timezone).strftime('%Y-%m-%d %H:%M'))
                campaigns_list.append({
                    'campaign_id': campaign.id,
                    'image_url': s3_url + campaign.image_url,
                    'campaign_url': campaign.campaign_url,
                    'campaign_name': campaign.campaign_name,
                    'campaign_type': campaign.campaign_type,
                    'campaign_date': f"{campaigns_start_date},{campaign_end_date}",
                    'campaign_start_oneday': campaign_start_firstday,
                    'campaign_end_oneday': campaign_end_firstday,
                    'device_types': [device.type for device in campaign.device_type_list],
                })
            return response.json(0, {
                'campaigns': campaigns_list
            })
        except Exception as e:
            return response.json(173)

    def record_user_behavior(cls, request_dict, response):
        """
        记录用户行为
        @param request_dict: 请求参数
        @param response: 响应对象
        @return: 响应对象
        """
        status = request_dict.get('status', None)  # 1.未跳过 2.已跳过 3.点击广告
        campaign_id = request_dict.get('campaign_id', None)
        token = request_dict.get('token', None)
        if not token:
            return response.json(444)
        token = TokenObject(token)
        if token.code != 0:
            return response.json(token.code)
        user_id = token.userID

        if not all([status, campaign_id]):
            return response.json(444)

        # 当日时间(各个时区都不同)戳区间获取
        current_date = datetime.now().date()
        start_of_day = datetime.combine(current_date, Time.min)
        end_of_day = datetime.combine(current_date, Time.max)
        start_timestamp = int(start_of_day.timestamp())
        end_timestamp = int(end_of_day.timestamp())

        try:
            # 筛选符合条件的记录
            open_screen_campaigns = OpenScreenCampaign.objects.filter(
                user_id=user_id,
                create_time__gt=start_timestamp,
                create_time__lt=end_timestamp,
                status=0,
            )

            # 检查是否存在记录
            if not open_screen_campaigns.exists():
                # 如果不存在，则创建新记录
                OpenScreenCampaign.objects.create(
                    user_id=user_id,
                    status=status,
                    campaign_id_id=campaign_id,
                    create_time=int(time.time()),
                    update_time=int(time.time())
                )

            else:
                # 如果存在，则更新最新的记录
                latest_campaign = open_screen_campaigns.latest("create_time")
                latest_campaign.status = status
                latest_campaign.campaign_id_id = campaign_id
                latest_campaign.update_time = int(time.time())
                latest_campaign.save()

        except Exception as e:
            return response.json(177, {'error': '更新错误'})

        return response.json(0)

    def get_user_behavior_log(self, request_dict, response):
        status = request_dict.get('status', None)
        user_id = request_dict.get('user_id', None)
        behavior_stime = request_dict.get('start_time', None)
        behavior_etime = request_dict.get('end_time', None)
        page = request_dict.get('page', 1)  # 默认为第一页
        page_size = request_dict.get('page_size', 10)  # 默认每页显示10条记录
        behavior_log_list = []

        open_screen_campaigns = OpenScreenCampaign.objects.all()
        if user_id is not None:
            open_screen_campaigns = open_screen_campaigns.filter(user_id=user_id)
        if status is not None:
            open_screen_campaigns = open_screen_campaigns.filter(status=status)
        # 时间过滤
        if behavior_stime is not None and behavior_etime is not None:
            open_screen_campaigns = open_screen_campaigns.filter(update_time__range=[behavior_stime, behavior_etime])
        elif behavior_stime is not None:
            open_screen_campaigns = open_screen_campaigns.filter(update_time__gte=behavior_stime)
        elif behavior_etime is not None:
            open_screen_campaigns = open_screen_campaigns.filter(update_time__lte=behavior_etime)

        if not open_screen_campaigns.exists():
            return response.json(0, {"behavior_log_list": [], "total": 0})
        # 分页
        open_screen_campaigns = open_screen_campaigns.order_by('id')
        paginator = Paginator(open_screen_campaigns, page_size)
        open_screen_campaigns_page = paginator.page(page)

        for campaign in open_screen_campaigns_page:
            campaign_name = campaign.campaign_id.campaign_name if campaign.campaign_id else '广告不存在'
            behavior_log_list.append({
                "campaign_name": campaign_name,
                "user_id": campaign.user_id,
                "status": campaign.status,
                "behavior_time": campaign.update_time
            })

        return response.json(0, {"behavior_log_list": behavior_log_list,
                                 "total": paginator.count})