cli2 4.0.6__tar.gz → 4.0.8__tar.gz

{cli2-4.0.6/cli2.egg-info → cli2-4.0.8}/PKG-INFO

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

{cli2-4.0.6 → cli2-4.0.8}/cli2/client.py

@@ -690,6 +690,7 @@ class ModelMetaclass(type):
     def __new__(cls, name, bases, attributes):
         if 'Paginator' in attributes:
             attributes['paginator'] = attributes['Paginator']
+
         cls = super().__new__(cls, name, bases, attributes)
         client = getattr(cls, 'client', None)
         if client:
@@ -700,11 +701,6 @@ class ModelMetaclass(type):
         client_class = getattr(cls, '_client_class', None)
         if client_class:
             client_class.models.append(cls)
-            client_cli = getattr(client_class, 'cli', None)
-            if client_cli:
-                group = ModelGroup(cls)
-                if len(group) > 1:
-                    cls.cli = client_cli[cls.__name__.lower()] = group
 
         cls._fields = dict()
 
@@ -939,6 +935,8 @@ class Model(metaclass=ModelMetaclass):
 
 
 class ClientMetaclass(type):
+    cli_kwargs = dict()
+
     def __new__(cls, name, bases, attributes):
         if 'Paginator' in attributes:
             attributes['paginator'] = attributes['Paginator']
@@ -947,20 +945,29 @@ class ClientMetaclass(type):
         # bind ourself as _client_class to any inherited model
         cls.Model = type('Model', (Model,), dict(_client_class=cls))
         cls.models = []
-
-        cli_kwargs = dict(
-            name=cls.__name__.lower().replace('client', '') or 'client',
-            doc=inspect.getdoc(cls),
-            overrides=dict(
-                cls=dict(factory=lambda: cls),
-                self=dict(factory=lambda: cls())
-            ),
-        )
-        cli_kwargs.update(cls.__dict__.get('cli_kwargs', dict()))
-        cls.cli = Group(**cli_kwargs)
-        cls.cli.load(cls)
         return cls
 
