python Screeps push script

push.py

import hashlib
import json
import glob
import os

import requests

auth = ('EMAIL', 'PASSWORD')

def push(ptr=False):
    URL = 'https://screeps.com/ptr/api/user/code' if ptr else 'https://screeps.com/api/user/code'
    j = {
        'branch': 'default',
        'modules': {},
    }

    for fn in glob.glob('*.js'):
        j['modules'][fn[:-3]] = open(fn).read()

    if ptr:
        j['modules']['main_real'] = j['modules']['main']
        j['modules']['main'] = j['modules']['ptr']

    hashes = '{%s}' % ', '.join('%s: "%s"' % (k, hashlib.md5(v).hexdigest()) for k, v in j['modules'].iteritems())
    j['modules']['main'] = j['modules']['main'].replace("'__HASHES__'", hashes)

    r = requests.post(URL,
                      auth=auth, data=json.dumps(j),
                      headers={'Content-Type': 'application/json; charset=utf-8'})

    print r.content
    print r.status_code

if __name__ == '__main__':
    import sys
    push(len(sys.argv) > 1)