samples/py_swtpm_localca/swtpm_utils.py

""" swtpm_logging.py
"""

# pylint: disable=W0703

import os
import sys


def append_to_file(filename, string):
    """" Append a string to a file """
    try:
        filedesc = os.open(filename, os.O_WRONLY|os.O_APPEND|os.O_CREAT|os.O_NOFOLLOW, 0o640)
        os.write(filedesc, string.encode('utf-8'))
        os.close(filedesc)
    except Exception as ex:
        sys.stdout.write("Error: %s\n" % ex)
        sys.stdout.write(string)
        try:
            if filedesc > 0:
                os.close(filedesc)
        except Exception:
            pass


def logit(logfile, string):
    """ Print the given string to stdout or into the logfile """
    if len(logfile) == 0:
        sys.stdout.write(string)
    else:
        append_to_file(logfile, string)


def logerr(logfile, string):
    """ Print the given string to stderr or into the logfile """
    if len(logfile) == 0:
        sys.stdout.write(string)
    else:
        append_to_file(logfile, string)