manyi/script/clean.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python
import shutil

from base import *

TESTCASE_DIR = os.path.join(PROJECT_ROOT, 'testcase')
path_joiner = lambda path, filename : os.path.join(path, filename)
testcase_path_joiner = lambda filename : path_joiner(TESTCASE_DIR, filename)

files_to_delete = {'ghostdriver.log', 'geckodriver.log'}
dirs_to_delete  = {'.cache', '__pycache__'}

def delete(path):
    if os.path.isfile(path):
        os.remove(path)
    elif os.path.isdir(path):
        shutil.rmtree(path)

@click.command()
def clean():
    for root, dirs, files in os.walk(PROJECT_ROOT):
        matches = ( set(files) & files_to_delete ) | ( set(dirs) & dirs_to_delete )
        if len(matches):
            map(delete, [ path_joiner(root, _) for _ in matches ])
    click.echo('success!')


if __name__ == "__main__":
    clean()