#!/usr/bin/env python3
import json

from sqlalchemy.orm.exc import NoResultFound, MultipleResultsFound

import pyenc
from pyenc.db import db
import pyenc.model as model

app = pyenc.create_app()
app.app_context().push()


"""
Fetch all found classes from redis, handle the data, and commit it to
our true database.
"""

# TODO this inserts already existing classes
# It shouldn't
for puppet_file in model.PuppetFile.query.all():
    data = json.loads(puppet_file.json)
    top = data['^']
    if top[0] == 'class':
        tmp = top[1]['#']
        idx = tmp.index('name')
        db.session.add(model.PuppetClass(
            class_name=tmp[idx + 1],
            comes_from=puppet_file))
        # print(tmp[idx + 1])
    elif top[0] == 'block':
        for element in top[1:]:
            if element['^'][0] == 'class':
                tmp = element['^'][1]['#']
                idx = tmp.index('name')
                db.session.add(model.PuppetClass(
                    class_name=tmp[idx + 1],
                    comes_from=puppet_file))
db.session.commit()