-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmain.py
98 lines (75 loc) · 3.29 KB
/
main.py
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import os
import logging
from logging.handlers import RotatingFileHandler
from mp.core import CoreService
from mp.settings import Settings
from mp.player import Player
import decky_plugin
# Create a logger
lvl = logging.DEBUG
_logger = logging.getLogger("magicpods")
_logger.setLevel(lvl)
# Create a file handler and set the level to DEBUG
file_handler = RotatingFileHandler(os.path.join(decky_plugin.DECKY_PLUGIN_LOG_DIR, "magicpodslog.txt"), mode='a', maxBytes=5*1024*1024, backupCount=0)
file_handler.setLevel(lvl)
# Create a console handler and set the level to INFO
console_handler = logging.StreamHandler()
console_handler.setLevel(lvl)
# Create a formatter and set it to the handlers
formatter = logging.Formatter('%(asctime)s %(levelname)-5s %(tag)s %(message)s', datefmt='%Y-%m-%d %H:%M:%S')
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)
# Add the handlers to the logger
_logger.addHandler(file_handler)
_logger.addHandler(console_handler)
logger = logging.LoggerAdapter(_logger, {"tag": "py"})
def bin_logging(msg):
_logger.info(msg, extra={"tag": "bi"})
class Plugin:
async def start_backed(self):
self.core.start()
async def restart_backend(self):
logger.info("Restarting")
self.core.restart()
async def logger_react(self, msg):
_logger.info(msg, extra={"tag": "re"})
async def load_setting(self, key):
return self.settings.load(key)
async def save_setting(self, key, value):
self.settings.save(key, value)
async def read_logs(self):
output = ""
with open(os.path.join(decky_plugin.DECKY_PLUGIN_LOG_DIR, "magicpodslog.txt")) as file:
for line in (file.readlines() [-150:]):
output += line
return output
async def debug_start_backed(self):
self.is_backend_allowed = True # do not forget to add this to debug methods
self.core.start()
async def debug_stop_backed(self):
self.is_backend_allowed = False # do not forget to add this to debug methods
self.core.stop()
# Used to prevent reconnecting websocket when backend off
async def backend_allowed(self):
return self.is_backend_allowed
async def play(self):
self.player.start()
async def stop(self):
self.player.stop()
# Asyncio-compatible long-running code, executed in a task when the plugin is loaded
async def _main(self):
logger.debug("_main starting")
self.settings = Settings(decky_plugin.DECKY_PLUGIN_SETTINGS_DIR)
self.core = CoreService(os.path.join(decky_plugin.DECKY_PLUGIN_DIR, "bin"), "MagicPodsCore", bin_logging)
self.is_backend_allowed = True # Allow reconnecting socket when user using plugin
self.player = Player(os.path.join(decky_plugin.DECKY_PLUGIN_DIR, "silence.mp3"),bin_logging)
self.core.start()
logger.info("_main finished")
# Function called first during the unload process, utilize this to handle your plugin being removed
async def _unload(self):
self.is_backend_allowed = False # Does not allow reconnecting socket when user delete plugin
self.core.stop()
logger.info("_unload finished")
# Migrations that should be performed before entering `_main()`.
async def _migration(self):
pass