如何封装现有Python模块？以跨平台兼容fcntl模块为例

问题根源 #

1. 导入逻辑缺陷：原代码通过临时修改sys.path规避导入自身，但这种方式不稳定，Python模块导入机制仍可能优先加载当前目录的自定义文件。
2. 常量获取错误：Python标准库的fcntl模块并未直接暴露LOCK_EX/LOCK_SH/LOCK_UN这些常量——它们是系统级flock调用的参数，不属于fcntl模块的公开属性。

修正后的代码 #

import sys
import os
import importlib.util

if os.name == 'posix':
    # 直接加载系统原生的fcntl模块，绕过当前目录的自定义文件
    spec = importlib.util.find_spec('fcntl')
    realfcntl = importlib.util.module_from_spec(spec)
    sys.modules['fcntl'] = realfcntl
    spec.loader.exec_module(realfcntl)

    # 定义flock所需的posix标准常量
    LOCK_EX = 1    # 排他锁
    LOCK_SH = 2    # 共享锁
    LOCK_UN = 8    # 释放锁

    def flock(fd, mode):
        return realfcntl.flock(fd, mode)
else:
    # Windows下模拟实现，不执行实际锁操作
    LOCK_EX = -1
    LOCK_SH = -1
    LOCK_UN = -1
    def flock(fd, mode):
        print(f'Pretending to manipulate locking on FD {fd}', file=sys.stderr)

关键改进点 #

• 可靠导入系统模块：使用importlib.util直接定位并加载系统原生fcntl模块，彻底避免导入自定义的fcntl.py自身。
• 常量直接定义：基于posix标准直接定义锁常量，无需依赖fcntl模块的非公开属性。
• 完全兼容原有代码：所有原有代码对fcntl.LOCK_EX、fcntl.flock的调用无需修改，接口保持一致。

内容的提问来源于stack exchange，提问作者Mikhail T.