simo 2.5.9__py3-none-any.whl → 2.5.10__py3-none-any.whl

simo/core/serializers.py

@@ -104,6 +104,34 @@ class HiddenSerializerField(serializers.CharField):
 class TextAreaSerializerField(serializers.CharField):
     pass
 
+class ComponentPrimaryKeyRelatedField(PrimaryKeyRelatedField):
+
+    def get_attribute(self, instance):
+        if self.queryset.model in (Icon, Zone, Category):
+            return super().get_attribute(instance)
+        return self.queryset.model.objects.filter(
+            pk=instance.config.get(self.source_attrs[0])
+        ).first()
+
+
+class ComponentManyToManyRelatedField(serializers.Field):
+
+    def __init__(self, *args, **kwargs):
+        self.queryset = kwargs.pop('queryset')
+        self.choices = {obj.pk: str(obj) for obj in self.queryset}
+        self.allow_blank = kwargs.pop('allow_blank', False)
+        super().__init__(*args, **kwargs)
+
+
+    def to_representation(self, value):
+        return [obj.pk for obj in value.all()]
+
+    def to_internal_value(self, data):
+        if data == [] and self.allow_blank:
+            return []
+        return self.queryset.filter(pk__in=data)
+
+
 
 class ComponentFormsetField(FormSerializer):
 
@@ -114,6 +142,10 @@ class ComponentFormsetField(FormSerializer):
         field_mapping = {
             HiddenField: HiddenSerializerField,
             Select2ListChoiceField: serializers.ChoiceField,
+            Select2ModelChoiceField: ComponentPrimaryKeyRelatedField,
+            forms.ModelMultipleChoiceField: ComponentManyToManyRelatedField,
+            Select2ListMultipleChoiceField: ComponentManyToManyRelatedField,
+            Select2ModelMultipleChoiceField: ComponentManyToManyRelatedField,
             forms.ModelChoiceField: FormsetPrimaryKeyRelatedField,
             forms.TypedChoiceField: serializers.ChoiceField,
             forms.FloatField: serializers.FloatField,
@@ -167,8 +199,18 @@ class ComponentFormsetField(FormSerializer):
     def _get_field_kwargs(self, form_field, serializer_field_class):
         kwargs = super()._get_field_kwargs(form_field, serializer_field_class)
         kwargs['style'] = {'form_field': form_field}
-        if serializer_field_class == FormsetPrimaryKeyRelatedField:
-            kwargs['queryset'] = form_field.queryset
+
+        if serializer_field_class in (
+            ComponentPrimaryKeyRelatedField, ComponentManyToManyRelatedField
+        ):
+            qs = form_field.queryset
+            if hasattr(qs.model, 'instance'):
+                qs = qs.filter(instance=self.context['instance'])
+            elif hasattr(qs.model, 'instances'):
+                qs = qs.filter(instances=self.context['instance'])
+            elif qs.model == Component:
+                qs = qs.filter(zone__instance=self.context['instance'])
+            kwargs['queryset'] = qs
 
         attrs = find_matching_class_kwargs(form_field, serializer_field_class)
         if 'choices' in attrs:
@@ -176,6 +218,7 @@ class ComponentFormsetField(FormSerializer):
 
         return kwargs
 
+
     def to_representation(self, instance):
         """
         Object instance -> Dict of primitive datatypes.
@@ -211,32 +254,6 @@ class ComponentFormsetField(FormSerializer):
         return validated_data
 
 
-class ComponentPrimaryKeyRelatedField(PrimaryKeyRelatedField):
-
-    def get_attribute(self, instance):
-        if self.queryset.model in (Icon, Zone, Category):
-            return super().get_attribute(instance)
-        return self.queryset.model.objects.filter(
-            pk=instance.config.get(self.source_attrs[0])
-        ).first()
-
-
-class ComponentManyToManyRelatedField(serializers.Field):
-
-    def __init__(self, *args, **kwargs):
-        self.queryset = kwargs.pop('queryset')
-        self.choices = {obj.pk: str(obj) for obj in self.queryset}
-        self.allow_blank = kwargs.pop('allow_blank', False)
-        super().__init__(*args, **kwargs)
-
-
-    def to_representation(self, value):
-        return [obj.pk for obj in value.all()]
-
-    def to_internal_value(self, data):
-        if data == [] and self.allow_blank:
-            return []
-        return self.queryset.filter(pk__in=data)
 
 
 class ComponentSerializer(FormSerializer):
@@ -291,7 +308,6 @@ class ComponentSerializer(FormSerializer):
                 self.instance = Component.objects.filter(id=res[0]).first()
 
     def get_fields(self):
-
         self.set_form_cls()
 
         ret = OrderedDict()

{simo-2.5.9.dist-info → simo-2.5.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: simo
-Version: 2.5.9
+Version: 2.5.10
 Summary: Smart Home on Steroids!
 Author-email: Simanas Venčkauskas <simanas@simo.io>
 Project-URL: Homepage, https://simo.io

{simo-2.5.9.dist-info → simo-2.5.10.dist-info}/RECORD

@@ -55,7 +55,7 @@ simo/core/middleware.py,sha256=hExD7Vmw7eitk0vAjOwKzkwrtuw8YxpflF92j_CA2YY,3193
 simo/core/models.py,sha256=Z6WLvU4K-LWIXghCBwTyZAOW5n2hCFNU-pteiPnpOAQ,22617
 simo/core/permissions.py,sha256=v0iJM4LOeYoEfMiw3OLPYio272G1aUEAg_z9Wd1q5m0,2993
 simo/core/routing.py,sha256=X1_IHxyA-_Q7hw1udDoviVP4_FSBDl8GYETTC2zWTbY,499
-simo/core/serializers.py,sha256=JoMsGoeeTfnoNSdLPMlx4ZbaFyhtw5Kh4KpbAIUovX8,20869
+simo/core/serializers.py,sha256=i6AWIfhJq86AHs8jq3GrCfW3iSQDS3fQiLmcjHd3K5E,21593
 simo/core/signal_receivers.py,sha256=qCpzEUv5Bl9--K8fe08GVDmE6EBOj292YBia1TYDdSE,9267
 simo/core/socket_consumers.py,sha256=trRZvBGTJ7xIbfdmVvn7zoiWp_qssSkMZykDrI5YQyE,9783
 simo/core/storage.py,sha256=_5igjaoWZAiExGWFEJMElxUw55DzJG1jqFty33xe8BE,342
@@ -88,7 +88,7 @@ simo/core/__pycache__/middleware.cpython-38.pyc,sha256=iOSTXSQl3sEsa-9kx_6w5zbEB
 simo/core/__pycache__/models.cpython-38.pyc,sha256=Nw2Fb11EwBdMVTHCpzkY6x0hRNCqRo0RtLNlRxS1VZE,18534
 simo/core/__pycache__/permissions.cpython-38.pyc,sha256=fH4iyqd9DdzRLEu2b621-FeM-napR0M7hzBUTHo9Q3g,2972
 simo/core/__pycache__/routing.cpython-38.pyc,sha256=3T3FPJ8Cn99xZCGvMyg2xjl7al-Shm9CelbSpkJtNP8,599
-simo/core/__pycache__/serializers.cpython-38.pyc,sha256=S4eNvJkMZO98FQXgNGbeKBJihG5LXdnhl0tm2tjjXTE,19266
+simo/core/__pycache__/serializers.cpython-38.pyc,sha256=fN3kkfxFbQczNkAoghp_mjuXEuFhpiecd87MB3cXW6k,19483
 simo/core/__pycache__/signal_receivers.cpython-38.pyc,sha256=lBVca6zNPVn3Ev98ekjPGzBR1MJk4xI19CyMcm4lf6A,7056
 simo/core/__pycache__/socket_consumers.cpython-38.pyc,sha256=KqbO1cOewodVPcy0-htVefyUjCuELKV0o7fOfYqfgPc,8490
 simo/core/__pycache__/storage.cpython-38.pyc,sha256=9R1Xu0FJDflfRXUPsqEgt0SpwiP7FGk7HaR8s8XRyI8,721
@@ -10551,9 +10551,9 @@ simo/users/templates/invitations/expired_msg.html,sha256=47DEQpj8HBSa-_TImW-5JCe
 simo/users/templates/invitations/expired_suggestion.html,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 simo/users/templates/invitations/taken_msg.html,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 simo/users/templates/invitations/taken_suggestion.html,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-simo-2.5.9.dist-info/LICENSE.md,sha256=M7wm1EmMGDtwPRdg7kW4d00h1uAXjKOT3HFScYQMeiE,34916
-simo-2.5.9.dist-info/METADATA,sha256=seKSLClAKBnMi8ejXmRi6A8gouqbRobDzkzLi4yiFvc,1923
-simo-2.5.9.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-simo-2.5.9.dist-info/entry_points.txt,sha256=S9PwnUYmTSW7681GKDCxUbL0leRJIaRk6fDQIKgbZBA,135
-simo-2.5.9.dist-info/top_level.txt,sha256=GmS1hrAbpVqn9OWZh6UX82eIOdRLgYA82RG9fe8v4Rs,5
-simo-2.5.9.dist-info/RECORD,,
+simo-2.5.10.dist-info/LICENSE.md,sha256=M7wm1EmMGDtwPRdg7kW4d00h1uAXjKOT3HFScYQMeiE,34916
+simo-2.5.10.dist-info/METADATA,sha256=c5kqqHmuLRew5BhrH8PMdeuw6jE_D0ehl-R0xv0Guk0,1924
+simo-2.5.10.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+simo-2.5.10.dist-info/entry_points.txt,sha256=S9PwnUYmTSW7681GKDCxUbL0leRJIaRk6fDQIKgbZBA,135
+simo-2.5.10.dist-info/top_level.txt,sha256=GmS1hrAbpVqn9OWZh6UX82eIOdRLgYA82RG9fe8v4Rs,5
+simo-2.5.10.dist-info/RECORD,,