如何从该代码段返回用户名(WP)？

def get_username(code):
    # 假设代码段中的变量名为code
    # 假设代码段中的用户名存储在字典中，键为'username'
    user_data = {
        'username': 'WP',
        'email': 'wp@example.com',
        'age': 25
    }
    
    # 检查代码段中的变量是否在字典中存在
    if code in user_data:
        return user_data[code]
    else:
        return None