Add more examples for primary_keys

2014-04-21 11:44:00 +03:00
parent 56f1fa1bb2
commit c5918545e3
1 changed files with 6 additions and 0 deletions
--- a/sqlalchemy_utils/functions/orm.py
+++ b/sqlalchemy_utils/functions/orm.py
@@ -30,6 +30,12 @@ def primary_keys(mixed):

        primary_keys(User.__table__)

+        primary_keys(User.__mapper__)
+
+        primary_keys(sa.orm.aliased(User))
+
+        primary_keys(sa.orm.alised(User.__table__))
+

    .. versionchanged: 0.25.3
        Made the function return an ordered dictionary instead of generator.