def main(): if len(sys.argv) != 2: print "USAGE: python %s <dump-file>" % (sys.argv[0], ) return dump_file = sys.argv[1] mock_am_conf() service_mgr = SM.ServiceManager(g_zk_server, g_zk_root, g_zk_user, g_zk_pass) bh_mgr = BH.BHManager(g_bhcli_path, g_bhcli_cluster) task_context = AT.TaskContext(service_mgr, bh_mgr) tool = BST.SchedulerTaskHelper("task_debug", task_context) ret, err_msg, all_apps, all_machines, all_resource = tool.step_dump(g_service_id, dump_file) if ret != 0: print "FAIL:%s" % (err_msg, ) else: print "SUCC"
class TestRestore: def __init__(self): ret = LOG.init_log(am_global.log_prefix) if ret != 0: print "[FATAL] InitLogger failed" sys.exit(1) ret = am_conf.LoadConf( os.path.join(am_global.conf_dir, am_global.conf_file)) if ret != 0: print "[FATAL] LoadConf failed" sys.exit(1) try: self.__service_mgr = service_manager.ServiceManager( am_conf.g_conf.zk_server, am_conf.g_conf.zk_root) except Exception, e: print "init service_mananger failed" print e traceback.print_exc() sys.exit(1) print "init service_manager sucessfully" self.__service_mgr.Init() #TODO: move to upper level try: bh_mgr = bhcli_helper.BHManager("./fake_bhcli_for_restore.py", am_conf.g_conf.bh_cluster) #bh_mgr = bhcli_helper.BHManager(am_conf.g_conf.bh_loc, am_conf.g_conf.bh_cluster) logging.info("init bhcli manager successfully") self.task_context_ = am_task.TaskContext(self.__service_mgr, bh_mgr) logging.info("init task context successfully") self.queue_lock_ = threading.Lock() self.task_queue_ = {} logging.info("init task lock and queue successfully") except Exception, e: print "init task context failed" print e print traceback.print_exc() sys.exit(1)
if __name__ == "__main__": ret = am_log.init_log(os.path.join(am_global.log_dir, "test")) if ret != 0: print "[FATAL] InitLogger failed" sys.exit(1) conf_files = list() conf_files.append(os.path.join(am_global.conf_dir, "am.conf")) conf_files.append(os.path.join(am_global.conf_dir, "ip.conf.sample")) conf_files.append(os.path.join(am_global.conf_dir, "bhcli.conf.sample")) ret = am_conf.LoadConf(conf_files) if ret != 0: print "[FATAL] LoadConf failed" sys.exit(1) bh_mgr = bhcli_helper.BHManager(am_conf.g_conf.bh_loc, am_conf.g_conf.bh_cluster) #describe app app_id = "bs_vip_0" bh_app = bh_mgr.get_app(app_id) ret, app_descr = bh_app.describe() print("describe app finished, app_id=[%s], ret=[%d]") % (app_id, ret) bh_resource = bh_mgr.get_resource(app_id) ret, group_descr = bh_resource.describe_container_group(app_id) print("describe resource finished, app_id=[%s], ret=[%d]") % (app_id, ret) container_group = json.loads(group_descr) ret, out = bh_resource.update_container_group(app_id, container_group) print("update resource finished, app_id=[%s], ret=[%d]") % (app_id, ret)