add validation of doc_class on insert

2012-02-21 17:00:21 -05:00 · 2012-02-21 17:00:21 -05:00 · 2f3825d3e2
commit 2f3825d3e2
parent 211e40e211
1 changed files with 3 additions and 0 deletions
--- a/oyster/core.py
+++ b/oyster/core.py
@ -94,6 +94,9 @@ class Kernel(object):
        **kwargs
            any keyword args will be added to the document's metadata
        """
+        if doc_class not in self.doc_classes:
+            raise ValueError('unregistered doc_class %s' % doc_class)
+
        tracked = self.db.tracked.find_one({'url': url})

        # if data is already tracked and this is just a duplicate call