Small fix

9b19fff4 · Serge S. Koval · 7b7064fe · 9b19fff4
Commit 9b19fff4 authored Mar 05, 2014 by Serge S. Koval
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 4 deletions

tools.py flask_admin/contrib/sqla/tools.py +5 -4

No files found.
--- a/flask_admin/contrib/sqla/tools.py
+++ b/flask_admin/contrib/sqla/tools.py
@@ -33,10 +33,11 @@ def get_primary_key(model):
                else:
                    pks.append(p.key)
        else:
-            for c in p.columns:
-                if c.primary_key:
-                    pks.append(p.key)
-                    break
+            if hasattr(p, 'columns'):
+                for c in p.columns:
+                    if c.primary_key:
+                        pks.append(p.key)
+                        break

    if len(pks) == 1:
        return pks[0]