LumAPI 1.1.2__tar.gz → 1.1.3__tar.gz

lumapi-1.1.3/LumAPI/__init__.py

@@ -0,0 +1,21 @@
+# LumAPI/__init__.py
+
+# 显式导入所有需要对外暴露的类和函数
+from .lumapi import (
+    lumapi, lumerical, LumFuncBase,
+    FDTD, MODE, DEVICE, INTERCONNECT,
+    savemat, loadmat, 
+    create_cmap, set_colorbar_range,
+    Estimate_focal, Kirchhoff, 
+    RayleighSommerfeld_Scalar, RayleighSommerfeld_Vector, AngularSpectrum_Vector
+)
+
+# 定义 import * 时对外暴露的接口白名单（这是静态检查器识别 import * 的关键）
+__all__ = [
+    'lumapi', 'lumerical', 'LumFuncBase',
+    'FDTD', 'MODE', 'DEVICE', 'INTERCONNECT',
+    'savemat', 'loadmat', 
+    'create_cmap', 'set_colorbar_range',
+    'Estimate_focal', 'Kirchhoff', 
+    'RayleighSommerfeld_Scalar', 'RayleighSommerfeld_Vector', 'AngularSpectrum_Vector'
+]

{lumapi-1.1.2 → lumapi-1.1.3}/LumAPI/lumapi.py

@@ -1081,6 +1081,10 @@ class lumerical:
         except Exception:
             self.lumapi = None
 
+    def __bool__(self):
+        """判断配置是否成功，允许直接使用 if lumapi: 来判断"""
+        return self.lumapi is not None
+
     def _check_config_and_prompt(self):
         """核心逻辑：检查配置状态，若无效则引导用户使用 LumAPI 命令"""
         if self.lumapi is None: