def get_screen_geometry():
config = ConfigParser.RawConfigParser()
config_path = os.path.join(common.config.path, 'config.txt')
config.read(config_path)
width = config.getint('ScreenSize', 'width')
height = config.getint('ScreenSize', 'height')
return (width, height)
def __init__(self):
self.redis_util = redis_utils()
self.result = []
self.threshold = (config.getint("default", "period") * 60 *
config.getfloat("default", "conn_failure_rate") /
config.getint("default", "frequency"))
self.fail_num = 0
self.initialize_data()
_logger.logger.info(self.threshold)
def run(self):
differ_time = (self.current_timer - self.timer).total_seconds()
_logger.logger.info("differ_time is %.2f" % (differ_time / 60))
if differ_time / 60 <= config.getint("default", "period"):
self.check()
else:
_logger.logger.info("Check period time reaches out %s mins!" %
(config["default"]["period"]))
self.initialize_data()
def __init__(self):
self.script_ttl = config.getint("script.ttl", 60) # minutes
self.scripts_folder = config.get("script.folder", "scripts")
self.master_host = config.get("script.master_host")
self.api_user = config.get("script.api_user")
self.api_password = config.get("script.api_password")
self.file_db = ScriptFileDB()
def __init__(self):
self.workers_count = config.getint("main.workers_count", 1)
self.amqp_address = config.get("main.amqp_server")
self.amqp_user = config.get("main.amqp_user", "guest")
self.amqp_pass = config.get("main.amqp_pass", "guest")
self.input_queue = config.get("main.input_queue", "spikeTasks")
self.output_queue = config.get("main.output_queue")
self.running = True
self.threads = []
self.worker = SpikeWorker(ExecutorFactory())
def __init__(self):
self.workers_count = config.getint("main.workers_count", 1)
self.amqp_address = config.get("main.amqp_server")
self.amqp_user = config.get("main.amqp_user", "guest")
self.amqp_pass = config.get("main.amqp_pass", "guest")
self.input_queue = config.get("main.input_queue", "spikeTasks")
self.output_queue = config.get("main.output_queue")
self.running = True
self.threads = []
self.worker = SpikeWorker(ExecutorFactory())
def collect_result(self):
"""
从redis重获取每个机器人执行测试线路连通性的失败结果，并将其结果存储在self.result中
e.g. self.result = [{'韩国2': {'ip':'164.52.56.58','result':'Fail'}}]
"""
self.result = []
key_name = config["redis"]["prefix_result_key"]
robot_num = config.getint("default", "total_robot_num")
for _index in range(1, robot_num + 1):
_data = eval(
self.redis_util.get_redis("%s_%s" % (key_name, _index)))
if len(_data) != 0:
self.result.append(_data)
def update_vpn_info(self):
result = mysql_utils(config["mysql"]["mysql_host"],
_logger).get_server_info()
robot_num = config.getint("default", "total_robot_num")
if result:
avg_ip_num = int(len(result) / robot_num)
for _index in range(1, robot_num + 1):
_start = avg_ip_num * (_index - 1)
_end = avg_ip_num * _index
if _index >= robot_num:
self.redis_util.set_redis_key(
"%s_%s" % (config["redis"]["prefix_vpn_key"], _index),
result[_start:],
)
else:
self.redis_util.set_redis_key(
"%s_%s" % (config["redis"]["prefix_vpn_key"], _index),
result[_start:_end],
)
_logger.logger.info("updated at %s!" % (datetime.datetime.now()))
else:
_logger.logger.error("Fail: Got an empty SQL result!")
elif result.upper() == "TIMEOUT":
msg = ssh_connection_failure_msg % (
region,
ip,
datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
)
else:
msg = u"%s:%s 已经下架 原因:%s 时间:%s" % (
region,
ip,
result,
datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
)
self.send_mail(msg)
def send_mail(self, msg):
_logger.logger.info(msg)
sms_sender(msg).bulk_send_msg()
if __name__ == "__main__":
runner = Runner()
while True:
_logger.logger.info("Start...")
start_time = time.time()
runner.run()
end_time = time.time()
time_used = end_time - start_time
time.sleep(config.getint("default", "frequency"))