datachain 0.20.4__py3-none-any.whl → 0.21.1__py3-none-any.whl

datachain/data_storage/metastore.py

@@ -37,13 +37,9 @@ from datachain.dataset import (
 from datachain.error import (
     DatasetNotFoundError,
     DatasetVersionNotFoundError,
-    NamespaceNotFoundError,
-    ProjectNotFoundError,
     TableMissingError,
 )
 from datachain.job import Job
-from datachain.namespace import Namespace
-from datachain.project import Project
 from datachain.utils import JSONSerialize
 
 if TYPE_CHECKING:
@@ -65,8 +61,6 @@ class AbstractMetastore(ABC, Serializable):
     uri: StorageURI
 
     schema: "schema.Schema"
-    namespace_class: type[Namespace] = Namespace
-    project_class: type[Project] = Project
     dataset_class: type[DatasetRecord] = DatasetRecord
     dataset_list_class: type[DatasetListRecord] = DatasetListRecord
     dataset_list_version_class: type[DatasetListVersion] = DatasetListVersion
@@ -112,115 +106,14 @@ class AbstractMetastore(ABC, Serializable):
     def cleanup_for_tests(self) -> None:
         """Cleanup for tests."""
 
-    #
-    # Namespaces
-    #
-
-    @property
-    @abstractmethod
-    def default_namespace_name(self):
-        """Gets default namespace name"""
-
-    @property
-    def system_namespace_name(self):
-        return Namespace.system()
-
-    @abstractmethod
-    def create_namespace(
-        self,
-        name: str,
-        description: Optional[str] = None,
-        uuid: Optional[str] = None,
-        ignore_if_exists: bool = True,
-        **kwargs,
-    ) -> Namespace:
-        """Creates new namespace"""
-
-    @abstractmethod
-    def get_namespace(self, name: str, conn=None) -> Namespace:
-        """Gets a single namespace by name"""
-
-    @abstractmethod
-    def list_namespaces(self, conn=None) -> list[Namespace]:
-        """Gets a list of all namespaces"""
-
-    @property
-    @abstractmethod
-    def is_studio(self) -> bool:
-        """Returns True if this code is ran in Studio"""
-
-    def is_local_dataset(self, dataset_namespace: str) -> bool:
-        """
-        Returns True if this is local dataset i.e. not pulled from Studio but
-        created locally. This is False if we ran code in CLI mode but using dataset
-        names that are present in Studio.
-        """
-        return self.is_studio or dataset_namespace == Namespace.default()
-
-    @property
-    def namespace_allowed_to_create(self):
-        return self.is_studio
-
-    #
-    # Projects
-    #
-
-    @property
-    @abstractmethod
-    def default_project_name(self):
-        """Gets default project name"""
-
-    @property
-    def listing_project_name(self):
-        return Project.listing()
-
-    @cached_property
-    def default_project(self) -> Project:
-        return self.get_project(
-            self.default_project_name, self.default_namespace_name, create=True
-        )
-
-    @cached_property
-    def listing_project(self) -> Project:
-        return self.get_project(self.listing_project_name, self.system_namespace_name)
-
-    @abstractmethod
-    def create_project(
-        self,
-        namespace_name: str,
-        name: str,
-        description: Optional[str] = None,
-        uuid: Optional[str] = None,
-        ignore_if_exists: bool = True,
-        **kwargs,
-    ) -> Project:
-        """Creates new project in specific namespace"""
-
-    @abstractmethod
-    def get_project(
-        self, name: str, namespace_name: str, create: bool = False, conn=None
-    ) -> Project:
-        """
-        Gets a single project inside some namespace by name.
-        It also creates project if not found and create flag is set to True.
-        """
-
-    @abstractmethod
-    def list_projects(self, namespace_id: Optional[int], conn=None) -> list[Project]:
-        """Gets list of projects in some namespace or in general (in all namespaces)"""
-
-    @property
-    def project_allowed_to_create(self):
-        return self.is_studio
-
     #
     # Datasets
     #
+
     @abstractmethod
     def create_dataset(
         self,
         name: str,
-        project_id: Optional[int] = None,
         status: int = DatasetStatus.CREATED,
         sources: Optional[list[str]] = None,
         feature_schema: Optional[dict] = None,
@@ -280,22 +173,15 @@ class AbstractMetastore(ABC, Serializable):
         """
 
     @abstractmethod
-    def list_datasets(
-        self, project_id: Optional[int] = None
-    ) -> Iterator[DatasetListRecord]:
-        """Lists all datasets in some project or in all projects."""
+    def list_datasets(self) -> Iterator[DatasetListRecord]:
+        """Lists all datasets."""
 
     @abstractmethod
-    def list_datasets_by_prefix(
-        self, prefix: str, project_id: Optional[int] = None
-    ) -> Iterator["DatasetListRecord"]:
-        """
-        Lists all datasets which names start with prefix in some project or in all
-        projects.
-        """
+    def list_datasets_by_prefix(self, prefix: str) -> Iterator["DatasetListRecord"]:
+        """Lists all datasets which names start with prefix."""
 
     @abstractmethod
-    def get_dataset(self, name: str, project_id: Optional[int] = None) -> DatasetRecord:
+    def get_dataset(self, name: str) -> DatasetRecord:
         """Gets a single dataset by name."""
 
     @abstractmethod
@@ -316,10 +202,10 @@ class AbstractMetastore(ABC, Serializable):
     @abstractmethod
     def add_dataset_dependency(
         self,
-        source_dataset: "DatasetRecord",
+        source_dataset_name: str,
         source_dataset_version: str,
-        dep_dataset: "DatasetRecord",
-        dep_dataset_version: str,
+        dataset_name: str,
+        dataset_version: str,
     ) -> None:
         """Adds dataset dependency to dataset."""
 
@@ -418,8 +304,6 @@ class AbstractDBMetastore(AbstractMetastore):
     and has shared logic for all database systems currently in use.
     """
 
-    NAMESPACE_TABLE = "namespaces"
-    PROJECT_TABLE = "projects"
     DATASET_TABLE = "datasets"
     DATASET_VERSION_TABLE = "datasets_versions"
     DATASET_DEPENDENCY_TABLE = "datasets_dependencies"
@@ -438,62 +322,11 @@ class AbstractDBMetastore(AbstractMetastore):
     def cleanup_tables(self, temp_table_names: list[str]) -> None:
         """Cleanup temp tables."""
 
-    @classmethod
-    def _namespaces_columns(cls) -> list["SchemaItem"]:
-        """Namespace table columns."""
-        return [
-            Column("id", Integer, primary_key=True),
-            Column("uuid", Text, nullable=False, default=uuid4()),
-            Column("name", Text, nullable=False),
-            Column("description", Text),
-            Column("created_at", DateTime(timezone=True)),
-        ]
-
-    @cached_property
-    def _namespaces_fields(self) -> list[str]:
-        return [
-            c.name  # type: ignore [attr-defined]
-            for c in self._namespaces_columns()
-            if c.name  # type: ignore [attr-defined]
-        ]
-
-    @classmethod
-    def _projects_columns(cls) -> list["SchemaItem"]:
-        """Project table columns."""
-        return [
-            Column("id", Integer, primary_key=True),
-            Column("uuid", Text, nullable=False, default=uuid4()),
-            Column("name", Text, nullable=False),
-            Column("description", Text),
-            Column("created_at", DateTime(timezone=True)),
-            Column(
-                "namespace_id",
-                Integer,
-                ForeignKey(f"{cls.NAMESPACE_TABLE}.id", ondelete="CASCADE"),
-                nullable=False,
-            ),
-            UniqueConstraint("namespace_id", "name"),
-        ]
-
-    @cached_property
-    def _projects_fields(self) -> list[str]:
-        return [
-            c.name  # type: ignore [attr-defined]
-            for c in self._projects_columns()
-            if c.name  # type: ignore [attr-defined]
-        ]
-
     @classmethod
     def _datasets_columns(cls) -> list["SchemaItem"]:
         """Datasets table columns."""
         return [
             Column("id", Integer, primary_key=True),
-            Column(
-                "project_id",
-                Integer,
-                ForeignKey(f"{cls.PROJECT_TABLE}.id", ondelete="CASCADE"),
-                nullable=False,
-            ),
             Column("name", Text, nullable=False),
             Column("description", Text),
             Column("attrs", JSON, nullable=True),
@@ -612,16 +445,6 @@ class AbstractDBMetastore(AbstractMetastore):
     #
     # Query Tables
     #
-    @cached_property
-    def _namespaces(self) -> Table:
-        return Table(
-            self.NAMESPACE_TABLE, self.db.metadata, *self._namespaces_columns()
-        )
-
-    @cached_property
-    def _projects(self) -> Table:
-        return Table(self.PROJECT_TABLE, self.db.metadata, *self._projects_columns())
-
     @cached_property
     def _datasets(self) -> Table:
         return Table(self.DATASET_TABLE, self.db.metadata, *self._datasets_columns())
@@ -645,34 +468,6 @@ class AbstractDBMetastore(AbstractMetastore):
     #
     # Query Starters (These can be overridden by subclasses)
     #
-    @abstractmethod
-    def _namespaces_insert(self) -> "Insert": ...
-
-    def _namespaces_select(self, *columns) -> "Select":
-        if not columns:
-            return self._namespaces.select()
-        return select(*columns)
-
-    def _namespaces_update(self) -> "Update":
-        return self._namespaces.update()
-
-    def _namespaces_delete(self) -> "Delete":
-        return self._namespaces.delete()
-
-    @abstractmethod
-    def _projects_insert(self) -> "Insert": ...
-
-    def _projects_select(self, *columns) -> "Select":
-        if not columns:
-            return self._projects.select()
-        return select(*columns)
-
-    def _projects_update(self) -> "Update":
-        return self._projects.update()
-
-    def _projects_delete(self) -> "Delete":
-        return self._projects.delete()
-
     @abstractmethod
     def _datasets_insert(self) -> "Insert": ...
 
@@ -715,134 +510,6 @@ class AbstractDBMetastore(AbstractMetastore):
     def _datasets_dependencies_delete(self) -> "Delete":
         return self._datasets_dependencies.delete()
 
-    #
-    # Namespaces
-    #
-
-    def create_namespace(
-        self,
-        name: str,
-        description: Optional[str] = None,
-        uuid: Optional[str] = None,
-        ignore_if_exists: bool = True,
-        **kwargs,
-    ) -> Namespace:
-        query = self._namespaces_insert().values(
-            name=name,
-            uuid=uuid or str(uuid4()),
-            created_at=datetime.now(timezone.utc),
-            description=description,
-        )
-        if ignore_if_exists and hasattr(query, "on_conflict_do_nothing"):
-            # SQLite and PostgreSQL both support 'on_conflict_do_nothing',
-            # but generic SQL does not
-            query = query.on_conflict_do_nothing(index_elements=["name"])
-        self.db.execute(query)
-
-        return self.get_namespace(name)
-
-    def get_namespace(self, name: str, conn=None) -> Namespace:
-        """Gets a single namespace by name"""
-        n = self._namespaces
-
-        query = self._namespaces_select(
-            *(getattr(n.c, f) for f in self._namespaces_fields),
-        ).where(n.c.name == name)
-        rows = list(self.db.execute(query, conn=conn))
-        if not rows:
-            raise NamespaceNotFoundError(f"Namespace {name} not found.")
-        return self.namespace_class.parse(*rows[0])
-
-    def list_namespaces(self, conn=None) -> list[Namespace]:
-        """Gets a list of all namespaces"""
-        n = self._namespaces
-
-        query = self._namespaces_select(
-            *(getattr(n.c, f) for f in self._namespaces_fields),
-        )
-        rows = list(self.db.execute(query, conn=conn))
-
-        return [self.namespace_class.parse(*r) for r in rows]
-
-    #
-    # Projects
-    #
-
-    def create_project(
-        self,
-        namespace_name: str,
-        name: str,
-        description: Optional[str] = None,
-        uuid: Optional[str] = None,
-        ignore_if_exists: bool = True,
-        **kwargs,
-    ) -> Project:
-        try:
-            namespace = self.get_namespace(namespace_name)
-        except NamespaceNotFoundError:
-            namespace = self.create_namespace(namespace_name)
-
-        query = self._projects_insert().values(
-            namespace_id=namespace.id,
-            uuid=uuid or str(uuid4()),
-            name=name,
-            created_at=datetime.now(timezone.utc),
-            description=description,
-        )
-        if ignore_if_exists and hasattr(query, "on_conflict_do_nothing"):
-            # SQLite and PostgreSQL both support 'on_conflict_do_nothing',
-            # but generic SQL does not
-            query = query.on_conflict_do_nothing(
-                index_elements=["namespace_id", "name"]
-            )
-        self.db.execute(query)
-
-        return self.get_project(name, namespace.name)
-
-    def get_project(
-        self, name: str, namespace_name: str, create: bool = False, conn=None
-    ) -> Project:
-        """Gets a single project inside some namespace by name"""
-        n = self._namespaces
-        p = self._projects
-
-        query = self._projects_select(
-            *(getattr(n.c, f) for f in self._namespaces_fields),
-            *(getattr(p.c, f) for f in self._projects_fields),
-        )
-        query = query.select_from(n.join(p, n.c.id == p.c.namespace_id)).where(
-            p.c.name == name, n.c.name == namespace_name
-        )
-
-        rows = list(self.db.execute(query, conn=conn))
-        if not rows:
-            if create:
-                return self.create_project(namespace_name, name)
-            raise ProjectNotFoundError(
-                f"Project {name} in namespace {namespace_name} not found."
-            )
-        return self.project_class.parse(*rows[0])
-
-    def list_projects(self, namespace_id: Optional[int], conn=None) -> list[Project]:
-        """
-        Gets a list of projects inside some namespace, or in all namespaces
-        """
-        n = self._namespaces
-        p = self._projects
-
-        query = self._projects_select(
-            *(getattr(n.c, f) for f in self._namespaces_fields),
-            *(getattr(p.c, f) for f in self._projects_fields),
-        )
-        query = query.select_from(n.join(p, n.c.id == p.c.namespace_id))
-
-        if namespace_id:
-            query = query.where(n.c.id == namespace_id)
-
-        rows = list(self.db.execute(query, conn=conn))
-
-        return [self.project_class.parse(*r) for r in rows]
-
     #
     # Datasets
     #
@@ -850,7 +517,6 @@ class AbstractDBMetastore(AbstractMetastore):
     def create_dataset(
         self,
         name: str,
-        project_id: Optional[int] = None,
         status: int = DatasetStatus.CREATED,
         sources: Optional[list[str]] = None,
         feature_schema: Optional[dict] = None,
@@ -862,11 +528,9 @@ class AbstractDBMetastore(AbstractMetastore):
         **kwargs,  # TODO registered = True / False
     ) -> DatasetRecord:
         """Creates new dataset."""
-        project_id = project_id or self.default_project.id
-
+        # TODO abstract this method and add registered = True based on kwargs
         query = self._datasets_insert().values(
             name=name,
-            project_id=project_id,
             status=status,
             feature_schema=json.dumps(feature_schema or {}),
             created_at=datetime.now(timezone.utc),
@@ -882,10 +546,10 @@ class AbstractDBMetastore(AbstractMetastore):
         if ignore_if_exists and hasattr(query, "on_conflict_do_nothing"):
             # SQLite and PostgreSQL both support 'on_conflict_do_nothing',
             # but generic SQL does not
-            query = query.on_conflict_do_nothing(index_elements=["project_id", "name"])
+            query = query.on_conflict_do_nothing(index_elements=["name"])
         self.db.execute(query)
 
-        return self.get_dataset(name, project_id)
+        return self.get_dataset(name)
 
     def create_dataset_version(  # noqa: PLR0913
         self,
@@ -942,7 +606,7 @@ class AbstractDBMetastore(AbstractMetastore):
             )
         self.db.execute(query, conn=conn)
 
-        return self.get_dataset(dataset.name, dataset.project.id, conn=conn)
+        return self.get_dataset(dataset.name, conn=conn)
 
     def remove_dataset(self, dataset: DatasetRecord) -> None:
         """Removes dataset."""
@@ -1080,15 +744,13 @@ class AbstractDBMetastore(AbstractMetastore):
 
     def _parse_dataset_list(self, rows) -> Iterator["DatasetListRecord"]:
         # grouping rows by dataset id
-        for _, g in groupby(rows, lambda r: r[11]):
+        for _, g in groupby(rows, lambda r: r[0]):
             dataset = self._parse_list_dataset(list(g))
             if dataset:
                 yield dataset
 
     def _get_dataset_query(
         self,
-        namespace_fields: list[str],
-        project_fields: list[str],
         dataset_fields: list[str],
         dataset_version_fields: list[str],
         isouter: bool = True,
@@ -1099,81 +761,48 @@ class AbstractDBMetastore(AbstractMetastore):
         ):
             raise TableMissingError
 
-        n = self._namespaces
-        p = self._projects
         d = self._datasets
         dv = self._datasets_versions
 
         query = self._datasets_select(
-            *(getattr(n.c, f) for f in namespace_fields),
-            *(getattr(p.c, f) for f in project_fields),
             *(getattr(d.c, f) for f in dataset_fields),
             *(getattr(dv.c, f) for f in dataset_version_fields),
         )
-        j = (
-            n.join(p, n.c.id == p.c.namespace_id)
-            .join(d, p.c.id == d.c.project_id)
-            .join(dv, d.c.id == dv.c.dataset_id, isouter=isouter)
-        )
+        j = d.join(dv, d.c.id == dv.c.dataset_id, isouter=isouter)
         return query.select_from(j)
 
     def _base_dataset_query(self) -> "Select":
         return self._get_dataset_query(
-            self._namespaces_fields,
-            self._projects_fields,
-            self._dataset_fields,
-            self._dataset_version_fields,
+            self._dataset_fields, self._dataset_version_fields
         )
 
     def _base_list_datasets_query(self) -> "Select":
         return self._get_dataset_query(
-            self._namespaces_fields,
-            self._projects_fields,
-            self._dataset_list_fields,
-            self._dataset_list_version_fields,
-            isouter=False,
+            self._dataset_list_fields, self._dataset_list_version_fields, isouter=False
         )
 
-    def list_datasets(
-        self, project_id: Optional[int] = None
-    ) -> Iterator["DatasetListRecord"]:
+    def list_datasets(self) -> Iterator["DatasetListRecord"]:
         """Lists all datasets."""
-        d = self._datasets
         query = self._base_list_datasets_query().order_by(
             self._datasets.c.name, self._datasets_versions.c.version
         )
-        if project_id:
-            query = query.where(d.c.project_id == project_id)
         yield from self._parse_dataset_list(self.db.execute(query))
 
     def list_datasets_by_prefix(
-        self, prefix: str, project_id: Optional[int] = None, conn=None
+        self, prefix: str, conn=None
     ) -> Iterator["DatasetListRecord"]:
-        d = self._datasets
         query = self._base_list_datasets_query()
-        if project_id:
-            query = query.where(d.c.project_id == project_id)
         query = query.where(self._datasets.c.name.startswith(prefix))
         yield from self._parse_dataset_list(self.db.execute(query))
 
-    def get_dataset(
-        self,
-        name: str,  # normal, not full dataset name
-        project_id: Optional[int] = None,
-        conn=None,
-    ) -> DatasetRecord:
-        """
-        Gets a single dataset in project by dataset name.
-        """
-        project_id = project_id or self.default_project.id
+    def get_dataset(self, name: str, conn=None) -> DatasetRecord:
+        """Gets a single dataset by name"""
         d = self._datasets
         query = self._base_dataset_query()
-        query = query.where(d.c.name == name, d.c.project_id == project_id)  # type: ignore [attr-defined]
+        query = query.where(d.c.name == name)  # type: ignore [attr-defined]
         ds = self._parse_dataset(self.db.execute(query, conn=conn))
         if not ds:
-            raise DatasetNotFoundError(
-                f"Dataset {name} not found in project {project_id}"
-            )
+            raise DatasetNotFoundError(f"Dataset {name} not found.")
         return ds
 
     def remove_dataset_version(
@@ -1243,20 +872,23 @@ class AbstractDBMetastore(AbstractMetastore):
     #
     def add_dataset_dependency(
         self,
-        source_dataset: "DatasetRecord",
+        source_dataset_name: str,
         source_dataset_version: str,
-        dep_dataset: "DatasetRecord",
-        dep_dataset_version: str,
+        dataset_name: str,
+        dataset_version: str,
     ) -> None:
         """Adds dataset dependency to dataset."""
+        source_dataset = self.get_dataset(source_dataset_name)
+        dataset = self.get_dataset(dataset_name)
+
         self.db.execute(
             self._datasets_dependencies_insert().values(
                 source_dataset_id=source_dataset.id,
                 source_dataset_version_id=(
                     source_dataset.get_version(source_dataset_version).id
                 ),
-                dataset_id=dep_dataset.id,
-                dataset_version_id=dep_dataset.get_version(dep_dataset_version).id,
+                dataset_id=dataset.id,
+                dataset_version_id=dataset.get_version(dataset_version).id,
             )
         )
 
@@ -1298,8 +930,6 @@ class AbstractDBMetastore(AbstractMetastore):
     def get_direct_dataset_dependencies(
         self, dataset: DatasetRecord, version: str
     ) -> list[Optional[DatasetDependency]]:
-        n = self._namespaces
-        p = self._projects
         d = self._datasets
         dd = self._datasets_dependencies
         dv = self._datasets_versions
@@ -1311,16 +941,18 @@ class AbstractDBMetastore(AbstractMetastore):
         query = (
             self._datasets_dependencies_select(*select_cols)
             .select_from(
-                dd.join(d, dd.c.dataset_id == d.c.id, isouter=True)
-                .join(dv, dd.c.dataset_version_id == dv.c.id, isouter=True)
-                .join(p, d.c.project_id == p.c.id, isouter=True)
-                .join(n, p.c.namespace_id == n.c.id, isouter=True)
+                dd.join(d, dd.c.dataset_id == d.c.id, isouter=True).join(
+                    dv, dd.c.dataset_version_id == dv.c.id, isouter=True
+                )
             )
             .where(
                 (dd.c.source_dataset_id == dataset.id)
                 & (dd.c.source_dataset_version_id == dataset_version.id)
             )
         )
+        if version:
+            dataset_version = dataset.get_version(version)
+            query = query.where(dd.c.source_dataset_version_id == dataset_version.id)
 
         return [self.dependency_class.parse(*r) for r in self.db.execute(query)]