Fix pyrpoject.toml search

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyproject-autoflake"
-version = "1.0.0"
+version = "1.0.1"
 description = "pyproject-autoflake (pautoflake), a monkey patching wrapper to connect autoflake with pyproject.toml configuration."
 authors = ["quwac <53551867+quwac@users.noreply.github.com>"]
 readme = "README.md"

pyproject_autoflake/pautoflake.py

@@ -2,13 +2,17 @@
 import signal
 import sys
 from pathlib import Path
-from typing import Dict, List, Optional, Tuple, Union
+from typing import Dict, List, Tuple, Union
 
 import autoflake
 import toml
 
 
 def _get_pyproject_toml_path(path_candidates: List[str]) -> str:
+    path_in_current = os.path.join(os.getcwd(), "pyproject.toml")
+    if os.path.exists(path_in_current):
+        return path_in_current
+
     absolute_paths: List[Path] = []
     for path_candidate in path_candidates:
         path = Path(path_candidate)
@@ -30,9 +34,7 @@ def _get_pyproject_toml_path(path_candidates: List[str]) -> str:
     raise ValueError("pyproject.toml not found.")
 
 
-def _get_argv_from_toml(pyproject_path: Optional[str] = None) -> List[str]:
-    if pyproject_path is None:
-        pyproject_path = os.path.join(os.getcwd(), "pyproject.toml")
+def _get_argv_from_toml(pyproject_path: str) -> List[str]:
     assert (
         pyproject_path is not None and pyproject_path != "" and os.path.exists(pyproject_path)
     ), f"pyproject.toml not found. {pyproject_path}"