gcp-platforms-auto 0.6.4__py3-none-any.whl → 0.6.5__py3-none-any.whl

gcp_platforms_auto/git.py

@@ -87,7 +87,7 @@ def git_push(repo, commit_message, files_to_push=None, branch="main"):
         logger.exception(f"Error pushing to Git: {str(e)}")
         raise e
 
-def create_repo(org, repo, token, github_enterprise="github.com"):
+def create_repo(org, repo, token, github_enterprise="github.com", branches=None):
     try:
         logger.info("Creating repository...")
         is_github_cloud = github_enterprise is "github.com"
@@ -115,6 +115,25 @@ def create_repo(org, repo, token, github_enterprise="github.com"):
             logger.error(f"Failed to create repository: {res.status_code} {res.text}")
             res.raise_for_status()
 
+        if branches:
+            main_ref_url = f"{api_base}/repos/{org}/{repo}/git/refs/heads/main"
+            res = requests.get(main_ref_url, headers=headers)
+            res.raise_for_status()
+            sha = res.json()["object"]["sha"]
+
+            for br in branches:
+                if br == "main":
+                    continue
+                ref_url = f"{api_base}/repos/{org}/{repo}/git/refs"
+                data = {"ref": f"refs/heads/{br}", "sha": sha}
+                r = requests.post(ref_url, headers=headers, json=data)
+                if r.status_code == 201:
+                    logger.info(f"Created branch {br}")
+                elif r.status_code == 422 and "Reference already exists" in r.text:
+                    logger.warning(f"Branch {br} already exists, skipping")
+                else:
+                    logger.error(f"Failed to create branch {br}: {r.status_code} {r.text}")
+
     except Exception as e:
         logger.exception(f"Error creating repository: {str(e)}")
         raise e

{gcp_platforms_auto-0.6.4.dist-info → gcp_platforms_auto-0.6.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gcp_platforms_auto
-Version: 0.6.4
+Version: 0.6.5
 Summary: A brief description of your package
 Author-email: ofir4858 <ofirshasha10@gmail.com>
 License: MIT

gcp_platforms_auto-0.6.5.dist-info/RECORD

@@ -0,0 +1,6 @@
+gcp_platforms_auto/__init__.py,sha256=7CmPCQxUqAvZTHKCREhL2b4MKD8Pn7FnKZUcMxYC09g,92
+gcp_platforms_auto/git.py,sha256=9gn4voiUtxGbFXKlICR80jWt5L1x7V0N6vntqs_x3Jw,5106
+gcp_platforms_auto-0.6.5.dist-info/METADATA,sha256=7fl0_0Lje7E7_G_f8BxJEPc-ya7OQbDkwygPIkXt2S0,494
+gcp_platforms_auto-0.6.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+gcp_platforms_auto-0.6.5.dist-info/top_level.txt,sha256=4q-ofPMmvBaTnIbAzs-Wp_OwheAVxxmJ1fW9vl3-kyE,19
+gcp_platforms_auto-0.6.5.dist-info/RECORD,,

gcp_platforms_auto-0.6.4.dist-info/RECORD

@@ -1,6 +0,0 @@
-gcp_platforms_auto/__init__.py,sha256=7CmPCQxUqAvZTHKCREhL2b4MKD8Pn7FnKZUcMxYC09g,92
-gcp_platforms_auto/git.py,sha256=5eh4xqPwrjdYEDhK3mPM0G6-mQsvm6dhZ2kD80yoxI8,4178
-gcp_platforms_auto-0.6.4.dist-info/METADATA,sha256=fe_Amei_SerPTmc6sDwP74LZh3MCoq1lRC-cuojFyg4,494
-gcp_platforms_auto-0.6.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gcp_platforms_auto-0.6.4.dist-info/top_level.txt,sha256=4q-ofPMmvBaTnIbAzs-Wp_OwheAVxxmJ1fW9vl3-kyE,19
-gcp_platforms_auto-0.6.4.dist-info/RECORD,,