fix(controller+client): utf-8 encode only string types

Author: mboersma

client/deis.py

@@ -303,6 +303,14 @@ def dictify(args):
     return data
 
 
+def encode(obj):
+    """Return UTF-8 encoding for string objects."""
+    if isinstance(obj, basestring):
+        return obj.encode('utf-8')
+    else:
+        return obj
+
+
 def readable_datetime(datetime_str):
     """
     Return a human-readable datetime string from an ECMA-262 (JavaScript)
@@ -1123,14 +1131,13 @@ def config_list(self, args):
                 width = max(map(len, keys)) + 5
                 for k in keys:
                     v = values[k]
-                    k, v = k.encode('utf-8'), v.encode('utf-8')
-                    self._logger.info(("{k:<" + str(width) + "} {v}").format(**locals()))
+                    self._logger.info("{:<" + str(width) + "} {}".format(
+                        encode(k), encode(v)))
             else:
                 output = []
                 for k in keys:
                     v = values[k]
-                    k, v = k.encode('utf-8'), v.encode('utf-8')
-                    output.append("{k}={v}".format(**locals()))
+                    output.append("{}={}".format(encode(k), encode(v)))
                 self._logger.info(' '.join(output))
         else:
             raise ResponseError(response)
@@ -1175,8 +1182,7 @@ def config_set(self, args):
                 self._logger.info('No configuration')
                 return
             for k, v in values.items():
-                k, v = k.encode('utf-8'), v.encode('utf-8')
-                self._logger.info("{k}: {v}".format(**locals()))
+                self._logger.info("{}: {}".format(encode(k), encode(v)))
         else:
             raise ResponseError(response)
 

controller/registry/private.py

@@ -158,6 +158,13 @@ def _put_tag(image_id, repository_path, tag):
 
 # utility functions
 
+def encode(obj):
+    """Return UTF-8 encoding for string objects."""
+    if isinstance(obj, basestring):
+        return obj.encode('utf-8')
+    else:
+        return obj
+
 
 def _construct_env(env, config):
     "Update current environment with latest config"
@@ -168,10 +175,10 @@ def _construct_env(env, config):
         if k in config:
             # update values defined by config
             v = config.pop(k)
-        new_env.append("{}={}".format(k.encode('utf-8'), v.encode('utf-8')))
+        new_env.append("{}={}".format(encode(k), encode(v)))
     # add other config ENV items
     for k, v in config.items():
-        new_env.append("{}={}".format(k.encode('utf-8'), v.encode('utf-8')))
+        new_env.append("{}={}".format(encode(k), encode(v)))
     return new_env
 
 

tests/config_test.go

@@ -11,6 +11,7 @@ import (
 var (
 	configListCmd  = "config:list --app={{.AppName}}"
 	configSetCmd   = "config:set FOO=讲台 --app={{.AppName}}"
+	configSet2Cmd  = "config:set FOO=10 --app={{.AppName}}"
 	configUnsetCmd = "config:unset FOO --app={{.AppName}}"
 )
 
@@ -52,11 +53,13 @@ func configListTest(
 }
 
 func configSetTest(t *testing.T, params *utils.DeisTestConfig) {
-	utils.Execute(t, configSetCmd, params, false, "")
+	utils.Execute(t, configSetCmd, params, false, "讲台")
 	utils.CheckList(t, appsInfoCmd, params, "(v3)", false)
+	utils.Execute(t, configSet2Cmd, params, false, "10")
+	utils.CheckList(t, appsInfoCmd, params, "(v4)", false)
 }
 
 func configUnsetTest(t *testing.T, params *utils.DeisTestConfig) {
 	utils.Execute(t, configUnsetCmd, params, false, "")
-	utils.CheckList(t, appsInfoCmd, params, "(v4)", false)
+	utils.CheckList(t, appsInfoCmd, params, "(v5)", false)
 }