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

gcp_platforms_auto/git.py

@@ -63,7 +63,7 @@ def get_repo(path_of_git_repo, branch="main", path_to_local_repo="/tmp/repo"):
         raise e
 
 
-def git_push(repo, commit_message, files_to_push=None):
+def git_push(repo, commit_message, files_to_push=None, branch="main"):
     try:
         logger.info(f"Attempting to push changes to Git with commit message: '{commit_message}'")
         if files_to_push:
@@ -74,20 +74,20 @@ def git_push(repo, commit_message, files_to_push=None):
             repo.git.add(all=True)
         logger.info(f"Committing changes with message: '{commit_message}'")
         repo.index.commit(commit_message)
-        logger.info("Pushing changes to remote 'origin'")
-        push_info = repo.remote(name="origin").push()
+        logger.info(f"Pushing changes to remote 'origin', branch '{branch}'")
+        push_info = repo.remote(name="origin").push(refspec=f"{branch}:{branch}")
         for info in push_info:
-          if info.flags >0:
-            logger.info(f"Git push info: {info.summary}")
-          if info.flags == 0:
-            logger.info(f"Git push info: No update {info.summary}")
+            if info.flags > 0:
+                logger.info(f"Git push info: {info.summary}")
+            elif info.flags == 0:
+                logger.info(f"Git push info: No update {info.summary}")
 
         logger.info("Successfully pushed changes to Git.")
     except Exception as e:
         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.3.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.3
+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.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-gcp_platforms_auto/__init__.py,sha256=7CmPCQxUqAvZTHKCREhL2b4MKD8Pn7FnKZUcMxYC09g,92
-gcp_platforms_auto/git.py,sha256=5slBDPEis29kHisdAaQy9rbV9petduB2sZwgQuD8z0E,4100
-gcp_platforms_auto-0.6.3.dist-info/METADATA,sha256=-g0eLryWuNtlWedUbjYJyOXVjpTbil1FuVZPwmEvtY0,494
-gcp_platforms_auto-0.6.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-gcp_platforms_auto-0.6.3.dist-info/top_level.txt,sha256=4q-ofPMmvBaTnIbAzs-Wp_OwheAVxxmJ1fW9vl3-kyE,19
-gcp_platforms_auto-0.6.3.dist-info/RECORD,,