46 lines
1.9 KiB
Python
46 lines
1.9 KiB
Python
import os
|
|
from django.core.management.commands.runserver import Command as BaseCommand
|
|
from watchdog.observers import Observer
|
|
from watchdog.events import FileSystemEventHandler
|
|
from django.utils.autoreload import restart_with_reloader
|
|
|
|
class WatchdogEventHandler(FileSystemEventHandler):
|
|
def __init__(self, command):
|
|
super().__init__()
|
|
self.command = command
|
|
self.allowed_extensions = ['.py'] # 允许处理的文件后缀列表
|
|
|
|
def on_any_event(self, event):
|
|
if "__pycache__" in event.src_path:
|
|
return # 如果事件发生在 __pycache__ 目录中，则直接返回，不做任何处理
|
|
if event.is_directory:
|
|
return
|
|
file_extension = os.path.splitext(event.src_path)[1]
|
|
if file_extension not in self.allowed_extensions:
|
|
return # 如果文件后缀不在允许的列表中，则直接返回，不做任何处理
|
|
if event.event_type in ('created', 'deleted', 'modified'):
|
|
self.command.stdout.write(f"File {event.event_type}: {event.src_path}")
|
|
# 在此处可以执行你希望在文件变化时执行的操作
|
|
restart_with_reloader()
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
# 创建 watchdog 观察者对象
|
|
observer = Observer()
|
|
# 创建 watchdog 事件处理器对象
|
|
event_handler = WatchdogEventHandler(self)
|
|
# 监听 Django 项目根目录下的文件变化
|
|
observer.schedule(event_handler, '.', recursive=True)
|
|
# 启动 watchdog 观察者
|
|
observer.start()
|
|
|
|
try:
|
|
# 启动 Django 内置的 runserver 命令
|
|
super().handle(*args, **options)
|
|
finally:
|
|
# 在 Django 服务停止时停止 watchdog 观察者
|
|
observer.stop()
|
|
observer.join()
|
|
|
|
|
|
requires_system_checks = [] |