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
|
from flask import jsonify
from schema import Optional, Schema
from pysite.base_route import APIView
from pysite.constants import ValidationTypes
from pysite.decorators import api_key, api_params
from pysite.mixins import DBMixin
# todo: type safety
SETTINGS_KEYS_DEFAULTS = {
"defcon_enabled": False,
"defcon_days": 1
}
GET_SCHEMA = Schema({
Optional("keys"): str
})
def settings_schema():
schema_dict = {Optional(key): type(SETTINGS_KEYS_DEFAULTS[key]) for key in SETTINGS_KEYS_DEFAULTS.keys()}
return Schema(schema_dict)
class ServerSettingsView(APIView, DBMixin):
path = "/bot/settings"
name = "bot.settings"
@api_key
@api_params(schema=GET_SCHEMA, validation_type=ValidationTypes.params)
def get(self, params=None):
keys_raw = None
if params:
keys_raw = params.get("keys")
keys = filter(lambda key: key in SETTINGS_KEYS_DEFAULTS,
keys_raw.split(",")) if keys_raw else SETTINGS_KEYS_DEFAULTS.keys()
result = {key: (self.db.get("bot_settings", key) or {}).get("value") or SETTINGS_KEYS_DEFAULTS[key] for key in
keys}
return jsonify(result)
@api_key
@api_params(schema=settings_schema(), validation_type=ValidationTypes.json)
def put(self, json_data):
# update in database
for key, value in json_data.items():
self.db.insert("bot_settings", {
"key": key,
"value": value
}, conflict="update")
return jsonify({
"success": True
})
|
