38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from django.db import models
|
||
from django.contrib.auth import get_user_model
|
||
|
||
Employee = get_user_model()
|
||
|
||
|
||
class Request(models.Model):
|
||
TYPE_CHOICES = [
|
||
('vacation', 'Отпуск'),
|
||
('sick_leave', 'Больничный'),
|
||
('transfer', 'Перевод'),
|
||
('dismissal', 'Увольнение'),
|
||
('other', 'Другое'),
|
||
]
|
||
STATUS_CHOICES = [
|
||
('pending', 'На рассмотрении'),
|
||
('approved', 'Одобрено'),
|
||
('rejected', 'Отклонено'),
|
||
]
|
||
employee = models.ForeignKey(
|
||
Employee, on_delete=models.CASCADE, related_name='requests')
|
||
type = models.CharField('Тип заявления', choices=TYPE_CHOICES, max_length=32)
|
||
date_from = models.DateField('Дата начала', null=True, blank=True)
|
||
date_to = models.DateField('Дата окончания', null=True, blank=True)
|
||
comment = models.TextField('Комментарий', blank=True)
|
||
status = models.CharField(
|
||
'Статус', choices=STATUS_CHOICES, max_length=16, default='pending')
|
||
created_at = models.DateTimeField(auto_now_add=True)
|
||
|
||
class Meta:
|
||
verbose_name = 'Заявление'
|
||
verbose_name_plural = 'Заявления'
|
||
ordering = ['-created_at']
|
||
|
||
def __str__(self):
|
||
return f"{self.get_type_display()} — {self.employee} ({self.get_status_display()})"
|
||
|