def set(self, key, value, description=None):
"""Return the settings with the given key.
:param key: the key identifying the setting
:param value: the value for the setting
:param description: optional setting description
"""
from aiida.backends.sqlalchemy.models.settings import DbSetting
self.validate_table_existence()
validate_attribute_key(key)
other_attribs = dict()
if description is not None:
other_attribs['description'] = description
DbSetting.set_value(key, value, other_attribs=other_attribs)
def test_settings(self):
"""
Test the settings table (similar to Attributes, but without the key.
"""
from aiida.backends.sqlalchemy.models.settings import DbSetting
from aiida.backends.sqlalchemy import get_scoped_session
session = get_scoped_session()
from pytz import UTC
from aiida.utils import timezone
from sqlalchemy.exc import IntegrityError
DbSetting.set_value(key='pippo', value=[1, 2, 3])
# s1 = DbSetting.objects.get(key='pippo')
s1 = DbSetting.query.filter_by(key='pippo').first()
self.assertEqual(s1.getvalue(), [1, 2, 3])
s2 = DbSetting(key='pippo')
s2.time = timezone.datetime.now(tz=UTC)
with self.assertRaises(IntegrityError):
with session.begin_nested():
# same name...
session.add(s2)
# Should replace pippo
DbSetting.set_value(key='pippo', value="a")
s1 = DbSetting.query.filter_by(key='pippo').first()
self.assertEqual(s1.getvalue(), "a")
def set_global_setting(key, value, description=None):
"""
Set a global setting in the DbSetting table (therefore, stored at the DB
level).
"""
DbSetting.set_value(key, value, other_attribs={"description": description})