inject attributes into methods

Author: vvcheremushkin

setup.py

@@ -29,7 +29,7 @@ def get_packages(package):
         "starlette<=1",
         "apispec<3",
     ],
-    version="0.0.6",
+    version="0.0.7",
     url="https://github.com/slv0/start_resty",
     license="BSD",
     description="The web framework",

star_resty/inject.py

@@ -0,0 +1,32 @@
+import operator
+from typing import TypeVar, Type, Optional, Generic
+
+from star_resty import Method
+
+__all__ = ('attr',)
+
+T = TypeVar('T')
+
+
+class InjectAttr(Generic[T]):
+    __slots__ = ('_func',)
+
+    def __init__(self, name=None):
+        if name:
+            self._func = operator.attrgetter(name)
+        else:
+            self._func = None
+
+    def __set_name__(self, owner, name):
+        if self._func is None:
+            self._func = operator.attrgetter(name)
+
+    def __get__(self, instance: Optional[Method], owner: Type[Method]) -> T:
+        if instance is None:
+            return self
+
+        return self._func(instance.request.app)
+
+
+def attr(_: Optional[Type[T]] = None, *, name: Optional[str] = None) -> T:
+    return InjectAttr[T](name)

star_resty/method/meta.py

@@ -45,6 +45,8 @@ def dump(content):
                 return response_schema.dump(content)
             except MarshmallowError as e:
                 raise DumpError(e) from e
+            except (ValueError, TypeError) as e:
+                raise DumpError(e) from e
 
         return dump
 

tests/test_inject.py

@@ -0,0 +1,58 @@
+import pytest
+from starlette.applications import Starlette
+from starlette.testclient import TestClient
+
+from star_resty import Method, inject
+from tests.utils.method import SearchUserResponse
+
+
+class Repository:
+
+    @staticmethod
+    async def get_user():
+        return {'id': 2}
+
+
+class GetUser(Method):
+    response_schema = SearchUserResponse
+
+    users_repository = inject.attr(Repository)
+
+    async def execute(self):
+        return await self.users_repository.get_user()
+
+
+class GetUserInjByName(Method):
+    response_schema = SearchUserResponse
+
+    users_repository = inject.attr(Repository, name='user_api')
+
+    async def execute(self):
+        return await self.users_repository.get_user()
+
+
+@pytest.fixture()
+def app():
+    return Starlette()
+
+
+def test_inject_attr(app):
+    app.users_repository = Repository()
+    app.add_route('/user', GetUser.as_endpoint(), methods=['GET'])
+
+    client = TestClient(app)
+    resp = client.get('/user')
+    assert resp.status_code == 200
+    body = resp.json()
+    assert body == {'id': 2}
+
+
+def test_inject_attr_by_name(app):
+    app.user_api = Repository()
+    app.add_route('/user', GetUserInjByName.as_endpoint(), methods=['GET'])
+
+    client = TestClient(app)
+    resp = client.get('/user')
+    assert resp.status_code == 200
+    body = resp.json()
+    assert body == {'id': 2}