feat: Suggest poetry init when pyproject.toml does not exist

flake8_nitpick/__init__.py

@@ -24,11 +24,10 @@
 # Constants
 NAME = "flake8-nitpick"
 ERROR_PREFIX = "NIP"
-PYPROJECT_TOML = "pyproject.toml"
 NITPICK_STYLE_TOML = "nitpick-style.toml"
 DEFAULT_NITPICK_STYLE_URL = "https://raw.githubusercontent.com/andreoliwa/flake8-nitpick/master/nitpick-style.toml"
 ROOT_PYTHON_FILES = ("setup.py", "manage.py", "autoapp.py")
-ROOT_FILES = (PYPROJECT_TOML, "setup.cfg", "requirements*.txt", "Pipfile") + ROOT_PYTHON_FILES
+ROOT_FILES = ("requirements*.txt", "Pipfile") + ROOT_PYTHON_FILES
 
 LOG = logging.getLogger("flake8.nitpick")
 
@@ -111,9 +110,11 @@ def __init__(self, root_dir: Path) -> None:
 
     def load_toml(self) -> YieldFlake8Error:
         """Load TOML configuration from files."""
-        self.pyproject_path: Path = self.root_dir / PYPROJECT_TOML
+        self.pyproject_path: Path = self.root_dir / PyProjectTomlChecker.file_name
         if not self.pyproject_path.exists():
-            yield self.flake8_error(1, f"{PYPROJECT_TOML} does not exist")
+            yield self.flake8_error(
+                1, f"{PyProjectTomlChecker.file_name} does not exist. Run 'poetry init' to create one."
+            )
             return
 
         self.pyproject_toml = toml.load(str(self.pyproject_path))
@@ -241,7 +242,9 @@ def find_root_dir(python_file: str) -> Optional[Path]:
             LOG.info("Loading cached root dir: %s", root_dir)
             return root_dir
 
-        found_files = climb_directory_tree(python_file, ROOT_FILES)
+        found_files = climb_directory_tree(
+            python_file, ROOT_FILES + (PyProjectTomlChecker.file_name, SetupCfgChecker.file_name)
+        )
         if not found_files:
             LOG.error("No files found while climbing directory tree from %s", python_file)
             return None

tests/test_nitpick.py

@@ -1,5 +1,5 @@
 """Nitpick tests."""
-from flake8_nitpick import ROOT_PYTHON_FILES
+from flake8_nitpick import ROOT_PYTHON_FILES, PyProjectTomlChecker
 from tests.helpers import ProjectMock
 
 
@@ -45,3 +45,10 @@ def test_comma_separated_keys_on_style_file(request):
 [food]
 eat = ham,salt"""
     )
+
+
+def test_suggest_poetry_init(request):
+    """Suggest poetry init when pyproject.toml does not exist."""
+    assert ProjectMock(request, pyproject_toml=False).lint().errors == {
+        f"NIP201 {PyProjectTomlChecker.file_name} does not exist. Run 'poetry init' to create one."
+    }