Created command to collect discord user ids

renzo · renzon · commit 707322fe6d12 · 2024-02-28T15:21:29.000-03:00
Partial of #4791
diff --git a/pythonpro/celery.py b/pythonpro/celery.py
@@ -5,5 +5,5 @@
 
 os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'pythonpro.settings')
 app = Celery('pythonpro.celery')
-app.config_from_object('django.conf:settings', namespace='CELERY')
-app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)
+app.config_from_object(settings, namespace='CELERY')
+app.autodiscover_tasks()
diff --git a/pythonpro/discord/admin.py b/pythonpro/discord/admin.py
@@ -1,6 +1,6 @@
 from django.contrib import admin
 
-from pythonpro.discord.models import DiscordUser
+from pythonpro.discord.models import DiscordUser, DiscordLead
 
 
 @admin.register(DiscordUser)
@@ -18,3 +18,20 @@ def has_change_permission(self, request, obj=None):
 
     def has_delete_permission(self, request, obj=None):
         return False
+
+
+@admin.register(DiscordLead)
+class DiscordLeadAdmin(admin.ModelAdmin):
+    fields = ["discord_id", "status", "created_at", "updated_at"]
+    list_display = fields
+    list_filter = ['status']
+    ordering = ['-updated_at']
+
+    def has_add_permission(self, request):
+        return False
+
+    def has_change_permission(self, request, obj=None):
+        return False
+
+    def has_delete_permission(self, request, obj=None):
+        return False
diff --git a/pythonpro/discord/api_client.py b/pythonpro/discord/api_client.py
@@ -35,6 +35,15 @@ class DiscordBotClient:
     def __init__(self, bot_token: str):
         self._bot_token = bot_token
 
