Bläddra i källkod

Fix load & dump commands

theenglishway (time) 6 år sedan
förälder
incheckning
b9b5298dc6
1 ändrade filer med 7 tillägg och 0 borttagningar
  1. 7 0
      pyplanner/database.py

+ 7 - 0
pyplanner/database.py

@@ -47,6 +47,8 @@ class Database:
         instances_dict = {}
 
         for k, v in self.models.items():
+            if v.__subclasses__():
+                continue
             instances = [i for i in session.query(v).all()]
             instances_dict[k] = instances
             yaml.register_class(v)
@@ -62,3 +64,8 @@ class Database:
         for k, v_list in instances_dict.items():
             model = self.models[k]
             instances = [model(**v) for v in v_list]
+
+            for i in instances:
+                session.merge(i)
+
+        session.commit()