swh:1:snp:7ce5f1105410d5ee1ad6abfdc873986c25b579e5
Tip revision: 572f88f3e28c80c8419ee9f29784a926fe783e5c authored by Dirk Roorda on 03 February 2021, 09:32:17 UTC
export of node tuples from browser
export of node tuples from browser
Tip revision: 572f88f
lib.py
"""
# Uitility functions
Read and write TF node sets from/to file.
"""
import os
import pickle
import gzip
from .parameters import PICKLE_PROTOCOL, GZIP_LEVEL
from .core.helpers import console
def writeSets(sets, dest):
"""Writes a dictionary of named sets to file.
The dictionary will be written as a gzipped, pickled data structure.
Intended use: if you have constructed custom node sets that you want to use
in search templates that run in the TF browser.
Parameters
----------
sets: dict of sets
The keys are names for the values, which are sets of nodes.
dest: string
A file path. You may use `~` to refer to your home directory.
The result will be written from this file.
Returns
-------
boolean
`True` if all went well, `False` otherwise.
"""
destPath = os.path.expanduser(dest)
(baseDir, fileName) = os.path.split(destPath)
if not os.path.exists(baseDir):
try:
os.makedirs(baseDir, exist_ok=True)
except Exception:
console(f'Cannot create directory "{baseDir}"', error=True)
return False
with gzip.open(destPath, "wb", compresslevel=GZIP_LEVEL) as f:
pickle.dump(sets, f, protocol=PICKLE_PROTOCOL)
return True
def readSets(source):
"""Reads a dictionary of named sets from file.
This is used by the TF browser, when the user has passed a
`--sets=fileName` argument to it.
Parameters
----------
source: string
A file path. You may use `~` to refer to your home directory.
This file must contain a gzipped, pickled data structure.
Returns
-------
data | boolean
The data structure contained in the file if all went well, False otherwise.
"""
sourcePath = os.path.expanduser(source)
if not os.path.exists(sourcePath):
console(f'Sets file "{source}" does not exist.')
return False
with gzip.open(sourcePath, "rb") as f:
sets = pickle.load(f)
return sets