main.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

from github import Github
from github.Issue import Issue
from github.Repository import Repository
import os
import time
import urllib.parse
import codecs

user: Github
my_notes: Repository
cur_time: str


def format_issue(issue: Issue):
    return '- [%s](%s)  %s  \t :alarm_clock:%s \n' % (
        issue.title, issue.html_url, sup('%s :speech_balloon:' % issue.comments), sub(issue.created_at))


def sup(text: str):
    return '<sup>%s</sup>' % text


def sub(text: str):
    return '<sub>%s</sub>' % text


def update_readme_md_file(contents):
    with codecs.open('README.md', 'w', encoding='utf-8') as f:
        f.writelines(contents)
        f.flush()
        f.close()


def login():
    global user
    username = os.environ.get('GITHUB_LOGIN')
    password = os.environ.get('GITHUB_PASSWORD')
    user = Github(username, password)


def get_my_notes():
    global my_notes
    my_notes = user.get_repo('%s/my-notes' % user.get_user().login)


def bundle_summary_section():
    global my_notes
    global cur_time
    global user

    total_label_count = my_notes.get_labels().totalCount
    total_issue_count = my_notes.get_issues().totalCount
    labels_html_url = 'https://github.com/%s/my-notes/labels' % user.get_user().login
    issues_html_url = 'https://github.com/%s/my-notes/issues' % user.get_user().login

    summary_section = '''
# GitHub Issues Blog :tada::tada::tada:
    
> :alarm_clock: 上次更新: %s
    
共 [%s](%s) 个标签, [%s](%s) 篇博文.
''' % (cur_time, total_label_count, labels_html_url, total_issue_count, issues_html_url)

    return summary_section


def bundle_pinned_issues_section():
    global my_notes

    pinned_label = my_notes.get_label(':+1:置顶')
    pinned_issues = my_notes.get_issues(labels=(pinned_label,))

    pinned_issues_section = '\n## 置顶 :thumbsup: \n'

    for issue in pinned_issues:
        pinned_issues_section += format_issue(issue)

    return pinned_issues_section


def format_issue_with_labels(issue: Issue):
    global user

    labels = issue.get_labels()
    labels_str = ''
    if labels:
        labels_str = '\n :label: \t' + sub('|')

    for label in labels:
        labels_str += sub('[%s](https://github.com/%s/my-notes/labels/%s)\t|\t' % (
            label.name, user.get_user().login, urllib.parse.quote(label.name)))

    return '- [%s](%s) %s  \t\t\t :alarm_clock:%s %s\n\n' % (
        issue.title, issue.html_url, sup('%s :speech_balloon:' % issue.comments), sub(issue.created_at), labels_str)


def bundle_new_created_section():
    global my_notes

    new_5_created_issues = my_notes.get_issues()[:5]

    new_created_section = '## 最新 :new: \n'

    for issue in new_5_created_issues:
        new_created_section += format_issue_with_labels(issue)

    return new_created_section


def bundle_list_by_labels_section():
    global my_notes
    global user

    list_by_labels_section = '## 分类  :card_file_box: \n'

    all_labels = my_notes.get_labels()

    for label in all_labels:
        temp = ''
        # 这里的count是用来计算该label下有多少issue的, 按理说应该是取issues_in_label的totalCount, 但是不知道为什么取出来的一直都是
        # 所有的issue数量, 之后再优化.
        count = 0
        issues_in_label = my_notes.get_issues(labels=(label,))
        for issue in issues_in_label:
            temp += format_issue(issue)
            count += 1

        list_by_labels_section += '''
<details>
<summary>%s\t<sup>%s:newspaper:</sup></summary>
%s
</details>
''' % (label.name, count, temp)

    return list_by_labels_section


def bundle_about_me_section():
    global user

    about_me_section = '''
## 关于:boy: 
[<img alt="%s" src="%s" width="233"/>](%s)
**%s**
:round_pushpin: %s
:black_flag: %s
''' % (user.get_user().name, user.get_user().avatar_url, user.get_user().html_url, user.get_user().name,
       user.get_user().location,
       user.get_user().bio)

    return about_me_section


def execute():
    global cur_time
    # common
    cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())

    # 1. login
    login()

    # 2. get my_notes
    get_my_notes()

    # 3. summary section
    summary_section = bundle_summary_section()
    print(summary_section)

    # 4. pinned issues section
    pinned_issues_section = bundle_pinned_issues_section()
    print(pinned_issues_section)

    # 5. new created section
    new_created_section = bundle_new_created_section()
    print(new_created_section)

    # 6. list by labels section
    list_by_labels_section = bundle_list_by_labels_section()
    print(list_by_labels_section)

    # 7. about me section
    about_me_section = bundle_about_me_section()
    print(about_me_section)

    contents = [summary_section, pinned_issues_section, new_created_section, list_by_labels_section, about_me_section]
    update_readme_md_file(contents)

    print('README.md updated successfully!!!')


if __name__ == '__main__':
    execute()