isotopic_logging.AutoprefixInjector

Complexity

Total Complexity

1

Size/Duplication

Total Lines	5
Duplicated Lines	0 %

Test Coverage

Coverage

100%

1 Method

Rating	Name	Duplication	Size	Complexity
A	__init__()	0	3	1

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

import datetime
import time

from .defaults import ELAPSED_TIME_FORMAT
from .generators import generate_oid
from .prefix import make_prefix, join_prefix


class DirectPrefixInjector(object):

    __slots__ = ['prefix', 'enter_time', ]

    def __init__(self, prefix):
        self.prefix = prefix

        # `enter_time` will be set by context manager
        self.enter_time = None

    def mark(self, message):
        # Use `format` as it will automatically convert parameters to strings
        return "{0}{1}".format(self.prefix, message)

    @property
    def elapsed_time(self):
        try:
            return time.time() - self.enter_time
        except TypeError:
            raise ValueError(
                "Prefix injector '{injector}' is out of context, hence has no "
                "elapsed time."
                .format(injector=self))

    def format_elapsed_time(self, fmt=None):
        dt = datetime.datetime.utcfromtimestamp(self.elapsed_time)
        return dt.strftime(fmt or ELAPSED_TIME_FORMAT)

    def __repr__(self):
        return """<{module}.{name}("{prefix}")>""".format(
            module=self.__class__.__module__,
            name=self.__class__.__name__,
            prefix=self.prefix)


class StaticPrefixInjector(DirectPrefixInjector):

    def __init__(self, prefix, delimiter=None):
        prefix = make_prefix(prefix, delimiter)
        super(StaticPrefixInjector, self).__init__(prefix)


class AutoprefixInjector(StaticPrefixInjector):

    def __init__(self, oid_generator=None, delimiter=None):
        autopart = generate_oid(oid_generator)
        super(AutoprefixInjector, self).__init__(autopart, delimiter)


class HybridPrefixInjector(DirectPrefixInjector):

    def __init__(self, prefix, oid_generator=None, delimiter=None):
        autopart = generate_oid(oid_generator)
        prefix = join_prefix([autopart, prefix, ], delimiter)
        super(HybridPrefixInjector, self).__init__(prefix)


def merge_injectors(*args):
    prefix = ''.join([x.prefix for x in args])
    return DirectPrefixInjector(prefix)