feat: move to ai4os-hub/modules-catalog

ai4os · Feb 28, 2024 · 8487d8c · 8487d8c
1 parent 9e8b998
commit 8487d8c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 10 deletions.
diff --git a/ai4papi/routers/v1/catalog/common.py b/ai4papi/routers/v1/catalog/common.py
@@ -247,8 +247,8 @@ def get_metadata(
                 "license": "",
                 "date_creation": "",
                 "sources": {
-                    "dockerfile_repo": f"https://github.com/deephdc/{item_name}",
-                    "docker_registry_repo": f"deephdc/{item_name}",
+                    "dockerfile_repo": f"https://github.com/ai4os-hub/{item_name}",
+                    "docker_registry_repo": f"ai4os-hub/{item_name}",
                     "code": "",
                 }
             }
@@ -271,7 +271,7 @@ def get_config(
 
 def retrieve_docker_tags(
     image: str,
-    repo: str = 'deephdc',
+    repo: str = 'ai4oshub',
     ):
     """
     Retrieve tags from Dockerhub image

diff --git a/ai4papi/routers/v1/catalog/modules.py b/ai4papi/routers/v1/catalog/modules.py
@@ -14,7 +14,7 @@
 @cached(cache=TTLCache(maxsize=1024, ttl=6*60*60))
 def get_items(
     ):
-    gitmodules_url = "https://raw.githubusercontent.com/deephdc/deep-oc/master/.gitmodules"
+    gitmodules_url = "https://raw.githubusercontent.com/ai4os-hub/modules-catalog/master/.gitmodules"
     r = requests.get(gitmodules_url)
 
     cfg = configparser.ConfigParser()
@@ -24,6 +24,7 @@ def get_items(
     for section in cfg.sections():
         items = dict(cfg.items(section))
         key = items.pop('path').lower()
+        items['url'] = items['url'].replace('.git', '')  # remove `.git`, if present
         modules[key] = items
 
     return modules
@@ -63,8 +64,8 @@ def get_metadata(
             "license": "",
             "date_creation": "",
             "sources": {
-                "dockerfile_repo": f"https://github.com/deephdc/{item_name}",
-                "docker_registry_repo": f"deephdc/{item_name}",
+                "dockerfile_repo": f"https://github.com/ai4os-hub/{item_name}",
+                "docker_registry_repo": f"ai4oshub/{item_name}",
                 "code": "",
             }
         }
@@ -97,7 +98,7 @@ def get_config(
     registry = metadata['sources']['docker_registry_repo']
     repo, image = registry.split('/')[:2]
     if repo not in ['deephdc', 'ai4oshub']:
-        repo = 'deephdc'
+        repo = 'ai4oshub'
 
     # Fill with correct Docker image
     conf["general"]["docker_image"]["value"] = f"{repo}/{image}"

diff --git a/ai4papi/routers/v1/catalog/tools.py b/ai4papi/routers/v1/catalog/tools.py
@@ -63,8 +63,8 @@ def get_metadata(
             "license": "",
             "date_creation": "",
             "sources": {
-                "dockerfile_repo": f"https://github.com/deephdc/{item_name}",
-                "docker_registry_repo": f"deephdc/{item_name}",
+                "dockerfile_repo": f"https://github.com/ai4os-hub/{item_name}",
+                "docker_registry_repo": f"ai4oshub/{item_name}",
                 "code": "",
             }
         }
@@ -95,7 +95,7 @@ def get_config(
     registry = metadata['sources']['docker_registry_repo']
     repo, image = registry.split('/')[:2]
     if repo not in ['deephdc', 'ai4oshub']:
-        repo = 'deephdc'
+        repo = 'ai4oshub'
 
     # Fill with correct Docker image
     conf["general"]["docker_image"]["value"] = f"{repo}/{image}"