aap-utils 0.1__tar.gz → 0.1.2__tar.gz

{aap_utils-0.1 → aap_utils-0.1.2}/PKG-INFO

@@ -1,10 +1,15 @@
 Metadata-Version: 2.1
 Name: aap_utils
-Version: 0.1
+Version: 0.1.2
 Summary: A Python library for registering an IP address for AAP
+Home-page: UNKNOWN
 Author: Minh Dang
 Author-email: danghoangminh86@gmail.com
+License: UNKNOWN
+Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Dist: requests
+
+UNKNOWN
+

{aap_utils-0.1 → aap_utils-0.1.2}/README.md

@@ -4,13 +4,13 @@ A Python library for registering an IP address on a specific server.
 
 ## Installation
 
-To install AAP lib from PyPI, run: pip install pmcd_aap
+To install AAP lib from PyPI, run: pip install aap_utils
 
 
 ## Usage
 
 ```python
-from aap import register_demo
+from aap_utils import register_demo
 
 register_demo(debug=True)
 

aap_utils-0.1.2/aap_utils/register.py

@@ -0,0 +1,55 @@
+import socket
+import requests
+import time
+import os
+
+def register_demo(debug=False):
+    for attempt in range(10):
+        try:
+            local_ip = socket.gethostbyname(socket.gethostname())
+            response = requests.post(
+                "http://hocmay-svc/aiplatform-api/demo/register/", 
+                json={"demo_ip": local_ip}, 
+                headers={"Content-Type": "application/json"}
+            )
+            if response.ok:
+                url, namespace = create_url()
+                # print(f"The demo service is at {namespace}.your_custorm_domain/demo/. Eg: {url}")
+                print(f"The demo service is at {url}")
+                if debug:
+                    print("IP registered successfully.")
+                return True
+            else:
+                if debug:
+                    print(f"Attempt {attempt + 1}: Failed: {response.text}")
+        except (socket.error, requests.exceptions.RequestException) as e:
+            if debug:  # Print errors only if debug is True
+                print(f"Attempt {attempt + 1}: Error: {e}")
+        
+        time.sleep(2)  # Wait for 2 seconds before retrying
+
+    if debug:
+        print("Failed to register IP after 10 attempts.")    
+    return False
+
+
+
+def create_url(domain_name = "aiplatform.vcntt.tech"):
+    # Define the static domain name    
+    
+    # Read the KERNEL_NAMESPACE environment variable
+    kernel_namespace = os.getenv("KERNEL_NAMESPACE")
+    
+    # Validate the environment variable
+    if not kernel_namespace:
+        kernel_namespace="undefined"
+    
+    # Replace 'machinelearning' with 'appmachinepublic' in KERNEL_NAMESPACE
+    if kernel_namespace.startswith("machinelearning"):
+        transformed_namespace = kernel_namespace.replace("machinelearning", "appmachinepublic", 1)
+    else:
+        raise ValueError(f"The KERNEL_NAMESPACE '{kernel_namespace}' does not follow the expected format.")
+    
+    # Create the final URL
+    url = f"https://{transformed_namespace}.{domain_name}/demo/"
+    return url, transformed_namespace

{aap_utils-0.1 → aap_utils-0.1.2}/aap_utils.egg-info/PKG-INFO

@@ -1,10 +1,15 @@
 Metadata-Version: 2.1
-Name: aap_utils
-Version: 0.1
+Name: aap-utils
+Version: 0.1.2
 Summary: A Python library for registering an IP address for AAP
+Home-page: UNKNOWN
 Author: Minh Dang
 Author-email: danghoangminh86@gmail.com
+License: UNKNOWN
+Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
-Requires-Dist: requests
+
+UNKNOWN
+

{aap_utils-0.1 → aap_utils-0.1.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aap_utils",  # The name of your library
-    version="0.1",
+    version="0.1.2",
     description="A Python library for registering an IP address for AAP",
     author="Minh Dang",
     author_email="danghoangminh86@gmail.com",

aap_utils-0.1/aap_utils/register.py

@@ -1,29 +0,0 @@
-import socket
-import requests
-import time
-
-def register_demo(debug=False):
-    for attempt in range(10):
-        try:
-            local_ip = socket.gethostbyname(socket.gethostname())
-            response = requests.post(
-                "http://hocmay-svc/aiplatform-api/demo/register/", 
-                json={"demo_ip": local_ip}, 
-                headers={"Content-Type": "application/json"}
-            )
-            if response.ok:
-                if debug:
-                    print("IP registered successfully.")
-                return True
-            else:
-                if debug:
-                    print(f"Attempt {attempt + 1}: Failed: {response.text}")
-        except (socket.error, requests.exceptions.RequestException) as e:
-            if debug:  # Print errors only if debug is True
-                print(f"Attempt {attempt + 1}: Error: {e}")
-        
-        time.sleep(2)  # Wait for 2 seconds before retrying
-
-    if debug:
-        print("Failed to register IP after 10 attempts.")
-    return False