+    @property
+    def cli(cls):
+        if '_cli' not in cls.__dict__:
+            cli_kwargs = dict(
+                name=cls.__name__.lower().replace('client', '') or 'client',
+                doc=inspect.getdoc(cls),
+                overrides=dict(
+                    cls=dict(factory=lambda: cls),
+                    self=dict(factory=lambda: cls())
+                ),
+            )
+            cli_kwargs.update(cls.cli_kwargs)
+            cli = Group(**cli_kwargs)
+            cli.load(cls)
+            for model in cls.models:
+                group = ModelGroup(model)
+                if len(group) > 1:
+                    cli[model.__name__.lower()] = group
+            cls._cli = cli
+        return cls._cli
+
 
 class Handler:
     """
@@ -1204,6 +1211,22 @@ class Client(metaclass=ClientMetaclass):
 
         List of keys to mask in logging, ie.: ``['password', 'secret']``
 
+    .. py:attribute:: cli
+
+        Generated :py:class:`~cli2.cli.Group` for this client.
+        Uses :py:attr:`cli_kwargs` to pass kwargs to the generated group.
+        Note that this is a cached property.
+
+    .. py:attribute:: cli_kwargs
+
+        Dict of overrides for the generated :py:class:`~cli2.cli.Group`.
+        Example:
+
+        .. code-block:: python
+
+            class YourClient(cli2.Client):
+                cli_kwargs = dict(cmdclass=YourCommandClass)
+
     .. py:attribute:: debug
 
         Enforce full logging: quiet requests are logged, masking does not

{cli2-4.0.6 → cli2-4.0.8/cli2.egg-info}/PKG-INFO

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

{cli2-4.0.6 → cli2-4.0.8}/setup.py

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

{cli2-4.0.6 → cli2-4.0.8}/tests/test_client.py

@@ -37,6 +37,17 @@ def client_class():
 
 @pytest.mark.asyncio
 async def test_client_cli(client_class, httpx_mock):
+    class TestModel(client_class.Model):
+        url_list = '/'
+
+    class TestModel2(client_class.Model):
+        pass
+
+    class TestModel3(client_class.Model):
+        @cli2.cmd
+        def find():
+            pass
+
     assert client_class.cli
     factory = client_class.cli.overrides['self']['factory']
     assert isinstance(factory(), client_class)
@@ -45,8 +56,6 @@ async def test_client_cli(client_class, httpx_mock):
     response = await client_class.cli['get'].async_call('http://lol')
     assert response.json() == [1]
 
-    class TestModel(client_class.Model):
-        url_list = '/'
     assert 'testmodel' in client_class.cli
     assert client_class.cli['testmodel']['get'].overrides['cls']['factory']
     assert not inspect.ismethod(client_class.cli['testmodel']['get'].target)
@@ -58,16 +67,8 @@ async def test_client_cli(client_class, httpx_mock):
     httpx_mock.add_response(url='http://lol/?page=2', json=[])
     await client_class.cli['testmodel']['find'].async_call()
 
-    class TestModel2(client_class.Model):
-        pass
-
     assert 'testmodel2' not in client_class.cli
 
-    class TestModel3(client_class.Model):
-        @cli2.cmd
-        def find():
-            pass
-
     assert list(client_class.cli['testmodel3'].keys()) == ['help', 'find']
 
 
@@ -85,7 +86,6 @@ async def test_client_cli_override(client_class, httpx_mock):
         @cli2.cmd
         async def find(cls, foo):
             return cls.url_list
-    assert await Client.cli['testmodel']['find'].async_call('bar') == 'bar/foo'
 
     class TestModel2(Client.Model):
         @classmethod
@@ -93,12 +93,15 @@ async def test_client_cli_override(client_class, httpx_mock):
         async def find(cls):
             return 'foo'
 
-    assert await Client.cli['testmodel2']['find'].async_call() == 'foo'
-    assert 'get' not in Client.cli['testmodel2']
-
     class TestModel3(Client.Model):
         url_list = '{client.test}/foo'
         create = None
+
+    assert await Client.cli['testmodel']['find'].async_call('bar') == 'bar/foo'
+
+    assert await Client.cli['testmodel2']['find'].async_call() == 'foo'
+    assert 'get' not in Client.cli['testmodel2']
+
     assert 'create' not in Client.cli['testmodel3']
 
 
@@ -123,13 +126,13 @@ async def test_async_factory(httpx_mock, client_class):
         async def factory(cls):
             return cls(base_url='http://bar')
 
+    class TestModel(TestClient.Model):
+        url_list = '/2'
+
     TestClient.cli.overrides['self']['factory'] = TestClient.factory
     httpx_mock.add_response(url='http://bar/', json=[dict(a=1)])
     assert await TestClient.cli['get'].async_call('/')
 
-    class TestModel(TestClient.Model):
-        url_list = '/2'
-
     httpx_mock.add_response(url='http://bar/2', json=[dict(a=1)])
     httpx_mock.add_response(url='http://bar/2?page=2', json=[])
     await TestClient.cli['testmodel']['find'].async_call()
@@ -777,6 +780,10 @@ async def test_model_crud(httpx_mock, client_class):
 
 @pytest.mark.asyncio
 async def test_client_cli2(httpx_mock, client_class):
+    class Foo(client_class.Model):
+        id = cli2.Field()
+        url_list = '/foo'
+
     assert client_class.cli.name == 'test'
     assert client_class.cli.doc == 'doc'
 
@@ -785,10 +792,6 @@ async def test_client_cli2(httpx_mock, client_class):
     resp = await meth.async_call('/foo')
     assert resp.json() == [1]
 
-    class Foo(client_class.Model):
-        id = cli2.Field()
-        url_list = '/foo'
-
     meth = client_class.cli['foo']['get']
     httpx_mock.add_response(url='http://lol/foo/1', json=dict(id=1, a=2))
     result = await meth.async_call('id=1')