submit_awcy.py 2.49 KB
Newer Older
Thomas Daede's avatar
Thomas Daede committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
#!/usr/bin/env python

from __future__ import print_function

import requests
import argparse
import os
import subprocess
import sys
from datetime import datetime

#our timestamping function, accurate to milliseconds
#(remove [:-3] to display microseconds)
def GetTime():
    return datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]

if "check_output" not in dir( subprocess ): # duck punch it in!
    def f(*popenargs, **kwargs):
        if 'stdout' in kwargs:
            raise ValueError('stdout argument not allowed, it will be overridden.')
        process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
        output, unused_err = process.communicate()
        retcode = process.poll()
        if retcode:
            cmd = kwargs.get("args")
            if cmd is None:
                cmd = popenargs[0]
            raise subprocess.CalledProcessError(retcode, cmd)
        return output
    subprocess.check_output = f

key = None
try:
    with open('secret_key','r') as keyfile:
        key = keyfile.read().strip()
except:
    print(GetTime(), "Could not open your secret_key file!")
    print(GetTime(), "Please create a file called secret_key in the current directory")
    print(GetTime(), "containing your AreWeCompressedYet key.")
    sys.exit(1)

parser = argparse.ArgumentParser(description='Submit test to arewecompressedyet.com')
parser.add_argument('-branch',default=None)
parser.add_argument('-prefix',default=None)
parser.add_argument('-master',action='store_true',default=False)
parser.add_argument('-set',default='objective-1-fast')
47
parser.add_argument('-extra_options',default='')
Thomas Daede's avatar
Thomas Daede committed
48 49 50 51 52 53 54 55 56 57 58 59
args = parser.parse_args()

if args.branch is None:
    try:
        args.branch = subprocess.check_output('git symbolic-ref -q --short HEAD',shell=True).strip()
    except:
        args.branch = None

if args.prefix is None:
    args.prefix = args.branch

commit = subprocess.check_output('git rev-parse HEAD',shell=True).strip()
60 61 62
short = subprocess.check_output('git rev-parse --short HEAD',shell=True).strip().decode("utf-8")
date = GetTime().strip()
date_short = date.split()[0]+'_'+date.split()[1].split('.')[0].replace(':', '')
Thomas Daede's avatar
Thomas Daede committed
63 64 65 66 67
user = args.prefix
is_master = args.master
run_id = user+'-'+date_short+'-'+short

print(GetTime(), 'Creating run '+run_id)
68 69 70 71 72 73 74 75 76
r = requests.post("https://beta.arewecompressedyet.com/submit/job", {
    'run_id': run_id,
    'commit': commit,
    'master': is_master,
    'key': key,
    'task': args.set,
    'codec': 'rav1e',
    'extra_options': args.extra_options,
})
Thomas Daede's avatar
Thomas Daede committed
77
print(GetTime(), r)