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
from nasa_client import NasaClient
from word_cloud import WordCloudGenerator

user: Github
username: str
ghiblog: Repository
cur_time: str


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


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
    github_repo_env = os.environ.get('GITHUB_REPOSITORY')
    username = github_repo_env[0:github_repo_env.index('/')]
    print("======check username:")
    print(github_repo_env)
    password = os.environ.get('GITHUB_TOKEN')
    user = Github(username, password)


def get_ghiblog():
    global ghiblog
    ghiblog = user.get_repo(os.environ.get('GITHUB_REPOSITORY'))


def bundle_summary_section():
    global ghiblog
    global cur_time
    global user
    global username

    total_label_count = ghiblog.get_labels().totalCount
    total_issue_count = ghiblog.get_issues().totalCount

    pic_of_the_day = NasaClient().get_picture_of_the_day()

    summary_section = '''

<p align='center'>
    <img src="https://badgen.net/badge/labels/{1}"/>
    <img src="https://badgen.net/github/issues/{0}/ghiblog"/>
    <img src="https://badgen.net/badge/last-commit/{2}"/>
    <img src="https://badgen.net/github/forks/{0}/ghiblog"/>
    <img src="https://badgen.net/github/stars/{0}/ghiblog"/>
    <img src="https://badgen.net/github/watchers/{0}/ghiblog"/>
    <img src="https://badgen.net/github/release/{0}/ghiblog"/>
</p>

<p align='center'>
    <a href="https://github.com/jwenjian/visitor-count-badge">
        <img src="https://visitor-badge.glitch.me/badge?page_id=jwenjian.ghiblog"/>
    </a>
</p>

'''.format(username, total_label_count, cur_time)

    return summary_section


def bundle_pinned_issues_section():
    global ghiblog

    pinned_label = ghiblog.get_label(':+1:置顶')
    # pinned_issues = ghiblog.get_issues(labels=(pinned_label,))
    pinned_issues = ghiblog.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, username

    labels = issue.get_labels()
    labels_str = ''

    for label in labels:
        labels_str += '[%s](https://github.com/%s/ghiblog/labels/%s), ' % (
            label.name, username, urllib.parse.quote(label.name))

    # check issue body
    if issue.body is not None:
        if '---' in issue.body:
            body_summary = issue.body[:issue.body.index('---')]
        else:
            body_summary = issue.body[:150]
            # 如果前150个字符中有代码块，则在 150 个字符中重新截取代码块之前的部分作为 summary
            if '```' in body_summary:
                body_summary = body_summary[:body_summary.index('```')]
    else:
        body_summary = ""
        
    return '''
#### [{0}]({1}) {2} \t {3}

:label: : {4}

{5}

[更多>>>]({1})

---

'''.format(issue.title, issue.html_url, sup('%s :speech_balloon:' % issue.comments), issue.created_at, labels_str[:-2],
           body_summary)


def bundle_new_created_section():
    global ghiblog

    new_5_created_issues = ghiblog.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 ghiblog
    global user

    # word cloud
    wordcloud_image_url = WordCloudGenerator(ghiblog).generate()

    list_by_labels_section = """
## 分类  :card_file_box: 

<details open="open">
    <summary>
        <img src="%s" title="词云, 点击展开详细分类" alt="词云， 点击展开详细分类">
        <p align="center">:cloud: 词云 :cloud: <sub>点击词云展开详细分类:point_down: </sub></p>
    </summary>

""" % (wordcloud_image_url,)

    all_labels = ghiblog.get_labels()

    for label in all_labels:
        temp = ''
        # 这里的count是用来计算该label下有多少issue的, 按理说应该是取issues_in_label的totalCount, 但是不知道为什么取出来的一直都是
        # 所有的issue数量, 之后再优化.
        count = 0
        issues_in_label = ghiblog.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)

    list_by_labels_section += """

</details>    
"""
    return list_by_labels_section


def bundle_cover_image_section() -> str:
    global ghiblog
    cover_label = ghiblog.get_label(':framed_picture:封面')
    if cover_label is None:
        return ''
    cover_issues = ghiblog.get_issues(labels=[cover_label])
    if cover_issues is None or cover_issues.totalCount == 0:
        return ''
    comments = cover_issues[0].get_comments()
    if comments is None or comments.totalCount == 0:
        return ''
    c = comments[comments.totalCount - 1]
    img_md = None
    img_desc = ''
    if '---' in c.body:
        img_md = c.body.split('---')[0]
        img_desc = c.body.split('---')[1]
    else:
        img_md = c.body
    if img_md is None:
        return ''
    img_url = img_md[(img_md.index('(') + 1):img_md.index(')')]
    print(img_url)
    return '''

<p align='center'>
<a href='{0}'>
<img src='{1}' width='50%' alt='{2}'>
</a>
</p>
<p align='center'>
<span>{2}</span>
</p>

    '''.format(c.html_url, img_url, img_desc)


def bundle_projects_section() -> str:
    global ghiblog, username
    project_label = ghiblog.get_label('开源')
    if not project_label:
        return ''
    issues = ghiblog.get_issues(labels=[project_label])
    if not issues or issues.totalCount == 0:
        return ''
    content = ''
    for (idx, i) in enumerate(issues):
        content += '''
| [{1}](https://github.com/{0}/{1}) | {2} | ![](https://badgen.net/github/stars/{0}/{1}) ![](https://badgen.net/github/forks/{0}/{1}) ![](https://badgen.net/github/watchers/{0}/{1}) |'''.format(
            username, i.title, i.body)
        if idx == 0:
            content += '\n| --- | --- | --- |'
    return '''
# 开源项目

{}

'''.format(content)


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

    # 1. login
    login()

    # 2. get ghiblog
    get_ghiblog()

    # 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. cover image section
    cover_image_section = bundle_cover_image_section()
    print(cover_image_section)

    # 8. projects section
    projects_section = bundle_projects_section()
    print(projects_section)

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

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

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


if __name__ == '__main__':
    execute()