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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
"""
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('/api/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('/api/classes-for')
def classes_for():
fqdn = request.args.get('fqdn')
classes = [cls.class_name for cls in
model.Host.query.where(model.Host.fqdn==fqdn).first().classes]
return Response(json.dumps(classes),
mimetype='application/json')
@app.route('/api/change-classes', methods=['POST'])
def change_classes():
j = request.json
host = model.Host.query.where(model.Host.fqdn==j['fqdn']).first()
remove_set = set(j['removed'])
new_cls = []
for cls in host.classes:
if cls.class_name in remove_set:
continue
new_cls.append(cls)
host.classes = new_cls
cls = model.PuppetClass.query \
.where(model.PuppetClass.class_name.in_(j['added'])) \
.all()
host.classes.extend(cls)
print(remove_set, db.db.session.dirty)
return flask.redirect(url_for('classes_for', fqdn=j['fqdn']))
@app.route('/api/hosts')
def list_hosts():
return Response(flask.json.dumps([x.serialize() for x in model.Host.query.all()]),
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
|
