Source code for pacifica.metadata.orm.instrument_group

# -*- coding: utf-8 -*-
"""InstrumentGroup links Groups and Instruments and objects."""
from peewee import ForeignKeyField, CompositeKey
from .utils import index_hash
from .base import DB
from import CherryPyAPI
from .groups import Groups
from .instruments import Instruments

[docs]class InstrumentGroup(CherryPyAPI): """ InstrumentGroup attributes are foreign keys. Attributes: +-------------------+-------------------------------------+ | Name | Description | +===================+=====================================+ | instrument | Link to the Instrument model | +-------------------+-------------------------------------+ | group | Link to the Group model | +-------------------+-------------------------------------+ """ instrument = ForeignKeyField(Instruments, backref='groups') group = ForeignKeyField(Groups, backref='instrument_members') # pylint: disable=too-few-public-methods class Meta(object): """PeeWee meta class contains the database and the primary key.""" database = DB primary_key = CompositeKey('instrument', 'group') # pylint: enable=too-few-public-methods
[docs] def to_hash(self, **flags): """Convert the object to a hash.""" obj = super(InstrumentGroup, self).to_hash(**flags) obj['_id'] = index_hash(int(self.__data__['group']), int(self.__data__['instrument'])) obj['instrument'] = int(self.__data__['instrument']) obj['group'] = int(self.__data__['group']) return obj
[docs] def from_hash(self, obj): """Convert the hash into the object.""" super(InstrumentGroup, self).from_hash(obj) self._set_only_if( 'instrument', obj, 'instrument', lambda: Instruments.get( == obj['instrument']) ) self._set_only_if( 'group', obj, 'group', lambda: Groups.get( == obj['group']) )
[docs] @classmethod def where_clause(cls, kwargs): """Where clause for the various elements.""" where_clause = super(InstrumentGroup, cls).where_clause(kwargs) attrs = ['instrument', 'group'] return cls._where_attr_clause(where_clause, kwargs, attrs)