123456789101112131415161718192021222324252627282930 |
- # -*- 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()
|