1 files changed, 23 insertions, 17 deletions
diff --git a/vyroute/basic_function/DeleteRoute.py b/vyroute/basic_function/DeleteRoute.py
index 048406c..175b712 100644
--- a/vyroute/basic_function/DeleteRoute.py
+++ b/vyroute/basic_function/DeleteRoute.py
@@ -1,10 +1,9 @@
 # Copyright (c) 2016 Hochikong
-def deleteroute(obj, data):
+def deleteroute(obj, route_type):
     """This method provide a router configuration delete function
 
-    Parameter data example:
-    {'config':'rip'/'static'/'ospf'/'all'
-    }
+    Parameter example:
+    'rip'/'static'/'ospf'/'bgp'/'all'
 
     WARNING!
     When you use this function,please don't forget this func will delete all same type
@@ -12,43 +11,50 @@ def deleteroute(obj, data):
     If you do not want your setting disappear,you can delete router configuration manually or rewrite
     this func.
 
-    :param obj: a connection object
-    :param data: a python dictionary
-    :return: a python dictionary
+    :param obj: A connection object
+    :param route_type: Route type
+    :return: A message or an error
     """
     delete_basic_configuration = "delete protocols %s"
     delete_all_protocols = "delete protocols"
 
     try:
-        if data['config'] == "all":
+        if route_type == "all":
             obj.sendline(delete_all_protocols)
             obj.prompt()
             if len(obj.before) > obj.before.index('\r\n') + 2:
                 return obj.before
             else:
-                return {"Result": "Delete successfully."}
-        elif data['config'] == 'rip':
+                return "Result : Delete successfully."
+        elif route_type == 'rip':
             obj.sendline(delete_basic_configuration % 'rip')
             obj.prompt()
             if len(obj.before) > obj.before.index('\r\n') + 2:
                 return obj.before
             else:
-                return {"Result": "Delete successfully."}
-        elif data['config'] == 'static':
+                return "Result : Delete successfully."
+        elif route_type == 'static':
             obj.sendline(delete_basic_configuration % 'static')
             obj.prompt()
             if len(obj.before) > obj.before.index('\r\n') + 2:
                 return obj.before
             else:
-                return {"Result": "Delete successfully."}
-        elif data['config'] == 'ospf':
+                return "Result : Delete successfully."
+        elif route_type == 'bgp':
+            obj.sendline(delete_basic_configuration % 'bgp')
+            obj.prompt()
+            if len(obj.before) > obj.before.index('\r\n') + 2:
+                return obj.before
+            else:
+                return "Result : Delete successfully."
+        elif route_type == 'ospf':
             obj.sendline(delete_basic_configuration % 'ospf')
             obj.prompt()
             if len(obj.before) > obj.before.index('\r\n') + 2:
                 return obj.before
             else:
-                return {"Result": "Delete successfully."}
+                return "Result : Delete successfully."
         else:
-            return {"Error": "Nonsupport protocols type."}
+            return "Error : Nonsupport protocols type."
     except Exception as e:
-        return {"Error": e}
+        return e