diff --git a/portier/settings.py b/portier/settings.py index 76b1b8e..c2aa042 100644 --- a/portier/settings.py +++ b/portier/settings.py @@ -28,6 +28,7 @@ ALLOWED_HOSTS = os.environ.get("DJANGO_ALLOWED_HOSTS", default="*").split(" ") DEFAULT_GROUP = 'default' +DEFAULT_RTMP_APPPLICATION = 'live' # Application definition diff --git a/rtmp/forms.py b/rtmp/forms.py new file mode 100644 index 0000000..80e381b --- /dev/null +++ b/rtmp/forms.py @@ -0,0 +1,16 @@ +from django.forms import ModelForm +from guardian.shortcuts import get_objects_for_user +from . import models + + +class StreamFilteredApplicationForm(ModelForm): + class Meta: + model = models.Stream + fields = ['name', 'application'] + + def __init__(self, *args, **kwargs): + user = kwargs.pop('user', None) + super().__init__(*args, **kwargs) + + # limit the stream selection to user-accessible streams + self.fields['application'].queryset = get_objects_for_user(user, 'rtmp.view_application') diff --git a/rtmp/management/commands/createdefaultapplication.py b/rtmp/management/commands/createdefaultapplication.py new file mode 100644 index 0000000..0e77a56 --- /dev/null +++ b/rtmp/management/commands/createdefaultapplication.py @@ -0,0 +1,16 @@ +from django.core.management.base import BaseCommand +from django.contrib.auth.models import Group +from django.conf import settings +from guardian.shortcuts import assign_perm + +from rtmp import models + + +class Command(BaseCommand): + help = 'Creates a default RTMP application that is available to all users in the default group' + + def handle(self, *args, **options): + default_group, _ = Group.objects.get_or_create(name=settings.DEFAULT_GROUP) + default_app, _ = models.Application.objects.get_or_create(name=settings.DEFAULT_RTMP_APPPLICATION) + + assign_perm('view_application', default_group, default_app) diff --git a/rtmp/views.py b/rtmp/views.py index afc9df2..d96ab60 100644 --- a/rtmp/views.py +++ b/rtmp/views.py @@ -13,6 +13,7 @@ from guardian.decorators import permission_required_or_403 from guardian.shortcuts import assign_perm from . import models +from . import forms logger = logging.getLogger(__name__) @@ -70,16 +71,26 @@ class StreamDetail(DetailView): name='dispatch') class StreamChange(UpdateView): model = models.Stream - fields = ['application', 'name'] + form_class = forms.StreamFilteredApplicationForm template_name_suffix = '_update_form' + def get_form_kwargs(self): + kwargs = super().get_form_kwargs() + kwargs['user'] = self.request.user + return kwargs + @method_decorator(login_required, name='dispatch') @method_decorator(permission_required_or_403('rtmp.add_stream'), name='dispatch') class StreamCreate(CreateView): model = models.Stream - fields = ["name", "application"] + form_class = forms.StreamFilteredApplicationForm + + def get_form_kwargs(self): + kwargs = super().get_form_kwargs() + kwargs['user'] = self.request.user + return kwargs def form_valid(self, form): valid = super().form_valid(form)