style: Format watch.py with black and add newlines between functions

This commit is contained in:
Paul Gauthier (aider) 2024-10-25 09:41:33 -07:00
parent 81903598a8
commit 21876e72fe

View file

@ -1,7 +1,9 @@
from pathlib import Path from pathlib import Path
from typing import Set from typing import Set
from watchfiles import watch from watchfiles import watch
def is_source_file(path: Path) -> bool: def is_source_file(path: Path) -> bool:
""" """
Check if a file is a source file that uses # or // style comments. Check if a file is a source file that uses # or // style comments.
@ -9,13 +11,35 @@ def is_source_file(path: Path) -> bool:
""" """
COMMENT_STYLE_EXTENSIONS = { COMMENT_STYLE_EXTENSIONS = {
# # style comments # # style comments
'.py', '.r', '.rb', '.pl', '.pm', '.sh', '.bash', '.yaml', '.yml', ".py",
".r",
".rb",
".pl",
".pm",
".sh",
".bash",
".yaml",
".yml",
# // style comments # // style comments
'.js', '.ts', '.jsx', '.tsx', '.cpp', '.c', '.h', '.hpp', '.java', ".js",
'.swift', '.kt', '.cs', '.go', '.rs', '.php' ".ts",
".jsx",
".tsx",
".cpp",
".c",
".h",
".hpp",
".java",
".swift",
".kt",
".cs",
".go",
".rs",
".php",
} }
return path.suffix.lower() in COMMENT_STYLE_EXTENSIONS return path.suffix.lower() in COMMENT_STYLE_EXTENSIONS
def watch_source_files(directory: str) -> Set[str]: def watch_source_files(directory: str) -> Set[str]:
""" """
Watch for changes to source files in the given directory and its subdirectories. Watch for changes to source files in the given directory and its subdirectories.
@ -33,6 +57,7 @@ def watch_source_files(directory: str) -> Set[str]:
changed_files = {str(Path(change[1])) for change in changes} changed_files = {str(Path(change[1])) for change in changes}
yield changed_files yield changed_files
def main(): def main():
"""Example usage of the file watcher""" """Example usage of the file watcher"""
import sys import sys
@ -52,5 +77,6 @@ def main():
except KeyboardInterrupt: except KeyboardInterrupt:
print("\nStopped watching files") print("\nStopped watching files")
if __name__ == "__main__": if __name__ == "__main__":
main() main()