|
6 | 6 | from flask import Flask, render_template, request, jsonify
|
7 | 7 |
|
8 | 8 | app = Flask(__name__)
|
9 |
| -sigma_versions = [os.path.basename(it.path) for it in os.scandir('../backend/') if it.is_dir()] |
| 9 | +sigma_versions = [ |
| 10 | + os.path.basename(it.path) for it in os.scandir("../backend/") if it.is_dir() |
| 11 | +] |
| 12 | + |
| 13 | + |
| 14 | +def version_key(version): |
| 15 | + return tuple(map(int, version.split("."))) |
| 16 | + |
10 | 17 |
|
11 | 18 | def get_port_from_version(version):
|
12 |
| - pattern = r'^\d+\.\d+\.\d+$' |
| 19 | + pattern = r"^\d+\.\d+\.\d+$" |
13 | 20 | if re.match(pattern, version):
|
14 |
| - return int(f'8{version.replace(".","")}') |
| 21 | + return int(f'8{version.replace(".", "")}') |
15 | 22 | else:
|
16 | 23 | return None
|
17 | 24 |
|
| 25 | + |
18 | 26 | @app.route("/")
|
19 | 27 | def home():
|
20 |
| - versions = sigma_versions |
21 |
| - latest_version = sigma_versions[-1:][0] |
22 |
| - port = get_port_from_version(latest_version) |
23 |
| - targets = requests.get(f'http://localhost:{port}/api/v1/targets').json() |
24 |
| - formats = requests.get(f'http://localhost:{port}/api/v1/formats').json() |
25 |
| - pipelines = requests.get(f'http://localhost:{port}/api/v1/pipelines').json() |
| 28 | + return render_template("index.html") |
26 | 29 |
|
27 |
| - return render_template( |
28 |
| - "index.html" |
29 |
| - ) |
30 | 30 |
|
31 | 31 | @app.route("/api/v1/sigma-versions", methods=["GET"])
|
32 | 32 | def get_versions():
|
33 |
| - return jsonify(sigma_versions[::-1]) |
| 33 | + return jsonify(sorted(sigma_versions, key=version_key, reverse=True)) |
| 34 | + |
34 | 35 |
|
35 | 36 | @app.route("/api/v1/<version>/targets", methods=["GET"])
|
36 | 37 | def get_targets(version):
|
37 | 38 | port = get_port_from_version(version)
|
38 |
| - return requests.get(f'http://localhost:{port}/api/v1/targets', params=dict(request.args)).json() |
| 39 | + return requests.get( |
| 40 | + f"http://localhost:{port}/api/v1/targets", params=dict(request.args) |
| 41 | + ).json() |
| 42 | + |
39 | 43 |
|
40 | 44 | @app.route("/api/v1/<version>/formats", methods=["GET"])
|
41 | 45 | def get_formats(version):
|
42 | 46 | port = get_port_from_version(version)
|
43 |
| - return requests.get(f'http://localhost:{port}/api/v1/formats', params=dict(request.args)).json() |
| 47 | + return requests.get( |
| 48 | + f"http://localhost:{port}/api/v1/formats", params=dict(request.args) |
| 49 | + ).json() |
| 50 | + |
44 | 51 |
|
45 | 52 | @app.route("/api/v1/<version>/pipelines", methods=["GET"])
|
46 | 53 | def get_pipelines(version):
|
47 | 54 | port = get_port_from_version(version)
|
48 |
| - return requests.get(f'http://localhost:{port}/api/v1/pipelines', params=dict(request.args)).json() |
| 55 | + return requests.get( |
| 56 | + f"http://localhost:{port}/api/v1/pipelines", params=dict(request.args) |
| 57 | + ).json() |
| 58 | + |
49 | 59 |
|
50 | 60 | @app.route("/api/v1/<version>/convert", methods=["POST"])
|
51 | 61 | def convert(version):
|
52 | 62 | port = get_port_from_version(version)
|
53 | 63 | payload = request.json
|
54 |
| - return requests.post(f'http://localhost:{port}/api/v1/convert', json=payload).text |
| 64 | + return requests.post(f"http://localhost:{port}/api/v1/convert", json=payload).text |
| 65 | + |
55 | 66 |
|
56 | 67 | if __name__ == "__main__":
|
57 | 68 | app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 8000)))
|
0 commit comments