add the ability to easily run on a local project

hauntsaninja · hauntsaninja · commit fc26cf1d7ec1 · 2020-10-15T15:50:08.000-07:00
This makes bisection easier as well
diff --git a/README.md b/README.md
@@ -27,10 +27,10 @@ mypy_primer, it'll check out projects as they were today and things should work!
 λ mypy_primer --help
 usage: mypy_primer [-h] [--new NEW] [--old OLD] [--repo REPO]
                    [--custom-typeshed-repo CUSTOM_TYPESHED_REPO] [--new-typeshed NEW_TYPESHED]
-                   [--old-typeshed OLD_TYPESHED] [-k PROJECT_SELECTOR] [--expected-success]
-                   [--project-date PROJECT_DATE] [-o {full,diff,concise}] [--old-success]
-                   [--coverage] [--bisect] [--bisect-error BISECT_ERROR] [-j CONCURRENCY]
-                   [--debug] [--base-dir BASE_DIR] [--clear]
+                   [--old-typeshed OLD_TYPESHED] [-k PROJECT_SELECTOR] [-p LOCAL_PROJECT]
+                   [--expected-success] [--project-date PROJECT_DATE] [-o {full,diff,concise}]
+                   [--old-success] [--coverage] [--bisect] [--bisect-error BISECT_ERROR]
+                   [-j CONCURRENCY] [--debug] [--base-dir BASE_DIR] [--clear]
 
 optional arguments:
   -h, --help            show this help message and exit
@@ -54,6 +54,9 @@ mypy:
 project selection:
   -k PROJECT_SELECTOR, --project-selector PROJECT_SELECTOR
                         regex to filter projects (matches against location)
+  -p LOCAL_PROJECT, --local-project LOCAL_PROJECT
+                        run only on the given file or directory. if a single file, supports a '#
+                        flags: ...' comment, like mypy unit tests
   --expected-success    filter to hardcoded subset of projects where some recent mypy version
                         succeeded aka are committed to the mypy way of life. also look at: --old-
                         success
@@ -112,6 +115,11 @@ Figure out what commit is causing a specific error in the project you care about
 mypy_primer -k pandas --bisect-error 'Incompatible types in assignment'
 ```
 
+Or on a local file / directory:
+```
+mypy_primer -p test.py --bisect --old v0.770
+```
+
 Find out what the hell mypy_primer is doing:
 ```
 mypy_primer --debug
diff --git a/mypy_primer.py b/mypy_primer.py
@@ -371,6 +371,19 @@ async def source_paths(self, mypy_python: str) -> List[Path]:
         )
         return [ARGS.projects_dir / self.name / p for p in proc.stdout.splitlines()]
 
+    @classmethod
+    def from_location(cls, location: str) -> Project:
+        additional_flags = ""
+        if Path(location).is_file():
+            with open(location) as f:
+                header = f.readline().strip()
+                if header.startswith("# flags:"):
+                    additional_flags = header[len("# flags:") :]
+        return Project(
+            location=location,
+            mypy_cmd=f"{{mypy}} {location} {additional_flags}",
+        )
+
 
 @dataclass(frozen=True)
 class MypyResult:
@@ -474,6 +487,8 @@ def format_full(self) -> str:
 
 
 def select_projects() -> Iterator[Project]:
+    if ARGS.local_project:
+        return iter([Project.from_location(ARGS.local_project)])
     project_iter = (p for p in PROJECTS)
     if ARGS.project_selector:
         project_iter = (
@@ -648,6 +663,14 @@ def parse_options(argv: List[str]) -> argparse.Namespace:
     proj_group.add_argument(
         "-k", "--project-selector", help="regex to filter projects (matches against location)"
     )
+    proj_group.add_argument(
+        "-p",
+        "--local-project",
+        help=(
+            "run only on the given file or directory. if a single file, supports a "
+            "'# flags: ...' comment, like mypy unit tests"
+        ),
+    )
     proj_group.add_argument(
         "--expected-success",
         action="store_true",
diff --git a/test.sh b/test.sh
@@ -2,6 +2,6 @@ set -ex
 
 isort --diff --check --quiet .
 black --diff --check --quiet .
-flake8 --max-line-length 100
+flake8 --max-line-length 100 --ignore E203,W503
 mypy -m mypy_primer --strict
 python -c 'import mypy_primer'
