1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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()
|
