breakdown_gpo.py

#! /bin/python3
# Version 3.0.2

import sys, time, shutil, os
from modules.breakdown import breakdown
from modules.arguments import parse
from modules.help import help

def main():
    # Pre-Breakdown Logic
    try:
        if len(sys.argv)==1 or "--help" in sys.argv or "-h" in sys.argv:
            help()
        else:
            input_path, output_path = parse(sys.argv[1:])
    except KeyboardInterrupt:
        print()
        quit()
    except Exception as e:
        print(e)
    # Breakdown Logic
    try:
        start_time = time.time()
        breakdown(input_path, output_path)
        print(f"Breakdown took {time.time() - start_time :.2f} seconds.")
    except KeyboardInterrupt:
        if(os.path.exists(output_path)):
            print("Interupt detected.")
            shutil.rmtree(output_path)
    except Exception as e:
        print(e)

if __name__=='__main__':
    main()