[Tool] Add tool to split log to tc wise logs

Adding tool to split glusto-tests logs
into tc wise logs.

usage: log_splitter [-h] -f LOG_FILE [-d DESTINATION_DIR]

Tool to split glusto logs to individual testcase logs.

optional arguments:
  -h, --help            show this help message and exit
  -f LOG_FILE, --log_file LOG_FILE
                        Glusto test log file
  -d DESTINATION_DIR, --dist-dir DESTINATION_DIR
                        Path were individual test logs are to be stored.

Change-Id: I776a1455f9f70c13ae6ad9d11f23a4b5366c5f6f
Signed-off-by: kshithijiyer <kshithij.ki@gmail.com>

1 files changed, 100 insertions, 0 deletions

diff --git a/tools/log_splitter/log_splitter.py b/tools/log_splitter/log_splitter.py
new file mode 100644
index 000000000..e433b3ee1
--- /dev/null
+++ b/tools/log_splitter/log_splitter.py
@@ -0,0 +1,100 @@
+#  Copyright (C) 2020 Red Hat, Inc. <http://www.redhat.com>
+#
+#  This program is free software; you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation; either version 3 of the License, or
+#  any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY :or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License along
+#  with this program; if not, write to the Free Software Foundation, Inc.,
+#  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+
+# Imports needed by the script.
+import argparse
+import os
+import sys
+
+
+def check_and_create_dir_if_not_present(directory):
+    """
+    A function to check and create directory if not present
+
+    Args:
+        directory(str): Directory to be created if not present
+
+    Retuns:
+        bool: True if successful else False
+    """
+    if not os.path.isdir(directory):
+        cmd = "mkdir -p {}".format(directory)
+        ret = os.system(cmd)
+        if ret:
+            return False
+        print("[INFO]: Dir created successfully")
+    else:
+        print("[INFO]: The dir already exists")
+    return True
+
+
+def main():
+    """
+    Main function of the tool.
+    """
+    # Setting up command line arguments.
+    parser = argparse.ArgumentParser(
+        description="Tool to split glusto logs to individual testcase logs."
+        )
+    parser.add_argument(
+        '-f', '--log_file', type=str, dest='log_file', required=True,
+        help="Glusto test log file")
+    parser.add_argument(
+        '-d', '--dist-dir', type=str, default=".", dest="destination_dir",
+        help="Path were individual test logs are to be stored.")
+    args = parser.parse_args()
+
+    # Fetching the values from command line.
+    log_file = args.log_file
+    destination_dir = args.destination_dir
+
+    # Check and create dir if not present
+    if not check_and_create_dir_if_not_present(destination_dir):
+        sys.exit("[ERROR]: Unable to create dir")
+
+    with open(log_file, 'r', encoding="ISO-8859-1") as log_file_fd:
+
+        # Read lines and set flag to check if
+        # file is open
+        file_open_flag = False
+        while True:
+            line = log_file_fd.readline()
+            if not line:
+                break
+
+            # Check if line is starting line.
+            if '(setUp) Starting Test : ' in line:
+                if file_open_flag:
+                    file_open_flag = False
+
+                # Open new fd for individual test
+                # file
+                filename = line.split(' ')[7]
+                if destination_dir != '.':
+                    filename = os.path.join(destination_dir,
+                                            filename)
+                file_open_flag = True
+
+            # Write lines to individual test file
+            if file_open_flag:
+                with open(filename, 'w') as test_file:
+                    test_file.write(line)
+
+    print("[INFO]: Log file split completed")
+
+
+if __name__ == "__main__":
+    main()