Peewee Model 自动清理字段空格方法

本文介绍了如何在 Peewee 模型中自动去除字段中的空白字符。通过重写模型的 `__init__` 方法，可以在创建实例时，对指定的字段应用 `strip()` 方法，从而实现自动去除空白字符的功能，保证数据的整洁性。

在使用 Peewee 进行数据库操作时，我们经常会遇到需要处理字段中空白字符的情况，例如用户输入的数据中可能包含多余的空格，影响数据的准确性和一致性。为了解决这个问题，我们可以在 Peewee 模型中实现自动去除空白字符的功能。

实现方法

实现的关键在于重写模型的 __init__ 方法。__init__ 方法是 Python 类的构造函数，在创建类的实例时会被调用。我们可以在 __init__ 方法中，获取传递给构造函数的参数，并对需要去除空白字符的字段应用 strip() 方法。

以下是一个示例代码：

import peewee as pw

db = pw.SqliteDatabase(':memory:') # 使用内存数据库方便测试

class BaseModel(pw.Model):
    class Meta:
        database = db

class Person(BaseModel):
    email = pw.CharField()
    name = pw.CharField()

    def __init__(self, *args, **kwargs):
        kwargs["email"] = kwargs.get("email", "").strip()
        kwargs["name"] = kwargs.get("name", "").strip()
        super().__init__(*args, **kwargs)

db.create_tables([Person])

# 创建包含空白字符的实例
mom = Person(email="  <a class=\"__cf_email__\" data-cfemail=\"c2afadb6aaa7b082a5afa3abaeeca1adaf\" href=\"/cdn-cgi/l/email-protection\">[email&#160;protected]</a>  ", name=" Stella Bird  ")

# 打印去除空白字符后的字段值
print(mom.email)
print(mom.name)

代码解释

1. __init__(self, *args, **kwargs)*: 重写 Person 类的 __init__ 方法。`args和kwargs` 用于接收传递给构造函数的位置参数和关键字参数。
2. kwargs["email"] = kwargs.get("email", "").strip(): 从 kwargs 中获取 email 字段的值，如果 email 字段不存在，则使用空字符串 "" 作为默认值。然后，对获取到的值应用 strip() 方法，去除首尾的空白字符，并将结果重新赋值给 kwargs["email"]。name 字段的处理方式类似。
3. *`super().init(args, kwargs)`: 调用父类（pw.Model）的 __init__ 方法，完成模型的初始化。

注意事项

• kwargs.get("email", "") 的作用是防止在创建实例时，如果没有传递 email 参数，导致 kwargs["email"] 抛出 KeyError 异常。
• strip() 方法只会去除字符串首尾的空白字符，不会去除字符串中间的空白字符。如果需要去除字符串中间的空白字符，可以使用 replace() 方法。例如：kwargs["name"] = kwargs.get("name", "").replace(" ", "")。
• 如果你的模型有很多字段需要去除空白字符，可以考虑使用循环来简化代码。

总结

通过重写 Peewee 模型的 __init__ 方法，我们可以方便地实现自动去除字段中空白字符的功能，从而保证数据的整洁性和一致性。这种方法简单易用，适用于各种需要处理空白字符的场景。希望本文能够帮助你更好地使用 Peewee 进行数据库操作。