mutmut_helper

Complexity

Total Complexity

8

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	cache_files()	0	11	5
A	modified_files()	0	8	3

import os.path
import subprocess


def modified_files():
    for file_ in (
        subprocess.run(["hg", "status", "--modified"], capture_output=True)
        .stdout.decode()
        .split("\n")
    ):
        if file_:
            yield file_[2:]


def cache_files(file_list):
    for file_ in file_list:
        dirname, fn = os.path.split(file_)
        cache_dir = os.path.join(dirname, "__pycache__")
        try:
            cache_files = os.listdir(cache_dir)
        except FileNotFoundError:
            continue
        for cache_file in cache_files:
            if cache_file.startswith(fn):
                yield os.path.join(cache_dir, cache_file)


if __name__ == "__main__":
    for cache_file in cache_files(modified_files()):
        print(cache_file)
        os.remove(cache_file)