add aggregator app

aggregator/forms.py

@@ -0,0 +1,55 @@
+from django import forms
+from django.core.exceptions import ValidationError
+from django.core.files.storage import default_storage
+from django.forms import ModelForm
+from djconfig.forms import ConfigForm
+from pyrogram.api.errors import ChannelPrivate
+
+from config.utils import parse_mtproto_chat
+from .client import get_client
+from .models import AggregationSource, Chat
+
+
+class AggregationSourceForm(ModelForm):
+    invite_link = forms.CharField(help_text='Invite link (with joinchat) or username')
+
+    def clean(self):
+        cleaned_data = super(AggregationSourceForm, self).clean()
+        invite_link = cleaned_data.pop('invite_link')
+        if invite_link:
+            with get_client() as app:
+                try:
+                    upd = app.join_chat(invite_link)
+                    chat = parse_mtproto_chat(app, upd.chats[0])
+                    cleaned_data['chat_id'] = chat.id
+                    Chat.from_obj(chat, app)
+                except ChannelPrivate:
+                    raise ValidationError('I was banned in this chat')
+        return cleaned_data
+
+    class Meta:
+        model = AggregationSource
+        fields = 'title', 'invite_link',
+
+
+class AggregationSourceEditForm(AggregationSourceForm):
+    invite_link = forms.CharField(help_text='Invite link (with joinchat) or username; '
+                                            'leave empty if change is not needed', required=False)
+
+
+class AggregatorAppConfigForm(ConfigForm):
+    slug = 'aggregator'
+    title = 'Aggregator'
+
+    pyrogram_app_id = forms.CharField()
+    pyrogram_app_hash = forms.CharField()
+    pyrogram_session = forms.FileField()
+
+    def save_session(self):
+        session = self.cleaned_data.get('pyrogram_session')
+        if session:
+            session.name = default_storage.save(session.name, session)
+
+    def save(self):
+        self.save_session()
+        super(AggregatorAppConfigForm, self).save()