fix: fix oracledb ImportError

shengchenyang · Dec 30, 2023 · e363937 · e363937
1 parent fe914c9
commit e363937
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ayugespidertools/common/spiderconf.py b/ayugespidertools/common/spiderconf.py
@@ -1,7 +1,6 @@
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Generic, Optional, Tuple, TypeVar
 
-from oracledb.exceptions import DatabaseError
 from sqlalchemy.exc import OperationalError
 
 from ayugespidertools.common.typevars import (
@@ -18,6 +17,12 @@
 from ayugespidertools.common.utils import ToolsForAyu
 from ayugespidertools.config import logger
 
+try:
+    from oracledb.exceptions import DatabaseError
+except ImportError:
+    # pip install ayugespidertools[database]
+    pass
+
 __all__ = [
     "get_spider_conf",
     "MysqlConfCreator",