crm开发之用户ModelForm定制和密码加密

Posted 2021-02-17 chengege

写了这么多的定制 功能。终于可以定制一下了！
因为是 stark 和 rbac 两个组建。 一起使用。 所以在这里，再记录一下。需要注意的点：

先放出 目录结构：

先从  stark 开始。
使用stark组件，其实就是对， 自己 的 web 业务中的。 表进行增删改查。

首先再  webapp 下面， 增加一个  stark 的py文件：

from django import forms
from django.core.exceptions import ValidationError
from stark.servers.start_v1 import site, StartHandler, SearchGroupRow, get_choice_text, StarkModelForm
from web import models
from web.utils import pwdmd5


class SchoolHandler(StartHandler):
    # list_display = ["title", StartHandler.display_edit, StartHandler.display_del]
    list_display = ["title"]


class DepartmentHandler(StartHandler):
    list_display = ["title"]


class UserInfoADDModelForm(StarkModelForm):
    ‘‘‘定制用户添加 form。 增加一个 确认密码字段， 并且需要增加验证‘‘‘
    confirm_password = forms.CharField(label="确认密码")

    class Meta:
        model = models.UserInfo
        fields = ["name", "password", "confirm_password", "nickname", "gender", "telephone", "email", "depart", "roles"]

    def clean_confirm_password(self):
        password = self.cleaned_data.get("password")
        confirm_password = self.cleaned_data["confirm_password"]
        if password and confirm_password:
            if password != confirm_password:
                raise ValidationError("两次密码输入不一致")
            return confirm_password  # 验证完成一定要返回，才能添加到 cleaned_data 字典中. 其他地方才能通过cleaned_data获取到

    def clean(self):
        ‘‘‘对密码进行加密， 可以在clean 里面来做。‘‘‘
        password = self.cleaned_data["password"]
        self.cleaned_data["password"] = pwdmd5.creatr_md5(password)
        return self.cleaned_data


class UserInfoEditModelForm(StarkModelForm):
    ‘‘‘定制用户编辑 form。不显示密码字段。‘‘‘
    class Meta:
        model = models.UserInfo
        fields = ["name", "nickname", "gender", "telephone", "email", "depart", "roles"]


class UserInfoHandler(StartHandler):
    list_display = ["name", "nickname", "age", get_choice_text("性别", "gender"), "telephone", "email", "depart", "roles"]

    def get_model_form_class(self, is_add=False):
        if is_add:
            return UserInfoADDModelForm
        return UserInfoEditModelForm


site.register(models.School, SchoolHandler)
site.register(models.Department, DepartmentHandler)
site.register(models.UserInfo, UserInfoHandler)

在这里，因为要使用 rbac 组件。 所以相应的  我的UserInfo 表。 是继承的 Rbac 组件中的UserInfo 表。
所以  rbac 组件中的 userinfo 表。需要一点小小的，调整：

class UserInfo(models.Model):
    name = models.CharField(verbose_name=‘用户名‘, max_length=32)
    password = models.CharField(verbose_name=‘密码‘, max_length=64)
    email = models.CharField(verbose_name=‘邮箱‘, max_length=32)
    roles = models.ManyToManyField(verbose_name=‘拥有的所有角色‘, to=Role, blank=True, null=True)
    # 一定要记住， 如果被继承了！ 在别的地方创建关联关系的时候。 会在别的地方找 Role 这张表。
    # 但是 别的地方肯定是没有的， 所有 在指定关联表的时候，直接把表对象放进去 to=Role 不要再使用 to="Role"
    # 这样在继承的时候， 会连带这Role这张表的内存地址，一起继承过去。 否则，会报错。 说在你的业务中找不到 Role 这张表

    class Meta:
        # django以后再做数据库迁移的时候， 不再为UserInfo类，创建相关的表以及结构
        # 此类 可以当作 "父类", 被其他 model类继承。
        abstract = True

ok 这就可以了！

再来看一看定制方面的功能：
需求：  添加用户时， 需要增加一个  确认密码的字段。  而编辑用户时， 密码字段不应该显示。
实现：  自定义两个 form 类。 一个添加，一个删除。

class UserInfoADDModelForm(StarkModelForm):
    ‘‘‘定制用户添加 form。 增加一个 确认密码字段， 并且需要增加验证‘‘‘
    confirm_password = forms.CharField(label="确认密码")

    class Meta:
        model = models.UserInfo
        fields = ["name", "password", "confirm_password", "nickname", "gender", "telephone", "email", "depart", "roles"]

    def clean_confirm_password(self):
        password = self.cleaned_data.get("password")
        confirm_password = self.cleaned_data["confirm_password"]
        if password and confirm_password:
            if password != confirm_password:
                raise ValidationError("两次密码输入不一致")
            return confirm_password  # 验证完成一定要返回，才能添加到 cleaned_data 字典中. 其他地方才能通过cleaned_data获取到

    def clean(self):
        ‘‘‘对密码进行加密， 可以在clean 里面来做。‘‘‘
        password = self.cleaned_data["password"]
        self.cleaned_data["password"] = pwdmd5.creatr_md5(password)
        return self.cleaned_data

添加的 form 类。 增加confirm_password 字段。   使用 clean_..... 对该密码进行验证。
再全局 钩子clean中， 对密码进行加密。 因为只有前面的验证通过了， 才会走到这里。

加密写的简单：瞎搞：

import hashlib
def creatr_md5(origin):
    # ha = hashlib.md5(b"asdasdasdasdasd")  # 也可以在这里进行加盐
    ha = hashlib.md5()  # 也可以在这里进行加盐
    ha.update(origin.encode("utf-8"))
    return ha.hexdigest()

 

再来是编辑：

class UserInfoEditModelForm(StarkModelForm):
    ‘‘‘定制用户编辑 form。不显示密码字段。‘‘‘
    class Meta:
        model = models.UserInfo
        fields = ["name", "nickname", "gender", "telephone", "email", "depart", "roles"]

只是少显示一个字段， 没啥好说的！

自定制：

class UserInfoHandler(StartHandler):
    list_display = ["name", "nickname", "age", get_choice_text("性别", "gender"), "telephone", "email", "depart", "roles"]

    def get_model_form_class(self, is_add=False):
        if is_add:
            return UserInfoADDModelForm
        return UserInfoEditModelForm

重写StartHandler 基类中的  get_model_form_class()  方法。返回不同的值：
　　此方法是，有 add视图函数， 和 rdit 视图函数进行调用的。
再调用时， 传入参数， is_add=Ture  或者  is_add=False  拿到想要的 form 类。 就可以实现了！

list_display 此列表作用于， 展示页面。 传几个字段，就展示几个字段的信息。

其中的函数 get_choice_text("性别", "gender") 是一个使用，闭包完成的，展示 filed对象中， 有 choice 选项的， 让其显示中文的功能。