def command2all_slaves(ips, command):
result = []
for ip in ips:
mission = {
"mission": "cmd2docker",
"commands": {
"command": command,
"arg": []
}
}
result.append(json.loads(mt.command2slave(ip, json.dumps(mission))))
return result
def basic_menu():
label.basic_menu
choice = raw_input("""
==================
{}: 分配容器
{}: 回收容器
{}: 查看镜像
{}: 查看容器
{}: 检查连接
{}: 详细信息
{}: 返回
{}: 退出
==================
> """.format(*colored_choice(6)))
show_logo()
if choice == "0":
result = json.loads(mt.ip_assign(container_subnet))
if result["code"]:
print pt.put_color(u"[X]分配 ip 失败", "red")
print " [-]", result["msg"]
goto.basic_menu
container_ip = result["result"]
result = json.loads(mt.check_load())
if result["code"]:
print pt.put_color(u"[X]负载查询失败", "red")
print " [-]", result["msg"]
goto.basic_menu
min_load = 100
for i in result["result"]:
for j in i:
value = i[j]["cpu"] * 0.2 + i[j]["mem"] * 0.8
if value < min_load:
min_load = value
ip = j
image_list = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": []
}
})))
if image_list["code"]:
print pt.put_color(u"[X]获取虚拟机: %s 的所有镜像失败" % ip, "red")
print " [-]", image_list["msg"]
goto.basic_menu
print u"选择镜像"
label.image_list
print "-" * 50
for i, image in enumerate(image_list["result"]):
print "%s: %s" % (pt.put_color(str(i), "cyan"), image)
print "{}: 返回\n{}: 退出".format(*colored_choice(0))
print "-" * 50
choice_image = raw_input("> ")
if choice_image == "b":
show_logo()
goto.basic_menu
elif choice_image == "q":
exit()
elif choice_image not in [str(c) for c in range(i + 1)]:
show_logo()
print pt.put_color(u"[X]输入有误, 重新输入", "red")
goto.image_list
image_name = image_list["result"][int(choice_image)]
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "run",
"arg": [image_name, container_ip]
}
})))
if result["code"]:
print pt.put_color(u"[X]启动容器失败", "red")
print u" [-]", result["msg"]
goto.basic_menu
print pt.put_color(u"[+]启动镜像 %s 的容器成功" % image_name, "green")
print u" [-]容器位于虚拟机 %s 中" % pt.put_color(ip, "white")
print u" [-]给容器分配的 ip 为:", pt.put_color(result["result"]["ip"],
"white")
print u" [-]ID 为", pt.put_color(result["result"]["id"], "white")
elif choice == '1':
mission = {
"mission": "cmd2docker",
"commands": {
"command": "others_cmd",
"arg": []
}
}
label.slave_list
print u"选择虚拟机"
print "-" * 50
alive_slave = []
empty_slave = []
results = []
for i, ip in enumerate(ips):
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": []
}
})))
results.append(result)
if result["code"]:
print u"%s: 虚拟机: %s" % (i, pt.put_color(ip, "red"))
print " [X]error:", result["msg"]
alive_slave.append(i)
print "%s: %s" % (pt.put_color(str(i),
"cyan"), pt.put_color(ip, "white"))
if result["result"] == []:
print pt.put_color(" [!]Empty", "yellow")
empty_slave.append(i)
else:
for j, r in enumerate(result["result"]):
print " %s: [%s] [%s] [%s]" % (
pt.put_color(str(j), "cyan"),
pt.put_color(
r["status"], "green" if r["status"] == "running"
else "yellow"), pt.put_color(r["ip"], "white"),
pt.put_color(r["image name"], "white"))
print
print "{}: 返回\n{}: 退出".format(*colored_choice(0))
print "-" * 50
choice_slave = raw_input("> ")
if choice_slave == 'b':
show_logo()
goto.basic_menu
elif choice_slave == 'q':
exit()
elif not choice_slave.isdigit():
show_logo()
print pt.put_color(u"输入有误, 重新输入", "red")
goto.slave_list
choice_slave = int(choice_slave)
if choice_slave not in alive_slave:
show_logo()
print pt.put_color(u"此虚拟机无法连接, 重新选择", "red")
goto.slave_list
elif choice_slave in empty_slave:
show_logo()
print pt.put_color(u"此虚拟机无容器, 重新选择", "red")
goto.slave_list
ip = ips[choice_slave]
choice_container = raw_input(
"\n选择容器\n=========\n{}: 返回\n{}: 退出\n=========\n> ".format(
*colored_choice(0)))
if choice_container == 'b':
show_logo()
goto.slave_list
elif choice_container == 'q':
exit()
elif not choice_container.isdigit():
show_logo()
print pt.put_color(u"输入有误, 重新输入", "red")
goto.slave_list
choice_container = int(choice_container)
if choice_container not in range(len(results[choice_slave]["result"])):
show_logo()
print pt.put_color(u"虚拟机: %s 无此容器, 重新选择" % ip, "red")
goto.slave_list
id_or_name = results[choice_slave]["result"][choice_container]["id"]
show_logo()
print u"[+]回收容器:", id_or_name
mission["commands"]["arg"] = [id_or_name, "kill"]
print u" [-]停止容器 ...",
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
goto.basic_menu
print pt.put_color(u"成功", "green")
print u" [-]删除容器 ...",
mission["commands"]["arg"] = [id_or_name, "rm"]
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
goto.basic_menu
print pt.put_color(u"成功", "green")
print u"[!]完成"
elif choice == "2":
show_logo()
image_list = json.loads(
mt.command2slave(
"192.168.12.1",
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": []
}
})))
if image_list["code"]:
print pt.put_color(u"[X]获取虚拟机: 192.168.12.1 的所有镜像失败", "red")
print u" [-]", image_list["msg"]
return
images = image_list["result"]
for i, image in enumerate(images):
print "%s: %s" % (pt.put_color(str(i), "cyan"), image)
elif choice == "3":
mission = {
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": []
}
}
results = []
for i, ip in enumerate(ips):
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print u"[X]获取虚拟机 %s 的所有容器失败" % ip
continue
results.extend(result["result"])
for i, container in enumerate(results):
print "%s. %s" % (pt.put_color(
str(i), "cyan"), pt.put_color(container["id"][:6], "white"))
print " [-]id: " + container["id"]
print u" [-]状态: " + pt.put_color(
container["status"],
"green" if container["status"] == "running" else "yellow")
print u" [-]启动时间点: " + pt.put_color(container["start time"],
"blue")
print u" [-]容器 ip: " + pt.put_color(container["ip"], "white")
print u" [-]所在虚拟机 ip: " + pt.put_color(container["slave ip"],
"white")
print u" [-]镜像名: " + pt.put_color(container["image name"],
"white")
print "-" * 50, '\n'
elif choice == '4':
mission = {
"mission": "cmd2slave",
"commands": {
"command": "check_alive",
"arg": []
}
}
for ip in ips:
result = json.loads(
mt.command2slave(ip, json.dumps(mission), timeout=10))
if result["code"]:
print pt.put_color(ip, "red"), pt.put_color(
u"内网", "red"), pt.put_color(u"外网", "red")
else:
if result["result"]:
print pt.put_color(ip, "yellow"), pt.put_color(
u"内网", "green"), pt.put_color(u"外网", "red")
else:
print pt.put_color(ip, "green"), pt.put_color(
u"内网", "green"), pt.put_color(u"外网", "green")
elif choice == "5":
slaves_info = {}
mission = [{
"mission": "cmd2slave",
"commands": {
"command": "loads_ls",
"arg": [],
}
}, {
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": [],
}
}, {
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": [],
}
}]
for ip in ips:
slaves_info[ip] = {
"load": {
"code": 1,
"msg": "",
"loads": []
},
"image": {
"code": 1,
"msg": "",
"images": []
},
"container": {
"code": 1,
"msg": "",
"containers": []
}
}
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
print u"[+]虚拟机: " + pt.put_color(ip, "white")
if result[0]["code"]:
print pt.put_color(u" [X]负载情况", "red")
print " [-]error: " + result[0]["msg"]
else:
print u" [-]负载情况"
print " [-]cpu: " + pt.put_color(
str(result[0]["result"]["cpu"]) + "%", "white")
print " [-]mem: " + pt.put_color(
str(result[0]["result"]["mem"]) + "%", "white")
if result[1]["code"]:
print pt.put_color(u" [X]拥有镜像", "red")
print " [-]error: " + result[1]["msg"]
else:
print u"\n [-]拥有镜像(%s)" % pt.put_color(
str(len(result[1]["result"])), "cyan")
for image in result[1]["result"]:
print " [-]" + image
if result[2]["code"]:
print pt.put_color(u"\n [X]拥有容器", "red")
print " [-]error: " + result[2]["msg"]
else:
print u"\n [-]拥有容器(%s)" % pt.put_color(
str(len(result[2]["result"])), "cyan")
for container in result[2]["result"]:
print " [-]short id: " + pt.put_color(
container["id"][:6], "white")
print u" [-]启动时间点: " + pt.put_color(
container["start time"], "blue")
print u" [-]状态: " + pt.put_color(
container["status"], "green"
if container["status"] == "running" else "yellow")
print u" [-]容器 ip: " + pt.put_color(
container["ip"], "white")
print " [-]id: " + container["id"]
print u" [-]镜像名: " + pt.put_color(
container["image name"], "white"), "\n"
print "-" * 50, '\n'
elif choice == 'b':
return
elif choice == 'q':
exit()
else:
print pt.put_color(u"输入有误, 重新输入", "red")
goto.basic_menu
def pro_menu():
label.pro_menu
choice = raw_input("""
==================
{}: 网络相关
{}: 批量运行
{}: 按镜像停止
{}: 全部停止
{}: 返回
{}: 退出
==================
> """.format(*colored_choice(4)))
show_logo()
if choice == '0':
nk_menu()
elif choice == '1':
choice = raw_input("""
==================
输入容器数量
==================
> """)
if not choice.isdigit():
show_logo()
goto.pro_menu
container_num = int(choice)
image_list = json.loads(
mt.command2slave(
mt.setting["bridge"]["self_ip"],
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": []
}
})))
if image_list["code"]:
print pt.put_color(
u"[X]获取虚拟机: %s 的所有镜像失败" % mt.setting["bridge"]["self_ip"],
"red")
print " [-]", image_list["msg"]
print u"选择镜像"
label.image_list
print "-" * 50
for i, image in enumerate(image_list["result"]):
print "%s: %s" % (pt.put_color(str(i), "cyan"), image)
print "{}: 返回\n{}: 退出".format(*colored_choice(0))
print "-" * 50
choice_image = raw_input("> ")
if choice_image == "b":
show_logo()
goto.pro_menu
elif choice_image == "q":
exit()
elif choice_image not in [str(c) for c in range(i + 1)]:
show_logo()
print pt.put_color(u"[X]输入有误, 重新输入", "red")
goto.image_list
for _ in range(container_num):
result = json.loads(mt.ip_assign(container_subnet))
if result["code"]:
print pt.put_color(u"[X]分配 ip 失败", "red")
print " [-]", result["msg"]
continue
container_ip = result["result"]
result = json.loads(mt.check_load())
if result["code"]:
print pt.put_color(u"[X]负载查询失败", "red")
print " [-]", result["msg"]
continue
min_load = 100
for i in result["result"]:
for j in i:
value = i[j]["cpu"] * 0.2 + i[j]["mem"] * 0.8
if value < min_load:
min_load = value
ip = j
image_name = image_list["result"][int(choice_image)]
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "run",
"arg": [image_name, container_ip]
}
})))
if result["code"]:
print pt.put_color(u"[X]启动容器失败", "red")
print u" [-]", result["msg"]
continue
print pt.put_color(u"[+]启动镜像 %s 的容器成功" % image_name, "green")
print u" [-]容器位于虚拟机 %s 中" % pt.put_color(ip, "white")
print u" [-]给容器分配的 ip 为:", pt.put_color(result["result"]["ip"],
"white")
print u" [-]ID 为", pt.put_color(result["result"]["id"], "white")
elif choice == '2':
mission = {
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": []
}
}
containers = {}
image_order = []
results = []
for i, ip in enumerate(ips):
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print u"[X]获取虚拟机 %s 的所有容器失败" % ip
continue
results.extend(result["result"])
for i, container in enumerate(results):
if not containers.has_key(container["image name"]):
containers[container["image name"]] = {"containers": []}
containers[container["image name"]]["containers"].append({
"slave ip":
container["slave ip"],
"id":
container["id"],
})
for i, image in enumerate(containers):
print u"[%s]镜像名: %s" % (pt.put_color(
str(i), "cyan"), pt.put_color(image, "white"))
image_order.append(image)
for ip in containers[image]["containers"]:
print ip
print "-" * 10, pt.put_color(
str(len(containers[image]["containers"])), "cyan"), "-" * 10
print "\n"
choice = raw_input("""
=================="
选择要批量删除的镜像:
==================
> """)
if choice == "b":
return
elif choice == 'q':
exit()
if (not choice.isdigit()) or (int(choice) not in range(
len(image_order))):
print pt.put_color(u"输入有误, 重新输入", "red")
return
choice = image_order[int(choice)]
if raw_input(
pt.put_color(
"将清空镜像 " + choice.encode("utf8") +
" 的所有容器，确认？\nyes/[no]> ", "yellow")) != "yes":
print u"已放弃操作"
return
mission = {
"mission": "cmd2docker",
"commands": {
"command": "others_cmd",
"arg": []
}
}
print "\n[+]" + choice
for i, container in enumerate(containers[choice]):
for j, r in enumerate(containers[choice][container]):
id_or_name = r["id"]
ip = r["slave ip"]
print u" [-]回收容器:", id_or_name
mission["commands"]["arg"] = [id_or_name, "kill"]
print u" [-]停止容器 ...",
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
else:
print pt.put_color(u"成功", "green")
print u" [-]删除容器 ...",
mission["commands"]["arg"] = [id_or_name, "rm"]
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
else:
print pt.put_color(u"成功", "green")
print u"[!]完成"
elif choice == '3':
if raw_input(pt.put_color("确定删除所有容器？\nyes/[no]> ", "yellow")) != "yes":
print u"已放弃操作"
return
mission = {
"mission": "cmd2docker",
"commands": {
"command": "others_cmd",
"arg": []
}
}
for i, ip in enumerate(ips):
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": []
}
})))
if result["code"]:
print u"[+]虚拟机: %s" % pt.put_color(ip, "red")
print " [X]error:", result["msg"]
continue
print u"[+]虚拟机: %s" % pt.put_color(ip, "green")
if result["result"] == []:
print pt.put_color(" [!]Empty", "yellow")
else:
for j, r in enumerate(result["result"]):
id_or_name = r["id"]
print u" [-]回收容器:", id_or_name
mission["commands"]["arg"] = [id_or_name, "kill"]
print u" [-]停止容器 ...",
result = json.loads(
mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
print pt.put_color(u"成功", "green")
print u" [-]删除容器 ...",
mission["commands"]["arg"] = [id_or_name, "rm"]
result = json.loads(
mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
continue
print pt.put_color(u"成功", "green")
print u"[!]完成"
elif choice == 'b':
return
elif choice == 'q':
exit()
else:
print pt.put_color("输入有误, 重新输入", "red")
goto.pro_menu
def nk_menu():
label.nk_menu
choice = raw_input("""
==================
{}: 显示可用 ip
{}: 显示已用 ip
{}: 按镜像显示已用 ip
{}: 退出
==================
> """.format(*colored_choice(3)))
show_logo()
if choice == '0':
result = json.loads(mt.ip_ls(container_subnet))
pprint(result)
elif choice == '1':
result = json.loads(mt.ip_used(container_subnet))
for i in result:
print i
# pprint(result)
elif choice == "2":
mission = {
"mission": "cmd2docker",
"commands": {
"command": "containers_ls",
"arg": []
}
}
IPs = {}
results = []
for i, ip in enumerate(ips):
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print u"[X]获取虚拟机 %s 的所有容器失败" % ip
continue
results.extend(result["result"])
for i, container in enumerate(results):
if not IPs.has_key(container["image name"]):
IPs[container["image name"]] = {"ips": []}
IPs[container["image name"]]["ips"].append(container["ip"])
for image in IPs:
print u"[+]镜像名: %s" % pt.put_color(image, "white")
for ip in IPs[image]["ips"]:
print ip
print "-" * 10, pt.put_color(str(len(IPs[image]["ips"])),
"cyan"), "-" * 10, "\n"
# print "%s. %s" % (pt.put_color(str(i), "cyan"),
# pt.put_color(container["id"][:6], "white"))
# print u" [-]容器 ip: "+pt.put_color(container["ip"], "white")
# print u" [-]镜像名: "+pt.put_color(container["image name"], "white")
elif choice == 'b':
return
elif choice == 'q':
exit()
else:
print pt.put_color("输入有误, 重新输入", "red")
def nk_menu(): # ok
label.network
choice = raw_input("""
==================
输入数字以继续:
{}: 检查连接
{}: 显示可用 ip
{}: 显示已用 ip
{}: 返回
{}: 退出
==================
> """.format(*colored_choice(3)))
subnet = get_setting("bridge")["subnet"]
show_logo()
if choice == '1':
mission = {
"mission": "cmd2slave",
"commands": {
"command": "check_alive",
"arg": [subnet]
}
}
ips = get_setting("slave_ip")
for ip in ips:
result = json.loads(
mt.command2slave(ip, json.dumps(mission), timeout=10))
if result["code"]:
print pt.put_color(ip, "red"), pt.put_color(
u"内网", "red"), pt.put_color(u"外网", "red")
else:
if result["result"]:
print pt.put_color(ip, "yellow"), pt.put_color(
u"内网", "green"), pt.put_color(u"外网", "red")
else:
print pt.put_color(ip, "green"), pt.put_color(
u"内网", "green"), pt.put_color(u"外网", "green")
elif choice == '2':
result = json.loads(mt.ip_ls(subnet))
if result["code"]:
print pt.put_color(u"获取可用 ip 出错", "red")
print "原因如下:\n", result["msg"]
else:
print pt.put_color("获取可用 ip 成功", "green")
print "结果如下:"
pprint(result["result"])
elif choice == '3':
result = json.loads(mt.ip_used(subnet))
if result["code"]:
print pt.put_color(u"获取已用 ip 出错", "red")
print "原因如下:\n", result["msg"]
else:
print pt.put_color("获取已用 ip 成功", "green")
print "结果如下:"
pprint(result["result"])
elif choice == 'b':
return
elif choice == 'q':
abort(1, 1)
else:
print pt.put_color("输入有误, 重新输入", "red")
goto.network
def dk_more_menu():
label.dk_more_menu
choice = raw_input("""
=======
{}: 运行
{}: 暂停
{}: 恢复
{}: 停止
{}: 删除
{}: 列出
{}: 负载
{}: 返回
{}: 退出
=======
> """.format(*colored_choice(7)))
show_logo()
if choice == "1":
ip = raw_input(u"输入分配容器的虚拟机的 ip: ")
if ip == "":
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.dk_more_menu
image_name = raw_input(u"输入镜像名（必要时加上版本号）: ")
if image_name == "":
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.dk_more_menu
container_ip = raw_input(u"输入给容器分配的 ip: ")
if container_ip == "":
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.dk_more_menu
subnet = get_setting("bridge")["subnet"]
result = json.loads(mt.ip_assign(subnet, container_ip))
if result["code"]:
print pt.put_color(u"指定 ip 失败", "red")
print u"原因如下:\n", result["msg"]
else:
container_ip = result["result"]
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "run",
"arg": [image_name, container_ip]
}
})))
if result["code"]:
print pt.put_color("启动容器失败", "red")
print "原因如下:\n", result["msg"]
else:
print pt.put_color(u"[+]启动容器 %s 成功" % image_name, "green")
print u" [-]位于虚拟机 %s 中" % pt.put_color(ip, "white")
print u" [-]容器分配的 ip 为:", pt.put_color(
result["result"]["ip"], "white")
print u" [-]ID 为", pt.put_color(result["result"]["id"],
"white")
elif choice in ["2", "3", "4", "5"]:
command = {
"2": "pause",
"3": "unpause",
"4": "kill",
"5": "rm",
}
notrun(command[choice])
elif choice == "6":
label.container_or_image
choice = raw_input("""
=======
{}: 容器
{}: 镜像
{}: 返回
{}: 退出
=======
> """.format(*colored_choice(2)))
choice_list = {
"1": "containers_ls",
"2": "images_ls",
}
show_logo()
if choice not in ['b', '1', '2', 'q']:
print pt.put_color("输入有误, 重新输入", "red")
goto.container_or_image
if choice == 'b':
goto.dk_more_menu
elif choice == 'q':
abort(1, 1)
choice = choice_list[choice]
label.ms
ip_num = raw_input("""
=============
{}: 单台虚拟机
{}: 多台虚拟机
{}: 所有虚拟机
{}: 返回
{}: 退出
=============
> """.format(*colored_choice(3)))
show_logo()
if ip_num not in ['b', '1', '2', '3', 'q']:
print pt.put_color("输入有误, 重新输入", "red")
goto.ms
if ip_num == "1":
ips = [raw_input("输入虚拟机的 ip: ")]
if ips == [""]:
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.ms
elif ip_num == "2":
ips = raw_input("输入 ip(用单个空格隔开): ").split(" ")
if ips == [""]:
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.ms
elif ip_num == "3":
ips = get_setting("slave_ip")
elif ip_num == 'b':
goto.container_or_image
elif ip_num == 'q':
abort(1, 1)
results = command2all_slaves(ips, choice)
for i, result in enumerate(results):
if result["code"]:
print "[+]slave:", pt.put_color(ips[i], "red")
print " [-]error:", result["msg"]
else:
print "[+]slave:", pt.put_color(ips[i], "green")
print " [-]result:"
pprint(result["result"])
print
elif choice == "7":
result = json.loads(mt.check_load())
if result["code"]:
print pt.put_color("负载查询失败", "red")
print "原因如下:\n", result["msg"]
else:
print pt.put_color("负载查询成功", "green")
print u"结果如下"
pprint(result["result"])
elif choice == 'b':
return
elif choice == 'q':
abort(1, 1)
else:
print pt.put_color("输入有误, 重新输入", "red")
goto.dk_more_menu
def dk_menu(): # ok
label.docker
choice = raw_input("""
===========
{}: 分配容器
{}: 回收容器
{}: 查看信息
{}: 列出镜像
{}: 更多操作
{}: 返回
{}: 退出
===========
> """.format(*colored_choice(5)))
show_logo()
if choice == "1":
subnet = get_setting("bridge")["subnet"]
result = json.loads(mt.ip_assign(subnet))
if result["code"]:
print pt.put_color("分配 ip 失败", "red")
print "原因如下:\n", result["msg"]
goto.docker
container_ip = result["result"]
result = json.loads(mt.check_load())
if result["code"]:
print pt.put_color("负载查询失败", "red")
print "原因如下:\n", result["msg"]
goto.docker
min_load = 100
for i in result["result"]:
for j in i:
value = i[j]["cpu"] * 0.2 + i[j]["mem"] * 0.8
if value < min_load:
min_load = value
ip = j
image_list = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": []
}
})))
if image_list["code"]:
print pt.put_color(u"获取虚拟机: %s 的所有镜像失败" % ip, "red")
print u"原因如下:\n", image_list["msg"]
goto.docker
print u"选择镜像"
label.choice_image
print "============="
for i, image in enumerate(image_list["result"]):
print "%s: %s" % (pt.put_color(str(i), "blue"), image)
print "============="
choice_image = raw_input("> ")
if choice_image == "":
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.docker
if choice_image not in [str(c) for c in range(i + 1)]:
show_logo()
print pt.put_color("输入有误, 重新输入", "red")
goto.choice_image
image_name = image_list["result"][int(choice_image)]
result = json.loads(
mt.command2slave(
ip,
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "run",
"arg": [image_name, container_ip]
}
})))
if result["code"]:
print pt.put_color("启动容器失败", "red")
print "原因如下:\n", result["msg"]
goto.docker
print pt.put_color(u"[+]启动容器 %s 成功" % image_name, "green")
print u" [-]位于虚拟机 %s 中" % pt.put_color(ip, "white")
print u" [-]容器分配的 ip 为:", pt.put_color(result["result"]["ip"],
"white")
print u" [-]ID 为", pt.put_color(result["result"]["id"], "white")
elif choice == "2":
mission = {
"mission": "cmd2docker",
"commands": {
"command": "others_cmd",
"arg": []
}
}
ips = get_setting("slave_ip")
results = command2all_slaves(ips, "containers_ls")
print u"选择虚拟机"
label.choice_sla
alive_slave = []
empty_slave = []
print "=============",
for i, result in enumerate(results):
print
if result["code"]:
print "%s: slave: %s" % (i, pt.put_color(ips[i], "red"))
print " [X]error:", result["msg"]
goto.docker
alive_slave.append(i)
print "%s: %s" % (pt.put_color(
str(i), "blue"), pt.put_color(ips[i], "green"))
if result["result"] == []:
print pt.put_color(" [!]Empty", "yellow")
empty_slave.append(i)
else:
for j, r in enumerate(result["result"]):
print " %s: [%s] [%s] [%s]" % (
pt.put_color(str(j), "blue"),
pt.put_color(r["status"],
"white"), pt.put_color(r["ip"], "white"),
pt.put_color(r["image name"], "white"))
print "\n{}: 返回\n{}: 退出".format(*colored_choice(0))
print "============="
choice_slave = raw_input("> ")
if choice_slave == 'b':
show_logo()
goto.docker
if choice_slave == 'q':
abort(1, 1)
if not choice_slave.isdigit():
show_logo()
print pt.put_color("输入有误, 重新输入", "red")
goto.choice_sla
choice_slave = int(choice_slave)
if choice_slave not in alive_slave:
show_logo()
print pt.put_color("此虚拟机无法连接, 重新输入", "red")
goto.choice_sla
elif choice_slave in empty_slave:
show_logo()
print pt.put_color("此虚拟机无容器, 重新输入", "red")
goto.choice_sla
ip = ips[choice_slave]
choice_container = raw_input(
"\n选择容器\n=========\n{}: 返回\n{}: 退出\n=========\n> ".format(
*colored_choice(0)))
if choice_container == 'b':
show_logo()
goto.choice_sla
if choice_container == 'q':
abort(1, 1)
if not choice_container.isdigit():
show_logo()
print pt.put_color(u"输入有误, 重新输入", "red")
goto.choice_sla
else:
choice_container = int(choice_container)
if choice_container not in range(
len(results[choice_slave]["result"])):
show_logo()
print pt.put_color(u"虚拟机: %s 无此容器, 重新输入" % ip, "red")
goto.choice_sla
id_or_name = results[choice_slave]["result"][choice_container]["id"]
show_logo()
print u"[+]回收容器:", id_or_name
mission["commands"]["arg"] = [id_or_name, "kill"]
print u" [-]停止容器 ...",
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color(u"失败", "red")
print u" [x]" + result["msg"]
goto.docker
print pt.put_color(u"成功", "green")
print u" [-]删除容器 ...",
mission["commands"]["arg"] = [id_or_name, "rm"]
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color("失败", "red")
print u" [x]" + result["msg"]
goto.docker
print pt.put_color(u"成功", "green")
print u"[!]完成"
elif choice == "3":
slaves_info = {}
ips = get_setting("slave_ip")
results = command2all_slaves(ips, "images_ls")
for i, result in enumerate(results):
slaves_info[ips[i]] = {
"image": {
"code": 1,
"msg": "",
"images": []
},
"container": {
"code": 1,
"msg": "",
"containers": []
}
}
if result["code"]:
slaves_info[ips[i]]["image"]["msg"] = result["msg"]
else:
slaves_info[ips[i]]["image"]["code"] = 0
slaves_info[ips[i]]["image"]["images"] = result["result"]
results = command2all_slaves(ips, "containers_ls")
for i, result in enumerate(results):
if result["code"]:
slaves_info[ips[i]]["container"]["msg"] = result["msg"]
else:
slaves_info[ips[i]]["container"]["code"] = 0
slaves_info[
ips[i]]["container"]["containers"] = result["result"]
for ip in ips:
print "[+]slave: " + pt.put_color(ip, "white")
images_info = slaves_info[ip]["image"]
if images_info["code"]:
print pt.put_color(" [X]images", "red")
print " [-]error: " + images_info["msg"]
else:
print " [-]images(%s)" % pt.put_color(
str(len(images_info["images"])), "blue")
for image in images_info["images"]:
print " [-]" + image
containers_info = slaves_info[ip]["container"]
if containers_info["code"]:
print pt.put_color("\n [X]containers", "red")
print " [-]error: " + containers_info["msg"]
else:
print "\n [-]containers(%s)" % pt.put_color(
str(len(containers_info["containers"])), "blue")
for container in containers_info["containers"]:
print " [-]short id: " + pt.put_color(
container["id"][:6], "white")
print " [-]ip: " + pt.put_color(
container["ip"], "white")
print " [-]id: " + container["id"]
print " [-]status: " + pt.put_color(
container["status"], "green"
if container["status"] == "running" else "yellow")
print " [-]image name: " + pt.put_color(
container["image name"], "white")
print
print "-" * 50
elif choice == "4":
show_logo()
image_list = json.loads(
mt.command2slave(
"192.168.12.1",
json.dumps({
"mission": "cmd2docker",
"commands": {
"command": "images_ls",
"arg": []
}
})))
if image_list["code"]:
print pt.put_color(u"获取虚拟机: 192.168.12.1 的所有镜像失败", "red")
print u"原因如下:\n", image_list["msg"]
return
images = image_list["result"]
for i, image in enumerate(images):
print "%s: %s" % (pt.put_color(str(i), "blue"), image)
elif choice == "5":
dk_more_menu()
elif choice == 'b':
return
elif choice == 'q':
abort(1, 1)
else:
print pt.put_color(u"输入有误, 重新输入", "red")
goto.docker
def notrun(command): # ok
label.notrun
choice = raw_input("""
===========
{}: 单个容器
{}: 多个容器
{}: 所有容器
{}: 返回
{}: 退出
========+==
> """.format(*colored_choice(3)))
if choice not in ['b', '1', '2', '3', 'q']:
show_logo()
print pt.put_color("输入有误, 重新输入", "red")
goto.notrun
if choice == 'b':
return show_logo()
elif choice == 'q':
abort(1, 1)
ip = raw_input("输入容器所在的虚拟机的 ip：")
if ip == "":
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.notrun
mission = {
"mission": "cmd2docker",
"commands": {
"command": "others_cmd",
"arg": []
}
}
show_logo()
if choice == "1":
names = [raw_input("输入容器的 id 或名字: ")]
if names == [""]:
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.notrun
elif choice == "2":
names = raw_input("输入容器的 id 或名字 (用单个空格隔开): ").split(" ")
if names == [""]:
show_logo()
print pt.put_color(u"操作已取消", "yellow")
goto.notrun
elif choice == "3":
mission["commands"]["command"] = "containers_ls"
result = json.loads(mt.command2slave(ip, json.dumps(mission)))
if result["code"]:
print pt.put_color("获得 %s 的所有容器失败" % ip, "red")
print result["msg"]
else:
names = []
name = result["result"]
for i in name:
names.append(i["id"])
mission["commands"]["command"] = "others_cmd"
for name in names:
mission["commands"]["arg"] = [name, command]
pprint(mt.command2slave(ip, json.dumps(mission)))