feat: Allow loading from multiple repos (#6)

.codeclimate.yml

@@ -0,0 +1,13 @@
+engines:
+  bandit:
+    enabled: true
+    checks:
+      assert_used:
+        enabled: false
+
+checks:
+  argument-count:
+    enabled: false
+  method-complexity:
+    config:
+      threshold: 8

.github/workflows/tests.yaml

@@ -50,7 +50,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install dependencies
-        run: poetry install --no-root --with=tests --sync
+        run: poetry install --with=tests --sync
 
       - name: Install poetry version under test
         run: |

poetry_homebrew_formula/plugin.py

@@ -61,7 +61,6 @@ def handle(self) -> int:
 
         self.line_error("<info>Generating formula</info>")
         template = self.load_template()
-        self.repo = self.poetry.pool.repository("pypi")
         package = self.project_with_activated_groups_only()
         package_info = self.get_root_package_info(package)
         resources = [self.get_package_info(dependency) for dependency in self.resolve_dependencies(package).packages]
@@ -101,16 +100,17 @@ def render_formula(
             + "\n"
         )
 
-        if not (output := self.option("output")):
-            output = Path.cwd() / f"{package.name}.rb"
-        elif output == "-":
+        if (output := self.option("output")) == "-":
             self.line("Writing template to stdout", verbosity=Verbosity.VERBOSE)
             sys.stdout.write(formula)
             return
+        if not output:
+            output = f"{package.name}.rb"
 
-        self.line(f"Writing template to {output}", verbosity=Verbosity.VERBOSE)
-        output.parent.mkdir(parents=True, exist_ok=True)
-        output.write_text(formula)
+        output_path = Path(output)
+        self.line(f"Writing template to {output_path}", verbosity=Verbosity.VERBOSE)
+        output_path.parent.mkdir(parents=True, exist_ok=True)
+        output_path.write_text(formula)
 
     def _render_resources(self, resources: list[PackageInfo]) -> str:
         template = get_template(RESOURCES_TEMPLATE)
@@ -154,9 +154,10 @@ def get_root_package_info(self, package: Package) -> RootPackageInfo:
         )
 
     def pick_link(self, package: Package) -> Link:
-        for link in self.repo.find_links_for_package(package):
-            if link.is_sdist:
-                return link
+        for repo in self.poetry.pool.repositories:
+            for link in repo.find_links_for_package(package):
+                if link.is_sdist:
+                    return link
         raise RuntimeError(f"No valid link found for {package.name}. ")
 
 

poetry_homebrew_formula/templating.py

@@ -33,6 +33,7 @@ def list_templates(self) -> list[str]:
         ]
     ),
     trim_blocks=True,
+    autoescape=False,
 )
 
 

pyproject.toml

@@ -26,6 +26,7 @@ mypy = "^1.8.0"
 rich-codex = "^1.2.7"
 commitizen = "^3.13.0"
 pre-commit = "^3.6.0"
+ipdb = "^0.13.13"
 
 [tool.poetry.group.tests.dependencies]
 pytest = "^7.4.4"