diff --git a/harvester/harvester.py b/harvester/harvester.py index 24f19c7..06ee8d9 100644 --- a/harvester/harvester.py +++ b/harvester/harvester.py @@ -15,9 +15,12 @@ def sanitize_name(name: str) -> str: """Sanitize a name to be filesystem-friendly.""" return name.lower().replace(" ", "-").replace("_", "-") -def harvest_readmes(organization: str, branch: list[str | None] = ["main", "dev", None], projects: list[str] = [], output_path: str = "reference") -> None: +def harvest_readmes(organization: str | Organization, branch: list[str | None] = ["main", "dev", None], projects: list[str] = [], output_path: str = "reference") -> None: """Harvest README files from repositories.""" - org = Organization("https://dev.azure.com/" + organization, token=get_token()) + if isinstance(organization, str): + org = Organization("https://dev.azure.com/" + organization, token=get_token()) + else: + org = organization if projects: # Target specific projects