Add ruff as default linting + formatting option

ccds-help.json

@@ -202,6 +202,29 @@
             }
         ]
     },
+    {
+        "field": "linting_and_formatting",
+        "help": {
+            "description": "How to handle linting and formatting on your code.",
+            "more_information": ""
+        },
+        "choices": [
+            {
+                "choice": "ruff",
+                "help": {
+                    "description": "Use ruff for linting and formatting.",
+                    "more_information": ""
+                } 
+            },
+            {
+                "choice": "flake8+black+isort",
+                "help": {
+                    "description": "Use flake8 for linting and black+isort for formatting.",
+                    "more_information": ""
+                } 
+            }
+        ]
+    },
     {
         "field": "open_source_license",
         "help": {

ccds.json

@@ -26,6 +26,10 @@
         "none",
         "basic"
     ],
+    "linting_and_formatting": [
+        "ruff",
+        "flake8+black+isort"
+    ],
     "open_source_license": ["No license file", "MIT", "BSD-3-Clause"],
     "docs": ["mkdocs", "none"],
     "include_code_scaffold": ["Yes", "No"]

ccds/hook_utils/dependencies.py

@@ -1,11 +1,16 @@
 packages = [
+    "pip",
+    "python-dotenv",
+]
+
+flake8_black_isort = [
     "black",
     "flake8",
     "isort",
-    "pip",
-    "python-dotenv",
 ]
 
+ruff = ["ruff"]
+
 basic = [
     "ipython",
     "jupyterlab",

hooks/post_gen_project.py

@@ -5,7 +5,14 @@
 # https://github.com/cookiecutter/cookiecutter/issues/824
 #   our workaround is to include these utility functions in the CCDS package
 from ccds.hook_utils.custom_config import write_custom_config
-from ccds.hook_utils.dependencies import basic, packages, scaffold, write_dependencies
+from ccds.hook_utils.dependencies import (
+    basic,
+    flake8_black_isort,
+    packages,
+    ruff,
+    scaffold,
+    write_dependencies,
+)
 
 #
 #  TEMPLATIZED VARIABLES FILLED IN BY COOKIECUTTER
@@ -24,6 +31,13 @@
 packages_to_install += basic
 # {% endif %}
 
+# {% if cookiecutter.linting_and_formatting == "ruff" %}
+packages_to_install += ruff
+# Remove setup.cfg
+Path("setup.cfg").unlink()
+# {% elif cookiecutter.linting_and_formatting == "flake8+black+isort" %}
+packages_to_install += flake8_black_isort
+# {% endif %}
 # track packages that are not available through conda
 pip_only_packages = [
     "awscli",

tests/conda_harness.sh

@@ -34,5 +34,7 @@ make
 make create_environment
 conda activate $PROJECT_NAME
 make requirements
+make lint
+make format
 
 run_tests $PROJECT_NAME $MODULE_NAME

tests/conftest.py

@@ -63,6 +63,7 @@ def _is_valid(config):
         "dataset_storage",
         "open_source_license",
         "include_code_scaffold",
+        "linting_and_formatting",
         "docs",
     ]
     cyclers = {k: cycle(cookiecutter_json[k]) for k in cycle_fields}

tests/pipenv_harness.sh

@@ -26,6 +26,10 @@ make create_environment
 # can happen outside of environment since pipenv knows based on Pipfile
 make requirements
 
+# linting + formatting must happen inside environment
+pipenv run make lint
+pipenv run make format
+
 # test with pipenv run
 pipenv run python -c "import sys; assert \"$PROJECT_NAME\" in sys.executable"
 

tests/test_creation.py

@@ -49,7 +49,6 @@ def test_baking_configs(config, fast):
     with bake_project(config) as project_directory:
         verify_folders(project_directory, config)
         verify_files(project_directory, config)
-        lint(project_directory)
 
         if fast < 2:
             verify_makefile_commands(project_directory, config)
@@ -100,7 +99,6 @@ def verify_files(root, config):
         "Makefile",
         "README.md",
         "pyproject.toml",
-        "setup.cfg",
         ".env",
         ".gitignore",
         "data/external/.gitkeep",
@@ -120,6 +118,9 @@ def verify_files(root, config):
     if not config["open_source_license"].startswith("No license"):
         expected_files.append("LICENSE")
 
+    if config["linting_and_formatting"] == "flake8+black+isort":
+        expected_files.append("setup.cfg")
+
     if config["include_code_scaffold"] == "Yes":
         expected_files += [
             f"{config['module_name']}/config.py",
@@ -156,6 +157,8 @@ def verify_makefile_commands(root, config):
     - blank command listing commands
     - create_environment
     - requirements
+    - linting
+    - formatting
     Ensure that these use the proper environment.
     """
     test_path = Path(__file__).parent
@@ -184,23 +187,18 @@ def verify_makefile_commands(root, config):
         stdout=PIPE,
     )
 
-    stdout_output, _ = _decode_print_stdout_stderr(result)
+    stdout_output, stderr_output = _decode_print_stdout_stderr(result)
 
     # Check that makefile help ran successfully
     assert "Available rules:" in stdout_output
     assert "clean                    Delete all compiled Python files" in stdout_output
 
-    assert result.returncode == 0
-
-
-def lint(root):
-    """Run the linters on the project."""
-    result = run(
-        ["make", "lint"],
-        cwd=root,
-        stderr=PIPE,
-        stdout=PIPE,
-    )
-    _, _ = _decode_print_stdout_stderr(result)
+    # Check that linting and formatting ran successfully
+    if config["linting_and_formatting"] == "ruff":
+        assert "All checks passed!" in stdout_output
+        assert "reformatted" not in stdout_output
+    elif config["linting_and_formatting"] == "flake8+black+isort":
+        assert "All done!" in stderr_output
+        assert "reformatted" not in stderr_output
 
     assert result.returncode == 0

tests/virtualenv_harness.sh

@@ -55,6 +55,8 @@ else
 fi
 
 make requirements
+make lint
+make format
 
 run_tests $PROJECT_NAME $MODULE_NAME
 

{{ cookiecutter.repo_name }}/Makefile

@@ -11,7 +11,7 @@ PYTHON_INTERPRETER = python
 #################################################################################
 
 {% if cookiecutter.dependency_file != 'none' %}
-## Install Python Dependencies
+## Install Python dependencies
 .PHONY: requirements
 requirements:
 	{% if "requirements.txt" == cookiecutter.dependency_file -%}
@@ -31,7 +31,19 @@ clean:
 	find . -type f -name "*.py[co]" -delete
 	find . -type d -name "__pycache__" -delete
 
-## Lint using flake8 and black (use `make format` to do formatting)
+{% if cookiecutter.linting_and_formatting == 'ruff' %}
+## Lint using ruff (use `make format` to do formatting)
+.PHONY: lint
+lint:
+	ruff check {{ cookiecutter.module_name }}
+
+## Format source code with ruff
+.PHONY: format
+format:
+	ruff check --select I --fix {{ cookiecutter.module_name }}
+	ruff format {{ cookiecutter.module_name }}
+{% elif cookiecutter.linting_and_formatting == 'flake8+black+isort' %}
+## Lint using flake8, black, and isort (use `make format` to do formatting)
 .PHONY: lint
 lint:
 	flake8 {{ cookiecutter.module_name }}
@@ -41,7 +53,9 @@ lint:
 ## Format source code with black
 .PHONY: format
 format:
+	isort --profile black {{ cookiecutter.module_name }}
 	black --config pyproject.toml {{ cookiecutter.module_name }}
+{% endif %}
 
 {% if not cookiecutter.dataset_storage.none %}
 ## Download Data from storage system
@@ -72,7 +86,7 @@ sync_data_up:
 {% endif %}
 
 {% if cookiecutter.environment_manager != 'none' %}
-## Set up python interpreter environment
+## Set up Python interpreter environment
 .PHONY: create_environment
 create_environment:
 	{% if cookiecutter.environment_manager == 'conda' -%}
@@ -97,7 +111,7 @@ create_environment:
 #################################################################################
 
 {% if cookiecutter.include_code_scaffold == 'Yes' %}
-## Make Dataset
+## Make dataset
 .PHONY: data
 data: requirements
 	$(PYTHON_INTERPRETER) {{ cookiecutter.module_name }}/data/make_dataset.py

{{ cookiecutter.repo_name }}/pyproject.toml

@@ -17,6 +17,7 @@ classifiers = [
 ]
 requires-python = "~={{ cookiecutter.python_version_number }}"
 
+{% if cookiecutter.linting_and_formatting == 'flake8+black+isort' %}
 [tool.black]
 line-length = 99
 include = '\.pyi?$'
@@ -25,8 +26,13 @@ exclude = '''
     \.git
   | \.venv
 )/
-'''
 
-[tool.ruff.lint.isort]
-known_first_party = ["{{ cookiecutter.module_name }}"]
-force_sort_within_sections = true
+[tool.isort]
+profile = "black"
+'''
+{% endif %}
+{% if cookiecutter.linting_and_formatting == 'ruff' %}
+[tool.ruff]
+line-length = 99
+src = ["{{ cookiecutter.module_name }}"]
+{% endif %}