sync base branch with master (demisto#8536)

avidan-H · web-flow · commit 210c44d6da69 · 2020-08-18T19:36:14.000+03:00
diff --git a/.github/workflows/sync-contribution-base-branch.yml b/.github/workflows/sync-contribution-base-branch.yml
@@ -0,0 +1,29 @@
+name: Sync contributor base branch
+on:
+  schedule:
+    # runs at minute 0 of the 0th and 12th hour every day.
+    - cron:  '0 0,12 * * *'
+
+jobs:
+  sync_contributor_base_branch:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v2
+      - name: Setup Python
+        uses: actions/setup-python@v2
+        with:
+          python-version: '3.8'
+      - name: Install Python Dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install pipenv
+      - name: Sync Base Branch to Master
+        env:
+          CONTENTBOT_GH_ADMIN_TOKEN: ${{ secrets.CONTENTBOT_GH_ADMIN_TOKEN }}
+        run: |
+          echo "Updating contribution base branches (contrib/*)"
+          cd Utils/contribution_sync
+          pipenv install --dev
+          pipenv run ./sync_contrib_base.py
+          echo "Finished updating base branches"
diff --git a/Utils/contribution_sync/Pipfile b/Utils/contribution_sync/Pipfile
@@ -0,0 +1,13 @@
+[[source]]
+name = "pypi"
+url = "https://pypi.org/simple"
+verify_ssl = true
+
+[dev-packages]
+
+[packages]
+pygithub = "*"
+requests = "*"
+
+[requires]
+python_version = "3.8"
diff --git a/Utils/contribution_sync/Pipfile.lock b/Utils/contribution_sync/Pipfile.lock
diff --git a/Utils/contribution_sync/sync_contrib_base.py b/Utils/contribution_sync/sync_contrib_base.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import urllib3
+from github import Github, enable_console_debug_logging
+from github.Repository import Repository
+from typing import List
+
+
+urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+
+def get_master_commit_sha(repo: Repository) -> str:  # noqa: E999
+    '''Return the sha commit of the master branch
+
+    Args:
+        repo (Repository): The repository whose master branch will be queried
+
+    Returns:
+        (str): The commit sha of the master branch's HEAD
+    '''
+    branch_data = repo.get_branch('master')
+    commit_sha = branch_data.commit.sha
+    return commit_sha
+
+
+def get_branch_names_with_contrib(repo: Repository) -> List[str]:  # noqa: E999
+    '''Return the list of branches that have the prefix of "contrib/" and that are base branches of open PRs
+
+    Args:
+        repo (Repository): The repository whose branches will be searched and listed
+
+    Returns:
+        (List[str]): List of branch names that have the "contrib/" prefix and are base branches of open PRs
+    '''
+    branch_names = []
+    for branch in repo.get_branches():
+        if branch.name.startswith('contrib/'):
+            prs_with_branch_as_base = repo.get_pulls(state='OPEN', base=branch.name)
+            if prs_with_branch_as_base.totalCount >= 1:
+                branch_names.append(branch.name)
+    return branch_names
+
+
+def main():
+    debug_mode = len(sys.argv) >= 2 and 'debug' in sys.argv[1].casefold
+    if debug_mode:
+        enable_console_debug_logging()
+    gh = Github(os.getenv('CONTENTBOT_GH_ADMIN_TOKEN'), verify=False)
+    organization = 'demisto'
+    repo = 'content'
+    content_repo = gh.get_repo(f'{organization}/{repo}')
+
+    master_sha = get_master_commit_sha(content_repo)
+    contrib_base_branches = get_branch_names_with_contrib(content_repo)
+    for branch_name in contrib_base_branches:
+        git_ref = content_repo.get_git_ref(f'heads/{branch_name}')
+        print(f'Updating branch "{branch_name}" to sha "{master_sha}"')
+        git_ref.edit(master_sha, force=True)
+
+    if debug_mode:
+        print(f'{contrib_base_branches=}')
+
+
+if __name__ == "__main__":
+    main()
