Gecko [ goons4good.org ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2022-10-11 05:09
__init__.py	6.11 KB	-rw-r--r--	2022-05-16 09:16
calc.py	5.54 KB	-rw-r--r--	2022-05-16 09:16
check.py	10.61 KB	-rw-r--r--	2022-05-16 09:16
file.py	1.92 KB	-rw-r--r--	2022-05-16 09:16
key.py	2.22 KB	-rw-r--r--	2022-05-16 09:16
local.py	942 B	-rw-r--r--	2022-05-16 09:16
reg.py	4 KB	-rw-r--r--	2022-05-16 09:16
runner.py	908 B	-rw-r--r--	2022-05-16 09:16
status.py	1.03 KB	-rw-r--r--	2022-05-16 09:16
timer.py	829 B	-rw-r--r--	2022-05-16 09:16
wheel.py	551 B	-rw-r--r--	2022-05-16 09:16

Rename

"""
Writes matches to disk to verify activity, helpful when testing

Normally this is used by giving the name of the file (without a path) that the
data will be saved to. If for instance you use ``foo`` as the name:

.. code-block:: yaml

foo:
      file.save

Then the file will be saved to:

.. code-block:: bash

<salt cachedir>/thorium/saves/foo

You may also provide an absolute path for the file to be saved to:

.. code-block:: yaml

/tmp/foo.save:
        file.save

Files will be saved in JSON format. However, JSON does not support ``set()``s.
If you are saving a register entry that contains a ``set()``, then it will fail
to save to JSON format. However, you may pass data through a filter which makes
it JSON compliant:

.. code-block:: yaml

foo:
      file.save:
        filter: True

Be warned that if you do this, then the file will be saved, but not in a format
that can be re-imported into Python.
"""

import os

import salt.utils.data
import salt.utils.files
import salt.utils.json

def save(name, filter=False):
    """
    Save the register to <salt cachedir>/thorium/saves/<name>, or to an
    absolute path.

If an absolute path is specified, then the directory will be created
    non-recursively if it doesn't exist.

USAGE:

.. code-block:: yaml

foo:
          file.save

/tmp/foo:
          file.save
    """
    ret = {"name": name, "changes": {}, "comment": "", "result": True}
    if name.startswith("/"):
        tgt_dir = os.path.dirname(name)
        fn_ = name
    else:
        tgt_dir = os.path.join(__opts__["cachedir"], "thorium", "saves")
        fn_ = os.path.join(tgt_dir, name)
    if not os.path.isdir(tgt_dir):
        os.makedirs(tgt_dir)
    with salt.utils.files.fopen(fn_, "w+") as fp_:
        if filter is True:
            salt.utils.json.dump(salt.utils.data.simple_types_filter(__reg__), fp_)
        else:
            salt.utils.json.dump(__reg__, fp_)
    return ret