#820

Committed 01 Oct 2023 02:22AM UTC coverage: 0.0% (-73.7%) from 73.748%

Build # #820

Build Type

push

circle-ci

Committed by Jiashen Cao

Commit Message

fix lint error

Run Details

0 of 12361 relevant lines covered (0.0%)

0.0 hits per line

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

0.0

/evadb/third_party/databases/interface.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 importlib
import os
from contextlib import contextmanager

from evadb.executor.executor_utils import ExecutorError


def _get_database_handler(engine: str, **kwargs):
    """
    Return the database handler. User should modify this function for
    their new integrated handlers.
    """

    # Dynamically import the top module.
    try:
        mod = dynamic_import(engine)
    except ImportError:
        req_file = os.path.join(os.path.dirname(__file__), engine, "requirements.txt")
        if os.path.isfile(req_file):
            with open(req_file) as f:
                raise ImportError(f"Please install the following packages {f.read()}")

    if engine == "postgres":
        return mod.PostgresHandler(engine, **kwargs)
    elif engine == "sqlite":
        return mod.SQLiteHandler(engine, **kwargs)
    elif engine == "mysql":
        return mod.MysqlHandler(engine, **kwargs)
    elif engine == "mariadb":
        return mod.MariaDbHandler(engine, **kwargs)
    else:
        raise NotImplementedError(f"Engine {engine} is not supported")


@contextmanager
def get_database_handler(engine: str, **kwargs):
    handler = _get_database_handler(engine, **kwargs)
    try:
        resp = handler.connect()
        if not resp.status:
            raise ExecutorError(f"Cannot establish connection due to {resp.error}")
        yield handler
    finally:
        handler.disconnect()


def dynamic_import(handler_dir):
    import_path = f"evadb.third_party.databases.{handler_dir}.{handler_dir}_handler"
    return importlib.import_module(import_path)

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 importlib	×
16	import os	×
17	from contextlib import contextmanager	×
18
19	from evadb.executor.executor_utils import ExecutorError	×
20
21
22	def _get_database_handler(engine: str, **kwargs):	×
23	"""
24	Return the database handler. User should modify this function for
25	their new integrated handlers.
26	"""
27
28	# Dynamically import the top module.
29	try:	×
30	mod = dynamic_import(engine)	×
31	except ImportError:
32	req_file = os.path.join(os.path.dirname(__file__), engine, "requirements.txt")
33	if os.path.isfile(req_file):
34	with open(req_file) as f:
35	raise ImportError(f"Please install the following packages {f.read()}")
36
37	if engine == "postgres":	×
38	return mod.PostgresHandler(engine, **kwargs)	×
39	elif engine == "sqlite":	×
40	return mod.SQLiteHandler(engine, **kwargs)	×
41	elif engine == "mysql":	×
42	return mod.MysqlHandler(engine, **kwargs)	×
43	elif engine == "mariadb":	×
44	return mod.MariaDbHandler(engine, **kwargs)	×
45	else:
46	raise NotImplementedError(f"Engine {engine} is not supported")
47
48
49	@contextmanager	×
50	def get_database_handler(engine: str, **kwargs):	×
51	handler = _get_database_handler(engine, **kwargs)	×
52	try:	×
53	resp = handler.connect()	×
54	if not resp.status:	×
55	raise ExecutorError(f"Cannot establish connection due to {resp.error}")
56	yield handler	×
57	finally:
58	handler.disconnect()	×
59
60
61	def dynamic_import(handler_dir):	×
62	import_path = f"evadb.third_party.databases.{handler_dir}.{handler_dir}_handler"	×
63	return importlib.import_module(import_path)	×

georgia-tech-db / eva / #820

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