Python中为DBus get_object设置超时及用户TOTP状态代码处理咨询

嘿，我来帮你搞定这个问题！下面是针对需求的完整解决方案，包括DBus超时设置、代码翻译和逻辑完善：

一、DBus调用超时设置说明 #

首先要说明：dbus-python的get_object方法本身并没有直接的超时参数，但我们可以通过两种方式实现超时控制，覆盖从总线连接到方法调用的全流程：

• 全局默认超时：设置总线对象的默认超时，对所有后续DBus操作生效
• 方法级单独超时：在具体的接口方法调用时指定超时，优先级高于全局设置

两种方式我都会在完善后的代码里体现。

二、代码翻译与完善（保留技术术语与代码原样） #

原代码的核心功能是通过SSSD Infopipe的DBus接口获取指定用户的**TOTP（基于时间的一次性密码）**启用状态。下面是翻译后的完整代码，同时补全了逻辑漏洞、添加了超时处理和全面的异常捕获：

import dbus
import sys

def get_user_totp_status(user_name=''):
    # 定义要获取的属性键名：TFA（双因素认证）启用状态
    key_name = 'tfaEnable'
    
    try:
        # 连接到系统DBus总线
        bus = dbus.SystemBus()
        # 设置总线全局默认超时：5000毫秒（5秒），适用于get_object等总线操作
        bus.set_default_timeout(5000)
        
        # 获取SSSD Infopipe的DBus对象（超时会触发DBusException）
        infopipe_obj = bus.get_object(
            "org.freedesktop.sssd.infopipe", 
            "/org/freedesktop/sssd/infopipe"
        )
        
        # 获取Infopipe的接口实例，用于调用具体方法
        ifp = dbus.Interface(
            infopipe_obj, 
            dbus_interface='org.freedesktop.sssd.infopipe'
        )
        
        # 调用GetUserAttr方法获取用户属性，单独指定超时3000毫秒（3秒）
        # 方法级超时优先级高于全局默认超时
        result = ifp.GetUserAttr(user_name, [key_name], timeout=3000)
        
        # 解析并返回TOTP状态
        user_totp_status = 'False'
        if result and key_name in result:
            # SSSD返回的属性值通常是列表格式，取第一个值即可
            for status in result[key_name]:
                # 兼容大小写不同的返回值（比如"True"/"true"）
                user_totp_status = 'True' if str(status).lower() == 'true' else 'False'
                break
        
        return user_totp_status
    
    except dbus.exceptions.DBusException as e:
        print(f"DBus调用错误: {str(e)}")
        sys.exit(1)
    except Exception as e:
        print(f"未知错误: {str(e)}")
        sys.exit(1)

# 示例：调用函数获取指定用户的TOTP状态
if __name__ == '__main__':
    if len(sys.argv) != 2:
        print("使用方式: python3 totp_status.py <用户名>")
        sys.exit(1)
    status = get_user_totp_status(sys.argv[1])
    print(f"用户{sys.argv[1]}的TOTP状态: {status}")

代码关键改进点： #

• 超时控制：同时设置了全局总线超时和方法级超时，避免DBus调用无响应
• 异常修复：原代码错误地在正常流程中打印错误信息，现已移至异常捕获块
• 逻辑完善：补全了TOTP状态的解析逻辑，兼容SSSD返回的不同字符串格式
• 健壮性提升：增加了参数检查、全面的异常捕获，确保程序不会意外崩溃
• 易用性优化：添加了命令行参数支持，方便直接测试

三、原代码片段的翻译（保留技术术语） #

对应你提供的不完整代码，中文翻译如下（保留代码原样和技术术语）：

def get_user_totp_status (user_name = ''):
# 属性键名：TFA启用状态
key_name = 'tfaEnable'
try:
# 连接系统DBus总线
bus = dbus.SystemBus()
except:
print "连接DBus错误!"
sys.exit(1)
# 获取SSSD Infopipe的DBus对象
infopipe_obj = bus.get_object("org.freedesktop.sssd.infopipe", "/org/freedesktop/sssd/infopipe")
# 获取Infopipe接口实例
ifp = dbus.Interface(infopipe_obj,dbus_interface='org.freedesktop.sssd.infopipe')
# 原代码此处错误地在正常流程打印错误，已修正
print "从DBus获取用户TOTP状态错误!"
# 调用GetUserAttr方法获取用户属性
result = ifp.GetUserAttr(user_name, [key_name])
user_totp_status = 'True' if result:
for status in result[key_name]:
user_totp_...（原代码未完成，已在完整代码中补全）

注：如果使用Python 2，请将代码中的print语句调整为Python 2语法（如去掉括号或使用%格式化），并确保安装了对应版本的dbus-python库。

内容的提问来源于stack exchange，提问作者Yun Peng Zhao