mongo get_or_create doesn't like lists in queries. Workaround.

glamkit · Nov 3, 2010 · 8e79e35 · 8e79e35
1 parent bf63cd5
commit 8e79e35
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/xmltools/lib/xml2dict.py b/xmltools/lib/xml2dict.py
@@ -4,10 +4,10 @@ def xml2dict(tag):
     r = {}
 
     #value
+    # tag.text is None, for empty tags.
     if tag.text is not None:
         v = tag.text.strip()
-        if v:
-            r['_value'] = v 
+        r['_value'] = v
 
     #attributes
     if tag.keys():

diff --git a/xmltools/processors/mongo.py b/xmltools/processors/mongo.py
@@ -30,4 +30,12 @@ def __call__(self, tag):
         d = xml2dict(tag)
         d = self.clean(d)
 
-        m, created = self.model.objects.get_or_create(**d)
+        if d is not None:
+            try:
+                m = self.model.objects.get(id=d['id'])
+                for k, v in d.iteritems():
+                    setattr(m, k, v)
+                m.save()
+            except self.model.DoesNotExist:
+                m = self.model.objects.create(**d)
+                m.save()