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
113
114
115
116
117
118
119
120
121
|
"""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():
"""
Create new flask app.
Should bind everything needed to it.
"""
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():
data = [x.serialize() for x in model.Host.query.all()]
return Response(flask.json.dumps(data),
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
|