craftbeerpi4-pione/cbpi/http_endpoints/http_system.py
2023-01-27 12:26:55 +01:00

240 lines
7.5 KiB
Python

import re
from aiohttp import web
from aiohttp import streamer
from cbpi.job.aiohttp import get_scheduler_from_app
import logging
from cbpi.api import request_mapping
from cbpi.utils import json_dumps
from cbpi import __version__, __codename__
import pathlib
import os
from cbpi.controller.system_controller import SystemController
class SystemHttpEndpoints:
def __init__(self,cbpi):
self.cbpi = cbpi
self.controller : SystemController = cbpi.system
self.cbpi.register(self, url_prefix="/system")
@request_mapping("/", method="GET", auth_required=False)
async def state(self, request):
"""
---
description: Get complete system state
tags:
- System
responses:
"200":
description: successful operation
"""
plugin_list = await self.cbpi.plugin.load_plugin_list("cbpi4gui")
try:
version= plugin_list[0].get("Version", "not detected")
except:
version="not detected"
return web.json_response(data=dict(
actor=self.cbpi.actor.get_state(),
fermenter=self.cbpi.fermenter.get_state(),
sensor=self.cbpi.sensor.get_state(),
kettle=self.cbpi.kettle.get_state(),
step=self.cbpi.step.get_state(),
fermentersteps=self.cbpi.fermenter.get_fermenter_steps(),
config=self.cbpi.config.get_state(),
version=__version__,
guiversion=version,
codename=__codename__)
, dumps=json_dumps)
@request_mapping(path="/logs", auth_required=False)
async def http_get_log(self, request):
result = []
file_pattern = re.compile("^(\w+.).log(.?\d*)")
for filename in sorted(os.listdir(self.cbpi.logsFolderPath), reverse=True):
if file_pattern.match(filename):
result.append(filename)
return web.json_response(result)
@request_mapping(path="/logs/{name}", method="DELETE", auth_required=False)
async def delete_log(self, request):
log_name = request.match_info['name']
self.cbpi.log.delete_log(log_name)
@request_mapping(path="/logs", method="DELETE", auth_required=False)
async def delete_all_logs(self, request):
self.cbpi.log.delete_logs()
return web.Response(status=204)
@request_mapping("/events", method="GET", name="get_all_events", auth_required=False)
def get_all_events(self, request):
"""
---
description: Get list of all registered events
tags:
- System
responses:
"200":
description: successful operation
"""
return web.json_response(data=self.cbpi.bus.dump())
@request_mapping("/jobs", method="GET", name="get_jobs", auth_required=False)
def get_all_jobs(self, request):
"""
---
description: Get all running Jobs
tags:
- System
responses:
"200":
description: successful operation
"""
scheduler = get_scheduler_from_app(self.cbpi.app)
result = []
for j in scheduler:
try:
result.append(dict(name=j.name, type=j.type, time=j.start_time))
except:
pass
return web.json_response(data=result)
@request_mapping("/restart", method="POST", name="RestartServer", auth_required=False)
async def restart(self, request):
"""
---
description: Restart System
tags:
- System
responses:
"200":
description: successful operation
"""
await self.controller.restart()
return web.Response(text="RESTART")
@request_mapping("/shutdown", method="POST", name="ShutdownSerer", auth_required=False)
async def shutdown(self, request):
"""
---
description: Shutdown System
tags:
- System
responses:
"200":
description: successful operation
"""
await self.controller.shutdown()
return web.Response(text="SHUTDOWN")
@request_mapping("/backup", method="GET", name="BackupConfig", auth_required=False)
async def backup(self, request):
"""
---
description: Zip and download Config Folder
tags:
- System
responses:
"200":
description: successful operation
content: # Response body
application/zip: # Media type
"""
await self.controller.backupConfig()
filename = "cbpi4_config.zip"
file_name = pathlib.Path(os.path.join(".", filename))
response = web.StreamResponse(
status=200,
reason='OK',
headers={'Content-Type': 'application/zip'},
)
await response.prepare(request)
with open(file_name, 'rb') as file:
for line in file.readlines():
await response.write(line)
await response.write_eof()
return response
@request_mapping("/log/{logtime}/", method="GET", name="BackupConfig", auth_required=False)
async def downloadlog(self, request):
"""
---
description: Zip and download craftbeerpi.service log
tags:
- System
responses:
"200":
description: successful operation
content: # Response body
application/zip: # Media type
"""
logtime = request.match_info['logtime']
await self.controller.downloadlog(logtime)
filename = "cbpi4_log.zip"
file_name = pathlib.Path(os.path.join(".", filename))
response = web.StreamResponse(
status=200,
reason='OK',
headers={'Content-Type': 'application/zip'},
)
await response.prepare(request)
with open(file_name, 'rb') as file:
for line in file.readlines():
await response.write(line)
await response.write_eof()
os.remove(file_name)
return response
@request_mapping("/restore", method="POST", name="RestoreConfig", auth_required=False)
async def restore(self, request):
"""
---
description: Restore Config
tags:
- System
responses:
"200":
description: successful operation
"""
logging.info("Restore Config")
data = await request.post()
logging.info("Data received")
await self.controller.restoreConfig(data)
return web.Response(status=200)
@request_mapping("/systeminfo", method="GET", name="SystemInfo", auth_required=False)
async def systeminfo(self, request):
"""
---
description: System Information
tags:
- System
responses:
"200":
description: successful operation
"""
systeminfo = await self.controller.systeminfo()
return web.json_response(data=systeminfo)
@request_mapping("/uploadsvg", method="POST", name="UploadSVG", auth_required=False)
async def uploadSVG(self, request):
"""
---
description: Upload SVG file to widgets folder
tags:
- System
responses:
"200":
description: successful operation
"""
logging.info("Upload SVG file")
data = await request.post()
logging.info("Data received")
await self.controller.uploadSVG(data)
return web.Response(status=200)