oafuncs 0.0.98.44__py3-none-any.whl → 0.0.98.45__py3-none-any.whl

oafuncs/__init__.py

@@ -42,4 +42,6 @@ from .oa_tool import *
 from .oa_date import *
 # ------------------- 2025-03-27 16:56:57 -------------------
 from .oa_geo import *
-# ------------------- 2025-09-04 14:08:26 -------------------
+# ------------------- 2025-09-04 14:08:26 -------------------
+from .oa_linux import *
+# ------------------- 2025-09-14 12:30:00 -------------------

oafuncs/_script/email.py

@@ -1,14 +1,9 @@
-import random
-import smtplib
-from email.header import Header
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
-
 from rich import print
 
 __all__ = ["send"]
 
 def _email_info():
+    import random
     email_dict = {
         "liukun0312@vip.qq.com": [4, 13, -10, 2, -10, 4, -7, -8, 8, -1, 3, -2, -11, -6, -9, -7],
         "756866877@qq.com": [4, -2, -3, 13, 12, 8, -6, 9, -12, 13, -10, -12, -11, -12, -4, -11],
@@ -26,6 +21,10 @@ def _decode_password(password):
 
 
 def _send_message(title, content, msg_to):
+    from email.header import Header
+    from email.mime.multipart import MIMEMultipart
+    from email.mime.text import MIMEText
+    import smtplib
     # 1. 连接邮箱服务器
     con = smtplib.SMTP_SSL("smtp.qq.com", 465)
 

oafuncs/oa_file.py

@@ -9,13 +9,14 @@ from rich import print
 __all__ = ["find_file", "link_file", "copy_file", "rename_file", "move_file", "clear_folder", "remove_empty_folder", "remove", "file_size", "mean_size", "make_dir", "replace_content"]
 
 
-def find_file(parent_dir: Union[str, os.PathLike], file_pattern: str, return_mode: str = "path") -> List[str]:
+def find_file(parent_dir: Union[str, os.PathLike], file_pattern: str, return_mode: str = "path", deep_find: bool = False) -> List[str]:
     """Finds files matching a specified pattern.
 
     Args:
         parent_dir: The parent directory where to search for files
         file_pattern: The file name pattern to search for
         return_mode: Return mode, 'path' to return full file paths, 'file' to return only file names. Defaults to 'path'
+        deep_find: Whether to search recursively in subdirectories. Defaults to False
 
     Returns:
         A list of file paths or file names if files are found, otherwise an empty list
@@ -24,9 +25,12 @@ def find_file(parent_dir: Union[str, os.PathLike], file_pattern: str, return_mod
     def natural_sort_key(s: str) -> List[Union[int, str]]:
         """Generate a key for natural sorting."""
         return [int(text) if text.isdigit() else text.lower() for text in re.split("([0-9]+)", s)]
-
-    search_pattern = os.path.join(str(parent_dir), file_pattern)
-    matched_files = glob.glob(search_pattern)
+    
+    if deep_find:
+        search_pattern = os.path.join(str(parent_dir), "**", file_pattern)
+    else:
+        search_pattern = os.path.join(str(parent_dir), file_pattern)
+    matched_files = glob.glob(search_pattern, recursive=deep_find)
 
     if not matched_files:
         return []

oafuncs/oa_geo.py

@@ -135,14 +135,64 @@ def mask_land_ocean(
     """
     mask = _land_sea_mask(lon, lat, keep)
 
-    # 用 apply_ufunc 自动对齐并广播掩膜
-    return xr.apply_ufunc(
-        lambda x, m: x.where(m),
-        data,
-        xr.DataArray(mask, dims=("lat", "lon")),
-        dask="parallelized",
-        keep_attrs=True,
-    )
+    # 将布尔掩膜转换为 xarray.DataArray
+    mask_da = xr.DataArray(mask, dims=("lat", "lon"))
+
+    # 如果输入已经是 xarray 对象，直接使用 where
+    if isinstance(data, (xr.DataArray, xr.Dataset)):
+        return data.where(mask_da)
+
+    # 如果输入是 numpy 数组，则假定最后两个维度是 (lat, lon)
+    if isinstance(data, np.ndarray):
+        arr = data
+        if arr.ndim < 2:
+            raise ValueError("numpy array 数据至少应包含 2 个维度 (lat, lon)")
+
+        if arr.ndim == 2:
+            lat_arr = np.asarray(lat)
+            lon_arr = np.asarray(lon)
+            # 支持 lat/lon 为 1D 或 2D
+            if lat_arr.ndim == 1 and lon_arr.ndim == 1:
+                da = xr.DataArray(arr, dims=("lat", "lon"), coords={"lat": lat_arr, "lon": lon_arr})
+            elif lat_arr.ndim == 2 and lon_arr.ndim == 2:
+                if lat_arr.shape != arr.shape or lon_arr.shape != arr.shape:
+                    raise ValueError("提供的二维经纬度数组形状必须匹配数据的 (lat, lon) 维度")
+                da = xr.DataArray(arr, dims=("lat", "lon"), coords={"lat": (("lat", "lon"), lat_arr), "lon": (("lat", "lon"), lon_arr)})
+            else:
+                raise ValueError("lat/lon 必须同时为 1D 或同时为 2D")
+        else:
+            # 为前面的维度生成占位名称，例如 dim_0, dim_1, ...
+            leading_dims = [f"dim_{i}" for i in range(arr.ndim - 2)]
+            dims = leading_dims + ["lat", "lon"]
+            coords = {f"dim_{i}": np.arange(arr.shape[i]) for i in range(arr.ndim - 2)}
+
+            lat_arr = np.asarray(lat)
+            lon_arr = np.asarray(lon)
+            # 如果 lat/lon 为 1D
+            if lat_arr.ndim == 1 and lon_arr.ndim == 1:
+                if lat_arr.shape[0] != arr.shape[-2] or lon_arr.shape[0] != arr.shape[-1]:
+                    raise ValueError("一维 lat/lon 长度必须匹配数据的最后两个维度")
+                coords.update({"lat": lat_arr, "lon": lon_arr})
+            # 如果 lat/lon 为 2D，要求其形状与数据最后两个维度一致
+            elif lat_arr.ndim == 2 and lon_arr.ndim == 2:
+                if lat_arr.shape != (arr.shape[-2], arr.shape[-1]) or lon_arr.shape != (arr.shape[-2], arr.shape[-1]):
+                    raise ValueError("二维 lat/lon 的形状必须匹配数据的最后两个维度")
+                coords.update({"lat": (("lat", "lon"), lat_arr), "lon": (("lat", "lon"), lon_arr)})
+            else:
+                raise ValueError("lat/lon 必须同时为 1D 或同时为 2D")
+
+            da = xr.DataArray(arr, dims=dims, coords=coords)
+
+        masked = da.where(mask_da)
+        # 返回与输入相同的类型：numpy -> numpy
+        return masked.values
+
+    # 其他类型尝试转换为 DataArray
+    try:
+        da = xr.DataArray(data)
+        return da.where(mask_da)
+    except Exception:
+        raise TypeError("data must be xr.DataArray, xr.Dataset, or numpy.ndarray")
 
 if __name__ == "__main__":
     pass

oafuncs/oa_linux.py

@@ -0,0 +1,53 @@
+from rich import print
+
+
+__all__ = ["os_command", "get_queue_node"]
+
+
+# 负责执行命令并返回输出
+def os_command(cmd):
+    import subprocess
+    print(f'🔍 执行命令: {cmd}')
+    result = subprocess.run(
+        cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True
+    )
+    # 打印错误信息（若有，方便排查问题）
+    if result.stderr:
+        print(f'❌ 错误输出: {result.stderr.strip()}')
+    # 检查命令是否执行成功（非0为失败）
+    if result.returncode != 0:
+        print(f'❌ 命令执行失败，退出码: {result.returncode}')
+        return None
+    return result.stdout
+
+# 返回“队列名:节点数”的字典
+def get_queue_node():
+    import re
+    # 执行 sinfo | grep "idle" 获取空闲队列数据
+    cmd = 'sinfo | grep "idle"'
+    output = os_command(cmd)
+    if not output:  # 命令执行失败或无输出，返回空字典
+        return {}
+    
+    # 初始化结果字典：键=队列名，值=节点数
+    queue_node_dict = {}
+    # 按行解析命令输出
+    for line in output.strip().split('\n'):
+        line = line.strip()
+        if not line:  # 跳过空行
+            continue
+        
+        # 正则匹配：仅捕获“队列名”（第1组）和“节点数”（第2组）
+        # 末尾用 .* 忽略节点列表，不影响匹配
+        pattern = r"^(\S+)\s+\S+\s+\S+\s+(\d+)\s+idle\s+.*$"
+        match = re.match(pattern, line)
+        
+        if match:
+            queue_name = match.group(1)    # 提取队列名作为字典的键
+            node_count = int(match.group(2))# 提取节点数作为字典的值（转为整数）
+            queue_node_dict[queue_name] = node_count  # 存入字典
+    
+    return queue_node_dict
+
+if __name__ == "__main__":
+    pass

{oafuncs-0.0.98.44.dist-info → oafuncs-0.0.98.45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: oafuncs
-Version: 0.0.98.44
+Version: 0.0.98.45
 Summary: Oceanic and Atmospheric Functions
 Home-page: https://github.com/Industry-Pays/OAFuncs
 Author: Kun Liu
@@ -187,4 +187,4 @@ query()
 <img title="OAFuncs" src="https://raw.githubusercontent.com/Industry-Pays/OAFuncs/main/oafuncs/_data/oafuncs.png" alt="OAFuncs">
 
 ## Wiki
-更多内容，查看[wiki](https://opendeep.wiki/Industry-Pays/OAFuncs/introduction)
+更多内容，查看[wiki_old](https://opendeep.wiki/Industry-Pays/OAFuncs/introduction) or [wiki_new](https://deepwiki.com/Industry-Pays/OAFuncs)

{oafuncs-0.0.98.44.dist-info → oafuncs-0.0.98.45.dist-info}/RECORD

@@ -1,11 +1,12 @@
-oafuncs/__init__.py,sha256=7630YgWbWGClu4Us1H2SAq-_eh9WzFHGxXkIXMcxRu0,1542
+oafuncs/__init__.py,sha256=G523BFVPxmODwq8j_88NYEiKbCzdQ3jfy51cmLeh7kM,1630
 oafuncs/oa_cmap.py,sha256=JwZMJ36uNwiCnzXqEtH2_PpeLtEaRaXP9YeGSl0PJSU,13886
 oafuncs/oa_data.py,sha256=CG2YHY_R6MFrPw3UznT4T8BE8yXdgBMnmdUAEdh9GAo,6506
 oafuncs/oa_date.py,sha256=aU2wVIWXyWoRiSQ9dg8sHvShFTxw86RrgbV3Q6tDjD4,6841
 oafuncs/oa_draw.py,sha256=zal0Y3RPpN0TCGN4Gw9qLtjQdT6V0ZqpSUBFVOPL0x4,13952
-oafuncs/oa_file.py,sha256=l9HTAK8iC1Bp_K7Mk3AX1UKuTFZZ-2yq5Hq71hnigbo,17299
-oafuncs/oa_geo.py,sha256=BWkvV6nW_c-UKqbgaoy4U1YQYUMzAQOJlK--XppNIms,4371
+oafuncs/oa_file.py,sha256=j9NOjxPOeAJsD5Zk4ODmFdVSSgr1CHVPvM1IHXy9RQA,17546
+oafuncs/oa_geo.py,sha256=UbzvUqgT2QP_9B7XSJRL1HDmGu0HnLC5nSP6ZrA5WH4,7177
 oafuncs/oa_help.py,sha256=0J5VaZX-cB0c090KxgmktQJBc0o00FsY-4wB8l5y00k,4178
+oafuncs/oa_linux.py,sha256=1fCmpgM33x3fLdQm8DOmaE4Pz4okXVNO0e-3svYnLgo,1850
 oafuncs/oa_nc.py,sha256=j501NlTuvrDIwNLXbMfE7nPPXdbbL7u9PGDj2l5AtnI,16277
 oafuncs/oa_python.py,sha256=xYMQnM0cGq9xUCtcoMpnN0LG5Rc_s94tai5nC6CNJ3E,4831
 oafuncs/oa_tool.py,sha256=VHx15VqpbzNlVXh0-3nJqcDgLVaECMD1FvxJ_CrV39E,8046
@@ -13,7 +14,7 @@ oafuncs/_data/hycom.png,sha256=MadKs6Gyj5n9-TOu7L4atQfTXtF9dvN9w-tdU9IfygI,10945
 oafuncs/_data/oafuncs.png,sha256=o3VD7wm-kwDea5E98JqxXl04_78cBX7VcdUt7uQXGiU,3679898
 oafuncs/_script/cprogressbar.py,sha256=BZi3MzF4q2Yl6fdZcLnW8MdpgpLeldI5NvnWMr-ZS94,16023
 oafuncs/_script/data_interp.py,sha256=gr1coA2N1mxzS4iv6S0C4lZpEQbuuHHNW-08RrhgPAA,4774
-oafuncs/_script/email.py,sha256=l5xDgdVj8O5V0J2SwjsHKdUuxOH2jZvwdMO_P0dImHU,2684
+oafuncs/_script/email.py,sha256=57jhRflm5QsyIshGMqtlfC6qn3b86GyiL4RQxdCOgxU,2702
 oafuncs/_script/netcdf_merge.py,sha256=tM9ePqLiEsE7eIsNM5XjEYeXwxjYOdNz5ejnEuI7xKw,6066
 oafuncs/_script/netcdf_modify.py,sha256=XDlAEToe_lwfAetkBSENqU5df-wnH7MGuxNTjG1gwHY,4178
 oafuncs/_script/netcdf_write.py,sha256=EDNycnhlrW1c6zcpmpObQeszDRX_lRxjTL-j0G4HqjI,17420
@@ -38,8 +39,8 @@ oafuncs/oa_sign/__init__.py,sha256=JSx1fcWpmNhQBvX_Bmq3xysfSkkFMrjbJASxV_V6aqE,1
 oafuncs/oa_sign/meteorological.py,sha256=3MSjy7HTcvz2zsITkjUMr_0Y027Gas1LFE9pk99990k,6110
 oafuncs/oa_sign/ocean.py,sha256=3uYEzaq-27yVy23IQoqy-clhWu1I_fhPFBAQyT-OF4M,5562
 oafuncs/oa_sign/scientific.py,sha256=moIl2MEY4uitbXoD596JmXookXGQtQsS-8_1NBBTx84,4689
-oafuncs-0.0.98.44.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
-oafuncs-0.0.98.44.dist-info/METADATA,sha256=yWxBsUGF1rlJBn42pXZyCUrgqhXWpyqc-l_CTyBEnSk,4384
-oafuncs-0.0.98.44.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-oafuncs-0.0.98.44.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
-oafuncs-0.0.98.44.dist-info/RECORD,,
+oafuncs-0.0.98.45.dist-info/licenses/LICENSE.txt,sha256=rMtLpVg8sKiSlwClfR9w_Dd_5WubTQgoOzE2PDFxzs4,1074
+oafuncs-0.0.98.45.dist-info/METADATA,sha256=JgC1BWV311BvpDhuyBam21v_fnaplfmdxKJkmakfM4s,4446
+oafuncs-0.0.98.45.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+oafuncs-0.0.98.45.dist-info/top_level.txt,sha256=bgC35QkXbN4EmPHEveg_xGIZ5i9NNPYWqtJqaKqTPsQ,8
+oafuncs-0.0.98.45.dist-info/RECORD,,