work

1 files changed, 6 insertions, 6 deletions

diff --git a/import_yaml.py b/import_yaml.py
index 35558c8..d602cad 100755
--- a/import_yaml.py
+++ b/import_yaml.py
@@ -2,12 +2,10 @@
 
 """Import extisting nodes.yaml into database"""
 
-import json
 import yaml
 
 import pyenc
-from pyenc.db import db
-import pyenc.model as model
+from pyenc.app import model
 
 app = pyenc.create_app()
 app.app_context().push()
@@ -21,8 +19,10 @@ for fqdn, val in data.items():
     h = model.Host.query.where(model.Host.fqdn == fqdn).first()
     if not h:
         h = model.Host(fqdn=fqdn)
-    h.environment = val.get('environment')
-    print(h)
+    h.environment = model.PuppetEnvironment \
+        .query \
+        .where(model.PuppetEnvironment.name == val.get('environment', 'production')) \
+        .one()
 
     classes = val['classes']
     if type(classes) == dict:
@@ -34,4 +34,4 @@ for fqdn, val in data.items():
     h.classes.extend(cls)
 
 
-db.session.commit()
+model.db.session.commit()