aboutsummaryrefslogtreecommitdiff
path: root/pyenc/__init__.py
blob: 6323fbe194db9cc3d7da7ffa5ebe07ac59342a86 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
"""
app object setup for application
"""

import flask
from flask import (
        Flask,
        request,
        Response,
        flash,
        redirect,
        url_for
        )

import random
import json

from . import model

import yaml

def create_app():
    app = Flask(__name__, instance_relative_config=True)

    app.config.from_pyfile('settings.py')

    from . import db
    db.init_app(app)

    from . import enc
    enc.init_app(app)

    # not API
    @app.route('/')
    def root_page():
        return flask.render_template('start_page.html',
                hosts=model.Host.query.order_by(model.Host.fqdn),
                random=random,
                str=str,
                )

    # API?
    @app.route('/remove', methods=['POST'])
    def remove_classes():
        print(request.form['fqdn'])
        print(request.form.getlist('cls'))
        flash('Classes removed')
        return redirect(url_for('root_page'))

    # API
    @app.route('/list-classes')
    def list_classes():
        q = request.args.get('q', '')
        qq = '%{}%'.format('%'.join(q.split(' ')))

        results = model.PuppetClass.query.where(model.PuppetClass.class_name.like(qq)).all()
        print(qq)
        return Response(json.dumps([x.class_name for x in results]),
                mimetype='application/json')

    @app.route('/enc')
    def enc():
        fqdn = request.args.get('fqdn', 'default')
        host = model.Host.query.where(model.Host.fqdn==fqdn).first()
        if not host:
            return Response(f"No host with name {fqdn}",
                    status=404,
                    )

        out = {
            'environment': host.environment,
            'classes': [cls.class_name for cls in host.classes],
        }
        return Response(yaml.dump(out),
                mimetype='application/x-yaml')




    return app