Skip to content

Commit

Permalink
Create clean.py
Browse files Browse the repository at this point in the history
add clean.py
  • Loading branch information
hanyul-ryu authored Aug 6, 2024
1 parent c176105 commit 268ae01
Showing 1 changed file with 29 additions and 0 deletions.
29 changes: 29 additions & 0 deletions clean.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
import pathlib
import shutil


def clean_project():
"""
Remove common build directories and *.so files in a Python project.
"""
# Directories to remove
dirs_to_remove = [
"__pycache__", ".pytest_cache",
"build", "*.egg-info", # "dist",
]
# File patterns to remove
files_to_remove = ["*.so"]

for path in pathlib.Path(".").rglob("*"):
# Remove specified directories
if path.is_dir() and any(path.match(d) for d in dirs_to_remove):
shutil.rmtree(path, ignore_errors=True)
print(f"Removed directory: {path}")
# Remove specified files
elif path.is_file() and any(path.match(f) for f in files_to_remove):
path.unlink()
print(f"Removed file: {path}")


if __name__ == "__main__":
clean_project()

0 comments on commit 268ae01

Please sign in to comment.