main.py

from github import Github, GithubException

from constants import TOKEN, LABELS


def create_or_update_labels(owner, repo):
    print('Creating or updating labels!!')
    g = Github(TOKEN)

    repo = g.get_repo(f"{owner}/{repo}")

    for label in LABELS:
        try:
            repo.create_label(**label)
        except GithubException as e:
            # print(e)
            repo_label = repo.get_label(label.get('name'))
            repo_label.edit(
                name=repo_label.name,
                color=label.get('color'),
                description=label.get('description')
            )
        except Exception as e:
            # print(e)
            raise e


def delete_existing_labels(owner, repo):
    print('Deleting old labels!!!')
    g = Github(TOKEN)

    repo = g.get_repo(f"{owner}/{repo}")
    for label in repo.get_labels():
        label.delete()