mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-23 05:54:58 +01:00
a7c9a5f0e6
Added functions to store current dashboard info in settings Info can be also retrieved from ui to start with the dashboard that was used last time
66 lines
2.4 KiB
Python
66 lines
2.4 KiB
Python
from cbpi.api.dataclasses import NotificationType
|
|
import logging
|
|
import json
|
|
import os
|
|
from os import listdir
|
|
from os.path import isfile, join
|
|
from cbpi.api.base import CBPiBase
|
|
from cbpi.api.config import ConfigType
|
|
|
|
from voluptuous.schema_builder import message
|
|
|
|
class DashboardController:
|
|
|
|
|
|
def __init__(self, cbpi):
|
|
self.caching = False
|
|
self.cbpi = cbpi
|
|
self.logger = logging.getLogger(__name__)
|
|
self.cbpi.register(self)
|
|
|
|
self.path = os.path.join(".", 'config', "cbpi_dashboard_1.json")
|
|
|
|
async def init(self):
|
|
pass
|
|
|
|
async def get_content(self, dashboard_id):
|
|
try:
|
|
self.path = os.path.join(".", 'config', "cbpi_dashboard_"+ str(dashboard_id) +".json")
|
|
logging.info(self.path)
|
|
with open(self.path) as json_file:
|
|
data = json.load(json_file)
|
|
return data
|
|
except:
|
|
return {'elements': [], 'pathes': []}
|
|
|
|
async def add_content(self, dashboard_id, data):
|
|
print(data)
|
|
self.path = os.path.join(".", 'config', "cbpi_dashboard_" + str(dashboard_id)+ ".json")
|
|
with open(self.path, 'w') as outfile:
|
|
json.dump(data, outfile, indent=4, sort_keys=True)
|
|
self.cbpi.notify(title="Dashboard {}".format(dashboard_id), message="Saved Successfully", type=NotificationType.SUCCESS)
|
|
return {"status": "OK"}
|
|
|
|
async def delete_content(self, dashboard_id):
|
|
self.path = os.path.join(".", 'config', "cbpi_dashboard_"+ str(dashboard_id)+ ".json")
|
|
if os.path.exists(self.path):
|
|
os.remove(self.path)
|
|
self.cbpi.notify(title="Dashboard {}".format(dashboard_id), message="Deleted Successfully", type=NotificationType.SUCCESS)
|
|
|
|
|
|
async def get_custom_widgets(self):
|
|
path = os.path.join(".", 'config', "dashboard", "widgets")
|
|
onlyfiles = [os.path.splitext(f)[0] for f in listdir(path) if isfile(join(path, f)) and f.endswith(".svg")]
|
|
return onlyfiles
|
|
|
|
async def get_dashboard_numbers(self):
|
|
max_dashboard_number = self.cbpi.config.get("max_dashboard_number", 4)
|
|
return max_dashboard_number
|
|
|
|
async def get_current_dashboard(self):
|
|
current_dashboard_number = self.cbpi.config.get("current_dashboard_number", 1)
|
|
return current_dashboard_number
|
|
|
|
async def set_current_dashboard(self, dashboard_id=1):
|
|
await self.cbpi.config.set("current_dashboard_number", dashboard_id)
|
|
return {"status": "OK"}
|