pluto/core/views/ads.py

from django.contrib.auth.mixins import LoginRequiredMixin
from django.shortcuts import render
from django.urls import reverse
from django.views import View
from mixins.views import ObjectCreate, ObjectDelete, ObjectList, ObjectUpdate
from two_factor.views.mixins import OTPRequiredMixin

from core.clients.platforms.agora import AgoraClient
from core.forms import AdForm
from core.models import Ad, LinkGroup
from core.views.helpers import synchronize_async_helper


def get_linkgroups(user):
    return LinkGroup.objects.filter(user=user, enabled=True)


def get_platforms(user):
    groups = get_linkgroups(user)
    platforms = []
    for group in groups:
        for platform in group.platforms.all():
            platforms.append(platform)

    return platforms


class Cheat(LoginRequiredMixin, OTPRequiredMixin, View):
    template_name = "mixins/partials/notify.html"

    def get(self, request):
        platforms = get_platforms(request.user)
        for platform in platforms:
            run = synchronize_async_helper(AgoraClient(platform))
            synchronize_async_helper(run.cheat())

        context = {"class": "success", "message": "Cheat run"}
        return render(request, self.template_name, context)


class AdNuke(LoginRequiredMixin, OTPRequiredMixin, View):
    template_name = "mixins/partials/notify.html"

    def get(self, request):
        platforms = get_platforms(request.user)
        for platform in platforms:
            run = synchronize_async_helper(AgoraClient(platform))
            synchronize_async_helper(run.nuke_ads())

        context = {"class": "success", "message": "Ads nuked"}
        return render(request, self.template_name, context)


class AdDist(LoginRequiredMixin, OTPRequiredMixin, View):
    template_name = "mixins/partials/notify.html"

    def get(self, request):
        ads = Ad.objects.filter(user=request.user, enabled=True)
        for ad in ads:
            if ad.link_group is not None:
                for platform in ad.link_group.platforms.all():
                    run = synchronize_async_helper(AgoraClient(platform))
                    synchronize_async_helper(run.dist_countries(ad))

        context = {"class": "success", "message": "Ads distributed"}
        return render(request, self.template_name, context)


class AdRedist(LoginRequiredMixin, OTPRequiredMixin, View):
    template_name = "mixins/partials/notify.html"

    def get(self, request):
        ads = Ad.objects.filter(user=request.user, enabled=True)
        for ad in ads:
            if ad.link_group is not None:
                for platform in ad.link_group.platforms.all():
                    run = synchronize_async_helper(AgoraClient(platform))
                    synchronize_async_helper(run.redist_countries(ad))

        context = {"class": "success", "message": "Ads updated"}
        return render(request, self.template_name, context)


class AdDedup(LoginRequiredMixin, OTPRequiredMixin, View):
    template_name = "mixins/partials/notify.html"

    def get(self, request):
        platforms = get_platforms(request.user)
        for platform in platforms:
            run = synchronize_async_helper(AgoraClient(platform))
            synchronize_async_helper(run.strip_duplicate_ads())

        context = {"class": "success", "message": "Ads deduplicated"}
        return render(request, self.template_name, context)


class AdList(LoginRequiredMixin, OTPRequiredMixin, ObjectList):
    list_template = "partials/ad-list.html"
    model = Ad
    page_title = "List of ads"

    list_url_name = "ads"
    list_url_args = ["type"]

    submit_url_name = "ad_create"

    def get_context_data(self):
        context = super().get_context_data()
        self.extra_buttons = [
            {
                "url": reverse("ad_dist"),
                "action": "distribute",
                "method": "get",
                "label": "Distribute ads",
                "icon": "fa-solid fa-upload",
            },
            {
                "url": reverse("ad_redist"),
                "action": "redistribute",
                "method": "get",
                "label": "Update ads",
                "icon": "fa-solid fa-refresh",
            },
            {
                "url": reverse("cheat"),
                "action": "cheat",
                "method": "get",
                "label": "Run cheat",
                "icon": "fa-solid fa-bolt",
            },
            {
                "url": reverse("ad_nuke"),
                "action": "nuke",
                "method": "get",
                "label": "Nuke ads",
                "icon": "fa-solid fa-bomb",
                "confirm": True,
            },
            {
                "url": reverse("ad_dedup"),
                "action": "deduplicate",
                "method": "get",
                "label": "Deduplicate ads",
                "icon": "fa-thin fa-copy",
                "confirm": True,
            },
        ]
        return context


class AdCreate(LoginRequiredMixin, OTPRequiredMixin, ObjectCreate):
    model = Ad
    form_class = AdForm

    submit_url_name = "ad_create"


class AdUpdate(LoginRequiredMixin, OTPRequiredMixin, ObjectUpdate):
    model = Ad
    form_class = AdForm

    submit_url_name = "ad_update"


class AdDelete(LoginRequiredMixin, OTPRequiredMixin, ObjectDelete):
    model = Ad