+    def get_member(self, discord_user_id: str):
+        headers = {
+            'Authorization': f'Bot {self._bot_token}'
+        }
+        r = requests.get(f'{_BASE_ENDPOINT_URI}/users/{discord_user_id}', headers=headers)
+        r.raise_for_status()
+        discord_user_dict = r.json()
+        return discord_user_dict
+
     def list_guild_members(self, guild_id, limit=100, after=0) -> dict:
         """
         :param guild_id: the discord server id
@@ -138,10 +147,3 @@ def generate_api_client(self, autorization_code: str) -> Union[DiscordAppClient,
         if self._bot_token is None:
             return DiscordAppClient(access_token)
         return DiscordAppAndBotClient(access_token, self._bot_token)
-
-
-if __name__ == '__main__':
-    import decouple
-    c = DiscordBotClient(decouple.config('DISCORD_APP_BOT_TOKEN'))
-    members = c.list_guild_members(971162582624903288)
-    print(members)
diff --git a/pythonpro/discord/facade.py b/pythonpro/discord/facade.py
@@ -0,0 +1,21 @@
+from django.conf import settings
+
+from pythonpro.discord.api_client import DiscordBotClient
+from pythonpro.discord.tasks import clean_discord_user
+
+discord_bot_client = DiscordBotClient(settings.DISCORD_APP_BOT_TOKEN)
+
+
+def clean_discord_users():
+    discord_user_id = 0
+    while True:
+        discord_members = discord_bot_client.list_guild_members(settings.DISCORD_GUILD_ID, after=discord_user_id)
+        if len(discord_members) == 0:
+            break
+        for member in discord_members:
+            discord_user = member['user']
+            discord_user_id = discord_user['id']
+            is_bot = discord_user.get('bot', False)
+            if is_bot:
+                continue
+            clean_discord_user.delay(discord_user_id)
diff --git a/pythonpro/discord/management/__init__.py b/pythonpro/discord/management/__init__.py
diff --git a/pythonpro/discord/management/commands/__init__.py b/pythonpro/discord/management/commands/__init__.py
diff --git a/pythonpro/discord/management/commands/clean_discord_users.py b/pythonpro/discord/management/commands/clean_discord_users.py
@@ -0,0 +1,13 @@
+from django.core.management import BaseCommand
+
+from pythonpro.discord import facade
+
+
+class Command(BaseCommand):
+    help = 'Sincronizar usuários do Discord com Painel da DevPro'
+
+    def add_arguments(self, parser):
+        pass
+
+    def handle(self, *args, **options):
+        facade.clean_discord_users()
diff --git a/pythonpro/discord/migrations/0003_create_discord_lead.py b/pythonpro/discord/migrations/0003_create_discord_lead.py
@@ -0,0 +1,23 @@
+# Generated by Django 4.2.10 on 2024-02-28 18:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('discord', '0002_discord_user_created_at_desc_index'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='DiscordLead',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('discord_id', models.CharField(max_length=64, unique=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('status', models.CharField(choices=[('A', 'Ativa'), ('I', 'Inativa')], default='I', max_length=32)),
+            ],
+        ),
+    ]
diff --git a/pythonpro/discord/models.py b/pythonpro/discord/models.py
@@ -15,3 +15,18 @@ class Meta:
 
     def __str__(self):
         return self.discord_email
+
+
+class DiscordLead(models.Model):
+    class Status(models.TextChoices):
+        ACTIVE = 'A', 'Ativa'
+        INACTIVE = 'I', 'Inativa'
+
+    discord_id = models.CharField(max_length=64, unique=True)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    status = models.CharField(
+        max_length=32,
+        choices=Status.choices,
+        default=Status.INACTIVE
+    )
diff --git a/pythonpro/discord/tasks.py b/pythonpro/discord/tasks.py
@@ -0,0 +1,27 @@
+from celery import shared_task
+
+from pythonpro.discord.models import DiscordLead
+from pythonpro.memberkit.models import Subscription
+
+
+@shared_task(
+    rate_limit=1,
+    max_retries=5,
+    retry_backoff=True,
+    retry_backoff_max=700,
+    retry_jitter=True
+)
+def clean_discord_user(discord_user_id):
+    has_discord_access = Subscription.objects.filter(
+        status=Subscription.Status.ACTIVE,
+        subscription_types__has_discord_access=True,
+        subscriber__discorduser__discord_id=discord_user_id
+    ).exists()
+
+    lead_status = DiscordLead.Status.ACTIVE if has_discord_access else DiscordLead.Status.INACTIVE
+    DiscordLead.objects.update_or_create(
+        defaults={'status': lead_status},
+        discord_id=discord_user_id
+    )
+
+    return print(f'Clean discord user: {discord_user_id}')
diff --git a/pythonpro/discord/tests/test_clean_discord_user.py b/pythonpro/discord/tests/test_clean_discord_user.py
@@ -0,0 +1,57 @@
+from pythonpro.discord.models import DiscordLead, DiscordUser
+from pythonpro.discord.tasks import clean_discord_user
+from model_bakery import baker
+
+from pythonpro.memberkit.models import SubscriptionType, Subscription
+
+
+def test_active_user(db):
+    discord_user_id = '1055109241507160165'
+    django_user = baker.make(DiscordUser, discord_id=discord_user_id).user
+    subscription_type = baker.make(SubscriptionType, has_discord_access=True)
+    baker.make(
+        Subscription,
+        subscription_types=[subscription_type],
+        subscriber=django_user,
+        status=Subscription.Status.ACTIVE
+    )
+    clean_discord_user(discord_user_id)
+    user = DiscordLead.objects.get(discord_id=discord_user_id)
+    assert user.status == DiscordLead.Status.ACTIVE
+
+
+def test_no_discord_user(db):
+    discord_user_id = '1055109241507160165'
+    clean_discord_user(discord_user_id)
+    user = DiscordLead.objects.get(discord_id=discord_user_id)
+    assert user.status == DiscordLead.Status.INACTIVE
+
+
+def test_subscription_inactive(db):
+    discord_user_id = '1055109241507160165'
+    django_user = baker.make(DiscordUser, discord_id=discord_user_id).user
+    subscription_type = baker.make(SubscriptionType, has_discord_access=True)
+    baker.make(
+        Subscription,
+        subscription_types=[subscription_type],
+        subscriber=django_user,
+        status=Subscription.Status.INACTIVE
+    )
+    clean_discord_user(discord_user_id)
+    user = DiscordLead.objects.get(discord_id=discord_user_id)
+    assert user.status == DiscordLead.Status.INACTIVE
+
+
+def test_subscription_type_has_no_discord_access(db):
+    discord_user_id = '1055109241507160165'
+    django_user = baker.make(DiscordUser, discord_id=discord_user_id).user
+    subscription_type = baker.make(SubscriptionType, has_discord_access=False)
+    baker.make(
+        Subscription,
+        subscription_types=[subscription_type],
+        subscriber=django_user,
+        status=Subscription.Status.ACTIVE
+    )
+    clean_discord_user(discord_user_id)
+    user = DiscordLead.objects.get(discord_id=discord_user_id)
+    assert user.status == DiscordLead.Status.INACTIVE
diff --git a/pythonpro/discord/tests/test_commands.py b/pythonpro/discord/tests/test_commands.py
@@ -0,0 +1,6 @@
+from django.core import management
+
+
+def test_clean_discord_users_command(mocker):
+    mocker.patch('pythonpro.discord.facade.discord_bot_client.list_guild_members', return_value=[])
+    management.call_command('clean_discord_users')
