fix get_pk function

uralbash · Aug 4, 2014 · af88ebc · af88ebc
1 parent 84f1863
commit af88ebc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -21,3 +21,4 @@ nohup.out
 .tox
 __pycache__
 _build
+.ropeproject
diff --git a/sqlalchemy_mptt/mixins.py b/sqlalchemy_mptt/mixins.py
@@ -65,7 +65,7 @@ def get_pk(cls):
     @classmethod
     def get_db_pk(cls):
         pk = getattr(cls, cls.get_pk())
-        return pk.name
+        return pk.name or cls.get_pk()
 
     @classmethod
     def get_class_pk(cls):