AlkantarClanX12
Current Path : /opt/cloudlinux/venv/lib64/python3.11/site-packages/dill/ |
Current File : //opt/cloudlinux/venv/lib64/python3.11/site-packages/dill/__init__.py |
#!/usr/bin/env python # # Author: Mike McKerns (mmckerns @caltech and @uqfoundation) # Copyright (c) 2008-2016 California Institute of Technology. # Copyright (c) 2016-2023 The Uncertainty Quantification Foundation. # License: 3-clause BSD. The full license text is available at: # - https://github.com/uqfoundation/dill/blob/master/LICENSE # author, version, license, and long description try: # the package is installed from .__info__ import __version__, __author__, __doc__, __license__ except: # pragma: no cover import os import sys parent = os.path.dirname(os.path.abspath(os.path.dirname(__file__))) sys.path.append(parent) # get distribution meta info from version import (__version__, __author__, get_license_text, get_readme_as_rst) __license__ = get_license_text(os.path.join(parent, 'LICENSE')) __license__ = "\n%s" % __license__ __doc__ = get_readme_as_rst(os.path.join(parent, 'README.md')) del os, sys, parent, get_license_text, get_readme_as_rst from ._dill import ( dump, dumps, load, loads, copy, Pickler, Unpickler, register, pickle, pickles, check, DEFAULT_PROTOCOL, HIGHEST_PROTOCOL, HANDLE_FMODE, CONTENTS_FMODE, FILE_FMODE, PickleError, PickleWarning, PicklingError, PicklingWarning, UnpicklingError, UnpicklingWarning, ) from .session import ( dump_module, load_module, load_module_asdict, dump_session, load_session # backward compatibility ) from . import detect, logger, session, source, temp # get global settings from .settings import settings # make sure "trace" is turned off logger.trace(False) objects = {} # local import of dill._objects #from . import _objects #objects.update(_objects.succeeds) #del _objects # local import of dill.objtypes from . import objtypes as types def load_types(pickleable=True, unpickleable=True): """load pickleable and/or unpickleable types to ``dill.types`` ``dill.types`` is meant to mimic the ``types`` module, providing a registry of object types. By default, the module is empty (for import speed purposes). Use the ``load_types`` function to load selected object types to the ``dill.types`` module. Args: pickleable (bool, default=True): if True, load pickleable types. unpickleable (bool, default=True): if True, load unpickleable types. Returns: None """ from importlib import reload # local import of dill.objects from . import _objects if pickleable: objects.update(_objects.succeeds) else: [objects.pop(obj,None) for obj in _objects.succeeds] if unpickleable: objects.update(_objects.failures) else: [objects.pop(obj,None) for obj in _objects.failures] objects.update(_objects.registered) del _objects # reset contents of types to 'empty' [types.__dict__.pop(obj) for obj in list(types.__dict__.keys()) \ if obj.find('Type') != -1] # add corresponding types from objects to types reload(types) def extend(use_dill=True): '''add (or remove) dill types to/from the pickle registry by default, ``dill`` populates its types to ``pickle.Pickler.dispatch``. Thus, all ``dill`` types are available upon calling ``'import pickle'``. To drop all ``dill`` types from the ``pickle`` dispatch, *use_dill=False*. Args: use_dill (bool, default=True): if True, extend the dispatch table. Returns: None ''' from ._dill import _revert_extension, _extend if use_dill: _extend() else: _revert_extension() return extend() def license(): """print license""" print (__license__) return def citation(): """print citation""" print (__doc__[-491:-118]) return # end of file