muck/models/exit.py

@@ -1,6 +1,6 @@
-from muck.models.dbref import Dbref
+from muck.models.dbref import Dbref, ObjectType
 
 class Exit(Dbref):
     __mapper_args__ = {
-        "polymorphic_identity": "exit"
+        "polymorphic_identity": ObjectType.EXIT
     }

muck/models/player.py

@@ -1,6 +1,6 @@
-from muck.models.dbref import Dbref
+from muck.models.dbref import Dbref, ObjectType
 
 class Player(Dbref):
     __mapper_args__ = {
-        "polymorphic_identity": "player"
+        "polymorphic_identity": ObjectType.PLAYER
     }

muck/models/program.py

@@ -1,6 +1,6 @@
-from muck.models.dbref import Dbref
+from muck.models.dbref import Dbref, ObjectType
 
 class Program(Dbref):
     __mapper_args__ = {
-        "polymorphic_identity": "program"
+        "polymorphic_identity": ObjectType.PROGRAM
     }

muck/models/room.py

@@ -1,6 +1,6 @@
-from muck.models.dbref import Dbref
+from muck.models.dbref import Dbref, ObjectType
 
 class Room(Dbref):
     __mapper_args__ = {
-        "polymorphic_identity": "room"
+        "polymorphic_identity": ObjectType.ROOM
     }

muck/models/thing.py

@@ -1,6 +1,6 @@
-from muck.models.dbref import Dbref
+from muck.models.dbref import Dbref, ObjectType
 
 class Thing(Dbref):
     __mapper_args__ = {
-        "polymorphic_identity": "thing"
+        "polymorphic_identity": ObjectType.THING
     }