ASJServer/Controller/CampaignController/CampaignController.py

import time
import json

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

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

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


class CampaignView(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 == 'getCountryList':
            return self.get_country_list(response)
        else:
            tko = TokenObject(
                request.META.get('HTTP_AUTHORIZATION'),
                returntpye='pc')
            if tko.code != 0:
                return response.json(tko.code)
            response.lang = tko.lang
            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 == '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

        try:
            # 连接并获取国家和设备类型
            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)

            # 添加设备名和地区返回
            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)

                s3_url, _, _ = self.s3_server()

                banner_campaign = campaign.banner_campaign
                for sort in banner_campaign:
                    banner_campaign[sort]["image"] = s3_url + banner_campaign[sort]["image"]
                campaign_data = {
                    'id': campaign.id,
                    'image_url': s3_url + campaign.image_url,
                    'banner_campaign': banner_campaign,
                    '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,
                    'app_bundle_type': campaign.app_bundle_type,
                    '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)
        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def add_campaign(self, request, request_dict, response):
        """
        添加新的广告活动
        @param request: 包含文件信息
        @param request_dict: 包含所有请求参数的字典
        @param response: 响应对象
        @return: 响应对象
        """
        try:
            campaign_name = request_dict.get('campaign_name', None)
            campaign_url = request_dict.get('campaign_url', None)
            status = request_dict.get('status', 2)
            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)
            app_bundle_type = request_dict.get('app_bundle_type', None)
            poster_file = request.FILES.get('posterFile', None)
            banner_files = request.FILES.getlist('bannerFiles', None)
            banner_campaign_urls = json.loads(request_dict.get('banner_campaign_url', "[]"))
            device_type_names = json.loads(request_dict.get('device_type_list', "[]"))  # 设备类型名称列表
            country_name_list = json.loads(request_dict.get('country_name_list', "[]"))  # 地区列表

            required_fields = [campaign_name, campaign_url, campaign_start_time, campaign_end_time,
                               app_bundle_type]
            list_fields = [device_type_names, country_name_list, banner_campaign_urls]

            # 检查基本字段是否为None
            if any(field is None for field in required_fields):
                return response.json(444)

            # 检查列表类型的字段是否为空
            if any(not field for field in list_fields):
                return response.json(444)

            # 确保至少提供了一个文件
            if not poster_file and not banner_files:
                return response.json(444)

            # 针对特殊地区的处理，表没设计好用这个处理挽救一下
            unknown_country = 0
            if "未知地区" in country_name_list:
                unknown_country = 1
                country_name_list.remove("未知地区")

            # 上传文件到S3

            banner_image_urls = self.upload_files_to_s3(banner_files,
                                                        "BannerAdvertise") if banner_files is not None else []
            poster_image_url = self.upload_files_to_s3(poster_file,
                                                       "OpenScreenAdvertise")[0] if poster_file is not None else ""
            campaign_type = []
            if banner_image_urls is not []:
                campaign_type.append(1)
                banner_campaign = {
                    str(index): {"image": image_url, "url": campaign_url}
                    for index, (image_url, campaign_url) in
                    enumerate(zip(banner_image_urls, banner_campaign_urls), start=1)
                }
            else:
                banner_campaign = {}

            if poster_image_url != "":
                campaign_type.append(2)
            create_time = int(time.time())
            update_time = int(time.time())
            # 创建 AppAdvertiseCampaign 实例
            new_campaign = AppAdvertiseCampaign.objects.create(
                image_url=poster_image_url,
                campaign_name=campaign_name,
                campaign_url=campaign_url,
                banner_campaign=banner_campaign,
                campaign_type=campaign_type,
                status=status,
                unknown_country=unknown_country,
                app_bundle_type=app_bundle_type,
                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)

            return response.json(0)

        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def update_campaign(self, request, request_dict, response):
        """
        更新广告活动
        @param request: 包含文件信息
        @param request_dict: 包含所有请求参数的字典
        @param response: 响应对象
        @return: 响应对象
        """
        try:
            campaign_id = request_dict.get('id', None)
            campaign_name = request_dict.get('campaign_name', None)
            campaign_url = request_dict.get('campaign_url', 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', None)
            campaign_show_etime = request_dict.get('campaign_show_etime', None)
            app_bundle_type = request_dict.get('app_bundle_type', None)
            campaign_type = json.loads(request_dict.get('campaign_type', "[]"))  # 广告类型列表
            device_type_names = json.loads(request_dict.get('device_type_list', "[]"))  # 设备类型名称列表
            country_name_list = json.loads(request_dict.get('country_name_list', "[]"))  # 地区列表
            poster_file = request.FILES.get('posterFile', None)

            # 更改轮播图广告图片
            sort_files = json.loads(request_dict.get('sort_files', "[]"))
            banner_files = request.FILES.getlist('bannerFiles', None)

            # 更改轮播图链接
            sort_urls = json.loads(request_dict.get('sort_url', "[]"))
            banner_urls = json.loads(request_dict.get('banner_urls', "[]"))

            if not campaign_id:
                return response.json(444)

            update_time = int(time.time())
            campaign = AppAdvertiseCampaign.objects.filter(pk=campaign_id).first()

            # 未知地区特殊处理
            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 poster_file is not None:
                self.del_file_to_s3(campaign.image_url)
                poster_image_url = self.upload_files_to_s3(poster_file,
                                                           "OpenScreenAdvertise")[0]
                campaign.image_url = poster_image_url

            # 更改轮播广告图片
            if banner_files is not None:
                image_num = 0
                banner_image_urls = self.upload_files_to_s3(banner_files, "BannerAdvertise")
                for sort in sort_files:
                    old_image = campaign.banner_campaign[str(sort)]["image"]
                    self.del_file_to_s3(old_image)
                    campaign.banner_campaign[str(sort)]["image"] = banner_image_urls[image_num]
                    image_num = image_num + 1

            # 更改轮播广告链接
            if banner_urls:
                url_num = 0
                for sort in sort_urls:
                    campaign.banner_campaign[str(sort)]["url"] = banner_urls[url_num]
                    url_num = url_num + 1

            # 常规字段
            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 app_bundle_type is not None:
                campaign.app_bundle_type = app_bundle_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()

            return response.json(0)
        except Exception as e:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def delete_campaign(self, request_dict, response):
        """
        删除广告活动
        @param request_dict: 包含所有请求参数的字典
        @param response: 响应对象
        @return: 响应对象
        """
        campaign_id = request_dict.get('id')
        if not campaign_id:
            return response.json(444)
        try:
            campaign = AppAdvertiseCampaign.objects.get(pk=campaign_id)
            if campaign.image_url != "":
                self.del_file_to_s3(campaign.image_url)
            if campaign.banner_campaign is not {}:
                banner_campaign = []
                for sort in campaign.banner_campaign:
                    banner = campaign.banner_campaign[sort]
                    banner_campaign.append(banner)
                self.del_file_to_s3(banner_campaign)
            # 清除多对多关系
            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:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def switch_campaign(self, request_dict, response):
        """
        广告活动 开启/关闭
        @param request_dict: 包含所有请求参数的字典
        @param response: 响应对象
        @return: 响应对象
        """
        campaign_id = request_dict.get('id')
        status = request_dict.get('status')
        if not all([campaign_id, status]):
            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:
            print(e)
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_user_behavior_log(self, request_dict, response):
        campaign_ids = request_dict.get('campaign_ids', None)
        start_time = request_dict.get('start_time', None)
        end_time = request_dict.get('end_time', None)
        if not all([start_time, end_time]):
            return response.json(444)
        try:
            open_screen_campaign_qs = OpenScreenCampaign.objects.filter(
                update_time__range=[int(start_time), int(end_time)])

            if campaign_ids is not None:
                open_screen_campaign_qs = open_screen_campaign_qs.filter(
                    Q(campaign_id_id__isnull=True) | Q(campaign_id_id__in=json.loads(campaign_ids)))

            open_screen_campaign_qs = open_screen_campaign_qs.select_related('campaign_id').annotate(
                campaign_name=F('campaign_id__campaign_name'),
                campaign_type=F('campaign_id__campaign_type'),
                campaign_status=F('campaign_id__status'),
                start_date=F('campaign_id__campaign_start_date'),
                end_date=F('campaign_id__campaign_end_date'),
                start_time=F('campaign_id__campaign_show_stime'),
                end_time=F('campaign_id__campaign_show_etime')
            ).values('id', 'user_id', 'status', 'update_time', 'create_time', 'campaign_id', 'campaign_name',
                     'campaign_type', 'campaign_status', 'start_date', 'end_date', 'start_time', 'end_time')
            if not open_screen_campaign_qs.exists():
                return response.json(0, {'list': []})
            campaigns_list = list(open_screen_campaign_qs)
            return response.json(0, {'list': campaigns_list})

        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    def get_country_list(self, response):
        try:
            if SERVER_TYPE == 'Ansjer.us_config.formal_settings':
                region_api = 'https://www.dvema.com/'
            elif SERVER_TYPE == 'Ansjer.eur_config.formal_settings':
                region_api = 'https://api.zositeche.com/'
            elif SERVER_TYPE == 'Ansjer.cn_config.formal_settings':
                region_api = 'https://www.zositechc.cn/'
            else:
                region_api = 'https://test.zositechc.cn/'
            country_qs = CountryModel.objects.filter(region__api=region_api).values('country_name')
            if not country_qs.exists():
                return response.json(173)

            country_list = []
            for country in country_qs:
                country_list.append(country['country_name'])
            return response.json(0, {'list': country_list})
        except Exception as e:
            return response.json(500, 'error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def upload_files_to_s3(files, path_prefix):
        """
        广告图片文件上传到S3存储桶
        @param files: 文件 request.FILES获取
        @param path_prefix: 存储路径
        @return image_urls: 列表
        """
        try:
            _, regin, AWS_SES_ACCESS_REGION = CampaignView.s3_server()

            # 确保files是一个列表，统一处理单个文件和多个文件
            if not isinstance(files, list):
                files = [files]

            image_urls = []
            bucket_name = "ansjerfilemanager"
            s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[regin], AWS_SECRET_ACCESS_KEY[regin], AWS_SES_ACCESS_REGION)
            for file in files:
                timestamp = int(time.time())
                file_key = f'app/campaign/{path_prefix}/{timestamp}_{file.name}'
                s3.upload_file_obj(bucket_name, file_key, file,
                                   {'ContentType': file.content_type, 'ACL': 'public-read'})
                image_urls.append(f"{path_prefix}/{timestamp}_{file.name}")

            return image_urls
        except Exception as e:
            LOGGER.info('存储桶添加异常:error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
            return None

    @staticmethod
    def del_file_to_s3(path_list):
        """
        S3存储桶 删除文件
        @param path_list: 单一字符串或者列表
        @return
        """
        try:
            _, regin, AWS_SES_ACCESS_REGION = CampaignView.s3_server()
            if not isinstance(path_list, list):
                path_list = [path_list]

            for path in path_list:
                # 删除存储桶原来的图片
                s3 = AmazonS3Util(AWS_ACCESS_KEY_ID[regin], AWS_SECRET_ACCESS_KEY[regin], AWS_SES_ACCESS_REGION)
                bucket_name = 'ansjerfilemanager'
                s3.delete_obj(bucket_name, f"app/campaign/{path}")
        except Exception as e:
            LOGGER.info('存储桶删除异常:error_line:{}, error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))

    @staticmethod
    def s3_server():
        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/"
            regin = 0
            AWS_SES_ACCESS_REGION = "cn-northwest-1"
        else:
            s3_url = "https://ansjerfilemanager.s3.amazonaws.com/app/campaign/"
            regin = 1
            AWS_SES_ACCESS_REGION = 'us-east-1'
        return s3_url, regin, AWS_SES_ACCESS_REGION