Source code for tool_shed.webapp.model.mapping

"""
Details of how the data model objects are mapped onto the relational database
are encapsulated here.
"""
import logging

import tool_shed.webapp.model
import tool_shed.webapp.util.shed_statistics as shed_statistics
from galaxy.model.base import SharedModelMapping
from galaxy.model.orm.engine_factory import build_engine
from tool_shed.webapp.model import mapper_registry
from tool_shed.webapp.security import CommunityRBACAgent

log = logging.getLogger(__name__)

metadata = mapper_registry.metadata


[docs]class ToolShedModelMapping(SharedModelMapping): security_agent: CommunityRBACAgent shed_counter: shed_statistics.ShedCounter create_tables: bool
[docs]def init(file_path, url, engine_options=None, create_tables=False) -> ToolShedModelMapping: """Connect mappings to the database""" engine_options = engine_options or {} # Create the database engine engine = build_engine(url, engine_options) # Connect the metadata to the database. metadata.bind = engine result = ToolShedModelMapping([tool_shed.webapp.model], engine=engine) if create_tables: metadata.create_all() result.create_tables = create_tables result.security_agent = CommunityRBACAgent(result) result.shed_counter = shed_statistics.ShedCounter(result) return result