poly-hammer-utils 0.0.24__py3-none-any.whl → 0.0.25__py3-none-any.whl

poly_hammer_utils/constants.py

@@ -1,5 +1,8 @@
+import os
 from pathlib import Path
 
 RESOURCES_FOLDER = Path(__file__).parent / 'resources'
 
-BLENDER_STARTUP_SCRIPT = RESOURCES_FOLDER / 'scripts' / 'blender' / 'startup.py'
+BLENDER_STARTUP_SCRIPT = RESOURCES_FOLDER / 'scripts' / 'blender' / 'startup.py'
+
+ENVIRONMENT = os.environ.get('ENVIRONMENT', 'staging')

poly_hammer_utils/extension/meta_human_dna.py

@@ -3,11 +3,18 @@ import logging
 import zipfile
 import tempfile
 from pathlib import Path
+from poly_hammer_utils.constants import ENVIRONMENT
 from poly_hammer_utils.utilities import download_and_unzip_to_folder, download_release_file
 from poly_hammer_utils.github.release import GitHubRelease
 from poly_hammer_utils.extension.packager import package_extension, get_addon_version
 from poly_hammer_utils.extension.server import update_extension_index, sync_extensions_from_s3
 
+BLENDER_EXTENSION_SERVER_S3_BUCKET = 'poly-hammer-portal-staging-app-data'
+if ENVIRONMENT == 'production':
+    BLENDER_EXTENSION_SERVER_S3_BUCKET = 'poly-hammer-portal-production-app-data'
+
+BLENDER_EXTENSION_SERVER_S3_FOLDER = 'products/blender-extensions/meta_human_dna'
+
 logger = logging.getLogger(__name__)
 
 PREFIX = 'meta_human_dna_core.'
@@ -122,8 +129,8 @@ def create_release(addon_folder: Path, releases_folder: Path):
     # First, sync existing extensions from S3
     sync_extensions_from_s3(
         repo_folder=releases_folder,
-        bucket=os.environ['BLENDER_EXTENSION_SERVER_S3_BUCKET'],
-        s3_prefix=os.environ['BLENDER_EXTENSION_SERVER_S3_FOLDER'],
+        bucket=BLENDER_EXTENSION_SERVER_S3_BUCKET,
+        s3_prefix=BLENDER_EXTENSION_SERVER_S3_FOLDER,
     )
 
     # Create the new .zip files for the addon's various platforms
@@ -168,8 +175,8 @@ def create_release(addon_folder: Path, releases_folder: Path):
     # Generate the updated extension index and upload it and the addons .zip file to S3
     update_extension_index(
         repo_folder=releases_folder,
-        bucket=os.environ['BLENDER_EXTENSION_SERVER_S3_BUCKET'],
-        s3_folder=os.environ['BLENDER_EXTENSION_SERVER_S3_FOLDER'],
+        bucket=BLENDER_EXTENSION_SERVER_S3_BUCKET,
+        s3_folder=BLENDER_EXTENSION_SERVER_S3_FOLDER,
         blender_version=os.environ['BLENDER_VERSION'],
         docker=True
     )

poly_hammer_utils/extension/server.py

@@ -2,7 +2,9 @@ import os
 import sys
 import logging
 import boto3
+import httpx
 from pathlib import Path
+from poly_hammer_utils.constants import ENVIRONMENT
 from poly_hammer_utils.utilities import shell, get_blender_executable
 
 logger = logging.getLogger(__name__)
@@ -11,6 +13,11 @@ ACCESS_KEY = os.environ.get('AWS_ACCESS_KEY_ID')
 SECRET_KEY = os.environ.get('AWS_SECRET_ACCESS_KEY')
 REGION = os.environ.get('AWS_REGION')
 
