GitHubのコントリビューション数をグラフ化してみた

先日、ブログのトップページにGitHubの草を表示してみたという記事をあげました。

今回はちょっとした小ネタです。お気づきの方もいるかもしれませんが、PC限定でブログのページ上部にGithubのコントリビューションを表示してみました。俗にいう「草を生やす」というやつです。特に深い意図はありませんが、これがあるだけでぐっと技術系ブログ色が強まるので、遊び半分で組み込んでみました。そこまで難しくないので、その方法を紹介しようと思います。生やすだけなら簡単Githubから情報を引っ張ってくるというと、パッと見は難しそうに見えますが、実は草を生やすだけなら簡単です。上記のGrass-Graphというサービス...

GitHubの草（コントリビューション）は日々の活動が可視化されるので、個人的にはとても良い仕組みだと思っています。しかし、週末くらいしか家でコードを書かない私のような人にとっては、土日ばかりに草が生えて残りはスカスカという悲しい感じに(´・ω・)

そこで、コントリビューションを週単位でまとめて表示する仕組みがあればよいのでは！ということで作ってみることにしました。

どこからデータを取ってくるか？

コントリビューションのデータは、GitHubが公開しているAPIを使えば取得可能です。しかし、サポートしているのが公開リポジトリに対するコントリビューションのみらしく、プライベートでひっそりと開発している人間にはあまり嬉しくありません。

そこで代替として、GitHubのページからデータを拾い上げて集計することにしました。

https://github.com/users/<UserID>/contributionsにアクセスすると以下のようなページが表示されます。UserIDは適宜ご自身のGitHubアカウント名に置き換えてください。このうち、上部にあるグラフにコントリビューション情報が埋め込まれているので、それを拾い集める形としました。

Pythonで実装してみる

軽く調べたところ、データの抽出にはPythonのBeautifulSoupというライブラリを使うと楽とのこと。肝心のページの取得にはurllib、グラフ描画はPillowというライブラリでできることも分かったので、全てPythonで完結できそうです。

ページの取得

ページの取得にはurllibのrequestモジュールを使います。指定したURLから取ってくるだけであれば、こんな簡単に実装できます。

from urllib import request

username = "corgi0901"
url = "https://github.com/users/" + username + "/contributions"
response = request.urlopen(url)

データ抽出

続いて、取得したページから必要なデータを取り出します。上述したコード中のresponseを引数に渡してBeautifulSoupを初期化したあと、検索条件を指定すれば、該当する要素をかき集めることができます。

今回対象としているページでは、コントリビューション情報は「rectかつclassがday」となっている要素です。そして、さらにその要素を紐解くと、data-dateに日付、data-countにコントリビューションの数が入っています。

試しに、各日付のコントリビューション数を表示するプログラムを書くとこんな感じになります。

from urllib import request
from bs4 import BeautifulSoup

# ページを取得
username = "corgi0901"
url = "https://github.com/users/" + username + "/contributions"
response = request.urlopen(url)

# BeautifulSoup初期化
soup = BeautifulSoup(response, "html.parser")
response.close()

# コントリビューション情報を抽出
contributions = soup.find_all("rect", class_="day")

# 各日付毎にコントリビューション数を表示
for day in contributions:
    date = day.get("data-date")
    count = day.get("data-count")
    print(date, ":", count)

実行結果はこんな感じです。過去1年分のコントリビューション数が取得できました。

2019-05-12 : 0
2019-05-13 : 0
2019-05-14 : 0
2019-05-15 : 2
2019-05-16 : 0
～途中省略～
2020-05-07 : 7
2020-05-08 : 0
2020-05-09 : 0
2020-05-10 : 2
2020-05-11 : 0