68 lines
3.1 KiB
Python
68 lines
3.1 KiB
Python
|
from django.conf.global_settings import MEDIA_ROOT
|
||
|
|
from django.db import models
|
||
|
|
import os
|
||
|
|
|
||
|
|
|
||
|
|
def user_directory_path(instance, filename):
|
||
|
|
ext = filename.split('.').pop()
|
||
|
|
filename = '{0}{1}.{2}'.format(instance.username, instance.phone, ext)
|
||
|
|
return os.path.join("avatar", filename) # 系统路径分隔符差异，增强代码重用性
|
||
|
|
|
||
|
|
|
||
|
|
# Create your models here.
|
||
|
|
class User(models.Model):
|
||
|
|
"""
|
||
|
|
用户表
|
||
|
|
"""
|
||
|
|
id = models.AutoField(verbose_name="id", primary_key=True, unique=True)
|
||
|
|
username = models.CharField(max_length=50, unique=True, verbose_name="用户名", null=True, blank=True)
|
||
|
|
pwd = models.CharField(max_length=1000, verbose_name="密码", null=True, blank=True)
|
||
|
|
email = models.EmailField(max_length=100, verbose_name="电子邮件")
|
||
|
|
phone = models.CharField(max_length=11, verbose_name="手机", null=True, blank=True)
|
||
|
|
create_time = models.DateTimeField(verbose_name="注册时间")
|
||
|
|
avatar = models.ImageField(verbose_name="头像", upload_to=user_directory_path)
|
||
|
|
last_login_time = models.DateTimeField(verbose_name="最后登录时间", auto_now=True)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.username
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
verbose_name = '用户表'
|
||
|
|
|
||
|
|
# 这里定义一个方法，作用是当用户注册时没有上传照片，模板中调用 [ModelName].[ImageFieldName].url 时赋予一个默认路径
|
||
|
|
def avatar_url(self):
|
||
|
|
if self.avatar and hasattr(self.avatar, 'url'):
|
||
|
|
return self.avatar.url
|
||
|
|
else:
|
||
|
|
return '/media/avatar/default.jpg'
|
||
|
|
|
||
|
|
|
||
|
|
class Menu(models.Model):
|
||
|
|
id = models.AutoField(verbose_name="id", primary_key=True, unique=True)
|
||
|
|
menu_name = models.CharField(verbose_name="菜单名称", max_length=50, null=False, blank=False)
|
||
|
|
parent_id = models.IntegerField(verbose_name="父菜单")
|
||
|
|
path = models.CharField(verbose_name="路由地址", max_length=128)
|
||
|
|
order = models.IntegerField(verbose_name="排序", default=0)
|
||
|
|
create_time = models.DateTimeField(verbose_name="创建时间")
|
||
|
|
update_time = models.DateTimeField(verbose_name="最后更新时间", auto_now=True)
|
||
|
|
|
||
|
|
|
||
|
|
class Role(models.Model):
|
||
|
|
id = models.AutoField(verbose_name="id", primary_key=True, unique=True)
|
||
|
|
role_name = models.CharField(verbose_name="角色名称", max_length=50, null=False, blank=False)
|
||
|
|
role_name_en = models.CharField(verbose_name="角色英文名称", max_length=50)
|
||
|
|
create_time = models.DateTimeField(verbose_name="创建时间")
|
||
|
|
update_time = models.DateTimeField(verbose_name="最后更新时间", auto_now=True)
|
||
|
|
|
||
|
|
|
||
|
|
class RoleMenu(models.Model):
|
||
|
|
id = models.AutoField(verbose_name="id", primary_key=True, unique=True)
|
||
|
|
menu_id = models.ForeignKey(verbose_name="菜单ID", to="Menu", to_field="id", on_delete=models.CASCADE)
|
||
|
|
role_id = models.ForeignKey(verbose_name="角色ID", to="Role", to_field="id", on_delete=models.CASCADE)
|
||
|
|
|
||
|
|
|
||
|
|
class RoleUser(models.Model):
|
||
|
|
id = models.AutoField(verbose_name="id", primary_key=True, unique=True)
|
||
|
|
role_id = models.ForeignKey(verbose_name="角色ID", to="Role", to_field="id", on_delete=models.CASCADE)
|
||
|
|
user_id = models.ForeignKey(verbose_name="用户ID", to="User", to_field="id", on_delete=models.CASCADE)
|