diff --git a/py-scripts/check_argparse.py b/py-scripts/check_argparse.py index 56331928..cdd65ce7 100644 --- a/py-scripts/check_argparse.py +++ b/py-scripts/check_argparse.py @@ -5,13 +5,6 @@ import pandas as pd import argparse -def get_tag(x, tag): - try: - return x[tag] - except: - return False - - def main(): parser = argparse.ArgumentParser( prog="check_argparse.py", @@ -48,7 +41,9 @@ def main(): 'description', 'epilog', 'usage']: - df[tag] = [get_tag(x, tag) for x in df['results']] + for result in df['results']: + if tag in result: + df[tag] = df['results'][tag] df['details'] = df['description'] + df['epilog'] + df['usage'] df.to_csv(args.output + '.csv', index=False)