Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2025-12-25 05:14:40 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

flame: fixing contextmanager for save and load preferences file

Browse source
This commit is contained in:
Jakub Jezek 2021-10-26 21:24:11 +02:00
parent 25557e7be8
commit d01ffd9373
No known key found for this signature in database
GPG key ID: D8548FBF690B100A
1 changed files with 16 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

48
openpype/hosts/flame/api/lib.py
View file

@ -10,37 +10,15 @@ log = Logger().get_logger(__name__)
@contextlib.contextmanager
def load_preferences_file(klass, filepath, attribute):
def io_preferences_file(klass, filepath, write=False):
try:
with open(filepath, "r") as prefs_file:
setattr(klass, attribute, pickle.load(prefs_file))
yield
flag = "w" if write else "r"
yield open(filepath, flag)
except IOError:
klass.log.info("Unable to load preferences from {}".format(
klass.log.info("Unable to work with preferences `{}`".format(
filepath))
finally:
klass.log.info("Preferences loaded from {}".format(filepath))
@contextlib.contextmanager
def save_preferences_file(klass, filepath, attribute):
try:
with open(filepath, "w") as prefs_file:
attr = getattr(klass, attribute)
pickle.dump(attr, prefs_file)
yield
except IOError:
klass.log.info("Unable to save preferences to {}".format(
filepath))
finally:
klass.log.info("Preferences saved to {}".format(filepath))
class FlameAppFramework(object):
# flameAppFramework class takes care of preferences
@ -170,19 +148,22 @@ class FlameAppFramework(object):
(proj_pref_path, user_pref_path,
glob_pref_path) = self.get_pref_file_paths()
with load_preferences_file(self, proj_pref_path, "prefs"):
with io_preferences_file(self, proj_pref_path) as prefs_file:
self.prefs = pickle.load(prefs_file)
self.log.info(
"Project - preferences contents:\n{}".format(
pformat(self.prefs)
))
with load_preferences_file(self, user_pref_path, "prefs_user"):
with io_preferences_file(self, user_pref_path) as prefs_file:
self.prefs_user = pickle.load(prefs_file)
self.log.info(
"User - preferences contents:\n{}".format(
pformat(self.prefs_user)
))
with load_preferences_file(self, glob_pref_path, "prefs_global"):
with io_preferences_file(self, glob_pref_path) as prefs_file:
self.prefs_global = pickle.load(prefs_file)
self.log.info(
"Global - preferences contents:\n{}".format(
pformat(self.prefs_global)
@ -204,19 +185,22 @@ class FlameAppFramework(object):
(proj_pref_path, user_pref_path,
glob_pref_path) = self.get_pref_file_paths()
with save_preferences_file(self, proj_pref_path, "prefs"):
with io_preferences_file(self, proj_pref_path, True) as prefs_file:
pickle.dump(self.prefs, prefs_file)
self.log.info(
"Project - preferences contents:\n{}".format(
pformat(self.prefs)
))
with save_preferences_file(self, user_pref_path, "prefs_user"):
with io_preferences_file(self, user_pref_path, True) as prefs_file:
pickle.dump(self.prefs_user, prefs_file)
self.log.info(
"User - preferences contents:\n{}".format(
pformat(self.prefs_user)
))
with save_preferences_file(self, glob_pref_path, "prefs_global"):
with io_preferences_file(self, glob_pref_path, True) as prefs_file:
pickle.dump(self.prefs_global, prefs_file)
self.log.info(
"Global - preferences contents:\n{}".format(
pformat(self.prefs_global)