This repository has been archived by the owner on Jan 29, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
6995008
commit 2250238
Showing
12 changed files
with
217 additions
and
201 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file was deleted.
Oops, something went wrong.
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,13 +1,151 @@ | ||
| import linters | ||
| import movers | ||
|
|
||
| import rich | ||
| # Steps to finalize the cookiecutter build | ||
| import shutil | ||
| import os | ||
| import pathlib | ||
| import importlib.util | ||
| import logging | ||
| import subprocess | ||
|
|
||
| def move_db_files(db_resource: str): | ||
| """ | ||
| Moves the correct db files to the correct location | ||
| Delete the remaining files in the db folder and the db folder itself | ||
| """ | ||
|
|
||
| if "postgres" in db_resource: | ||
| shutil.move( | ||
| "src/db/postgres_models.py", | ||
| "src/models.py" | ||
| ) | ||
| shutil.move( | ||
| "src/db/postgres_seeder.py", | ||
| "src/flask/flaskapp/seeder.py", | ||
| ) | ||
| if "mongo" in db_resource: | ||
| shutil.move( | ||
| "src/db/mongo_models.py", | ||
| "src/models.py" | ||
| ) | ||
| shutil.move( | ||
| "src/db/mongo_seeder.py", | ||
| "src/flask/flaskapp/seeder.py", | ||
| ) | ||
|
|
||
| def remove_aca_files() -> None: | ||
| """Removes unneeded files if aca is not selected""" | ||
| file_names = ( | ||
| "src/Dockerfile", | ||
| ) | ||
|
|
||
| for file_name in file_names: | ||
| os.remove(file_name) | ||
|
|
||
| def remove_flask_migration_files() -> None: | ||
| """ | ||
| Removes the flask migration files if postgres is not selected | ||
| This only applies to flask projects | ||
| """ | ||
| if "{{ cookiecutter.project_backend }}" == "flask" and "mongo" in "{{cookiecutter.db_resource }}": | ||
| shutil.rmtree("src/flask/flaskapp/migrations") | ||
| else: | ||
| pass | ||
|
|
||
| def rename_backend_files(): | ||
| """ | ||
| Rename the selected backend folder corresponding to the selected option. | ||
| remove the project_backend folders that are not selected | ||
| """ | ||
|
|
||
| selected_backend = "{{cookiecutter.project_backend}}" | ||
|
|
||
| project_backends = ["django", "fastapi", "flask"] | ||
| project_backends.remove(selected_backend) | ||
|
|
||
| src = pathlib.Path('src') | ||
|
|
||
| for unused_backend in project_backends: | ||
| shutil.rmtree(src / pathlib.Path(unused_backend)) | ||
|
|
||
| shutil.copytree( | ||
| src / pathlib.Path(selected_backend), | ||
| pathlib.Path.cwd() / src, | ||
| dirs_exist_ok=True, | ||
| ) | ||
| shutil.rmtree(src / pathlib.Path(selected_backend)) | ||
|
|
||
|
|
||
| def choose_web_bicep(): | ||
| """Selects the correct web.bicep file""" | ||
| bicep_files = { | ||
| "aca": "infra/aca.bicep", | ||
| "appservice": "infra/appservice.bicep", | ||
| } | ||
|
|
||
| shutil.move( | ||
| bicep_files.pop("{{cookiecutter.project_host}}"), | ||
| "infra/web.bicep", | ||
| ) | ||
|
|
||
| for file_name in bicep_files.values(): | ||
| os.remove(file_name) | ||
|
|
||
|
|
||
| def check_for_files() -> None: | ||
| """ | ||
| Iterate through the cookiecutter options. | ||
| remove the files corresponding to the results | ||
| TODO: Add task progress | ||
| """ | ||
|
|
||
| # DB Options | ||
| # The chosen db option is stored in src. | ||
| move_db_files("{{cookiecutter.db_resource}}") | ||
| shutil.rmtree("src/db") # Clean up remaining db folder | ||
|
|
||
| # Backend Options | ||
| remove_flask_migration_files() | ||
|
|
||
| # Azure Host Options | ||
| choose_web_bicep() | ||
| if "{{cookiecutter.project_host }}" != "aca": | ||
| remove_aca_files() | ||
|
|
||
| if "{{cookiecutter.project_host}}" == "appservice": | ||
| pass | ||
|
|
||
|
|
||
| rename_backend_files() | ||
|
|
||
|
|
||
| def error_msg(pkg: str) -> str: | ||
| return f"`{pkg}` is not installed. Run `pip install {pkg}` to install it." | ||
|
|
||
| def run_ruff_fix_and_black() -> None: | ||
| """checks if ruff and black are installed and runs them on the project""" | ||
| if importlib.util.find_spec("ruff"): | ||
| subprocess.run(["python3", "-m" "ruff", "--fix", "src"]) | ||
| else: | ||
| logging.warning(error_msg("ruff")) | ||
|
|
||
| if importlib.util.find_spec("black"): | ||
| subprocess.run(["python3", "-m", "black", "-q", "src"]) | ||
| else: | ||
| logging.warning(error_msg("black")) | ||
|
|
||
| def run_bicep_format() -> None: | ||
| """formats your bicep files""" | ||
| subprocess.run(["az", "bicep", "format", "--file", "infra/main.bicep"]) | ||
|
|
||
|
|
||
| def lint() -> None: | ||
| """Runs all linters""" | ||
| run_ruff_fix_and_black() | ||
| run_bicep_format() | ||
|
|
||
| if __name__ == "__main__": | ||
| # It's import to remove the unecessary files before moving the db files | ||
|
|
||
| rich.print("Removing unecessary files") | ||
| movers.check_for_files() | ||
| check_for_files() | ||
| rich.print("Linting files") | ||
| linters.lint() | ||
| lint() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.