cli2 3.3.6__tar.gz → 3.3.7__tar.gz

{cli2-3.3.6/cli2.egg-info → cli2-3.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cli2
-Version: 3.3.6
+Version: 3.3.7
 Summary: image:: https://yourlabs.io/oss/cli2/badges/master/pipeline.svg
 Home-page: https://yourlabs.io/oss/cli2
 Author: James Pic

{cli2-3.3.6 → cli2-3.3.7}/cli2/client.py

@@ -597,22 +597,33 @@ class Related(MutableField):
     .. py:attribute:: model
 
         *STRING* name of the related model class.
+
+    .. py:attribute:: many
+
+        Set this to True if you're expecting a list of models in the field.
     """
-    def __init__(self, model, *args, **kwargs):
+    def __init__(self, model, many=False, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.model = model
+        self.many = many
 
     def internalize(self, obj, data):
         """
         Return the related object's data.
         """
-        return data.data
+        if not self.many:
+            return data.data
+
+        return [item.data for item in data]
 
     def externalize(self, obj, value):
         """
         Instanciate the related model class with the value.
         """
-        return getattr(obj.client, self.model)(value)
+        model_class = getattr(obj.client, self.model)
+        if not self.many:
+            return model_class(value)
+        return [model_class(item) for item in value]
 
 
 class ModelCommand(Command):

{cli2-3.3.6 → cli2-3.3.7}/cli2/test_client.py

@@ -393,7 +393,7 @@ def test_model_inheritance():
     assert [*client.Model2._fields.keys()] == ['bar', 'foo']
 
 
-def test_relation():
+def test_relation_simple():
     class Child(Client.Model):
         foo = cli2.Field()
 
@@ -415,6 +415,29 @@ def test_relation():
     assert model.data['child']['foo'] == 3
 
 
+def test_relation_many():
+    class Child(Client.Model):
+        foo = cli2.Field()
+
+    class Father(Client.Model):
+        children = cli2.Related('Child', many=True)
+
+    client = Client()
+    model = client.Father(dict(children=[dict(foo=1)]))
+    assert model.children[0].foo == 1
+    assert model.data['children'][0]['foo'] == 1
+
+    model.children.append(client.Child(foo=2))
+    assert model.data['children'][1]['foo'] == 2
+    assert model.children[1].foo == 2
+
+    new = [client.Child(dict(foo=3))]
+    model.children = new
+    assert len(model.children) == 1
+    assert model.children[0].foo == 3
+    assert model.data['children'][0]['foo'] == 3
+
+
 @pytest.mark.asyncio
 async def test_python_expression(httpx_mock):
     class Model(Client.Model):

{cli2-3.3.6 → cli2-3.3.7/cli2.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cli2
-Version: 3.3.6
+Version: 3.3.7
 Summary: image:: https://yourlabs.io/oss/cli2/badges/master/pipeline.svg
 Home-page: https://yourlabs.io/oss/cli2
 Author: James Pic

{cli2-3.3.6 → cli2-3.3.7}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup
 
 setup(
     name='cli2',
-    version='3.3.6',
+    version='3.3.7',
     setup_requires='setupmeta',
     install_requires=[
         'docstring_parser',