mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-25 08:35:03 +08:00
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
import nonebot
|
||
|
|
from watchdog.observers import Observer
|
||
|
|
from watchdog.events import FileSystemEventHandler
|
||
|
|
|
||
|
|
from liteyuki.utils.base.config import get_config
|
||
|
|
from liteyuki.utils.base.reloader import Reloader
|
||
|
|
from liteyuki.utils.base.resource import load_resources
|
||
|
|
|
||
|
|
if get_config("debug", False):
|
||
|
|
|
||
|
|
src_directories = (
|
||
|
|
"liteyuki/liteyuki_main",
|
||
|
|
"liteyuki/plugins",
|
||
|
|
"liteyuki/utils",
|
||
|
|
)
|
||
|
|
src_excludes_extensions = (
|
||
|
|
"pyc",
|
||
|
|
)
|
||
|
|
|
||
|
|
res_directories = (
|
||
|
|
"liteyuki/resources",
|
||
|
|
"resources",
|
||
|
|
)
|
||
|
|
|
||
|
|
nonebot.logger.info("Liteyuki Reload is enable, watching for file changes...")
|
||
|
|
|
||
|
|
class CodeModifiedHandler(FileSystemEventHandler):
|
||
|
|
"""
|
||
|
|
Handler for code file changes
|
||
|
|
"""
|
||
|
|
def on_modified(self, event):
|
||
|
|
if event.src_path.endswith(src_excludes_extensions) or event.is_directory:
|
||
|
|
return
|
||
|
|
nonebot.logger.debug(f"{event.src_path} modified, reloading bot...")
|
||
|
|
Reloader.reload()
|
||
|
|
|
||
|
|
|
||
|
|
class ResourceModifiedHandler(FileSystemEventHandler):
|
||
|
|
"""
|
||
|
|
Handler for resource file changes
|
||
|
|
"""
|
||
|
|
def on_modified(self, event):
|
||
|
|
nonebot.logger.debug(f"{event.src_path} modified, reloading resource...")
|
||
|
|
load_resources()
|
||
|
|
|
||
|
|
|
||
|
|
code_modified_handler = CodeModifiedHandler()
|
||
|
|
resource_modified_handle = ResourceModifiedHandler()
|
||
|
|
|
||
|
|
observer = Observer()
|
||
|
|
for directory in src_directories:
|
||
|
|
observer.schedule(code_modified_handler, directory, recursive=True)
|
||
|
|
for directory in res_directories:
|
||
|
|
observer.schedule(resource_modified_handle, directory, recursive=True)
|
||
|
|
observer.start()
|