Если вы хотите, чтобы принятый ответ был обобщен, вы можете использовать:
import shelve
def save_workspace(filename, names_of_spaces_to_save, dict_of_values_to_save):
'''
filename = location to save workspace.
names_of_spaces_to_save = use dir() from parent to save all variables in previous scope.
-dir() = return the list of names in the current local scope
dict_of_values_to_save = use globals() or locals() to save all variables.
-globals() = Return a dictionary representing the current global symbol table.
This is always the dictionary of the current module (inside a function or method,
this is the module where it is defined, not the module from which it is called).
-locals() = Update and return a dictionary representing the current local symbol table.
Free variables are returned by locals() when it is called in function blocks, but not in class blocks.
Example of globals and dir():
>>> x = 3 #note variable value and name bellow
>>> globals()
{'__builtins__': <module '__builtin__' (built-in)>, '__name__': '__main__', 'x': 3, '__doc__': None, '__package__': None}
>>> dir()
['__builtins__', '__doc__', '__name__', '__package__', 'x']
'''
print 'save_workspace'
print 'C_hat_bests' in names_of_spaces_to_save
print dict_of_values_to_save
my_shelf = shelve.open(filename,'n') # 'n' for new
for key in names_of_spaces_to_save:
try:
my_shelf[key] = dict_of_values_to_save[key]
except TypeError:
#
# __builtins__, my_shelf, and imported modules can not be shelved.
#
#print('ERROR shelving: {0}'.format(key))
pass
my_shelf.close()
def load_workspace(filename, parent_globals):
'''
filename = location to load workspace.
parent_globals use globals() to load the workspace saved in filename to current scope.
'''
my_shelf = shelve.open(filename)
for key in my_shelf:
parent_globals[key]=my_shelf[key]
my_shelf.close()
an example script of using this:
import my_pkg as mp
x = 3
mp.save_workspace('a', dir(), globals())
чтобы получить / загрузить рабочее пространство:
import my_pkg as mp
x=1
mp.load_workspace('a', globals())
print x #print 3 for me
это сработало, когда я его запустил. Я признаю, что не понимаю dir() и globals() 100%, поэтому я не уверен, что может быть какое-то странное предостережение, но пока это работает. Комментарии приветствуются:)
после еще одного исследования, если вы вызовете save_workspace, как я предложил для глобалов, и save_workspace находится внутри функции, она не будет работать должным образом, если вы хотите сохранить проверяемые объекты в локальной области. Для этого используйте locals(). Это происходит потому, что глобальные переменные берут глобальные переменные из модуля, в котором определена функция, а не из того места, где она вызывается.