Source code for aiida.storage.psql_dos.models.settings

###########################################################################
# Copyright (c), The AiiDA team. All rights reserved.                     #
# This file is part of the AiiDA code.                                    #
#                                                                         #
# The code is hosted on GitHub at https://github.com/aiidateam/aiida-core #
# For further information on the license, see the LICENSE.txt file        #
# For further information please visit http://www.aiida.net               #
###########################################################################
"""Module to manage node settings for the SQLA backend."""

from sqlalchemy import Column
from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.sql.schema import Index
from sqlalchemy.types import DateTime, Integer, String, Text

from aiida.common import timezone
from aiida.storage.psql_dos.models.base import Base




class DbSetting(Base):
    """Database model to store global settings."""

    __tablename__ = 'db_dbsetting'

    id = Column(Integer, primary_key=True)
    key = Column(String(1024), nullable=False, unique=True)
    val = Column(JSONB, default={})

    # I also add a description field for the variables
    description = Column(Text, default='', nullable=False)
    time = Column(DateTime(timezone=True), default=timezone.now, onupdate=timezone.now, nullable=False)

    __table_args__ = (
        Index(
            'ix_pat_db_dbsetting_key', 'key', postgresql_using='btree', postgresql_ops={'key': 'varchar_pattern_ops'}
        ),
    )



    def __str__(self):
        return f"'{self.key}'={self.val}"