#763

Committed 06 Sep 2023 03:00AM UTC coverage: 74.923% (+5.1%) from 69.779%

Build # #763

Build Type

push

circle-ci

Committed by

jiashenC

Commit Message

fix lint

Run Details

10 of 10 new or added lines in 1 file covered. (100.0%)

8739 of 11664 relevant lines covered (74.92%)

0.75 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

57.14

/evadb/executor/create_database_executor.py

# coding=utf-8
# Copyright 2018-2023 EvaDB
#
# 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 pandas as pd

from evadb.database import EvaDBDatabase
from evadb.executor.abstract_executor import AbstractExecutor
from evadb.executor.executor_utils import ExecutorError
from evadb.models.storage.batch import Batch
from evadb.parser.create_statement import CreateDatabaseStatement
from evadb.third_party.databases.interface import get_database_handler
from evadb.utils.logging_manager import logger


class CreateDatabaseExecutor(AbstractExecutor):
    def __init__(self, db: EvaDBDatabase, node: CreateDatabaseStatement):
        super().__init__(db, node)

    def exec(self, *args, **kwargs):
        # TODO: handle if_not_exists

        logger.debug(
            f"Trying to connect to the provided engine {self.node.engine} with params {self.node.param_dict}"
        )

        # Check if database already exists.
        db_catalog_entry = self.catalog().get_database_catalog_entry(
            self.node.database_name
        )
        if db_catalog_entry is not None:
            raise ExecutorError(f"{self.node.database_name} already exists.")

        # Check the validity of database entry.
        handler = get_database_handler(self.node.engine, **self.node.param_dict)
        resp = handler.connect()
        if not resp.status:
            raise ExecutorError(f"Cannot establish connection due to {resp.error}")

        logger.debug(f"Creating database {self.node}")
        self.catalog().insert_database_catalog_entry(
            self.node.database_name, self.node.engine, self.node.param_dict
        )

        yield Batch(
            pd.DataFrame(
                [
                    f"The database {self.node.database_name} has been successfully created."
                ]
            )
        )

1	# coding=utf-8
2	# Copyright 2018-2023 EvaDB
3	#
4	# Licensed under the Apache License, Version 2.0 (the "License");
5	# you may not use this file except in compliance with the License.
6	# You may obtain a copy of the License at
7	#
8	# http://www.apache.org/licenses/LICENSE-2.0
9	#
10	# Unless required by applicable law or agreed to in writing, software
11	# distributed under the License is distributed on an "AS IS" BASIS,
12	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13	# See the License for the specific language governing permissions and
14	# limitations under the License.
15	import pandas as pd	1✔
16
17	from evadb.database import EvaDBDatabase	1✔
18	from evadb.executor.abstract_executor import AbstractExecutor	1✔
19	from evadb.executor.executor_utils import ExecutorError	1✔
20	from evadb.models.storage.batch import Batch	1✔
21	from evadb.parser.create_statement import CreateDatabaseStatement	1✔
22	from evadb.third_party.databases.interface import get_database_handler	1✔
23	from evadb.utils.logging_manager import logger	1✔
24
25
26	class CreateDatabaseExecutor(AbstractExecutor):	1✔
27	def __init__(self, db: EvaDBDatabase, node: CreateDatabaseStatement):	1✔
28	super().__init__(db, node)	1✔
29
30	def exec(self, args, *kwargs):	1✔
31	# TODO: handle if_not_exists
32
33	logger.debug(	×
34	f"Trying to connect to the provided engine {self.node.engine} with params {self.node.param_dict}"
35	)
36
37	# Check if database already exists.
38	db_catalog_entry = self.catalog().get_database_catalog_entry(	×
39	self.node.database_name
40	)
41	if db_catalog_entry is not None:	×
42	raise ExecutorError(f"{self.node.database_name} already exists.")
43
44	# Check the validity of database entry.
45	handler = get_database_handler(self.node.engine, **self.node.param_dict)	×
46	resp = handler.connect()	×
47	if not resp.status:	×
48	raise ExecutorError(f"Cannot establish connection due to {resp.error}")
49
50	logger.debug(f"Creating database {self.node}")	×
51	self.catalog().insert_database_catalog_entry(	×
52	self.node.database_name, self.node.engine, self.node.param_dict
53	)
54
55	yield Batch(	×
56	pd.DataFrame(
57	[
58	f"The database {self.node.database_name} has been successfully created."
59	]
60	)
61	)

georgia-tech-db / eva / #763

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous