Skip to content

Commit

Permalink
refactoring
Browse files Browse the repository at this point in the history
  • Loading branch information
l8556 committed Sep 12, 2024
1 parent c31a007 commit de18bdb
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions frameworks/report/report.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,23 +25,24 @@ def value_count(df: pd.DataFrame, column_name: str) -> str:

def insert_column(self, path: str, location: str, column_name: str, value: str, delimiter='\t') -> pd.DataFrame:
df = self.read(path, delimiter=delimiter)
if column_name not in df.columns:
if column_name not in df.columns:
df.insert(loc=df.columns.get_loc(location), column=column_name, value=value)
else:
print(f"[green]|INFO| Column `{column_name}` already exists in `{path}`")
print(f"[cyan]|INFO| Column `{column_name}` already exists in `{path}`")
return df

def merge(self, reports: list, result_csv_path: str, delimiter='\t') -> str | None:
if reports:
merge_reports = []
for csv_ in reports:
if isfile(csv_):
report = self.read(csv_, delimiter)
if report is not None:
merge_reports.append(report)
merge_reports = [
self.read(csv_, delimiter)
for csv_ in reports
if isfile(csv_) and (report := self.read(csv_, delimiter)) is not None
]

if merge_reports:
df = pd.concat(merge_reports, ignore_index=True)
df.to_csv(result_csv_path, index=False, sep=delimiter)
return result_csv_path

print('[green]|INFO| No files to merge')

@staticmethod
Expand Down

0 comments on commit de18bdb

Please sign in to comment.