main.py

# -*- coding: utf-8 -*-
import argparse
from datetime import datetime

from github import Github

from github_daily.config import LABEL_DICT, MY_BLOG_REPO
from github_daily.utils import (
    make_base_issues_comments_str,
    make_blog_issues_str,
    make_cook_issue_table,
    replace_readme_comments,
)


def get_me(user):
    return user.get_user().login


def login(token):
    return Github(token)


def main(github_token, repo_name, issue_number, issue_label_name):
    # issue_number for future use
    u = login(github_token)
    me = get_me(u)
    if issue_number:
        labels = LABEL_DICT.get(issue_label_name)
        if not labels:
            return
        issues = u.get_repo(repo_name).get_issues(labels=labels.get("label_list", []))
        parse_func = make_base_issues_comments_str
        # only Cook now, if one more, refactor it
        if issue_label_name == "Cook":
            parse_func = make_cook_issue_table
        comment_str = parse_func(me, issues)
        comments_name = labels.get("comment_name", "")
    else:
        # from 2021 just for me(yihong0618), if you want to use you can delete the lines below
        # if you are not me change here
        now_year = datetime.now().year
        since = datetime(now_year, 1, 1)
        issues = u.get_repo(MY_BLOG_REPO).get_issues(since=since, creator=me)
        comment_str = make_blog_issues_str(since, issues)
        comments_name = "my_blog"
    replace_readme_comments("README.md", comment_str, comments_name)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("github_token", help="github_token")
    parser.add_argument("repo_name", help="repo_name")
    parser.add_argument(
        "--issue_number", help="issue_number", default=None, required=False
    )
    parser.add_argument(
        "--issue_label_name", help="issue_label_name", default=None, required=False
    )
    options = parser.parse_args()
    main(
        options.github_token,
        options.repo_name,
        options.issue_number,
        options.issue_label_name,
    )