时间戳

首页 > 解决方案 > 有没有更有效的存储关键字参数的方法？

python - 有没有更有效的存储关键字参数的方法？

问题描述

所以这是我的代码：

def formateUserData(FirstName = None, MiddleName = None, LastName = None, DOB = None, Gender = None):
    formatedUserData = {}
    dataFieldKeys = ['First Name', 'Middle Name', 'Last Name', 'DOB', 'Gender']
    dataFieldValues = [FirstName, MiddleName, LastName, DOB, Gender]

    for key, value in zip(dataFieldKeys, dataFieldValues):
        if value: formatedUserData[key] = value

    return formatedUserData

如您所见，关键字参数必须重复 3 次（第 1 行第一次，第 3 行第二次，第 4 行第三次）。有没有办法只存储一次关键字参数就可以做同样的事情？如果可能的话，也许通过使用某种方式遍历关键字参数？顺便说一句，我希望不使用 **kwargs。

标签： pythonpython-3.xperformancefunctional-programmingdry

解决方案

也许这个划痕会激发您考虑我觉得非常方便的数据类

from dataclasses import dataclass, fields
from datetime import date
from typing import Literal, Optional

@dataclass
class User:
    first_name: Optional[str] = None
    middle_name: Optional[str] = None
    last_name: Optional[str] = None
    dob: Optional[date] = None
    gender: Optional[Literal['male', 'female']] = None


def not_none_dict(user: User) -> dict:
    not_none = {}
    for field in fields(user):
        value = getattr(user, field.name)
        if value:
            not_none[field.name] = value
    return not_none

print(
    not_none_dict(User(
        first_name='Joe',
        last_name='Doe'
    ))
)
>>> {'first_name': 'Joe', 'last_name': 'Doe'}

推荐阅读