openstack/deb-python-sqlalchemy-utils
Code Issues Proposed changes

Fixed generates decorator

Browse Source
This commit is contained in:
Konsta Vesterinen
2013-10-23 15:33:15 +03:00
parent dd8b5b2f13
commit 002b5b69cd
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
sqlalchemy_utils/decorators.py
View File

@@ -18,6 +18,9 @@ def generates(attr):
parts = attr.split('.') parts = attr.split('.')
generator_registry[parts[0]].append(wrapper) generator_registry[parts[0]].append(wrapper)
wrapper.__generates__ = parts[1] wrapper.__generates__ = parts[1]
elif isinstance(attr, sa.orm.attributes.InstrumentedAttribute):
generator_registry[attr.class_.__name__].append(wrapper)
wrapper.__generates__ = attr
else: else:
wrapper.__generates__ = attr wrapper.__generates__ = attr
return wrapper return wrapper

4
tests/test_generates.py
View File

@@ -5,8 +5,8 @@ from tests import TestCase
class GeneratesTestCase(TestCase): class GeneratesTestCase(TestCase):
def setup_method(self, method): def teardown_method(self, method):
TestCase.setup_method(self, method) TestCase.teardown_method(self, method)
decorators.generator_registry = defaultdict(list) decorators.generator_registry = defaultdict(list)
def test_generates_value_before_flush(self): def test_generates_value_before_flush(self):