Refactor get_primary_keys

2014-05-06 10:24:36 +03:00
parent 7f1a7b3907
commit e74541ad9d
1 changed files with 6 additions and 5 deletions
--- a/sqlalchemy_utils/functions/orm.py
+++ b/sqlalchemy_utils/functions/orm.py
@@ -76,11 +76,12 @@ def get_primary_keys(mixed):

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


 def get_columns(mixed):