#!/usr/bin/env python

import sciris as sc
import sys

clean_nb = False if 'fast' in sys.argv else True


T = sc.timer()

# Get files
here = sc.thispath()
nbfiles = sc.getfilelist(here / '*.ipynb')
files1 = sc.getfilelist(here / '*.obj')
files2 = sc.getfilelist(here / 'my-*.*')
files = sorted(set(files1 + files2))

# Remove auto-generated files
if len(files):
    print(f'Deleting {len(files)} files:')
    for i,f in enumerate(files):
        print(f'{i}. {f}')
    print('...in 2 seconds')
    sc.timedsleep(2)
    sc.rmpath(files)
else:
    print('No files to delete')

# Clean notebook files
def clear_notebook(file):
    cmd = f'jupyter nbconvert --clear-output --inplace {file}'
    sc.runcommand(cmd, printinput=True, printoutput=True)

if clean_nb:
    print('Clearing notebook output:')
    sc.parallelize(clear_notebook, iterarg=nbfiles)
else:
    print('\nNot clearing notebook output\n')

# Undo modifications to dancing_lines.gif
sc.runcommand('git checkout -- dancing_lines.gif')

T.toc('Done!\n\n')