Changeset 210 for trunk/dbpickle/dbpickle.py

Timestamp:

03/22/08 21:44:25 (2 years ago)

Author:

akaihola

Message:

[dbpickle] Dumping now skips missing db tables with a warning.

Files:

• trunk/dbpickle/dbpickle.py (modified) (2 diffs)

trunk/dbpickle/dbpickle.py

@@ -93 +93 @@
         m2ms = [m2m.name for m2m in meta.many_to_many]
 
-        for obj in model.objects.all():
-            logging.debug('dumping %s.%s %r' % (app, model_name, obj))
-            pk = obj._get_pk_val()
-            objects[app, model_name, pk] = obj
-
-            for m2m in m2ms:
-                # store many-to-many related objects for every
-                # many-to-many relation of this object
-                foreign_objs = getattr(obj, m2m).all()
-                logging.debug('dumping %s.%s.%s x %d' % (app, model_name, m2m, len(foreign_objs)))
-                m2m_lists.append((obj, m2m, tuple(foreign_objs)))
+        try:
+            for obj in model.objects.all():
+                logging.debug('dumping %s.%s %r' % (app, model_name, obj))
+                pk = obj._get_pk_val()
+                objects[app, model_name, pk] = obj
+
+                for m2m in m2ms:
+                    # store many-to-many related objects for every
+                    # many-to-many relation of this object
+                    foreign_objs = getattr(obj, m2m).all()
+                    logging.debug('dumping %s.%s.%s x %d' % (app, model_name, m2m, len(foreign_objs)))
+                    m2m_lists.append((obj, m2m, tuple(foreign_objs)))
+        except backend.Database.OperationalError:
+            logging.warning('table for %r not found' % model)
 
     # pickle all objects and many-to-many relations on disk
@@ -225 +228 @@
         os.environ['DJANGO_SETTINGS_MODULE'] = opts.settings
 
-    from django.db import transaction
-    from django.db import models
-
+    from django.db import transaction, models, backend
 
     if opts.action == 'dump':