137 lines
3.7 KiB
Python
Executable File
137 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
from __future__ import print_function
|
|
|
|
import arrow
|
|
import os
|
|
import peewee
|
|
import sys
|
|
|
|
import config
|
|
import models
|
|
|
|
from bottle import abort, get, post, request, run, static_file
|
|
|
|
|
|
@get("/")
|
|
def index():
|
|
return server_static("index.html")
|
|
|
|
|
|
@get("/module/<name>")
|
|
def module(*args, **kwargs):
|
|
return server_static("index.html")
|
|
|
|
|
|
@get("/api/v1/modules")
|
|
def get_api():
|
|
models.db.connect()
|
|
join_alias = models.ModuleStatus.alias()
|
|
module_status = [
|
|
{
|
|
"module": row.module,
|
|
"origin": row.origin,
|
|
"datetime": arrow.get(row.datetime).isoformat(),
|
|
"status": "good" if row.is_good else "bad"
|
|
}
|
|
# TODO: SQL query
|
|
for row in (models.ModuleStatus
|
|
.select()
|
|
.join(
|
|
join_alias,
|
|
peewee.JOIN.LEFT_OUTER,
|
|
on=(
|
|
(models.ModuleStatus.module == join_alias.module) &
|
|
(models.ModuleStatus.datetime < join_alias.datetime)
|
|
)
|
|
)
|
|
.where(
|
|
join_alias.datetime.is_null()
|
|
)
|
|
.group_by(models.ModuleStatus.module)
|
|
.order_by(models.ModuleStatus.module.asc()))
|
|
]
|
|
models.db.close()
|
|
return {
|
|
"data": module_status
|
|
}
|
|
|
|
|
|
@post("/api/v1/modules")
|
|
def post_api():
|
|
payload = request.json
|
|
if not payload:
|
|
abort(400, "Invalid JSON payload.")
|
|
models.db.connect()
|
|
try:
|
|
for status in ["good", "bad"]:
|
|
for module in payload["modules"][status]:
|
|
models.ModuleStatus.create(
|
|
module=module,
|
|
origin=payload["origin"],
|
|
is_good=(status == "good")
|
|
)
|
|
except KeyError:
|
|
abort(400, "Invalid JSON payload.")
|
|
models.db.close()
|
|
return "OK"
|
|
|
|
|
|
@get("/api/v1/modules/<module>/<datetime_start>/<datetime_end>")
|
|
def get_api_module_with_dates(module, datetime_start, datetime_end):
|
|
datetime_start = arrow.get(datetime_start).datetime
|
|
datetime_end = arrow.get(datetime_end).datetime
|
|
models.db.connect()
|
|
module_status = {
|
|
module: [
|
|
{
|
|
"origin": row.origin,
|
|
"datetime": arrow.get(row.datetime).isoformat(),
|
|
"status": "good" if row.is_good else "bad"
|
|
}
|
|
for row in (models.ModuleStatus
|
|
.select()
|
|
.where(
|
|
(models.ModuleStatus.module == module) &
|
|
(models.ModuleStatus.datetime >= datetime_start) &
|
|
(models.ModuleStatus.datetime < datetime_end)
|
|
)
|
|
.order_by(models.ModuleStatus.datetime.asc()))
|
|
]
|
|
}
|
|
models.db.close()
|
|
return {
|
|
"data": module_status
|
|
}
|
|
|
|
|
|
@get("/static/<filepath:path>")
|
|
def server_static(filepath):
|
|
script_directory = os.path.dirname(os.path.realpath(__file__))
|
|
return static_file(
|
|
filepath,
|
|
root=os.path.join(script_directory, "static")
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
DEBUG = config.DEBUG
|
|
if len(sys.argv) > 3:
|
|
HOST = sys.argv[1]
|
|
PORT = int(sys.argv[2])
|
|
else:
|
|
HOST = config.HOST
|
|
PORT = config.PORT
|
|
|
|
# Log SQL queries in debug mode
|
|
if DEBUG:
|
|
import logging
|
|
logger = logging.getLogger('peewee')
|
|
logger.setLevel(logging.DEBUG)
|
|
logger.addHandler(logging.StreamHandler())
|
|
|
|
# Init db
|
|
models.init_db()
|
|
|
|
run(host=HOST, port=PORT, debug=DEBUG)
|