Skip to content

Commit

Permalink
feat(stubs): make StubRepository.resolve_package return `StubReposi…
Browse files Browse the repository at this point in the history
…toryPackage`

Signed-off-by: Braden Mars <bradenmars@bradenmars.me>
  • Loading branch information
BradenM committed Dec 12, 2022
1 parent 1b6a46a commit 24ef2fa
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions micropy/stubs/repo.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ def latest_for_package(
return versions[0]
return max(versions, key=lambda x: x.package.version)

def resolve_package(self, name: str) -> str:
def resolve_package(self, name: str) -> StubRepositoryPackage:
"""Resolve a package name to a package path.
Args:
Expand All @@ -129,12 +129,12 @@ def resolve_package(self, name: str) -> str:
StubNotFound: When package cannot be resolved.
"""
for package in self.search(name):
for package in self.search(str(name)):
if package.match_exact(name) or package.match_exact(
"/".join([package.repo_name, name])
):
return package.url
return package
latest = self.latest_for_package(package)
if latest and latest.name == name:
return latest.url
return latest
raise exc.StubNotFound(name)

0 comments on commit 24ef2fa

Please sign in to comment.