Merge pull request #185 from youknowone/fix-key-class

Fix key override for class

ring/__version__.py

@@ -1 +1 @@
-__version__ = '0.9.0'
+__version__ = '0.9.1'

ring/func/base.py

@@ -708,7 +708,7 @@ def __init__(self, rope):
         self._rope = rope
 
     def key(self, func):
-        self._rope.compose_key = lambda pargs: func(*pargs.args, **pargs.kwargs)
+        self._rope.compose_key = lambda pargs: func(*(pargs.bounds + pargs.args), **pargs.kwargs)
 
     def encode(self, func):
         self._rope._encode = func

setup.cfg

@@ -7,4 +7,4 @@ norecursedirs=.git py ci
 [flake8]
 ignore=E501,E701
 [metadata]
-version_from_file: ring/__init__.py
+version_from_file: ring/__version__.py

tests/test_customize.py

@@ -2,7 +2,7 @@
 import ring
 
 
-def test_override_key():
+def test_override_key_function():
 
     @ring.dict({})
     def f(v):
@@ -29,3 +29,18 @@ def f_decode(v):
 
     assert f.storage.backend['test:5'] == ('encoded', 5)
     assert f.get(5) == 6
+
+
+def test_override_key_class():
+    # https://github.com/youknowone/ring/issues/184
+    class A:
+        @ring.dict({})
+        def get_entity(self, id):
+            return {"test": "ok"}
+
+        @get_entity.ring.key
+        def get_entity_key(self, id):
+            return "test:{id}".format(id=id)
+
+    a = A()
+    a.get_entity("123")