--- a/ForgeTracker/forgetracker/model/migrations.py
+++ b/ForgeTracker/forgetracker/model/migrations.py
@@ -1,7 +1,12 @@
+import cPickle as pickle
+from itertools import chain
+
+import pymongo
+from ming.orm import state
from pylons import c
from flyway import Migration
-from pyforge.model import Thread
+from pyforge.model import Thread, AppConfig, ArtifactReference
from forgetracker.model import Ticket
class V0(Migration):
@@ -24,16 +29,28 @@
def up(self):
for pg in self.ormsession.find(Ticket):
- for t in self.ormsession.find(Thread, dict(artifact_id=pg._id)):
- t.artifact_reference = pg.dump_ref()
+ q1 = self.ormsession.find(Thread, dict(artifact_id=pg._id))
+ q2 = self.ormsession.find(Thread, {'artifact_reference.artifact_id':pg._id})
+ for t in chain(q1, q2):
+ t.artifact_reference = self._dump_ref(pg)
t.artifact_id = None
+ self.ormsession.update_now(t, state(t))
self.ormsession.flush()
def down(self):
for pg in self.ormsession.find(Ticket):
- for t in self.ormsession.find(Thread, dict(artifact_reference=pg.dump_ref())):
+ for t in self.ormsession.find(Thread, dict(artifact_reference=self._dump_ref(pg))):
t.artifact_id = pg._id
t.artifact_reference = None
+ self.ormsession.update_now(t, state(t))
self.ormsession.flush()
+ def _dump_ref(self, art):
+ app_config = self.ormsession.get(AppConfig, art.app_config_id)
+ return ArtifactReference(dict(
+ project_id=app_config.project_id,
+ mount_point=app_config.options.mount_point,
+ artifact_type=pymongo.bson.Binary(pickle.dumps(art.__class__)),
+ artifact_id=art._id))
+