get_repo.py

import os
import sys
from github import Github
from github import Auth
from github.GithubException import UnknownObjectException, GithubException

lsts = []

reponaming = " Listing contents of   "
countrepolen = " Listing repos of  "
warninglen = " Warning!: Repository not found! "

maindir = os.path.dirname(os.path.abspath(__file__))

# Read the token from the file
try:
    with open(f"{maindir}/mykey.txt", "r") as file:
        key = file.read().strip()
except FileNotFoundError:
    print(
        "\033[93mError: Token file 'mykey.txt' not found.\033[0m"
    )  # Yellow text for error
    sys.exit(1)

auth = Auth.Token(key)
g = Github(auth=auth)
user = g.get_user()

# Color codes
GREEN = "\033[92m"
YELLOW = "\033[93m"
RESET = "\033[0m"
BLUE = "\033[94m"


def ls_inside_repo(lsts):

    for con in lsts:
        filecnt = 0
        dircnt = 0
        correct = True
        try:
            mycontent = g.get_user().get_repo(con)
            print()
            print(f"{BLUE}*{'='*(len(reponaming)+len(con)+1)}*{RESET}")
            print(f"{BLUE}| Listing contents of   {con} |{RESET}")
            print(f"{BLUE}*{'-'*(len(reponaming)+len(con)+1)}*{RESET}")
            print()
            contents = mycontent.get_contents("")
            if not contents:
                print(f"{YELLOW} Warning: Repository '{con}' is empty.{RESET}")
                print(f"{YELLOW}*{'-'*(len(warninglen)+len(con))}*{RESET}")
            while contents:
                file_content = contents.pop(0)
                if file_content.type == "dir":
                    dircnt += 1
                    print(f"{BLUE}      {file_content.path}/ {RESET}")
                else:
                    print(f"{GREEN}      {file_content.path} {RESET}")
                    filecnt += 1
        except UnknownObjectException:
            correct = False
            print(f"\n{YELLOW} Warning!: Repository '{con}' not found! {RESET}")
            print(f"{YELLOW}*{'-'*(len(warninglen)+len(con)+1)}*{RESET}")
        except GithubException as e:
            correct = False
            if e.status == 404 and "empty" in e.data.get("message", "").lower():
                print(f"\n{YELLOW} Warning!: Repository '{con}' is empty!{RESET}")
                print(f"{YELLOW}*{'-'*(len(warninglen)+len(con))}*{RESET}")
            else:
                print(
                    f"\n{YELLOW} Warning!: An unexpected error occurred while accessing '{con}' - {e}{RESET}"
                )
                print(f"{YELLOW}*{'-'*(len(warninglen)+len(con)+1)}*{RESET}")
        except Exception as e:
            correct = False
            print(
                f"\n{YELLOW} Warning!: An unexpected error occurred while accessing '{con}' - {e}{RESET}"
            )
            print(f"{YELLOW}*{'-'*(len(warninglen)+len(con)+1)}*{RESET}")
        if correct:
            print()
            print(f"{BLUE}      Dir: {dircnt} {GREEN}  File: {filecnt}")
    print()


def ls_repos():
    try:
        myrepos = g.get_user().get_repos()
        print()
        print(f"{BLUE}*{'='*(len(countrepolen)+len(user.login)+1)}*{RESET}")
        print(f"{BLUE}| Listing repos of  {user.login} |{RESET}")
        print(f"{BLUE}*{'-'*(len(countrepolen)+len(user.login)+1)}*{RESET}")
        print()

        # Set to track repositories (ignoring forks)
        repo_names = set()
        cnt = 0

        for repo in myrepos:
            if repo.name not in repo_names:  # Only count non-fork repositories
                cnt += 1
                print(f"{BLUE}     {repo.name}{RESET}")
            repo_names.add(repo.name)

        # Printing Total Repos in bold blue color
        print()
        print(f"{BLUE}     \033[1mTotal Repos: {cnt}{RESET}")
        print()

    except Exception as e:
        print(f"{YELLOW}Error: Unable to list repositories - {e}{RESET}")


# Main logic
lsts = sys.argv[1:]

if len(lsts) > 0:
    ls_inside_repo(lsts)
else:
    ls_repos()