scraper.py

# coding:utf-8

import datetime
import codecs
import requests
import os
import time
from pyquery import PyQuery as pq


# 实现git提交
def git_add_commit_push(date, filename):
    cmd_git_add = 'git add {filename}'.format(filename=filename)
    cmd_git_commit = 'git commit -m "{date}"'.format(date=date)
    cmd_git_push = 'git push -u origin master'

    os.system(cmd_git_add)
    os.system(cmd_git_commit)
    os.system(cmd_git_push)


def createMarkdown(date, filename):
    with open(filename, 'w') as f:
        f.write("## " + date + "\n")


def scrape(language, filename):
    HEADERS = {
        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10.7; rv:11.0) Gecko/20100101 Firefox/11.0',
        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
        'Accept-Encoding': 'gzip,deflate,sdch',
        'Accept-Language': 'zh-CN,zh;q=0.8'
    }

    url = 'https://github.com/trending/{language}'.format(language=language)
    r = requests.get(url, headers=HEADERS)
    assert r.status_code == 200

    d = pq(r.content)
    items = d('ol.repo-list li')

    # codecs to solve the problem utf-8 codec like chinese
    with codecs.open(filename, 'a', 'utf-8') as f:
        if language == '':
            language = 'general'

        f.write('\n### {language}\n'.format(language=language))

        for item in items:
            i = pq(item)
            title = i('h3 a').text()
            description = i('p.col-9').text()
            url = i('h3 a').attr('href')
            url = 'https://github.com' + url
            f.write(u'* [{title}]({url}):{description}\n'.format(
                title=title, url=url, description=description
            ))


def job():
    strdate = datetime.datetime.now().strftime('%Y-%m-%d')
    filename = '{date}.md'.format(date=strdate)

    # create markdown file
    createMarkdown(strdate, filename)

    print('start scraping')
    # write markdown
    # today's trending
    scrape('', filename)
    scrape('python', filename)
    scrape('javascript', filename)
    scrape('swift', filename)
    scrape('java', filename)
    scrape('go', filename)

    print('start git push')
    # git add commit push
    git_add_commit_push(strdate, filename)
    print('finish job in ' + strdate)

if __name__ == '__main__':
    while True:
        job()
        time.sleep(24 * 60 * 60)