weboob-ci/server.py

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)