diff options
author | Hochikong <michellehzg@gmail.com> | 2016-05-23 19:38:16 +0800 |
---|---|---|
committer | Hochikong <michellehzg@gmail.com> | 2016-05-23 19:38:16 +0800 |
commit | 8e2f9aa509a5d90897844c196675836a10068237 (patch) | |
tree | c5aaf9af194d730ecd1ce486e64b07cd2f39b96c /vyroute/basic_function | |
parent | 71db4ed81a299a8f4b1947272bccb7966360f98f (diff) | |
download | python-vyos-mgmt-8e2f9aa509a5d90897844c196675836a10068237.tar.gz python-vyos-mgmt-8e2f9aa509a5d90897844c196675836a10068237.zip |
add new file and modify some wrong
Diffstat (limited to 'vyroute/basic_function')
-rw-r--r-- | vyroute/basic_function/DeleteRoute.py | 57 | ||||
-rw-r--r-- | vyroute/basic_function/Modifylo.py | 50 | ||||
-rw-r--r-- | vyroute/basic_function/RIPRoute.py | 55 | ||||
-rw-r--r-- | vyroute/basic_function/StaticRoute.py | 79 | ||||
-rw-r--r-- | vyroute/basic_function/__init__.py | 0 |
5 files changed, 241 insertions, 0 deletions
diff --git a/vyroute/basic_function/DeleteRoute.py b/vyroute/basic_function/DeleteRoute.py new file mode 100644 index 0000000..87af263 --- /dev/null +++ b/vyroute/basic_function/DeleteRoute.py @@ -0,0 +1,57 @@ +# author=hochikong +from Exscript.protocols import SSH2 +from Exscript import Account + + +def deleteroute(data): + """This method provide a router configuration delete function + + Parameter data example: + {'router':'vyos@172.16.77.188','passwd':'vyos', + 'config':'rip' + } + + WARNING! + When you use this function,please don't forget this func will delete all same type + router configuration,when your 'config' in data is 'rip',it will delete all rip router setting. + If you do not want your setting disappear,you can delete router configuration manually or rewrite + this func. + + :param data: a python dictionary + :return: a python dictionary + """ + delete_basic_configuration = "delete protocols %s" + + try: + stringlist = list(data['router']) + divi = stringlist.index('@') + user = ''.join(stringlist[:divi]) + passwd = data['passwd'] + address = ''.join(stringlist[divi + 1:]) + account = Account(user, passwd) + conn = SSH2() + conn.connect(address) + conn.login(account) + + # configure mode + conn.execute("configure") + + # delete specific configuration + conn.execute(delete_basic_configuration % data['config']) + + # commit configuration + conn.execute("commit") + + # save configuration + conn.execute("save") + + # exit configure mode + conn.execute("exit") + + # close connection + conn.close(force=True) + + return {"Result": "Delete successfully"} + + except Exception, e: + return {"Error": e}
\ No newline at end of file diff --git a/vyroute/basic_function/Modifylo.py b/vyroute/basic_function/Modifylo.py new file mode 100644 index 0000000..6808c0d --- /dev/null +++ b/vyroute/basic_function/Modifylo.py @@ -0,0 +1,50 @@ +# author=hochikong +from Exscript.protocols import SSH2 +from Exscript import Account + + +def modifylo(data): + """This method provide a loopback address configuration function + + Parameter data example: + {'router':'vyos@172.16.77.188','passwd':'vyos','config':'1.1.1.1/32' + } + + :param data: a python dictionary + :return: a python dictionary + """ + lo_basic_configuration = "set interfaces loopback lo address %s" + + try: + stringlist = list(data['router']) + divi = stringlist.index('@') + user = ''.join(stringlist[:divi]) + passwd = data['passwd'] + address = ''.join(stringlist[divi+1:]) + account = Account(user, passwd) + conn = SSH2() + conn.connect(address) + conn.login(account) + +# configure mode + conn.execute("configure") + +# configure loopback interface lo address + conn.execute(lo_basic_configuration % data['config']) + + # commit configuration + conn.execute("commit") + + # save configuration + conn.execute("save") + + # exit configure mode + conn.execute("exit") + + # close connection + conn.close(force=True) + + return {"Result": "Configured successfully"} + + except Exception, e: + return {'Error': e} diff --git a/vyroute/basic_function/RIPRoute.py b/vyroute/basic_function/RIPRoute.py new file mode 100644 index 0000000..f559a17 --- /dev/null +++ b/vyroute/basic_function/RIPRoute.py @@ -0,0 +1,55 @@ +# author=hochikong +from Exscript.protocols import SSH2 +from Exscript import Account + + +def riproute(data): + """This method provide a RIP protocols router configuration function + + Parameter data example: + {'router':'vyos@172.16.77.188','passwd':'vyos', + 'config':['192.168.10.0/24','10.20.10.0/24'], + } + + :param data: a python dictionary + :return: a python dictionary + """ + rip_basic_configuration = "set protocols rip network %s" + redistribute_configuration = "set protocols rip redistribute connected" + try: + stringlist = list(data['router']) + divi = stringlist.index('@') + user = ''.join(stringlist[:divi]) + passwd = data['passwd'] + address = ''.join(stringlist[divi + 1:]) + account = Account(user, passwd) + conn = SSH2() + conn.connect(address) + conn.login(account) + + # configure mode + conn.execute("configure") + + # configure RIP router + for i in data['config']: + conn.execute(rip_basic_configuration % i) + + conn.execute(redistribute_configuration) + + # commit configuration + conn.execute("commit") + + # save configuration + conn.execute("save") + + # exit configure mode + conn.execute("exit") + + # close connection + conn.close(force=True) + + return {"Result": "Configured successfully"} + + except Exception, e: + return {"Error": e} + diff --git a/vyroute/basic_function/StaticRoute.py b/vyroute/basic_function/StaticRoute.py new file mode 100644 index 0000000..f9fa233 --- /dev/null +++ b/vyroute/basic_function/StaticRoute.py @@ -0,0 +1,79 @@ +# author=hochikong +from Exscript.protocols import SSH2 +from Exscript import Account + + +def staticroute(data): + """This method provide a basic static router configuration function + + Parameter data example: + {'router':'vyos@172.16.77.188','passwd':'vyos', + 'config':[{'target':'10.20.10.0/24','next-hop':'10.20.10.1','distance':'1'}, + {'target':"192.168.20.0/24','next-hop':'192.168.20.1','distance':'1'} + ] + } + + :param data: a python dictionary + :return:a python dictionary + """ + static_basic_configuration = "set protocols static route %s next-hop %s distance %s" + + try: + stringlist = list(data['router']) + divi = stringlist.index('@') + user = ''.join(stringlist[:divi]) + passwd = data['passwd'] + address = ''.join(stringlist[divi+1:]) + account = Account(user, passwd) + conn = SSH2() + conn.connect(address) + conn.login(account) + + # configure mode + conn.execute("configure") + + # configure static router + for i in data['config']: + conn.execute(static_basic_configuration % (i['target'], + i['next-hop'], + i['distance'])) + + # commit configuration + conn.execute("commit") + + # save configuration + conn.execute("save") + + # exit configure mode + conn.execute("exit") + + # close connection + conn.close(force=True) + + return {"Result": "Configured successfully"} + + except Exception, e: + return {'Error': e} + + + + + + + + + + + + + + + + + + + + + + + diff --git a/vyroute/basic_function/__init__.py b/vyroute/basic_function/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/vyroute/basic_function/__init__.py |