keep previous 'id' if present

Ghini · Apr 10, 2018 · cc788d2 · cc788d2
1 parent ef2e511
commit cc788d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/stream_d2w.py b/scripts/stream_d2w.py
@@ -202,15 +202,15 @@ def get_location(session, keys):
              'lon': v.coords['lon']}
         result['plants'].append(d)
 
-print 'db.gardens.update({uuid: %s}, %s, {upsert: true});' % (json.dumps(garden['uuid']), json.dumps(result['garden']))
+print 'db.gardens.update({uuid: "%s"}, {$set: %s}, {upsert: true});' % (garden['uuid'], json.dumps(garden))
 for i in result['species']:
     print 'db.taxa.update({name: %s}, %s, {upsert: true});' % (json.dumps(i['name']), json.dumps(i))
 print 'db.plants.deleteMany({garden: %s});' % json.dumps(garden['name'])
 print 'db.plants.insertMany(%s);' % json.dumps(result['plants'])
 
 print '''\
 db.gardens.find().sort({id:-1}).limit(1).forEach(function(g){
-    db.gardens.updateOne({uuid: "%(uuid)s"}, {$set: {id: g.id + 1}});
+    db.gardens.updateOne({uuid: "%(uuid)s", id: {$exists: false}}, {$set: {id: g.id + 1}});
 });
 db.gardens.find({uuid: "%(uuid)s"}).forEach(function (elem) {
     db.plants.updateMany({garden_uuid: elem.uuid}, {$set: {garden_id: elem.id}})