from pathlib import Path
import shutil, sys
from types import SimpleNamespace as sn

from .lib import data, log

def main():
    msgs_path = Path(__file__).parent / 'data/messages.json'
    msgs = sn(**{ key:val['message'] for key,val in data.json.read(msgs_path)['clean'].items() })
    targets = ['*.pyc']
    if '--py2' not in sys.argv:
        targets.extend(['dist', 'build', '*_cache', '__pycache__', '*.egg-info'])
    for target in targets:
        for path in Path('.').rglob(target):
            if path.is_dir():
                shutil.rmtree(path)
                log.info(f'{msgs.log_REMOVED} {path}/')
            elif path.is_file():
                path.unlink()
                log.info(f'{msgs.log_REMOVED} {path}')
    log.success(f'{msgs.log_CLEAN_COMPLETE}!')

if __name__ == '__main__' : main()