import
pandas as pd
from
pandas
import
ExcelWriter
import
os
def
get_data():
warnings
=
open
(
'warnings.txt'
,
'r'
)
suggestions
=
open
(
'suggestions.txt'
,
'r'
)
packages
=
open
(
'packages.txt'
,
'r'
)
shells
=
open
(
'shells.txt'
,
'r'
)
warn_data
=
warnings.readlines()
sugg_data
=
suggestions.readlines()
pack_data
=
packages.read()
shell_data
=
shells.readlines()
return
warn_data, sugg_data, pack_data, shell_data
def
clean_data():
warn, sugg, pack, shell
=
get_data()
warn_clean
=
[]
for
line
in
warn:
warn_clean.append(line.split(
'|'
))
for
i
in
range
(
len
(warn_clean)):
warn_clean[i]
=
warn_clean[i][:
2
]
sugg_clean
=
[]
for
line
in
sugg:
sugg_clean.append(line.split(
'|'
))
for
i
in
range
(
len
(sugg_clean)):
sugg_clean[i]
=
sugg_clean[i][:
2
]
pack_clean
=
[]
pack
=
pack.split(
'|'
)
pack_clean
=
pack
del
pack_clean[
0
]
shell_clean
=
[]
for
i
in
range
(
len
(shell)):
shell_clean.append(shell[i].rstrip(
'\n'
))
return
warn_clean, sugg_clean, pack_clean, shell_clean
def
convert_to_excel():
warnings, suggestions, packages, shells
=
clean_data()
try
:
os.mkdir(
'outputs'
)
except
(Exception):
pass
os.chdir(
'outputs'
)
warn_packages
=
[]
warn_text
=
[]
for
i
in
range
(
len
(warnings)):
warn_packages.append(warnings[i][
0
])
for
i
in
range
(
len
(warnings)):
warn_text.append(warnings[i][
1
])
print
(warn_packages, warn_text)
warn
=
pd.DataFrame()
warn[
'Packages'
]
=
warn_packages
warn[
'warnings'
]
=
warn_text
writer
=
ExcelWriter(
'warnings.xlsx'
)
warn.to_excel(writer,
'report1'
, index
=
False
)
workbook
=
writer.book
worksheet
=
writer.sheets[
'report1'
]
worksheet.set_column(
'A:A'
,
15
)
worksheet.set_column(
'B:B'
,
45
)
writer.save()
sugg_packages
=
[]
sugg_text
=
[]
for
i
in
range
(
len
(suggestions)):
sugg_packages.append(suggestions[i][
0
])
for
i
in
range
(
len
(suggestions)):
sugg_text.append(suggestions[i][
1
])
sugg
=
pd.DataFrame()
sugg[
'Packages'
]
=
sugg_packages
sugg[
'suggestions'
]
=
sugg_text
writer1
=
ExcelWriter(
'suggestions.xlsx'
)
sugg.to_excel(writer1,
'report2'
, index
=
False
)
workbook
=
writer1.book
worksheet
=
writer1.sheets[
'report2'
]
worksheet.set_column(
'A:A'
,
25
)
worksheet.set_column(
'B:B'
,
120
)
writer1.save()
pack_data
=
pd.DataFrame()
pack_data[
'Packages'
]
=
packages
writer1
=
ExcelWriter(
'packages.xlsx'
)
pack_data.to_excel(writer1,
'report3'
, index
=
False
)
workbook
=
writer1.book
worksheet
=
writer1.sheets[
'report3'
]
worksheet.set_column(
'A:A'
,
75
)
writer1.save()
os.chdir(
'..'
)
if
__name__
=
=
'__main__'
:
warnings, suggestions, packages, shells
=
clean_data()
convert_to_excel()