写了这么多的定制 功能。终于可以定制一下了!
因为是 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 选项的, 让其显示中文的功能。