gcp-platforms-auto 0.6.3__tar.gz → 0.6.5__tar.gz

{gcp_platforms_auto-0.6.3 → gcp_platforms_auto-0.6.5}/PKG-INFO

@@ -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.3 → gcp_platforms_auto-0.6.5}/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 → gcp_platforms_auto-0.6.5}/gcp_platforms_auto.egg-info/PKG-INFO

@@ -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.3 → gcp_platforms_auto-0.6.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gcp_platforms_auto"
-version = "0.6.3"
+version = "0.6.5"
 description = "A brief description of your package"
 readme = "README.md"
 authors = [