deal with OneToOneField

yifanzhang · Mar 12, 2012 · 4bb06cd · 4bb06cd
1 parent 7d87faf
commit 4bb06cd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/fixture_magic/management/commands/dump_object.py b/fixture_magic/management/commands/dump_object.py
@@ -1,3 +1,4 @@
+import sys
 from optparse import make_option
 
 from django.core.exceptions import FieldError, ObjectDoesNotExist
@@ -59,6 +60,8 @@ def handle(self, *args, **options):
                 for rel in related_fields:
                     try:
                         add_to_serialize_list(obj.__getattribute__(rel).all())
+                    except AttributeError:
+                        add_to_serialize_list(obj.__getattribute__(rel))
                     except FieldError:
                         pass
                     except ObjectDoesNotExist:

diff --git a/fixture_magic/utils.py b/fixture_magic/utils.py
@@ -1,4 +1,5 @@
 from django.db import models
+import collections
 
 serialize_me = []
 seen = {}
@@ -47,6 +48,9 @@ def serialize_fully():
 
 
 def add_to_serialize_list(objs):
+    if not isinstance(objs, collections.Iterable):
+        objs = [ objs ]
+
     for obj in objs:
         if obj is None:
             continue