Add support for ATRIS

Author: jwodder

src/find_datalad_repos/__main__.py

@@ -28,6 +28,14 @@ def set_mode(
     default=logging.INFO,
     help="Set logging level  [default: INFO]",
 )
+@click.option(
+    "--atris",
+    flag_value=RepoHost.ATRIS,
+    type=click.UNPROCESSED,
+    callback=set_mode,
+    expose_value=False,
+    help="Update ATRIS data",
+)
 @click.option(
     "--gin",
     flag_value=RepoHost.GIN,
@@ -96,6 +104,8 @@ def main(log_level: int, regen_readme: bool, mode: set[RepoHost] | None = None)
             reports.extend(
                 record.update_hub_datalad_org(os.environ["HUB_DATALAD_ORG_TOKEN"])
             )
+        if RepoHost.ATRIS in mode:
+            reports.extend(record.update_atris())
         with open(RECORD_FILE, "w") as fp:
             print(record.model_dump_json(indent=4), file=fp)
 

src/find_datalad_repos/core.py

@@ -18,6 +18,7 @@ class RepoHost(Enum):
     GIN = 2
     OSF = 3
     HUB_DATALAD_ORG = 4
+    ATRIS = 5
 
 
 class Updater(ABC, Generic[T, U, S]):

src/find_datalad_repos/diff.py

@@ -63,6 +63,7 @@ def main(
     old_osf_repos = {r.id for r in from_record.osf}
     old_gin_repos = {r.id for r in from_record.gin}
     old_hub_repos = {r.id for r in from_record.hub_datalad_org}
+    old_atris_repos = {r.id for r in from_record.atris}
     new_record = RepoRecord()
     for ghr in to_record.github:
         if ghr.name not in old_github_repos:
@@ -76,6 +77,9 @@ def main(
     for hubr in to_record.hub_datalad_org:
         if hubr.id not in old_hub_repos:
             new_record.hub_datalad_org.append(hubr)
+    for atrisr in to_record.atris:
+        if atrisr.id not in old_atris_repos:
+            new_record.atris.append(atrisr)
     mkreadmes(new_record, filename=readme_file, directory=readme_dir)
 
 

src/find_datalad_repos/gin.py

@@ -55,17 +55,20 @@ def as_table_row(self) -> TableRow:
 
 
 class GINSearcher(Client, Searcher[GINRepo]):
-    def __init__(self, token: str, url: str | None = None) -> None:
+    def __init__(self, token: str | None = None, url: str | None = None) -> None:
         if url is None:
             url = "https://gin.g-node.org"
+        headers = {}
+        if token is not None:
+            # Passing `token` directly to ghreq results in an Authorization
+            # header of "Bearer {token}", which GIN doesn't seem to support.
+            headers["Authorization"] = f"token {token}"
         super().__init__(
             api_url=f"{url}/api/v1",
             user_agent=USER_AGENT,
             accept=None,
             api_version=None,
-            # Passing `token` directly to ghreq results in an Authorization
-            # header of "Bearer {token}", which GIN doesn't seem to support.
-            headers={"Authorization": f"token {token}"},
+            headers=headers,
             # Don't retry on 500's until
             # <https://github.com/G-Node/gogs/issues/148> is resolved
             retry_config=RetryConfig(retry_statuses=range(501, 600)),

src/find_datalad_repos/readmes.py

@@ -37,6 +37,12 @@ def mkreadmes(
         base_url="https://hub.datalad.org",
         header="hub.datalad.org",
     )
+    (atris_block, atris_active, atris_gone) = make_gin_tables(
+        record.atris,
+        Path(directory, "atris"),
+        base_url="https://atris.fz-juelich.de",
+        header="ATRIS",
+    )
     (osf_block, osf_active, osf_gone) = make_osf_tables(record.osf)
     with open(filename, "w") as fp:
         print("# Summary", file=fp)
@@ -62,11 +68,17 @@ def mkreadmes(
             f" [{hub_gone}](#gone-3) gone",
             file=fp,
         )
+        print(
+            f"- [ATRIS](#atris): [{atris_active}](#active-3) active +"
+            f" [{atris_gone}](#gone-4) gone",
+            file=fp,
+        )
         print(file=fp)
         print(github_block, file=fp)
         print(osf_block, file=fp)
         print(gin_block, file=fp)
-        print(hub_block, end="", file=fp)
+        print(hub_block, file=fp)
+        print(atris_block, end="", file=fp)
 
 
 def make_github_tables(

src/find_datalad_repos/record.py

@@ -12,6 +12,7 @@ class RepoRecord(BaseModel):
     osf: list[OSFRepo] = Field(default_factory=list)
     gin: list[GINRepo] = Field(default_factory=list)
     hub_datalad_org: list[GINRepo] = Field(default_factory=list)
+    atris: list[GINRepo] = Field(default_factory=list)
 
     def update_github(self, token: str) -> list[str]:
         return update_collection(self.github, GitHubUpdater, token=token)
@@ -27,6 +28,11 @@ def update_hub_datalad_org(self, token: str) -> list[str]:
             self.hub_datalad_org, GINUpdater, token=token, url="https://hub.datalad.org"
         )
 
+    def update_atris(self) -> list[str]:
+        return update_collection(
+            self.atris, GINUpdater, url="https://atris.fz-juelich.de"
+        )
+
 
 def update_collection(
     collection: list[T], updater_cls: type[Updater[T, U, S]], **searcher_kwargs: Any