Tryag File Manager - Edit By Mr.SaHr

//opt/nydus/pyvenv/lib64/python3.5/site-packages/nydus/scripts/ipc.py

"""Constants and functions used during inter-process communication"""
from datetime import datetime
from functools import partial
from typing import Any, Dict
import json
import re

CANONICAL_TIMESTRING_FORMAT = '%Y-%m-%dT%H:%M:%S.%fZ'
CANONICAL_TIMESTRING_FORMAT_REGEX = re.compile(
    r"^(\d{4})-(\d{2})-(\d{2})T(\d{2}):(\d{2}):(\d{2}).(\d{6})Z$")

EXIT_METHOD_NOT_FOUND = 13
EXIT_PACKAGE_IMPORT_FAILED = 21
EXIT_PACKAGE_NOT_FOUND = 42

MARKER_BEGIN_OP_RESULT = "*|*BEGIN*|*"
MARKER_END_OP_RESULT = "*|*END*|*"

class DateTimeEncoder(json.JSONEncoder):
    """Uses our canonical format."""
    def default(self, obj: Any) -> Any:  # pylint: disable=E0202,W0221
        if isinstance(obj, datetime):
            return obj.strftime(CANONICAL_TIMESTRING_FORMAT)
        return json.JSONEncoder.default(self, obj)  # pylint: disable=assignment-from-no-return

class DateTimeDecoder(json.JSONDecoder):
    """Uses our canonical format."""
    def __init__(self, *args, **kwargs):
        super().__init__(*args, object_hook=self.dict_to_object, **kwargs)

def dict_to_object(self, d: Dict[Any, Any]) -> Any:  # pylint: disable=C0103
        for key, value in d.items():
            if isinstance(value, str) and CANONICAL_TIMESTRING_FORMAT_REGEX.match(value):
                d[key] = datetime.strptime(value, CANONICAL_TIMESTRING_FORMAT)
        return d

dumps = partial(json.dumps, cls=DateTimeEncoder, separators=(',', ':'))  # pylint: disable=invalid-name
load = partial(json.load, cls=DateTimeDecoder)  # pylint: disable=invalid-name
loads = partial(json.loads, cls=DateTimeDecoder)  # pylint: disable=invalid-name