Commands to enable talk and video system (#1107)

* Add Django command to enable video system for an event

* Add enable_talk_system

* Raise CommandError

Author: hongquan

src/pretix/eventyay_common/management/__init__.py

@@ -0,0 +1,50 @@
+from django.core.management.base import BaseCommand, CommandError
+from django_scopes import scope
+
+from pretix.base.models import Event, Organizer, User
+from pretix.eventyay_common.tasks import send_event_webhook
+
+
+class Command(BaseCommand):
+    help = 'Enable talk system for an event'
+
+    def add_arguments(self, parser):
+        parser.add_argument('event_slug', type=str, help='Event slug')
+        parser.add_argument('-o', dest='organizer_slug', type=str, required=True, help='Organizer slug')
+        parser.add_argument('-u', dest='user_email', type=str, required=True, help='User email address')
+
+    def handle(self, event_slug: str, organizer_slug: str, user_email: str, **options):
+
+        try:
+            user = User.objects.get(email=user_email)
+        except User.DoesNotExist:
+            raise CommandError(f'User with email {user_email} does not exist')
+
+        try:
+            organizer = Organizer.objects.get(slug=organizer_slug)
+        except Organizer.DoesNotExist:
+            raise CommandError(f'Organizer {organizer_slug} does not exist')
+
+        try:
+            with scope(organizer=organizer):
+                event = Event.objects.get(slug=event_slug, organizer=organizer)
+        except Event.DoesNotExist:
+            raise CommandError(f'Event {organizer_slug}/{event_slug} does not exist')
+
+        send_event_webhook(
+            user_id=user.id,
+            event={
+                'organiser_slug': event.organizer.slug,
+                'name': event.name.data,
+                'slug': event.slug,
+                'date_from': str(event.date_from),
+                'date_to': str(event.date_to),
+                'timezone': str(event.settings.timezone),
+                'locale': event.settings.locale,
+                'locales': event.settings.locales,
+                'is_video_creation': event.is_video_creation,
+            },
+            action='create',
+        )
+
+        self.stdout.write(f'🎉 Talk system enabled for event {organizer_slug}/{event_slug}.')

src/pretix/eventyay_common/management/commands/__init__.py

@@ -0,0 +1,63 @@
+from datetime import datetime, timedelta, timezone
+
+import jwt
+from django.conf import settings
+from django.core.management.base import BaseCommand, CommandError
+from django_scopes import scope
+
+from pretix.base.models import Event, Organizer, User
+from pretix.eventyay_common.tasks import create_world
+from pretix.eventyay_common.utils import encode_email
+
+
+class Command(BaseCommand):
+    help = 'Enable video system for an event'
+
+    def add_arguments(self, parser):
+        parser.add_argument('event_slug', type=str, help='Event slug')
+        parser.add_argument('-o', dest='organizer_slug', type=str, required=True, help='Organizer slug')
+        parser.add_argument('-u', dest='user_email', type=str, required=True, help='User email address')
+
+    def handle(self, event_slug: str, organizer_slug: str, user_email: str, **options):
+
+        try:
+            user = User.objects.get(email=user_email)
+        except User.DoesNotExist:
+            raise CommandError(f'User with email {user_email} does not exist')
+
+        try:
+            organizer = Organizer.objects.get(slug=organizer_slug)
+        except Organizer.DoesNotExist:
+            raise CommandError(f'Organizer {organizer_slug} does not exist')
+
+        try:
+            with scope(organizer=organizer):
+                event = Event.objects.get(slug=event_slug, organizer=organizer)
+        except Event.DoesNotExist:
+            raise CommandError(f'Event {organizer_slug}/{event_slug} does not exist')
+
+        # Generate token for video system (admin token)
+        iat = datetime.now(timezone.utc)
+        exp = iat + timedelta(days=30)
+        payload = {
+            'exp': exp,
+            'iat': iat,
+            'uid': encode_email(user.email),
+            'has_permission': True,
+        }
+        token = jwt.encode(payload, settings.SECRET_KEY, algorithm='HS256')
+
+        create_world(
+            is_video_creation=True,
+            event_data={
+                'id': event.slug,
+                'title': event.name.data,
+                'timezone': event.settings.timezone,
+                'locale': event.settings.locale,
+                'has_permission': True,
+                'token': token,
+            },
+        )
+
+        self.stdout.write(f'🎉 Video system enabled for event {organizer_slug}/{event_slug}.')
+

src/pretix/eventyay_common/management/commands/enable_talk_system.py

@@ -8,7 +8,7 @@
 from django.contrib import messages
 from django.core.exceptions import PermissionDenied
 from django.db import transaction
-from django.db.models import Case, F, Max, Min, Prefetch, Q, Sum, When, IntegerField
+from django.db.models import Case, F, IntegerField, Max, Min, Prefetch, Q, Sum, When
 from django.db.models.functions import Coalesce, Greatest
 from django.http import HttpRequest, HttpResponseRedirect, JsonResponse
 from django.shortcuts import redirect
@@ -24,7 +24,6 @@
 from pretix.base.models import Event, EventMetaValue, Organizer, Quota
 from pretix.base.services import tickets
 from pretix.base.services.quotas import QuotaAvailability
-
 from pretix.control.forms.event import EventWizardBasicsForm, EventWizardFoundationForm
 from pretix.control.forms.filter import EventFilterForm
 from pretix.control.permissions import EventPermissionRequiredMixin
@@ -40,8 +39,10 @@
     generate_token,
 )
 from pretix.helpers.plugin_enable import is_video_enabled
+
 from ..forms.event import EventUpdateForm
 
+
 class EventList(PaginationMixin, ListView):
     model = Event
     context_object_name = 'events'
@@ -441,7 +442,7 @@ def post(self, request, *args, **kwargs):
                         'is_video_creation': request.event.is_video_creation,
                     }
                     send_event_webhook.delay(user_id=self.request.user.id, event=event_dict, action='update')
-                messages.success(self.request, _("Your changes have been saved."))
+                messages.success(self.request, _('Your changes have been saved.'))
                 return self.form_valid(form)
             else:
                 messages.error(