Network Utils

网络工具包。

• 依赖

pip install tqdm

• 包引用（示例）

# package path: D:\workspace\network_utils
# ├─ anywhere
#    ├─ script.py
#    └─ cfgbackup_ft_fw01.conf

# script.py
import sys
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
sys.path.append(os.path.dirname(SCRIPT_DIR))

from network_utils.conf_parser import FortiGate
from network_utils.tools import dict2json

FT = FortiGate("cfgbackup_ft_fw01.conf")
dict2json(FT.parse_policy(), "output.json")

conf_parser.py

将网络配置文件解析为 dict 或者 json 格式，网络设备包括👇：

• 飞塔防火墙（图形界面 backup 文件）

# 主函数部分
if __name__ == "__main__":
    # 配置文件位置 和 解析输出位置
    conf_path = r"D:\workspace\network_utils\doc\cfgbackup_ft_fw01.conf"
    out_path = "output.json"

    FT = FortiGate(conf_path)
    p = FT.parse_policy()
    # print(p)

    from tools import dict2json
    dict2json(p, out_path)

• CheckPoint 网关（save configuration）

# 主函数部分
if __name__ == "__main__":
    conf_path = "doc/gaia_os_cp_gw.txt"
    CP = CheckPoint(conf_path)
    conf = CP.parse_configuration()
    print(conf["set"]["installer"]["policy"])

translator.py

翻译 api 调用器。

• 百度 API

# 主函数部分
if __name__ == "__main__":
    # 从文件读取 appid 和 secretKey
    with open("doc/bt_app_sec.txt", "r", encoding="utf-8") as f:
        appid, secretKey = f.readlines()[0].strip().split(",")
    
    bt = Baidu_Translator(appid, secretKey)
    
    # 翻译单词
    res = bt.translate("banana")
    print(res)

    # 翻译文件（逐行翻译，逐行输出）
    res = bt.translate_file("doc/words.txt", is_save=True)
    with open("doc/output.txt", "w", encoding="utf-8") as f:
        f.writelines([w + "\n" for w in res])

connection.py

ssh 远程连接脚本。

• 依赖（可选）

pip install getpass

• 直接调用

# 主函数部分
if __name__ == "__main__":
    ip = input("IP_ADDRESS: ")
    username = input("username: ")
    
    try:
        from getpass import getpass
        password = getpass("password: ")
    except ImportError:
        print("getpass is not installed.")
        password = input("password: ")
    
    client = Connection(ip, username=username, password=password)
    # 逐行读取在远程执行的命令（参考示例：doc/centos_nginx）
    command_file = "doc/centos_nginx"
    
    # 捕获额外需要输入的提示，返回对应的输入
    confirm_flag = {
        "'s password:": "centos",
        "Enter expert password:": "centos",
        "Are you sure you want to continue?(Y/N)[N]": "y"
        }
    # 实时返回远程输出信息
    for stdout in client.send_command_file(command_file, confirm_flag=confirm_flag):
        print(stdout)
    client.close()

filter.py

python过滤器。

• 依赖

pip install pandas

• 过滤表格的时间列

# 主函数部分
if __name__ == "__main__":
    out_data = pd.read_csv("互联网边界防火墙内对外下载数据.csv")
    out_data["上次命中时间"] = out_data["最近命中时间"].apply(date_filter)

    in_data = pd.read_csv("互联网边界防火墙外对内下载数据.csv")
    in_data["上次命中时间"] = in_data["最近命中时间"].apply(date_filter)

    with pd.ExcelWriter("互联网边界防火墙策略_命中统计.xlsx") as writer:
        out_data.to_excel(writer, sheet_name="内对外", index=False)
        in_data.to_excel(writer, sheet_name="外对内", index=False)