!511 check the legal of dims in the entrance of interface and make it more rigorous

Merge pull request !511 from wangshuide/wsd0727

mindinsight/datavisual/processors/tensor_processor.py

@@ -42,8 +42,7 @@ def convert_array_from_str(dims, limit=0):
     Raises:
         ParamValueError, If flexible dimensions exceed limit value.
     """
-    dims = dims.replace('[', '') \
-               .replace(']', '')
+    dims = dims.strip().lstrip('[').rstrip(']')
     dims_list = []
     count = 0
     for dim in dims.split(','):
@@ -131,6 +130,23 @@ class TensorProcessor(BaseProcessor):
             UrlDecodeError, If unquote train id error with strict mode.
         """
         Validation.check_param_empty(train_id=train_ids, tag=tags)
+        if dims is not None:
+            if not isinstance(dims, str):
+                raise ParamValueError('The type of dims must be str, but got {}.'.format(type(dims)))
+            dims = dims.strip()
+            if not (dims.startswith('[') and dims.endswith(']')):
+                raise ParamValueError('The value: {} of dims must be '
+                                      'start with `[` and end with `]`.'.format(dims))
+            for dim in dims[1:-1].split(','):
+                dim = dim.strip()
+                if dim == ":":
+                    continue
+                if dim.startswith('-'):
+                    dim = dim[1:]
+                if not dim.isdigit():
+                    raise ParamValueError('The value: {} of dims in the square brackets '
+                                          'must be int or `:`.'.format(dims))
+
         for index, train_id in enumerate(train_ids):
             try:
                 train_id = unquote(train_id, errors='strict')