diff options
| -rw-r--r-- | fuzzycat/cluster.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/fuzzycat/cluster.py b/fuzzycat/cluster.py index 9b97509..32faec6 100644 --- a/fuzzycat/cluster.py +++ b/fuzzycat/cluster.py @@ -24,13 +24,14 @@ import orjson as json import fuzzy DEFAULT_CACHE_DIR = os.path.join(os.path.expanduser("~"), ".cache", "fuzzycat") +TMP_PREFIX = 'fuzzycat-' -def sort_by_column(filename, mode="w", opts="-k 2", fast=True): +def sort_by_column(filename, mode="w", opts="-k 2", fast=True, prefix=TMP_PREFIX): """ Sort tabular file with sort(1), returns the filename of the sorted file. """ - with tempfile.NamedTemporaryFile(delete=False, mode=mode) as tf: + with tempfile.NamedTemporaryFile(delete=False, mode=mode, prefix=prefix) as tf: env = os.environ.copy() if fast: env["LC_ALL"] = "C" @@ -60,12 +61,12 @@ def cut(f=0, sep='\t'): """ return lambda v: v.split(sep)[f] -def cluster_by_title(args): +def cluster_by_title(args, prefix=TMP_PREFIX): """ Basic example for a three stage process: extract, sort, group. Speed is about: 20K/s (json roundtrip, sorting, grouping). """ - with tempfile.NamedTemporaryFile(delete=False, mode="w") as tf: + with tempfile.NamedTemporaryFile(delete=False, mode="w", prefix=prefix) as tf: for line in fileinput.input(files=args.files if len(args.files) > 0 else ('-', )): doc = json.loads(line) try: @@ -91,7 +92,7 @@ def cluster_by_title_normalized(args): Normalize title, e.g. analysisofheritability. 17k/s. """ pattern = re.compile('[\W_]+', re.UNICODE) - with tempfile.NamedTemporaryFile(delete=False, mode="w") as tf: + with tempfile.NamedTemporaryFile(delete=False, mode="w", prefix=prefix) as tf: for line in fileinput.input(files=args.files if len(args.files) > 0 else ('-', )): doc = json.loads(line) try: @@ -117,7 +118,7 @@ def cluster_by_title_nysiis(args): """ Soundex on title. """ - with tempfile.NamedTemporaryFile(delete=False, mode="w") as tf: + with tempfile.NamedTemporaryFile(delete=False, mode="w", prefix=prefix) as tf: for line in fileinput.input(files=args.files if len(args.files) > 0 else ('-', )): doc = json.loads(line) try: |