#36: add debugtalk.py local per-directory plugins

ate/__init__.py

@@ -1 +1 @@
-__version__ = '0.5.3'
+__version__ = '0.5.4'

ate/context.py

@@ -46,6 +46,9 @@ class Context(object):
         self.testcase_parser.bind_functions(self.testcase_functions_config)
         self.testcase_parser.bind_variables(self.testcase_variables_mapping)
 
+        if level == "testset":
+            self.import_module_functions(["ate.debugtalk"], "testset")
+
     def import_requires(self, modules):
         """ import required modules dynamicly
         """

ate/debugtalk.py

@@ -0,0 +1,26 @@
+import datetime
+import random
+import string
+import time
+
+from ate.exception import ParamsError
+
+
+def gen_random_string(str_len):
+    """ generate random string with specified length
+    """
+    return ''.join(
+        random.choice(string.ascii_letters + string.digits) for _ in range(str_len))
+
+def get_timestamp(str_len=13):
+    """ get timestamp string, length can only between 0 and 16
+    """
+    if isinstance(str_len, int) and 0 < str_len < 17:
+        return str(time.time()).replace(".", "")[:str_len]
+
+    raise ParamsError("timestamp length can only between 0 and 16.")
+
+def get_current_date(fmt="%Y-%m-%d"):
+    """ get current date, default format is %Y-%m-%d
+    """
+    return datetime.datetime.now().strftime(fmt)