Source code for korus.database.database

from .interface import (
    DeploymentInterface,
    AnnotationInterface,
    FileInterface,
    JobInterface,
    StorageInterface,
    TaxonomyInterface,
    LabelInterface,
    TagInterface,
    GranularityInterface,
)
from .backend import DatabaseBackend
from .backend.sqlite import SQLiteBackend


class Database:

    def __init__(self, backend: DatabaseBackend):
        self.backend = backend

        self._label = LabelInterface(self.backend.label)

        self.deployment = DeploymentInterface(self.backend.deployment)
        self.storage = StorageInterface(self.backend.storage)
        self.file = FileInterface(self.backend.file, self.storage)
        self.job = JobInterface(self.backend.job, self.file)
        self.taxonomy = TaxonomyInterface(self.backend.taxonomy, self._label)
        self.tag = TagInterface(self.backend.tag)
        self.granularity = GranularityInterface(self.backend.granularity)
        self.annotation = AnnotationInterface(
            self.backend.annotation,
            taxonomy=self.taxonomy,
            job=self.job,
            file=self.file,
            tag=self.tag,
            granularity=self.granularity,
        )

    @property
    def korus_version(self) -> str:
        """Get Korus version used to create the database.

        Returns:
            version: str
                The Korus version in the form x.y.z
        """
        return self.backend.korus_version

    @property
    def tables(self) -> dict:
        """Returns a dict with all the tables in the database"""
        return {
            "deployment": self.deployment,
            "storage": self.storage,
            "file": self.file,
            "job": self.job,
            "taxonomy": self.taxonomy,
            "tag": self.tag,
            "granularity": self.granularity,
            "annotation": self.annotation,
        }


class SQLiteDatabase(Database):

    def __init__(self, path: str, new: bool = False):
        super().__init__(SQLiteBackend(path, new))