cleaning up

This commit is contained in:
Eric Romano
2013-04-29 18:15:40 -04:00
parent 0c28913cd4
commit f7f2b55547
2 changed files with 98 additions and 53 deletions

View File

@ -1,3 +1,4 @@
#!/bin/env python
import os, sys, datetime, math, itertools
try:
import requests
@ -5,6 +6,69 @@ except:
print 'the requests module is required'
exit(1)
kitty=[
[0,0,0,4,0,0,0,0,4,0,0,0],
[0,0,4,2,4,4,4,4,2,4,0,0],
[0,0,4,2,2,2,2,2,2,4,0,0],
[2,2,4,2,4,2,2,4,2,4,2,2],
[0,0,4,2,2,3,3,2,2,4,0,0],
[2,2,4,2,2,2,2,2,2,4,2,2],
[0,0,0,3,4,4,4,4,3,0,0,0]]
oneup=[
[0,0,4,4,4,4,4,4,4,0,0],
[0,4,2,2,1,1,1,2,2,4,0],
[4,3,2,2,1,1,1,2,2,3,4],
[4,3,3,4,4,4,4,4,3,3,4],
[0,4,4,1,4,1,4,1,4,4,0],
[0,0,4,1,1,1,1,1,4,0,0],
[0,0,0,4,4,4,4,4,0,0,0]]
twoup=[
[0,0,4,4,4,4,4,4,4,0,0,0,0,0,4,4,4,4,4,4,4,0,0],
[0,4,2,2,1,1,1,2,2,4,0,0,0,4,2,2,1,1,1,2,2,4,0],
[4,3,2,2,1,1,1,2,2,3,4,0,4,3,2,2,1,1,1,2,2,3,4],
[4,3,3,4,4,4,4,4,3,3,4,0,4,3,3,4,4,4,4,4,3,3,4],
[0,4,4,1,4,1,4,1,4,4,0,0,0,4,4,1,4,1,4,1,4,4,0],
[0,0,4,1,1,1,1,1,4,0,0,0,0,0,4,1,1,1,1,1,4,0,0],
[0,0,0,4,4,4,4,4,0,0,0,0,0,0,0,4,4,4,4,4,0,0,0]]
hello=[
[0,1,0,0,0,0,0,0,0,1,0,1,0,0,0,0,0,4],
[0,2,0,0,0,0,0,0,0,2,0,2,0,0,0,0,0,4],
[0,3,3,3,0,2,3,3,0,3,0,3,0,1,3,1,0,3],
[0,4,0,4,0,4,0,4,0,4,0,4,0,4,0,4,0,3],
[0,3,0,3,0,3,3,3,0,3,0,3,0,3,0,3,0,2],
[0,2,0,2,0,2,0,0,0,2,0,2,0,2,0,2,0,0],
[0,1,0,1,0,1,1,1,0,1,0,1,0,1,1,1,0,4]]
hackerschool=[
[4,4,4,4,4,4],
[4,3,3,3,3,4],
[4,1,3,3,3,4],
[4,3,3,3,3,4],
[4,4,4,4,4,4],
[0,0,4,4,0,0]
[4,4,4,4,4,4]]
octocat=[
[0,0,0,4,0,0,0,4,0],
[0,0,4,4,4,4,4,4,4],
[0,0,4,1,3,3,3,1,4],
[4,0,3,4,3,3,3,4,3],
[0,4,0,0,4,4,4,0,0],
[0,0,4,4,4,4,4,4,4],
[0,0,4,0,4,0,4,0,4]]
hireme=[
[1,0,0,0,4,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[3,3,3,0,2,0,3,3,3,0,2,3,3,0,0,3,3,0,3,0,0,2,3,3],
[4,0,4,0,4,0,4,0,0,0,4,0,4,0,0,4,0,4,0,4,0,4,0,4],
[3,0,3,0,3,0,3,0,0,0,3,3,3,0,0,3,0,3,0,3,0,3,3,3],
[2,0,2,0,2,0,2,0,0,0,2,0,0,0,0,2,0,2,0,2,0,2,0,0],
[1,0,1,0,1,0,1,0,0,0,1,1,1,0,0,1,0,1,0,1,0,1,1,1]]
def get_calendar(username):
"""retrieves the github commit calendar data for a username"""
BASEURL='https://github.com/'
@ -22,15 +86,6 @@ def max_commits(input):
output.reverse()
return output[0]
def trim_calendar(input):
"""returns index of first sunday in the input"""
for i, j in enumerate(input):
day = input[i][0]
day = datetime.datetime.strptime(day, '%Y/%m/%d')
weekday = datetime.datetime.weekday(day)
if weekday == 6:
return i
def multiplier(max_commits):
"""calculates a multiplier to scale github colors to commit history"""
m = max_commits/4.0
@ -49,9 +104,10 @@ def get_start_date():
weekday = datetime.datetime.weekday(date)
return date
def date_gen(start_date):
'''generator to return the next day, requires a datetime object as input'''
for i in itertools.count():
def date_gen(start_date, offset=0):
'''generator to return the next day, requires a datetime object as input. The offset is weeks'''
start = offset * 7
for i in itertools.count(start):
yield start_date + datetime.timedelta(i)
def values_in_date_order(image, multiplier=1):
@ -61,59 +117,48 @@ def values_in_date_order(image, multiplier=1):
for w in range(width):
yield image[h][w]*multiplier
def fake_it(image, start_date, username, repo, multiplier=1):
template=('git init gitfiti\n'
def commit(content, commitdate):
template = '''echo %s >> gitfiti\nGIT_AUTHOR_DATE=%s GIT_COMMITTER_DATE= %s git commit -a -m "gitfiti"\n'''
return template % (content, commitdate.isoformat(), commitdate.isoformat())
def fake_it(image, start_date, username, repo, offset=0, multiplier=1):
template = ('#!/bin/bash\n'
'git init gitfiti\n'
'cd gitfiti\n'
'touch gitfiti\n'
'git add gitfiti\n'
'%s\n'
'git remote add origin git@github.com:%s/%s.git\n'
'git pull\n'
'git push -u origin master\n')
strings = []
for value, date in zip(values_in_date_order(image, multiplier), date_gen(start_date)):
for value, date in zip(values_in_date_order(image, multiplier), date_gen(start_date, offset)):
for i in range(value):
strings.append(commit(i, date))
return template % ("".join(strings), username, repo)
def commit(content, commitdate):
template='''echo %s >> gitfiti\nGIT_AUTHOR_DATE=%s GIT_COMMITTER_DATE= %s git commit -a -m "gitfiti"\n'''
return template % (content, commitdate.isoformat(), commitdate.isoformat())
def save(output, filename):
"""Saves the list to a given filename"""
f = open(filename, "w")
f.write('\n')
f.write(output)
f.close()
def main():
username='gelstudios'
print 'enter your github username:'
username = raw_input(">")
cal = get_calendar(username)
x = trim_calendar(cal)
m = max_commits(cal)
m = multiplier(m)
cal = cal[x:]
fake_it(kitty, get_start_date(), m)
m = multiplier(max_commits(cal))
print 'enter name of the repo to be used by gitfiti:'
repo = raw_input(">")
print 'enter weeks to offset the image:'
offset = raw_input(">")
output = fake_it(image, get_start_date(), username, repo, offset, m)
save(output, 'gitfiti.sh')
print 'gitfiti.sh saved, run it from inside the repo you want to gitfiti'
if __name__=='__main__':
main()
kitty=[
[0,0,0,4,0,0,0,0,4,0,0,0],
[3,3,4,2,4,4,4,4,2,4,3,3],
[0,0,4,2,2,2,2,2,2,4,0,0],
[3,3,4,2,4,2,2,4,2,4,3,3],
[0,0,4,2,2,3,3,2,2,4,0,0],
[3,3,4,2,2,2,2,2,2,4,3,3],
[0,0,0,4,4,4,4,4,4,0,0,0]]
oneup=[
[0,0,4,4,4,4,4,4,4,0,0],
[0,4,2,2,1,1,1,2,2,4,0],
[4,3,2,2,1,1,1,2,2,3,4],
[4,3,3,4,4,4,4,4,3,3,4],
[0,4,4,1,4,1,4,1,4,4,0],
[0,0,4,1,1,1,1,1,4,0,0],
[0,0,0,4,4,4,4,4,0,0,0]]
hello=[
[0,1,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,2,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,3,3,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,4,4,4,4,4,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,3,3,3,3,3,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,2,2,0,2,2,0,0,0,0,0,0,0,0,0,0,0,0,0],
[0,1,1,0,1,1,0,0,0,0,0,0,0,0,0,0,0,0,0]]
main()