def __init__(self, db_col, cfg_file="scorebotcfg.json", obj_id=None):
threading.Thread.__init__(self)
self.col = db_col
self.flag_queue = Queue.Queue()
self.flag_answer_queue = Queue.Queue()
self.msg_queue = Queue.Queue()
self.logger_obj = Logger("scorebot")
self.oqueue = Queue.Queue()
self.equeue = Queue.Queue()
self.logger = ThreadedLogger("ctfgame.log", self.oqueue, self.equeue)
self.logger.start()
self.pp = pprint.PrettyPrinter(indent=2)
self.blues_queues = {}
self.teams_rounds = {}
# todo - fix why this isn't actually the first round!
self.this_round = 1
self.interval = 120
self.inround = False
self.go_time = time.time()
self.save_time = time.time()
self.save_interval = 10
if obj_id:
bson_id = ObjectId(obj_id)
self.obj_id = bson_id
json_obj = self.col.find_one(self.obj_id)
if globalvars.debug:
self.pp.pprint(json_obj)
game = jaraco.modb.decode(decode_dict(json_obj))
if "config" in game:
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams,
self.injects) = self.json_cfg_obj.process(game["config"])
else:
raise Exception("No config section in restore object!")
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
if "scores" in game:
scores = game["scores"]
if "blueteams" in scores:
for team in self.blue_teams:
if team in scores["blueteams"]:
blue_score = scores["blueteams"][team]
self.blue_teams[team].set_scores(blue_score)
else:
raise Exception(
"Missing scores for team %s in restore object!"
% team)
else:
raise Exception(
"Missing blueteams score block in restore object!" %
team)
else:
raise Exception("Missing scores block in restore object!" %
team)
if "flags" in game:
all_flags = game["flags"]
self.flag_store.restore(all_flags)
else:
self.obj_id = obj_id
self.flag_store = FlagStore(self.logger_obj, self.flag_queue,
self.flag_answer_queue)
self.json_cfg_obj = JsonConfig(cfg_file, self.flag_store)
(self.blue_teams, self.injects) = self.json_cfg_obj.process()
self.message_store = MessageStore(self.logger_obj, self.msg_queue)
self.flag_server = FlagServer(self.logger_obj, self.flag_queue,
self.flag_answer_queue,
self.msg_queue)
for team in self.blue_teams.keys():
self.blue_teams[team].add_queue(self.flag_queue)
blue_queue = Queue.Queue()
self.blues_queues[team] = blue_queue
self.blue_teams[team].set_queue(blue_queue)
self.teams_rounds[team] = True
self.movies = Movies(self.logger_obj)
#self.movies.set_movie()
self.bottle_server = BottleServer(self.blue_teams, self.flag_store,
self.message_store, self.movies,
'0.0.0.0', 8090)
globalvars.binjitsu = True
elif o in ("-v", "--verbose"):
globalvars.verbose = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-q", "--quick"):
globalvars.quick = True
elif o in ("-n", "--nomovie"):
globalvars.nomovie = True
else:
assert False, "unhandled option"
flag_queue_obj = Queue.Queue()
message_queue_obj = Queue.Queue()
logger_obj = Logger("scorebot")
flag_store = FlagStore(logger_obj, flag_queue_obj)
message_store = MessageStore(logger_obj, message_queue_obj)
flag_server = FlagServer(logger_obj, flag_queue_obj, message_queue_obj)
t = threading.Thread(target=flag_server.serve_forever)
t.start()
blue_teams = read_config(cfg_file, flag_store)
myscoreboard = Scoreboard(blue_teams, flag_store, message_store)
myscoreboard.start()
for team in blue_teams.keys():
blue_teams[team].add_queue(flag_queue_obj)
blue_teams[team].start()
flag_store.start()
message_store.start()
injects.start()
if __name__ == "__main__":
