Package Gnumed :: Package timelinelib :: Package general :: Module methodcontainer |
[frames] | no frames]

# Source Code for Module Gnumed.timelinelib.general.methodcontainer

     1  # Copyright (C) 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018  Rickard Lindberg, Roger Lindberg
     2  #
     3  # This file is part of Timeline.
     4  #
     5  # Timeline is free software: you can redistribute it and/or modify
     6  # it under the terms of the GNU General Public License as published by
     7  # the Free Software Foundation, either version 3 of the License, or
     8  # (at your option) any later version.
     9  #
    10  # Timeline is distributed in the hope that it will be useful,
    11  # but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  # GNU General Public License for more details.
    14  #
    15  # You should have received a copy of the GNU General Public License
    16  # along with Timeline.  If not, see <http://www.gnu.org/licenses/>.
    17
    18
    19  from collections import defaultdict
    20
    21
    22 -class MethodContainer(object):
    23
    24 -    def __init__(self, methods_kvp, default_method=None):
    25          self._default_method = default_method
    26          if self._all_keys_are_booleans(methods_kvp):
    27              if self._first_truthy_method(methods_kvp):
    28                  self._container = defaultdict(self._default, [(True, self._first_truthy_method(methods_kvp))])
    29              else:
    30                  self._container = defaultdict(self._default, [])
    31          else:
    32              self._container = defaultdict(self._default, methods_kvp)
    33
    34 -    def select(self, key):
    35          return self._container[key]
    36
    37 -    def _all_keys_are_booleans(self, methods_kvp):
    38          return len([m for k, m in methods_kvp if not isinstance(k, bool)]) == 0
    39
    40 -    def _first_truthy_method(self, methods_kvp):
    41          for key, method in methods_kvp:
    42              if key:
    43                  return method
    44
    45 -    def _default(self):
    46          if self._default_method is None:
    47              return self._noop
    48          else:
    49              return self._default_method
    50
    51 -    def _noop(self, *args, **kwargs):
    52          pass
    53