diff --git a/tanks/run_tanks.py b/tanks/run_tanks.py index ab6a8d6..f680bbe 100755 --- a/tanks/run_tanks.py +++ b/tanks/run_tanks.py @@ -3,23 +3,45 @@ import optparse import shutil import time +import asyncore +import asynchat from tanks import Pflanzarr T = 60*5 MAX_HIST = 30 -HIST_STEP = 100 +HIST_STEP = 100 +key = 'tanks:::2bac5e912ff2e1ad559b177eb5aeecca' -parser = optparse.OptionParser('DATA_DIR easy|medium|hard MAX_TURNS') -opts, args = parser.parse_args() -if (len(args) != 3) or (args[1] not in ('easy', 'medium', 'hard')): - parser.error('Wrong number of arguments') -try: - turns = int(args[2]) -except: - parser.error('Invalid number of turns') +class Flagger(asynchat.async_chat): + """Use to connect to flagd and submit the current flag holder.""" -while True: - start = time.time() + def __init__(self, addr, auth): + asynchat.async_chat.__init__(self) + self.create_socket(socket.AF_INET, socket.SOCK_STREAM) + self.connect((addr, 6668)) + self.push(auth + '\n') + self.flag = None + + def handle_read(self): + msg = self.recv(4096) + raise ValueError("Flagger died: %r" % msg) + + def handle_error(self): + # If we lose the connection to flagd, nobody can score any + # points. Terminate everything. + asyncore.close_all() + asynchat.async_chat.handle_error(self) + + def set_flag(self, team): + if team: + eteam = team + else: + eteam = '' + self.push(eteam + '\n') + self.flag = team + + +def run_tanks(): p = Pflanzarr.Pflanzarr(args[0], args[1]) p.run(turns) @@ -38,6 +60,32 @@ while True: if highest - MAX_HIST > num and not (num % HIST_STEP == 0): shutil.rmtree(os.path.join(path, num)) - diff = time.time() - start - if diff - T > 0: - time.sleep( diff - T ) + try: + winner = open('/var/lib/tanks/winner').read().strip() + except: + winner = None + flagger.set_flag(winner) + + +def main(): + parser = optparse.OptionParser('DATA_DIR easy|medium|hard MAX_TURNS') + opts, args = parser.parse_args() + if (len(args) != 3) or (args[1] not in ('easy', 'medium', 'hard')): + parser.error('Wrong number of arguments') + try: + turns = int(args[2]) + except: + parser.error('Invalid number of turns') + + + flagger = Flagger('localhost', key) + lastrun = 0 + while True: + asyncore.loop(60, count=1) + now = time.time() + if now - lastrun >= 60: + run_tanks() + lastrun = now + +if __name__ == '__main__': + main()