40 lines
1.2 KiB
Python
Executable File
40 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""Generate CSV of pychopper stats."""
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
|
|
import pandas as pd
|
|
|
|
from .util import wf_parser # noqa: ABS101
|
|
|
|
|
|
def argparser():
|
|
"""Argument parser for entrypoint."""
|
|
parser = wf_parser("generate_pychopper_stats")
|
|
parser.add_argument("--data", required=True, help="")
|
|
parser.add_argument("--output_dir", required=True, help="")
|
|
|
|
return parser
|
|
|
|
|
|
def generate_pychopper_stats(tsv, output):
|
|
"""Make CSV of pychopper stats."""
|
|
classified_path = os.path.join(output, "pychopper_stats.csv")
|
|
df = pd.read_csv(tsv, sep="\t", index_col="Name")
|
|
classified = df.loc[df["Category"] == "Classification"]\
|
|
.copy().reset_index().rename(columns={'Name': 'Classification'})
|
|
classified["Percentage"] = \
|
|
100 * classified["Value"] / classified["Value"].sum()
|
|
tuning = df.loc[df["Category"] == "AutotuneSample"]\
|
|
.copy().reset_index().rename(columns={'Name': 'Filter'})
|
|
tuning.to_csv(classified_path)
|
|
|
|
|
|
def main(args):
|
|
"""Run entry point."""
|
|
assert os.path.isfile(args.data)
|
|
assert os.path.isdir(args.output_dir)
|
|
generate_pychopper_stats(tsv=args.data, output=args.output_dir)
|