Scheduled maintenance on Saturday, 27 September 2025, from 07:00 AM to 4:00 PM GMT (09:00 AM to 6:00 PM CEST) - some services may be unavailable -

Skip to content
Snippets Groups Projects
DeviceModel.py 2.43 KiB
Newer Older
  • Learn to ignore specific revisions
  • Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    # Copyright 2021-2023 H2020 TeraFlow (https://www.teraflow-h2020.eu/)
    #
    # Licensed under the Apache License, Version 2.0 (the "License");
    # you may not use this file except in compliance with the License.
    # You may obtain a copy of the License at
    #
    #      http://www.apache.org/licenses/LICENSE-2.0
    #
    # Unless required by applicable law or agreed to in writing, software
    # distributed under the License is distributed on an "AS IS" BASIS,
    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    # See the License for the specific language governing permissions and
    # limitations under the License.
    
    import operator
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    from sqlalchemy import Column, Enum, String
    from sqlalchemy.dialects.postgresql import ARRAY, UUID
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    from sqlalchemy.orm import relationship
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    from typing import Dict
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    from .enums.DeviceDriver import ORM_DeviceDriverEnum
    from .enums.DeviceOperationalStatus import ORM_DeviceOperationalStatusEnum
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    from ._Base import _Base
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    
    class DeviceModel(_Base):
        __tablename__ = 'device'
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
        device_uuid = Column(UUID(as_uuid=False), primary_key=True)
        device_name = Column(String, nullable=False)
        device_type = Column(String, nullable=False)
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
        device_operational_status = Column(Enum(ORM_DeviceOperationalStatusEnum), nullable=False)
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
        device_drivers = Column(ARRAY(Enum(ORM_DeviceDriverEnum), dimensions=1))
    
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
        #topology_devices = relationship('TopologyDeviceModel', back_populates='device')
        config_rules = relationship('ConfigRuleModel', passive_deletes=True) # lazy='joined', back_populates='device'
        endpoints    = relationship('EndPointModel', passive_deletes=True) # lazy='joined', back_populates='device'
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
    
        def dump_id(self) -> Dict:
            return {'device_uuid': {'uuid': self.device_uuid}}
    
        def dump(self) -> Dict:
            return {
                'device_id'                : self.dump_id(),
                'name'                     : self.device_name,
                'device_type'              : self.device_type,
                'device_operational_status': self.device_operational_status.value,
                'device_drivers'           : [driver.value for driver in self.device_drivers],
                'device_config'            : {'config_rules': [
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
                    config_rule.dump()
                    for config_rule in sorted(self.config_rules, key=operator.attrgetter('position'))
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
                ]},
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
                'device_endpoints'         : [
                    endpoint.dump()
                    for endpoint in self.endpoints
                ],
    
    Lluis Gifre Renom's avatar
    Lluis Gifre Renom committed
            }