码迷,mamicode.com
首页 > 其他好文 > 详细

Models模型(下)

时间:2018-09-14 21:27:29      阅读:241      评论:0      收藏:0      [点我收藏+]

标签:dir   save   作者   相对   war   return   update   python   password   

一、最基本的django模型

1、先看下一个新闻博客的Article模型。这个模型是最基本的django模型,里面包括了各个字段(fields),重写了显示文章对象名字的__str__方法(python内置的),并在Meta选项里给模型命名(verbose name)。建议每个django模型至少包括字段,重写的__str__方法和Meta选项。

from django.db import models
from django.contrib.auth.models import User
from django.urls import reverse
from django.utils.timezone import now

# Create your models here.

class Article(models.Model):

    STATUS_CHOICES = (
        (d, 草稿),
        (p, 发表),
    )

    title = models.CharField(标题, max_length=200, unique=True)
    slug = models.SlugField(slug, max_length=60)
    body = models.TextField(正文)
    pub_date = models.DateTimeField(发布时间, default=now, null=True)
    create_date = models.DateTimeField(创建时间, auto_now_add=True)
    mod_date = models.DateTimeField(修改时间, auto_now=True)
    status = models.CharField(文章状态, max_length=1, choices=STATUS_CHOICES, default=p)
    views = models.PositiveIntegerField(浏览量, default=0)
    author = models.ForeignKey(User, verbose_name=作者, on_delete=models.CASCADE)

    tags = models.ManyToManyField(Tag, verbose_name=标签集合, blank=True)

    def __str__(self):
        return self.title

    class Meta:
        verbose_name = "article"

2、基础模型很多时候并不能满足需求。试想打算使用django自带的通用视图创建文章,由于通用视图在完成对象创建后需要跳转到文章的absolute_url, 这时需要在模型里加入自定义的get_absolute_url方法。由于希望统计每篇文章浏览次数,还需自定义一个使浏览量自增1的viewed方法,并更新数据表。

def get_absolute_url(self):
    return reverse(blog:article_detail, args=[str(self.id)])

def viewed(self):
    self.views += 1
    self.save(update_fields=[views])

如果希望调用Article.objects.all()按时pub_date降序排列查询结果,可以在Meta里加入ordering选项即可。

class Meta:
    ordering = [-pub_date]
    verbose_name = "article"

二、模型中自定义图片和文件上传路径

  Django模型中的ImageField和FileField的upload_to选项是必填项,其存储路径是相对于MEIDA_ROOT而来的。然而可能希望动态定义上传路径,比如把文件上传到每个用户名下的文件夹里,并对上传文件重命名,这时可以定义一个user_directory_path方法。

from django.db import models
from django.contrib.auth.models import User
import uuid
import os

# Create your models here.


def user_directory_path(instance, filename):
    ext = filename.split(.)[-1]
    filename = {}.{}.format(uuid.uuid4().hex[:10], ext)
    # return the whole path to the file
    return os.path.join(instance.user.id, "avatar", filename)


class UserProfile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name=profile)
    avatar = models.ImageField(upload_to=user_directory_path, verbose_name="头像")

三、Django模型的Manager方法

  Django模型自带models.Manager方法,可以简化代码。如下面案例中,可以使用Person.objects.all()查询到所有人,而Person.authors.all和Person.editors.all()只返回所authors和editors。

from django.db import models

# Create your models here.


class AuthorManager(models.Manager):
    def get_queryset(self):
        return super().get_queryset().filter(role=A)

class EditorManager(models.Manager):
    def get_queryset(self):
        return super().get_queryset().filter(role=E)

class Person(models.Model):
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    role = models.CharField(max_length=1, choices=((A, _(Author)), (E, _(Editor))))
    objects = models.Manager()
    authors = AuthorManager()
    editors = EditorManager()

四、Django模型的save方法重写

在很多应用场景中需要重写django模型的save方法。

from django.db import models
import hashlib


class UserInfo(models.Model):
    username = models.CharField("用户名", max_length=64, unique=True)
    password = models.CharField("密码", max_length=64)
    uid = models.CharField(verbose_name=个人唯一ID, max_length=64, unique=True)

    wx_id = models.CharField(verbose_name="微信ID", max_length=128, blank=True, null=True, db_index=True)

    def save(self, *args, **kwargs):
        # 创建用户时,为用户自动生成个人唯一ID
        if not self.pk:
        # 存在就更新,不存在就创建
            m = hashlib.md5()
            m.update(self.username.encode(encoding="utf-8"))
            self.uid = m.hexdigest()
        super(UserInfo, self).save(*args, **kwargs)

五、完美的Django高级模型结构

from django.db import models
from django.urls import reverse

# Create your models here.

# 自定义Manager方法
class HighRatingManager(models.Manager):
    def get_queryset(self):
        return super().get_queryset().filter(rating=1)


class Product(models.Model):
    # CHOICES选项
    RATING_CHOICES = (
        ("1", Very good),
        ("2", Good),
        ("3", Bad),
    )

    # 数据表字段
    name = models.CharField(name, max_length=30)
    rating = models.CharField(max_length=1, choices=RATING_CHOICES)


# MANAGERS方法
objects = models.Manager()
high_rating_products = HighRatingManager()


# META类选项
class Meta:
    verbose_name = product
    verbose_name_plural = products


# __str__方法
def __str__(self):
    return self.name


# 重写save方法
def save(self, *args, **kwargs):
    pass

# 定义绝对路径
def get_absolute_url(self):
    return reverse(product_details, kwargs={pk: self.id})


# 定义其它方法
def do_something(self):
    pass

 

Models模型(下)

标签:dir   save   作者   相对   war   return   update   python   password   

原文地址:https://www.cnblogs.com/bijian/p/9648882.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!