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

Refactor get_primary_keys

Browse Source
This commit is contained in:
Konsta Vesterinen
2014-05-06 10:24:36 +03:00
parent 7f1a7b3907
commit e74541ad9d
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sqlalchemy_utils/functions/orm.py
View File

@@ -76,11 +76,12 @@ def get_primary_keys(mixed):
.. seealso:: :func:`get_columns` .. seealso:: :func:`get_columns`
""" """
columns = OrderedDict() return OrderedDict(
for key, column in get_columns(mixed).items(): (
if column.primary_key: (key, column) for key, column in get_columns(mixed).items()
columns[key] = column if column.primary_key
return columns )
)
def get_columns(mixed): def get_columns(mixed):