purge_dict.py
88e97ff5f796feature_greedy_coarseni…

purge_dict.py
View Options

			# Copyright (C) 2018 by the RROMPy authors
			#
			# This file is part of RROMPy.
			#
			# RROMPy is free software: you can redistribute it and/or modify
			# it under the terms of the GNU Lesser General Public License as published by
			# the Free Software Foundation, either version 3 of the License, or
			# (at your option) any later version.
			#
			# RROMPy is distributed in the hope that it will be useful,
			# but WITHOUT ANY WARRANTY; without even the implied warranty of
			# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
			# GNU Lesser General Public License for more details.
			#
			# You should have received a copy of the GNU Lesser General Public License
			# along with RROMPy. If not, see <http://www.gnu.org/licenses/>.
			#

			from rrompy.utilities.base.find_dict_str_key import findDictStrKey
			from rrompy.utilities.base.types import ListAny, DictAny
			from rrompy.utilities.warning_manager import warn

			__all__ = ['purgeDict']

			def purgeDict(dct:DictAny, allowedKeys : ListAny = [], silent : bool = False,
			complement : bool = False, dictname : str = "",
			baselevel : int = 0) -> DictAny:
			if dictname != "":
			dictname = " in " + dictname
			dctcp = {}
			for key in dct.keys():
			akey = findDictStrKey(key, allowedKeys)
			if (akey is None) != complement:
			if not silent:
			warn("Ignoring key {0}{2} with value {1}.".format(key,
			dct[key],
			dictname),
			baselevel)
			else:
			if akey is None:
			akey = key
			dctcp[akey] = dct[key]
			return dctcp