Source code for dtale.global_state
DATA = {}
DTYPES = {}
SETTINGS = {}
METADATA = {}
CONTEXT_VARIABLES = {}
HISTORY = {}
[docs]def get_data(data_id=None):
global DATA
if data_id is None:
return DATA
return DATA.get(data_id)
[docs]def get_dtypes(data_id=None):
global DTYPES
if data_id is None:
return DTYPES
return DTYPES.get(data_id)
[docs]def get_settings(data_id=None):
global SETTINGS
if data_id is None:
return SETTINGS
return SETTINGS.get(data_id)
[docs]def get_context_variables(data_id=None):
global CONTEXT_VARIABLES
if data_id is None:
return CONTEXT_VARIABLES
return CONTEXT_VARIABLES.get(data_id)
[docs]def get_history(data_id=None):
global HISTORY
if data_id is None:
return HISTORY
return HISTORY.get(data_id)
[docs]def set_data(data_id, val):
global DATA
DATA[data_id] = val
[docs]def set_dtypes(data_id, val):
global DTYPES
DTYPES[data_id] = val
[docs]def set_settings(data_id, val):
global SETTINGS
SETTINGS[data_id] = val
[docs]def set_context_variables(data_id, val):
global CONTEXT_VARIABLES
CONTEXT_VARIABLES[data_id] = val
[docs]def set_history(data_id, val):
global HISTORY
HISTORY[data_id] = val
[docs]def cleanup(data_id=None):
"""
Helper function for cleanup up state related to a D-Tale process with a specific port
:param port: integer string for a D-Tale process's port
:type port: str
"""
global DATA, DTYPES, SETTINGS, METADATA, CONTEXT_VARIABLES, HISTORY
if data_id is None:
DATA = {}
SETTINGS = {}
DTYPES = {}
METADATA = {}
CONTEXT_VARIABLES = {}
HISTORY = {}
else:
for data in [DATA, DTYPES, SETTINGS, METADATA, CONTEXT_VARIABLES, HISTORY]:
data.pop(data_id, None) # use dict.pop with a default so that KeyError won't occur