check.py

import json
import urllib2
import requests
from datetime import date, timedelta, datetime
from collections import OrderedDict
import sys
import logging

from flask import Flask, render_template, jsonify, request, redirect, url_for
from lxml import html
from lxml import etree
import pygal
from pygal.style import Style
from pygal import Config
import pytz
from pytz import timezone

app = Flask(__name__)

# Configure logging.
app.logger.setLevel(logging.DEBUG)
del app.logger.handlers[:]

handler = logging.StreamHandler(stream=sys.stdout)
handler.setLevel(logging.DEBUG)
handler.formatter = logging.Formatter(
    fmt=u"%(asctime)s level=%(levelname)s %(message)s",
    datefmt="%Y-%m-%dT%H:%M:%SZ",
)
app.logger.addHandler(handler)

start_date = date(2016, 4, 7)
cst = timezone('US/Central')
today = datetime.now(cst).date()
sorted_dict = None

def render_chart(commit_dict):
	style = Style(
		transition='200ms ease-in',
		font_family='googlefont:Josefin+Sans',
		plot_background='transparent',
		background='transparent',
		colors=('#F9A027', '#E8537A')
	)
	config = Config()
	params = {
		'show_y_guides':False,
		'show_x_guides':False,
		'style':style
	}
	config = pygal.Config(no_prefix=True, **params)
	chart = pygal.Line(config, height=400, x_label_rotation=20, range=(0, max([int(commit_num) for commit_num in commit_dict.values()])))
	chart.x_labels = [commit_date.strftime('%b %d').lstrip("0").replace(" 0", " ") for commit_date in commit_dict.keys()]
	chart.add('Commits', [int(commit_num) for commit_num in commit_dict.values()])
	chart.title = 'Commit History from %s' % start_date.strftime('%B %d, %Y').lstrip("0").replace(" 0", " ")
	chart.value_formatter = lambda x: "%.0f" % x
	chart = chart.render()
	unicode_chart=chart.decode('utf-8')
	return unicode_chart

def get_custom_message(streak, commit_dict, committed_today):
	days_left = get_days_left(date(2016, 5, 6), committed_today)
	if today - timedelta(days=streak) <= start_date:
		if committed_today:
			return 'You\'re in great shape overall! Keep going for %s more days after today (through May 6, 2016) and you\'ll get a custom Freetail Hackers Git Challenge shirt!' % (str(days_left)), True
		else:
			return 'You\'re in great shape overall! Keep going for %s more days, including today (through May 6, 2016) and you\'ll get a custom Freetail Hackers Git Challenge shirt!' % (str(days_left)), True
	else:
		start_day = start_date + timedelta(days=1)
		days_left = get_days_left(date(2016, 5, 7), committed_today)
		
		if today - timedelta(days=streak) <= start_date:
			return 'Looks like you started a day late! No worries, we still want you to commit for 30 days, just commit for %s more days (until May 7, 2016) to get your custom Freetail Hackers Git Challenge shirt!' % (str(days_left)), True
		else:
			return 'Our automated check isn\'t able to verify your completion towards a 30 day streak. If you believe this is a mistake, check to make sure all the repositories you committed to are public. If you still think there\'s an error, please reach out to <a href="mailto:hello@freetailhackers.com">hello@freetailhackers.com</a> so we can investigate further.', False

def get_days_left(end_date, committed_today):
	if committed_today:
		return (end_date - today).days
	else:
		return (end_date - today + timedelta(days=1)).days

@app.route('/', endpoint='index')
def index():
	return render_template('index.html')

def get_streak(username, page_tree):
	current_streaks = page_tree.xpath('//*[@id="contributions-calendar"]/div[5]/span[2]/text()')
	return current_streaks

def get_commits(username, streak):
	commit_dict = {}
	day_streak = 0 if int(streak.split()[0]) == 0 else int(streak.split()[0]) - 1
	current_iteration_day = start_date
	end_day = today + timedelta(days=1)
	page = requests.get('https://github.com/users/%s/contributions' % username)
	tree = html.fromstring(page.content)
	while current_iteration_day != end_day:
		current_commit = tree.xpath('//rect[@data-date="' + str(current_iteration_day) + '"]/@data-count')
		if current_commit:
			commit_dict[current_iteration_day] = current_commit[0]
		current_iteration_day += timedelta(days=1)
	sorted_dict = OrderedDict(sorted(commit_dict.items(), key=lambda t:t[0]))
	return sorted_dict

@app.route('/results', endpoint='get-results', methods=['GET', 'POST'])
def get_results():
	if request.method == 'GET':
		return redirect(url_for('index'))
	username = request.form['username']
	page = requests.get('https://github.com/'  + username)
	if page.status_code != 200:
		return render_template('error.html', title='Invalid Username', message='That doesn\'t seem to be a valid GitHub username')
	page_tree = html.fromstring(page.content)
	streak_list = get_streak(username, page_tree)
	if streak_list:
		streak = streak_list[0]
	else:
		return render_template('error.html', title='Invalid Username', message='That doesn\'t seem to be a valid GitHub username')
	commit_dict = get_commits(username, streak)
	committed_today = int(commit_dict[today]) > 0 if today in commit_dict else False
	message, valid = get_custom_message(int(streak.split()[0]), commit_dict, committed_today)
	chart = render_chart(commit_dict)
	return render_template('results.html', streak=streak, commits=commit_dict, chart=chart, message=message, valid=valid, today=committed_today)

@app.errorhandler(404)
def page_not_found(error):
    return render_template('error.html', title='Page Not Found', message='That page doesn\'t exist'), 404

@app.errorhandler(500)
def internal_error(error):
	return render_template('error.html', title='Internal Server Error', message='There appears to be an internal server error going on right now. Please contact <a href=\'mailto:tech@freetailhackers.com\'>tech@freetailhackers.com</a> to allow us to investigate further.'), 500

if __name__ == '__main__':
	app.run(debug=True)