From c5918545e3a3c4397fba770a5c9b5b4a91488016 Mon Sep 17 00:00:00 2001
From: Konsta Vesterinen <konsta.vesterinen@gmail.com>
Date: Mon, 21 Apr 2014 11:44:00 +0300
Subject: [PATCH] Add more examples for primary_keys

---
 sqlalchemy_utils/functions/orm.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/sqlalchemy_utils/functions/orm.py b/sqlalchemy_utils/functions/orm.py
index 5bc610a..6f6451e 100644
--- 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.