mirror of https://github.com/dirtbags/tanks.git
65 lines
2.1 KiB
Python
Executable File
65 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
""" Here's an example tank, in RFC822ish format:
|
|
|
|
From: Joe Cool <joe@cool.cc>
|
|
Tank-Name: Red Baron
|
|
Color: #c7e148
|
|
Sensor-0: 50 0 7 1
|
|
Sensor-1: 30 0 90 0
|
|
|
|
get-turret 12 + set-turret! ( Rotate turret )
|
|
37 40 set-speed! ( Go in circles )
|
|
0 sensor? { fire! } if ( Fire if turret sensor triggered )
|
|
1 sensor? { -50 50 set-speed! } if ( Turn if collision sensor triggered )
|
|
"""
|
|
|
|
import argparse
|
|
import sys
|
|
import email
|
|
import requests
|
|
import os
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="Upload forf tanks.",
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument('--infile', '-i', nargs='?', metavar='FILE',
|
|
help="filename containing tank program",
|
|
type=argparse.FileType('r'), default=sys.stdin)
|
|
parser.add_argument('--url', '-u', nargs='?', metavar='URL',
|
|
help="URL to submit your tank to",
|
|
default='http://woozle.org/tanks/upload.cgi')
|
|
return parser.parse_args()
|
|
|
|
def read_tank(infile):
|
|
msg = email.message_from_file(infile)
|
|
headers = dict(msg.items())
|
|
return headers, msg.get_payload()
|
|
|
|
def post_tank(headers, code, url):
|
|
token = os.environ.get('token')
|
|
if not token:
|
|
raise RuntimeError("Must provide token in 'token' environment variable")
|
|
request = {}
|
|
request['token'] = token
|
|
request['name'] = headers.get('Tank-Name', '')
|
|
request['author'] = headers.get('From', '')
|
|
request['color'] = headers.get('Color', '#c0c0c0')
|
|
for sn in range(0, 10):
|
|
sensor = 'Sensor-{}'.format(sn)
|
|
if sensor in headers:
|
|
sr, sa, sw, st = headers[sensor].split()
|
|
request['s{}r'.format(sn)] = sr
|
|
request['s{}a'.format(sn)] = sa
|
|
request['s{}w'.format(sn)] = sw
|
|
request['s{}t'.format(sn)] = st
|
|
request['program'] = code
|
|
r = requests.post(url, data=request, headers={'Accept': 'text/plain'})
|
|
print(r.text)
|
|
r.raise_for_status()
|
|
|
|
if __name__ == '__main__':
|
|
args = parse_args()
|
|
headers, code = read_tank(args.infile)
|
|
post_tank(headers, code, args.url)
|