georgia-tech-db / eva / #850

import contextlib

import pandas as pd

from evadb.database import EvaDBDatabase

from evadb.executor.abstract_executor import AbstractExecutor

from evadb.executor.executor_utils import (

from evadb.models.storage.batch import Batch

from evadb.plan_nodes.create_plan import CreatePlan

from evadb.storage.storage_engine import StorageEngine

from evadb.utils.errors import CatalogError

from evadb.utils.logging_manager import logger

class CreateExecutor(AbstractExecutor):

    def __init__(self, db: EvaDBDatabase, node: CreatePlan):

        super().__init__(db, node)

    def exec(self, *args, **kwargs):

        is_native_table = self.node.table_info.database_name is not None

        check_if_exists = handle_if_not_exists(

        name = self.node.table_info.table_name

        if check_if_exists:

            yield Batch(pd.DataFrame([f"Table {name} already exists"]))

            return

        create_table_done = False

        logger.debug(f"Creating table {self.node.table_info}")

        if not is_native_table:

            catalog_entry = self.catalog().create_and_insert_table_catalog_entry(

            catalog_entry = create_table_catalog_entry_for_native_table(

        storage_engine = StorageEngine.factory(self.db, catalog_entry)

        try:

            storage_engine.create(table=catalog_entry)

            create_table_done = True

            msg = f"The table {name} has been successfully created"

            if self.children != []:

                assert (

                child = self.children[0]

                rows = 0

                for batch in child.exec():

                    batch.drop_column_alias()

                    storage_engine.write(catalog_entry, batch)

                    rows += len(batch)

                msg = (

            yield Batch(pd.DataFrame([msg]))