Make mongoengine field for UsageKeys expect a Location

2014-05-21 14:43:51 -04:00
parent 9dc48c00b4
commit 4c2e612aeb
1 changed files with 2 additions and 2 deletions
--- a/common/lib/xmodule/xmodule/modulestore/mongoengine_fields.py
+++ b/common/lib/xmodule/xmodule/modulestore/mongoengine_fields.py
@@ -59,9 +59,9 @@ class UsageKeyField(mongoengine.StringField):
        """
        For now saves the usage key in the deprecated location i4x/c4x form
        """
-        assert isinstance(location, (NoneType, SlashSeparatedCourseKey))
+        assert isinstance(location, (NoneType, Location))
        if location is None:
-            return location
+            return None
        return super(UsageKeyField, self).to_mongo(location.to_deprecated_string())

    def to_python(self, location):