Proyecto LCX Dispatcharr multicuenta

apps/m3u/forms.py

@@ -0,0 +1,79 @@
+# apps/m3u/forms.py
+from django import forms
+from .models import M3UAccount, M3UFilter
+import re
+
+class M3UAccountForm(forms.ModelForm):
+    enable_vod = forms.BooleanField(
+        required=False,
+        initial=False,
+        label="Enable VOD Content",
+        help_text="Parse and import VOD (movies/series) content for XtreamCodes accounts"
+    )
+
+    class Meta:
+        model = M3UAccount
+        fields = [
+            'name',
+            'server_url',
+            'uploaded_file',
+            'server_group',
+            'max_streams',
+            'is_active',
+            'enable_vod',
+        ]
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        # Set initial value for enable_vod from custom_properties
+        if self.instance and self.instance.custom_properties:
+            custom_props = self.instance.custom_properties or {}
+            self.fields['enable_vod'].initial = custom_props.get('enable_vod', False)
+
+    def save(self, commit=True):
+        instance = super().save(commit=False)
+
+        # Handle enable_vod field
+        enable_vod = self.cleaned_data.get('enable_vod', False)
+
+        # Parse existing custom_properties
+        custom_props = instance.custom_properties or {}
+
+        # Update VOD preference
+        custom_props['enable_vod'] = enable_vod
+        instance.custom_properties = custom_props
+
+        if commit:
+            instance.save()
+        return instance
+
+    def clean_uploaded_file(self):
+        uploaded_file = self.cleaned_data.get('uploaded_file')
+        if uploaded_file:
+            if not uploaded_file.name.endswith('.m3u'):
+                raise forms.ValidationError("The uploaded file must be an M3U file.")
+        return uploaded_file
+
+    def clean(self):
+        cleaned_data = super().clean()
+        url = cleaned_data.get('server_url')
+        file = cleaned_data.get('uploaded_file')
+        # Ensure either `server_url` or `uploaded_file` is provided
+        if not url and not file:
+            raise forms.ValidationError("Either an M3U URL or a file upload is required.")
+        return cleaned_data
+
+
+class M3UFilterForm(forms.ModelForm):
+    class Meta:
+        model = M3UFilter
+        fields = ['m3u_account', 'filter_type', 'regex_pattern', 'exclude']
+
+    def clean_regex_pattern(self):
+        pattern = self.cleaned_data['regex_pattern']
+        try:
+            re.compile(pattern)
+        except re.error:
+            raise forms.ValidationError("Invalid regex pattern")
+        return pattern