获取当前登录用户并在Model |上设置为默认用户Django + Python3

时间:2017-02-23 19:50:09

标签: python django

我正在学习Django,并挑战自己创建一个小型票务系统作为研究案例。

完成了一些事情,但现在我遇到了一些问题。

如何保存“Ticket”,Django获取当前登录用户并在我的Ticket模型的“usuario”字段中设置为默认值?

这是我的models.py文件:

from django.contrib.auth.models import User
from django.db import models
from datetime import datetime

class Projeto(models.Model):
    """
        Classe que gere os Projetos
        Permite que se cadastre N usuários por Projeto
        Retorna:
            NOME_DO_PROJETO | SITE_DO_PROJETO
    """
    nome = models.CharField(max_length=100)
    site = models.CharField(max_length=200)
    informacoes = models.TextField(blank=True)
    usuarios = models.ManyToManyField(User, related_name='projetos')

    def __str__(self):
        return self.nome + " | " + self.site

class Ticket(models.Model):
    """
        Classe que gere os Tickets no sistema.
        Retorna:
            DATA HORA | TITULO DO CHAMADO
    """
    TIPOS_TICKET = (
        ('BUG', 'Bug'),
        ('URGENTE', 'Urgente'),
        ('FINANCEIRO', 'Financeiro'),
        ('MELHORIA', 'Melhoria')
    )
    STATUS_TICKET = (
        ('ABERTO', 'Aberto'),
        ('AGUARDANDO_CLIENTE', 'Aguardando Cliente'),
        ('EM_ANALISE', 'Em Análise'),
        ('FINALIZADO', 'Finalizado'),
        ('CANCELADO', 'Cancelado'),
    )
    titulo = models.CharField(max_length=200)
    conteudo = models.TextField()
    tipo = models.CharField(max_length=30, choices=TIPOS_TICKET, default='BUG')
    status = models.CharField(max_length=30, choices=STATUS_TICKET, default='ABERTO')
    projeto = models.ForeignKey(
        Projeto,
        on_delete=models.CASCADE,
        limit_choices_to={'usuarios':1}
    )
    usuario = models.ForeignKey(
        User,
        on_delete=models.CASCADE,
        null=True
    )
    data_abertura = models.DateTimeField('Data Abertura', auto_now_add=True)
    data_fechamento = models.DateTimeField('Data Fechamento', blank=True, null=True)

    def __str__(self):
        return str(datetime.strftime(self.data_abertura, "%d/%m/%y %H:%M") + " | " + self.titulo)

    def save(self, *args, **kwargs):
        self.usuario = User
        super(Ticket, self).save(*args, **kwargs)

class TicketMsg(models.Model):
    """
        Mensagens dos tickets
        Retorna:
            texto da mensagem
    """
    texto = models.TextField()
    ticket = models.ForeignKey(Ticket, on_delete=models.CASCADE)
    data_resposta = models.DateTimeField('Data Resposta')

    def __str__(self):
        return str(self.texto)

我的admin.py文件:

from django.contrib import admin
from django.contrib.auth.models import User
from .models import Ticket, TicketMsg, Projeto

# Register your models here.
class ProjetoAdmin(admin.ModelAdmin):
    model = Projeto
    filter_horizontal = ('usuarios',)

class TicketAdmin(admin.ModelAdmin):
    model = Ticket
    exclude = ('status', 'data_fechamento', 'data_abertura', 'usuario')
    list_display = ('titulo', 'tipo', 'status', 'projeto', 'data_abertura')
    list_filter = ('projeto', 'tipo', 'status')

    def get_ordering(self, request):
        return ['data_abertura']

admin.site.register(Projeto, ProjetoAdmin)
admin.site.register(Ticket, TicketAdmin)
admin.site.register(TicketMsg)

这是Django Admin的票据清单。

inserir a descrição da imagem aqui

这个想法是使用默认的Django用户系统来识别谁试图打开票证。

我成功地系统过滤了每个用户可以打开门票的项目,但是现在我无法将当前用户设置为新门票的默认值。

这实际上是我的表格。

inserir a descrição da imagem aqui

欢迎所有帮助! 谢谢你们!

1 个答案:

答案 0 :(得分:2)

当前登录的用户作为Djangos Request-Response-Lifecycle的一部分附加在请求对象上。您无法在模型的方法中访问请求(就像您希望在 save()方法中那样)。

但是,您可以覆盖管理表单的 save_model()方法,其中请求可用:

class TicketAdmin(admin.ModelAdmin):
    model = Ticket
    exclude = ('status', 'data_fechamento', 'data_abertura', 'usuario')
    list_display = ('titulo', 'tipo', 'status', 'projeto', 'data_abertura')
    list_filter = ('projeto', 'tipo', 'status')

    def get_ordering(self, request):
        return ['data_abertura']

    def save_model(self, request, obj, form, change):
        obj.usuario = request.user
        super(TicketAdmin, self).save_model(request, obj, form, change)

然后无需覆盖模型上的save()方法。

有关官方Django文档的ModelAdmin的更多信息:https://docs.djangoproject.com/en/1.10/ref/contrib/admin/#django.contrib.admin.ModelAdmin.save_model