diff options
Diffstat (limited to '')
-rwxr-xr-x | import_yaml.py | 12 |
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() |