feat: try archiving when repo deletion fails (#398)

Author: xuezhma

CHANGELOG.md

@@ -18,6 +18,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - update qs to 6.14.1 via npm override to address security vulnerability
 - pin @cdklabs/generative-ai-cdk-constructs to 0.1.311 to fix build compatibility
 - update starlette to 0.50.0 and fastapi to 0.128.0 to address security vulnerabilities
+- updated `sagemaker-templates` to try archiving when repo deletion fails
 
 
 ## v3.1.0

modules/sagemaker/sagemaker-templates/common/code_repo_construct.py

@@ -252,9 +252,38 @@ def lambda_handler(event, context):
                 if response_code == 204:
                     print(f"Repository {repo_name} deleted successfully")
         except urllib.error.HTTPError as e:
-            print(f"HTTP Error: {e.code} - {e.reason}")
-            print(e.read().decode())
-            raise
+            if e.code == 404:
+                # Repository doesn't exist - already deleted
+                print(f"Repository {repo_name} not found - treating as successful deletion")
+                cfnresponse.send(event, context, cfnresponse.SUCCESS, {})
+                return
+            elif e.code == 403:
+                # Forbidden - try to archive instead
+                print(f"HTTP Error deleting repository: 403 - Forbidden")
+                print(f"Attempting to archive repository instead...")
+                try:
+                    archive_data = json.dumps({'archived': True}).encode('utf-8')
+                    archive_req = urllib.request.Request(
+                        f"https://api.github.com/repos/{github_owner}/{repo_name}",
+                        data=archive_data,
+                        headers=headers,
+                        method='PATCH'
+                    )
+                    with urllib.request.urlopen(archive_req) as archive_response:
+                        if archive_response.getcode() == 200:
+                            print(f"Repository {repo_name} archived successfully")
+                            cfnresponse.send(event, context, cfnresponse.SUCCESS, {})
+                            return
+                except Exception as archive_error:
+                    print(f"Failed to archive repository: {str(archive_error)}")
+                    cfnresponse.send(event, context, cfnresponse.FAILED, {})
+                    return
+            else:
+                # Other errors - fail
+                print(f"HTTP Error deleting repository: {e.code} - {e.reason}")
+                print(e.read().decode())
+                cfnresponse.send(event, context, cfnresponse.FAILED, {})
+                return
 
         cfnresponse.send(event, context, cfnresponse.SUCCESS, {})
     else: