Refactored template presistence module

a10networks · Apr 16, 2020 · 8121dbd · 8121dbd
1 parent 9f127af
commit 8121dbd
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/acos_client/v30/slb/template/persistence.py b/acos_client/v30/slb/template/persistence.py
@@ -37,8 +37,10 @@ def exists(self, name):
     def create(self, name, **kwargs):
         if self.exists(name):
             raise acos_errors.Exists
-        cookie_name = kwargs.get("cookie_name", None)
-        self._post(self.prefix, self.get_params(name, cookie_name=cookie_name), **kwargs)
+        self._post(self.prefix,
+                   self.get_params(name,
+                                   cookie_name=kwargs.get("cookie_name")),
+                   **kwargs)
 
     def delete(self, name, **kwargs):
         self._delete(self.prefix + name, **kwargs)
@@ -51,11 +53,10 @@ def __init__(self, client):
         super(CookiePersistence, self).__init__(client)
 
     def get_params(self, name, **kwargs):
-        cookie_name = kwargs.get("cookie_name", None)
         return {
             "cookie": {
                 "name": name,
-                "cookie-name": cookie_name
+                "cookie-name": kwargs.get("cookie_name")
             }
         }