summaryrefslogtreecommitdiff
path: root/cloudinit/atomic_helper.py
blob: a3cfd942666acea8468836455463e4a8295d2380 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/usr/bin/python
# vi: ts=4 expandtab

import json
import os
import tempfile

_DEF_PERMS = 0o644


def write_file(filename, content, mode=_DEF_PERMS, omode="wb"):
    # open filename in mode 'omode', write content, set permissions to 'mode'
    tf = None
    try:
        tf = tempfile.NamedTemporaryFile(dir=os.path.dirname(filename),
                                         delete=False, mode=omode)
        tf.write(content)
        tf.close()
        os.chmod(tf.name, mode)
        os.rename(tf.name, filename)
    except Exception as e:
        if tf is not None:
            os.unlink(tf.name)
        raise e


def write_json(filename, data, mode=_DEF_PERMS):
    # dump json representation of data to file filename.
    return write_file(
        filename, json.dumps(data, indent=1, sort_keys=True) + "\n",
        omode="w", mode=mode)