这种写法可以替代配置文件的传参,效率更高
json文件:baittype.json
{
"bait_type":[
{
"type":"ssh",
"user":"root",
"password":"qazwsx",
"start_time":"14:00:00",
"end_time":"15:00:00"
},
{
"type":"telnet",
"user":"yvan",
"password":"password",
"start_time":"15:00:00",
"end_time":"16:00:00"
},
{
"type":"ftp",
"user":"myuser",
"password":"mypasswd",
"start_time":"16:00:00",
"end_time":"17:00:00"
}
]
}
源码:
#!/usr/bin/env python
# --*-- coding:UTF-8 --*--
import os
import json
bait_type_path = os.path.dirname(__file__)
with open(os.path.join(bait_type_path, 'baittype.json')) as fp:
data = json.load(fp)
bait_data = []
for entry in data["bait_type"]:
bait = {}
bait["type"] = entry["type"]
bait["start_time"] = entry["start_time"]
bait["end_time"] = entry["end_time"]
bait_data.append(bait)