feat: added advertiser and campaign apps

This commit is contained in:
ITQ
2025-02-20 22:54:28 +03:00
parent 4793973233
commit 0f77f9eb73
24 changed files with 1580 additions and 0 deletions
@@ -0,0 +1,14 @@
from django.contrib import admin
from apps.advertiser.models import Advertiser
class AdvertiserAdmin(admin.ModelAdmin):
readonly_fields = (Advertiser.id.field.name,)
fields = (
Advertiser.id.field.name,
Advertiser.name.field.name,
)
admin.site.register(Advertiser, AdvertiserAdmin)
@@ -0,0 +1,6 @@
from django.apps import AppConfig
class AdvertiserConfig(AppConfig):
name = "apps.advertiser"
label = "advertiser"
@@ -0,0 +1,25 @@
# Generated by Django 5.1.6 on 2025-02-13 21:41
import uuid
from django.db import migrations, models
class Migration(migrations.Migration):
initial = True
dependencies = [
]
operations = [
migrations.CreateModel(
name='Advertiser',
fields=[
('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
('name', models.TextField()),
],
options={
'abstract': False,
},
),
]
@@ -0,0 +1,149 @@
from uuid import UUID
from decimal import Decimal, ROUND_HALF_UP
from django.core.cache import cache
from django.db import models
from apps.core.models import BaseModel
class Advertiser(BaseModel):
name = models.TextField()
def __str__(self) -> str:
return self.name
@property
def advertiser_id(self) -> UUID:
return self.id
@advertiser_id.setter
def advertiser_id(self, value: UUID) -> None:
self.id = value
def get_statistics(self) -> dict[str, int | float]:
campaigns = self.campaigns.all()
total_impressions = 0
total_clicks = 0
total_spent_impressions = Decimal("0.0")
total_spent_clicks = Decimal("0.0")
for campaign in campaigns:
stats = campaign.get_statistics()
total_impressions += stats["impressions_count"]
total_clicks += stats["clicks_count"]
total_spent_impressions += Decimal(str(stats["spent_impressions"]))
total_spent_clicks += Decimal(str(stats["spent_clicks"]))
total_spent = total_spent_impressions + total_spent_clicks
conversion = (
(
Decimal(str(total_clicks))
/ Decimal(str(total_impressions))
* Decimal("100")
)
if total_impressions > 0
else Decimal("0")
)
return {
"impressions_count": total_impressions,
"clicks_count": total_clicks,
"conversion": float(
conversion.quantize(Decimal("0.01"), rounding=ROUND_HALF_UP)
),
"spent_impressions": float(
total_spent_impressions.quantize(
Decimal("0.01"), rounding=ROUND_HALF_UP
)
),
"spent_clicks": float(
total_spent_clicks.quantize(
Decimal("0.01"), rounding=ROUND_HALF_UP
)
),
"spent_total": float(
total_spent.quantize(Decimal("0.01"), rounding=ROUND_HALF_UP)
),
}
def get_daily_statistics(self) -> list[dict[str, int | float]]:
campaigns = self.campaigns.all()
daily_stats_map = {}
for campaign in campaigns:
daily_stats = campaign.get_daily_statistics()
for stat in daily_stats:
date = stat["date"]
if date not in daily_stats_map:
daily_stats_map[date] = {
"impressions_count": 0,
"clicks_count": 0,
"spent_impressions": Decimal("0.0"),
"spent_clicks": Decimal("0.0"),
}
daily_stats_map[date]["impressions_count"] += stat[
"impressions_count"
]
daily_stats_map[date]["clicks_count"] += stat["clicks_count"]
daily_stats_map[date]["spent_impressions"] += Decimal(
str(stat["spent_impressions"])
)
daily_stats_map[date]["spent_clicks"] += Decimal(
str(stat["spent_clicks"])
)
days_range = range(cache.get("current_date", 0) + 1)
for day in days_range:
if day not in daily_stats_map:
daily_stats_map[day] = {
"impressions_count": 0,
"clicks_count": 0,
"spent_impressions": Decimal("0.0"),
"spent_clicks": Decimal("0.0"),
}
daily_stats = []
for date, metrics in daily_stats_map.items():
total_spent = (
metrics["spent_impressions"] + metrics["spent_clicks"]
)
conversion = (
Decimal(str(metrics["clicks_count"]))
/ Decimal(str(metrics["impressions_count"]))
* Decimal("100")
if metrics["impressions_count"] > 0
else Decimal("0")
)
daily_stats.append({
"date": date,
"impressions_count": metrics["impressions_count"],
"clicks_count": metrics["clicks_count"],
"conversion": float(
conversion.quantize(
Decimal("0.01"), rounding=ROUND_HALF_UP
)
),
"spent_impressions": float(
metrics["spent_impressions"].quantize(
Decimal("0.0000000001"), rounding=ROUND_HALF_UP
)
),
"spent_clicks": float(
metrics["spent_clicks"].quantize(
Decimal("0.0000000001"), rounding=ROUND_HALF_UP
)
),
"spent_total": float(
total_spent.quantize(
Decimal("0.0000000001"), rounding=ROUND_HALF_UP
)
),
})
return sorted(daily_stats, key=lambda item: item["date"])
@@ -0,0 +1,48 @@
from uuid import uuid4
from django.test import TestCase, override_settings
from django.core.cache import cache
from apps.advertiser.models import Advertiser
from apps.campaign.models import Campaign
class AdvertiserModelTest(TestCase):
def setUp(self) -> None:
self.advertiser = Advertiser.objects.create(name="Test Advertiser")
def test_advertiser_creation(self):
self.assertIsInstance(self.advertiser, Advertiser)
self.assertEqual(self.advertiser.name, "Test Advertiser")
def test_advertiser_str_method(self):
self.assertEqual(str(self.advertiser), "Test Advertiser")
def test_advertiser_id_property(self):
self.assertEqual(self.advertiser.advertiser_id, self.advertiser.id)
new_id = uuid4()
self.advertiser.advertiser_id = new_id
self.assertEqual(self.advertiser.id, new_id)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advertiser_campaigns_relationship(self):
loll = cache.get("current_date", 0)
campaign = Campaign.objects.create(
advertiser=self.advertiser,
impressions_limit=0,
clicks_limit=0,
cost_per_impression=0,
cost_per_click=0,
ad_title="title",
ad_text="text",
start_date=15,
end_date=16,
)
self.assertIn(campaign, self.advertiser.campaigns.all())
@@ -0,0 +1,189 @@
from django.core.cache import cache
from django.test import TestCase, override_settings
from apps.advertiser.models import Advertiser
from apps.campaign.models import Campaign, CampaignImpression, CampaignClick
from apps.client.models import Client
class AdvertiserStatisticsTest(TestCase):
@classmethod
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def setUpTestData(cls):
cache.set("current_date", 1)
cls.advertiser = Advertiser.objects.create(name="Test Advertiser")
cls.campaign1 = Campaign.objects.create(
advertiser=cls.advertiser,
impressions_limit=1000,
clicks_limit=500,
cost_per_impression=0.05,
cost_per_click=0.10,
ad_title="Campaign 1",
ad_text="This is the first test campaign.",
start_date=1,
end_date=10,
)
cls.campaign2 = Campaign.objects.create(
advertiser=cls.advertiser,
impressions_limit=2000,
clicks_limit=1000,
cost_per_impression=0.04,
cost_per_click=0.08,
ad_title="Campaign 2",
ad_text="This is the second test campaign.",
start_date=2,
end_date=12,
)
cls.client_instance = Client.objects.create(
login="test_client",
age=30,
gender="MALE",
location="Test Location",
)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def setUp(self):
cache.clear()
cache.set("current_date", 5)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_get_statistics_no_data(self):
stats = self.advertiser.get_statistics()
expected_stats = {
"impressions_count": 0,
"clicks_count": 0,
"conversion": 0,
"spent_impressions": 0.0,
"spent_clicks": 0.0,
"spent_total": 0.0,
}
self.assertEqual(stats, expected_stats)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_get_statistics_with_data(self):
CampaignImpression.objects.create(
campaign=self.campaign1,
client=self.client_instance,
price=self.campaign1.cost_per_impression,
date=3,
)
CampaignClick.objects.create(
campaign=self.campaign1,
client=self.client_instance,
price=self.campaign1.cost_per_click,
date=3,
)
CampaignImpression.objects.create(
campaign=self.campaign2,
client=self.client_instance,
price=self.campaign2.cost_per_impression,
date=4,
)
stats = self.advertiser.get_statistics()
expected_stats = {
"impressions_count": 2,
"clicks_count": 1,
"conversion": 50.0,
"spent_impressions": 0.09,
"spent_clicks": 0.10,
"spent_total": 0.19,
}
self.assertEqual(stats, expected_stats)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_get_daily_statistics_no_data(self):
daily_stats = self.advertiser.get_daily_statistics()
expected_stats = [
{
"impressions_count": 0,
"clicks_count": 0,
"conversion": 0,
"spent_impressions": 0.0,
"spent_clicks": 0.0,
"spent_total": 0.0,
"date": day,
}
for day in range(cache.get("current_date", 0) + 1)
]
self.assertEqual(daily_stats, expected_stats)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_get_daily_statistics_with_data(self):
CampaignImpression.objects.create(
campaign=self.campaign1,
client=self.client_instance,
price=self.campaign1.cost_per_impression,
date=3,
)
CampaignClick.objects.create(
campaign=self.campaign1,
client=self.client_instance,
price=self.campaign1.cost_per_click,
date=3,
)
CampaignImpression.objects.create(
campaign=self.campaign2,
client=self.client_instance,
price=self.campaign2.cost_per_impression,
date=4,
)
daily_stats = self.advertiser.get_daily_statistics()
expected_stats = [
{
"impressions_count": 1 if day == 3 else 1 if day == 4 else 0,
"clicks_count": 1 if day == 3 else 0,
"conversion": 100.0 if day == 3 else 0.0,
"spent_impressions": 0.05
if day == 3
else 0.04
if day == 4
else 0.0,
"spent_clicks": 0.10 if day == 3 else 0.0,
"spent_total": 0.15 if day == 3 else 0.04 if day == 4 else 0.0,
"date": day,
}
for day in range(cache.get("current_date") + 1)
]
self.assertEqual(daily_stats, expected_stats)