function of generating workflow config file in server part of testing-scheduler

[bottlenecks.git] / testing-scheduler / server / src / conductor_processor / task.py

diff --git a/testing-scheduler/server/src/conductor_processor/task.py b/testing-scheduler/server/src/conductor_processor/task.py
new file mode 100644 (file)
index 0000000..6f25aef
--- /dev/null
+++ b/testing-scheduler/server/src/conductor_processor/task.py
@@ -0,0 +1,28 @@
+##############################################################################\r
+# Copyright (c) 2018 Huawei Technologies Co.,Ltd. and others\r
+#\r
+# All rights reserved. This program and the accompanying materials\r
+# are made available under the terms of the Apache License, Version 2.0\r
+# which accompanies this distribution, and is available at\r
+# http://www.apache.org/licenses/LICENSE-2.0\r
+##############################################################################\r
+\r
+import json\r
+import os\r
+\r
+\r
+class TaskFile(object):\r
+    def __init__(self, taskName='task_0'):\r
+        self._defaultConfFile = self._getFilePath("defaultTaskFile.json")\r
+        with open(self._defaultConfFile) as defaultConf:\r
+            self._jsonObj = json.load(defaultConf)\r
+        self._jsonObj['name'] = taskName\r
+\r
+    def generateFromStep(self, stepObject):\r
+        self._jsonObj['name'] = stepObject.getName()\r
+        print "taskFile:", self._jsonObj['name']\r
+        return self._jsonObj\r
+\r
+    def _getFilePath(self, fileName):\r
+        dirPath = os.path.dirname(os.path.realpath(__file__))\r
+        return os.path.join(dirPath, fileName)\r