DEVOPS-50 new files

Author: githubofkrishnadhas

add_a_repo_to_github_app.py

@@ -0,0 +1,46 @@
+import  requests
+import json
+import os
+import argparse
+from dotenv import load_dotenv
+from list_repos_in_org import list_repos_in_github_org
+from list_github_app_matching_name import list_github_apps_in_organization_matching_name
+
+def add_repository_to_github_app(organization:str,matching_repos: list[dict], app_short_list: list[dict] ):
+    """
+    add repos to github app using rest api
+    :return:
+    https://docs.github.com/en/rest/apps/installations?apiVersion=2022-11-28#add-a-repository-to-an-app-installation
+    """
+    # Assuming that the Github app name doesnot contain any special cgharacters and github app name is same as app_slug.
+    print(f'Matching repos are:  {matching_repos}')
+    print(f'GitHub App details are:  {app_short_list}')
+    add_repo_to_github_app_endpoint = f'https://api.github.com/user/installations/{id}/repositories/REPOSITORY_ID
+'
+    headers = {
+        "Accept": "application/vnd.github+json",
+        "Authorization": f"Bearer {os.getenv('GH_TOKEN')}",
+        "X-GitHub-Api-Version": "2022-11-28"
+    }
+
+
+def main():
+    """ to test the script"""
+    load_dotenv()
+    organization = os.getenv('ORGANIZATION')
+    parser = argparse.ArgumentParser(description="Add specific repos matching a string in repo names to a github teams")
+    parser.add_argument("--search_string", required=True, type=str, help="github repo name search string")
+    parser.add_argument("--github_app_name", required=True, type=str, help="github repo name search string")
+    args = parser.parse_args()
+    search_string = args.search_string
+    github_app_name = args.github_app_name
+    # Function call for repo list
+    matching_repos = list_repos_in_github_org(organization, search_string)
+    # Function call for github app details
+    app_short_list = list_github_apps_in_organization_matching_name(organization, github_app_name)
+    # Fucntion call for adding repo to github app
+    add_repository_to_github_app(organization, matching_repos, app_short_list)
+
+
+if __name__ == "__main__":
+    main()

list_github_app_matching_name.py

@@ -0,0 +1,74 @@
+# reference: https://docs.github.com/en/rest/orgs/orgs?apiVersion=2022-11-28#list-app-installations-for-an-organization
+import  requests
+import json
+import argparse
+import os
+from dotenv import load_dotenv
+
+def list_github_apps_in_organization_matching_name(organization: str, github_app_name: str):
+    """
+    list GitHub apps in the orgaization
+    :param organization:
+    :return:
+ reference: https://docs.github.com/en/rest/orgs/orgs?apiVersion=2022-11-28#list-app-installations-for-an-organization
+    """
+
+    list_github_app_endpoint = f'https://api.github.com/orgs/{organization}/installations'
+    headers = {
+        "Accept": "application/vnd.github+json",
+        "Authorization": f"Bearer {os.getenv('GH_TOKEN')}",
+        "X-GitHub-Api-Version": "2022-11-28"
+    }
+
+    response = requests.get(list_github_app_endpoint, headers=headers)
+    response_json = response.json()
+    response_code = response.status_code
+
+    if response_code == 200:
+        print(f'Listing GitHub Apps in {organization} organization successful')
+    else:
+        print(f'Listing GitHub Apps in {organization} failed')
+
+    total_num_of_github_app_installations = response_json['total_count']
+    print(f'Found {total_num_of_github_app_installations} GitHub App in {organization} organization')
+
+    # Extract the list of dictionaries
+    list_of_installations = response_json["installations"]
+
+    # Convert list of dictionaries to JSON string
+    json_data = json.dumps(list_of_installations, indent=4)
+
+    # Write JSON string to a file
+    with open("list_of_github_installations.json", "w") as file:
+        file.write(json_data)
+
+    app_short_list = []
+    for app in list_of_installations:
+        app_dict = {}
+        if app['app_slug'] == github_app_name:
+            app_dict['id'] = app['id']
+            app_dict['app_id'] = app['app_id']
+            app_dict['app_slug'] = app['app_slug']
+            app_short_list.append(app_dict)
+
+    return app_short_list
+
+def main():
+    """ To test the code """
+    load_dotenv()
+    GH_TOKEN = os.getenv('gh_token')
+    organization = os.getenv('ORGANIZATION')
+    # organization = 'devwithkrishna'
+    parser = argparse.ArgumentParser(description="Get the App installation id from name")
+    parser.add_argument("--github_app_name", required=True, type=str, help="github repo name search string")
+    args = parser.parse_args()
+    github_app_name = args.github_app_name
+    # Function call
+    app_short_list = list_github_apps_in_organization_matching_name(organization, github_app_name)
+
+
+if __name__ == '__main__':
+    main()
+
+
+

list_repos_in_org.py

@@ -0,0 +1,69 @@
+import  requests
+import os
+import argparse
+from dotenv import load_dotenv
+
+def list_repos_in_github_org(orgaization: str, search_string: str):
+    """ get the repo id of repos natching the search string """
+
+    # GitHub endpoint for listing repos under an organization
+    repo_url = f"https://api.github.com/orgs/{orgaization}/repos"
+    # print(f"github api endpoint url {repo_url}")
+
+    headers = {
+        "Accept": "application/vnd.github+json",
+        "Authorization": f"Bearer {os.getenv('GH_TOKEN')}",
+        "X-GitHub-Api-Version": "2022-11-28"
+    }
+
+    # Define pagination parameters
+    per_page = 100  # Number of records per page
+    page = 1  # Initial page number
+    list_of_repo_names = []
+
+    while True:
+        # Add pagination parameters to the URL
+        params = {'per_page': per_page, 'page': page}
+        response = requests.get(repo_url, headers=headers, params=params)
+        response_json = response.json()  ## Github repo details
+
+        # Checking the API status code
+        if response.status_code == 200:
+            print(f"API request successful on {repo_url}")
+            # print(response_json)
+        else:
+            print(f"API request failed with status code {response.status_code}:")
+            # print(response_json)
+            break
+
+        # Get the repo names from the list of dictionaries and add to another list
+        for repo in response_json:
+            repo_dict = {}
+            repo_dict['name'] = repo['full_name']
+            repo_dict['id'] = repo['id']
+            list_of_repo_names.append(repo_dict)
+
+        page += 1  # Move to the next page
+
+
+        # Finding repos starting with search string
+        matching_repos = [repo for repo in list_of_repo_names if repo['name'].startswith(f'{orgaization}/{search_string}')]
+        print(f"Matching repos {search_string} are: {matching_repos}")
+        return matching_repos
+        # Break the loop if no more pages
+        if len(response_json) < per_page:
+            break
+
+
+def main():
+    """ main function to test"""
+    load_dotenv()
+    organization = os.getenv('ORGANIZATION')
+    parser = argparse.ArgumentParser(description="Add specific repos matching a string in repo names to a github teams")
+    parser.add_argument("--search_string", required=True, type=str, help="github repo name search string")
+    args = parser.parse_args()
+    search_string = args.search_string
+    matching_repos = list_repos_in_github_org(organization, search_string)
+
+if __name__ == '__main__':
+    main()