Extend table_name to support attributes.
This commit is contained in:
@@ -90,14 +90,22 @@ def primary_keys(class_):
|
|||||||
yield column
|
yield column
|
||||||
|
|
||||||
|
|
||||||
def table_name(class_):
|
def table_name(obj):
|
||||||
"""
|
"""
|
||||||
Return table name of given declarative class.
|
Return table name of given target, declarative class or the
|
||||||
|
table name where the declarative attribute is bound to.
|
||||||
"""
|
"""
|
||||||
|
class_ = getattr(obj, 'class_', obj)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
return class_.__tablename__
|
return class_.__tablename__
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
try:
|
||||||
return class_.__table__.name
|
return class_.__table__.name
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
def non_indexed_foreign_keys(metadata, engine=None):
|
def non_indexed_foreign_keys(metadata, engine=None):
|
||||||
|
@@ -12,7 +12,14 @@ class TestTableName(TestCase):
|
|||||||
|
|
||||||
self.Building = Building
|
self.Building = Building
|
||||||
|
|
||||||
def test_table_name(self):
|
def test_class(self):
|
||||||
assert table_name(self.Building) == 'building'
|
assert table_name(self.Building) == 'building'
|
||||||
del self.Building.__tablename__
|
del self.Building.__tablename__
|
||||||
assert table_name(self.Building) == 'building'
|
assert table_name(self.Building) == 'building'
|
||||||
|
|
||||||
|
def test_attribute(self):
|
||||||
|
assert table_name(self.Building.id) == 'building'
|
||||||
|
assert table_name(self.Building.name) == 'building'
|
||||||
|
|
||||||
|
def test_target(self):
|
||||||
|
assert table_name(self.Building()) == 'building'
|
||||||
|
Reference in New Issue
Block a user