74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""Oracle 11.2 thick 连接。环境变量与 rpa_oms 相同。"""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
from pathlib import Path
|
||
|
||
_client_ready = False
|
||
_PROJECT_ROOT = Path(__file__).resolve().parent
|
||
_OMS_ENV = _PROJECT_ROOT.parent / "rpa_oms" / ".env"
|
||
|
||
|
||
class RepoUnavailable(Exception):
|
||
def __init__(self, message: str = "数据库不可用"):
|
||
self.message = message
|
||
super().__init__(message)
|
||
|
||
|
||
def load_runtime_env(paths: list[Path] | None = None) -> None:
|
||
"""先读返利 .env，空缺再补 rpa_oms/.env。不覆盖已有非空环境变量。"""
|
||
env_paths = paths if paths is not None else [_PROJECT_ROOT / ".env", _OMS_ENV]
|
||
for path in env_paths:
|
||
_fill_env(path)
|
||
|
||
|
||
def _fill_env(path: Path) -> None:
|
||
if not path.is_file():
|
||
return
|
||
for raw in path.read_text(encoding="utf-8").splitlines():
|
||
line = raw.strip()
|
||
if not line or line.startswith("#") or "=" not in line:
|
||
continue
|
||
key, _, value = line.partition("=")
|
||
key = key.strip()
|
||
value = value.strip()
|
||
if len(value) >= 2 and value[0] == value[-1] and value[0] in {'"', "'"}:
|
||
value = value[1:-1]
|
||
if not key or not value:
|
||
continue
|
||
current = os.environ.get(key, "")
|
||
if not current:
|
||
os.environ[key] = value
|
||
|
||
|
||
def oracle_dsn() -> tuple[str, str, str]:
|
||
user = os.getenv("RPA_ORACLE_USER", "")
|
||
password = os.getenv("RPA_ORACLE_PASSWORD", "")
|
||
host = os.getenv("RPA_ORACLE_HOST", "")
|
||
if not user or not password or not host:
|
||
raise RepoUnavailable("未配置 RPA_ORACLE_HOST / RPA_ORACLE_USER / RPA_ORACLE_PASSWORD")
|
||
port = os.getenv("RPA_ORACLE_PORT", "1521")
|
||
service = os.getenv("RPA_ORACLE_SERVICE", "rpa")
|
||
return user, password, f"{host}:{port}/{service}"
|
||
|
||
|
||
def connect_from_env():
|
||
user, password, dsn = oracle_dsn()
|
||
_ensure_client()
|
||
import oracledb
|
||
|
||
return oracledb.connect(user=user, password=password, dsn=dsn)
|
||
|
||
|
||
def _ensure_client() -> None:
|
||
global _client_ready
|
||
if _client_ready:
|
||
return
|
||
import oracledb
|
||
|
||
if os.getenv("RPA_ORACLE_THICK", "1") == "1":
|
||
lib_dir = os.getenv("RPA_ORACLE_CLIENT_LIB", "")
|
||
kwargs = {"lib_dir": lib_dir} if lib_dir else {}
|
||
oracledb.init_oracle_client(**kwargs)
|
||
_client_ready = True
|