Skip to content

Instantly share code, notes, and snippets.

@ertgl
Last active October 16, 2025 16:16
Show Gist options
  • Save ertgl/c70a9af5504b774a10b40bdac848059c to your computer and use it in GitHub Desktop.
Save ertgl/c70a9af5504b774a10b40bdac848059c to your computer and use it in GitHub Desktop.
Removes all Python files except '__init__.py' under any 'migrations' directory.
# MIT License
#
# Copyright (c) 2025 Ertuğrul Keremoğlu <[email protected]>
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
# Source: https://gist.github.com/ertgl/c70a9af5504b774a10b40bdac848059c
__all__ = ["main"]
import sys
from collections.abc import Iterable
from pathlib import Path
WORK_DIR = Path.cwd()
VENV_DIR = WORK_DIR / ".venv"
def iterate_migration_file_paths() -> Iterable[Path]:
for migration_file_path in WORK_DIR.rglob("**/migrations/*.py"):
if str(migration_file_path).startswith(str(VENV_DIR)):
continue
if migration_file_path.name == "__init__.py":
continue
yield migration_file_path
def delete_migration_files() -> int:
removed: set[str] = set()
for migration_file_path in iterate_migration_file_paths():
migration_file_path.unlink()
removed.add(str(migration_file_path))
num_removed = len(removed)
if num_removed == 0:
print("No migration files found.")
elif num_removed == 1:
print("1 migration file removed.")
else:
print(f"{num_removed} migration files removed.")
return 0
def main(
argv: list[str] | None = None,
) -> int:
return delete_migration_files()
if __name__ == "__main__":
sys.exit(main(argv=sys.argv.copy()))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment