ns3_setup.py

#!/usr/bin/env python
# coding=utf-8
"""
@author: Jiawei Wu
@create time: 2020-3-10 19:56
@edit time: 2020-04-20 20:59
"""
import argparse
import json
import os
from pathlib import Path
import sys


parser = argparse.ArgumentParser(description='安装参数')
parser.add_argument('--wafdir', default='/ns-3-dev', type=str, help='安装目录')
# 其他参数
parser.add_argument('--noaddon', default=False, action='store_true', help='是否执行addon代码')
parser.add_argument('--noenv', default=False, action='store_true', help='是否执行env代码')

args = parser.parse_args()
# 处理args.wafdir
if args.wafdir.startswith('~'):
    # 对于～开头的路径，要转为绝对路径
    args.wafdir = args.wafdir.replace('~', str(Path.home()))

cur_path = Path().resolve()


def exec_addon_setup():
    addon_path = cur_path / 'ns3-addon'
    os.chdir(addon_path)
    print(f'{sys.executable} addon_setup.py --wafdir={args.wafdir} --noreconf --norebuild')
    os.system(f'{sys.executable} addon_setup.py --wafdir={args.wafdir} --noreconf --norebuild')


def exec_env_setup():
    env_path = cur_path / 'ns3-env'
    os.chdir(env_path)
    print(f'{sys.executable} env_setup.py --wafdir={args.wafdir} ')
    os.system(f'{sys.executable} env_setup.py --wafdir={args.wafdir} ')


if __name__ == '__main__':
    if not args.noaddon:
        print("\n\033[1;36;49mExecuting addon_setup\033[0m\n")
        exec_addon_setup()
    if not args.noenv:
        print("\n\033[1;36;49mExecuting env_setup\033[0m\n")
        exec_env_setup()