+BASE_URL = 'https://api.portal.staging.polyhammer.com'
+if ENVIRONMENT == 'production':
+    BASE_URL = 'https://api.portal.polyhammer.com'
+
+PORTAL_API_KEY = os.environ.get('PORTAL_API_KEY')
 
 def sync_extensions_from_s3(
         repo_folder: Path,
@@ -164,6 +171,34 @@ def generate_extension_index(
 
     shell(command)
 
+def trigger_poly_hammer_portal_extension_index_refresh() -> httpx.Response:
+    """Trigger sync of all extensions from S3."""
+    logger.info("Triggering extension sync...")
+    
+    url = f"{BASE_URL}/api/v1/admin/extensions/sync"
+    headers = {"X-Admin-API-Key": PORTAL_API_KEY}
+
+    response = httpx.post(url, headers=headers, timeout=60.0)
+
+    if response.status_code == 200:
+        data = response.json()
+        logger.info("Sync complete!")
+        logger.info(f"Extensions processed: {data['extensions_processed']}")
+        logger.info(f"Platforms synced: {data['total_platforms_synced']}")
+        
+        if data.get("errors"):
+            logger.error(f"Errors: {data['errors']}")
+        
+        if data.get("details"):
+            logger.info("Extension details:")
+            for ext in data["details"]:
+                logger.info(f"- {ext['extension_id']}: {ext['versions_synced']} versions, {ext['platforms_synced']} platforms")
+    else:
+        logger.error(f"Failed with status {response.status_code}")
+        logger.error(f"Response: {response.text}")
+
+    return response
+
 def update_extension_index(
         repo_folder: Path, 
         bucket: str,
@@ -182,17 +217,19 @@ def update_extension_index(
         blender_version (str, optional): Blender version to use. Defaults to '4.5'.
         docker (bool, optional): Whether to use Docker for index generation. Defaults to False.
     """
-    # Step 2: Generate the extension index from all local .zip files
+    # Step 1: Generate the extension index from all local .zip files
     generate_extension_index(
         repo_folder=repo_folder,
         blender_version=blender_version,
         docker=docker,
     )
 
-    # Step 3: Upload new/modified .zip files and the updated index.json back to S3
+    # Step 2: Upload new/modified .zip files and the updated index.json back to S3
     upload_extensions_to_s3(
         repo_folder=repo_folder,
         bucket=bucket,
         s3_prefix=s3_folder,
     )
 
+    # Step 3: Trigger the Poly Hammer Portal to refresh its extension index
+    trigger_poly_hammer_portal_extension_index_refresh()

{poly_hammer_utils-0.0.24.dist-info → poly_hammer_utils-0.0.25.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: poly-hammer-utils
-Version: 0.0.24
+Version: 0.0.25
 Summary: 
 Author: Poly Hammer
 Author-email: info@polyhammer.com
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: PyGithub (==2.5.0)
 Requires-Dist: boto3 (>=1.42.35,<2.0.0)
+Requires-Dist: httpx (>=0.28.1,<0.29.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)
 Requires-Dist: requirements-parser (>=0.11.0,<0.12.0)
 Requires-Dist: tomlkit (>=0.14.0,<0.15.0)

{poly_hammer_utils-0.0.24.dist-info → poly_hammer_utils-0.0.25.dist-info}/RECORD

@@ -1,16 +1,16 @@
 poly_hammer_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 poly_hammer_utils/addon/packager.py,sha256=apoZhQpd2QeRucFq53u-L1FPmN4MYuiNrjTETAE9NQg,8435
 poly_hammer_utils/addon/release.py,sha256=EZirS8_c9S1OEZoOHStgciu_C7vMxBc_-O5P1fM9Qu4,9074
-poly_hammer_utils/constants.py,sha256=Q5cCA6H-AJYWdbndCwk3fIoBPHx_9NTgjvY7p-e0grA,162
-poly_hammer_utils/extension/meta_human_dna.py,sha256=PDwjC33fZWlWeNh7QxDfvf1MKUd4OoW8yD-aMkhSyuo,7471
+poly_hammer_utils/constants.py,sha256=4XGiwn7ip50V1zvmpVD9YZYmE6wPhj2Ark2AXQuJvSc,228
+poly_hammer_utils/extension/meta_human_dna.py,sha256=HP5BGYhxwjlPBAPnJxqtKjVI5uWyK3ZTFFLDJ3jNoaw,7740
 poly_hammer_utils/extension/packager.py,sha256=btYzyU_fa2Qaa6R9eZSojQ5uy3nQ0kDxCf49LnTZrHM,2115
-poly_hammer_utils/extension/server.py,sha256=G0hLs5DuzEcf-exMtMPy57Gi3JQmAxx5tPn58NLGCmI,6584
+poly_hammer_utils/extension/server.py,sha256=vuDMqEehgFiFiUy-R28oKzijHo5wIJzW0l66wwPU6I8,8072
 poly_hammer_utils/github/release.py,sha256=P4vrNvdgWuzJI0mRye09BP6vL5b7kQ5fpuDDnEqJbwg,9951
 poly_hammer_utils/helpers.py,sha256=f_2fNrydsbIvwD2LuoBj2gjETMP7dFzJtq1_7WMBwNI,1655
 poly_hammer_utils/launch.py,sha256=Nc9FlFuGtMSjgSLsk6MMMj-FECsOGtfUQlZaUZt_rGU,469
 poly_hammer_utils/resources/scripts/blender/startup.py,sha256=eUd0VwAHjTwIscVKCbk_0-F7p0jgIEsVYdzpIjlzOK0,2282
 poly_hammer_utils/resources/scripts/unreal/init_unreal.py,sha256=1-d9IU8ZSAIS3MUANrsGx4ZmqNJ5f8S2k8XJVg0Bghs,693
 poly_hammer_utils/utilities.py,sha256=7dmZo02HXbWhfmwDHkgVUE_gyq4e-gqb4OdBc2bokbs,6491
-poly_hammer_utils-0.0.24.dist-info/METADATA,sha256=Wn_uSbbeTk60PtG28IGtPt0b8CaOKPu38RFgzEavu_c,894
-poly_hammer_utils-0.0.24.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-poly_hammer_utils-0.0.24.dist-info/RECORD,,
+poly_hammer_utils-0.0.25.dist-info/METADATA,sha256=4qTBctRnioUIPFOWKdlMakIu8peOj5qzyrzs1FQoJo4,934
+poly_hammer_utils-0.0.25.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+poly_hammer_utils-0.0.25.dist-info/RECORD,,