232 lines
7.4 KiB
Python
232 lines
7.4 KiB
Python
from datetime import datetime
|
|
import os
|
|
from django import apps
|
|
from django.db import models
|
|
from django.forms import ValidationError
|
|
from base.models import Department, JobPosition, Tags
|
|
from django.utils.translation import gettext_lazy as _
|
|
from employee.models import Employee
|
|
from base.models import Company
|
|
from base.horilla_company_manager import HorillaCompanyManager
|
|
from django.db.models.signals import post_save, post_delete
|
|
from horilla_audit.methods import get_diff
|
|
|
|
from horilla_audit.models import HorillaAuditInfo, HorillaAuditLog
|
|
|
|
|
|
PRIORITY = [
|
|
("low", "Low"),
|
|
("medium", "Medium"),
|
|
("high", "High"),
|
|
]
|
|
MANAGER_TYPES = [
|
|
("department", "Department"),
|
|
("job_position", "Job Position"),
|
|
("individual", "Individual"),
|
|
]
|
|
|
|
TICKET_TYPES = [
|
|
("suggestion", "Suggestion"),
|
|
("complaint", "Complaint"),
|
|
("service_request", "Service Request"),
|
|
("meeting_request", "Meeting Request"),
|
|
("anounymous_complaint", "Anonymous Complaint"),
|
|
("others", "Others"),
|
|
]
|
|
|
|
TICKET_STATUS = [
|
|
("new","New"),
|
|
("in_progress","In Progress"),
|
|
("on_hold","On Hold"),
|
|
("resolved","Resolved"),
|
|
("canceled","Canceled"),
|
|
]
|
|
|
|
|
|
class DepartmentManager(models.Model):
|
|
manager =models.ForeignKey(
|
|
Employee,
|
|
verbose_name="Manager",
|
|
related_name="dep_manager",
|
|
on_delete=models.CASCADE
|
|
)
|
|
department = models.ForeignKey(
|
|
Department,
|
|
verbose_name="Department",
|
|
related_name="dept_manager",
|
|
on_delete=models.CASCADE
|
|
)
|
|
company_id = models.ForeignKey(
|
|
Company,
|
|
null=True, editable=False,
|
|
on_delete=models.PROTECT
|
|
)
|
|
|
|
class TicketType(models.Model):
|
|
title = models.CharField(max_length = 100,unique=True)
|
|
type = models.CharField(choices = TICKET_TYPES , max_length=50)
|
|
prefix = models.CharField(max_length=3,unique=True)
|
|
is_active = models.BooleanField(default=True)
|
|
company_id = models.ForeignKey(Company,null=True, editable=False, on_delete=models.PROTECT)
|
|
objects = HorillaCompanyManager(
|
|
related_company_field="company_id"
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
class Ticket(models.Model):
|
|
|
|
title = models.CharField(max_length=50)
|
|
employee_id = models.ForeignKey(
|
|
Employee,
|
|
on_delete=models.PROTECT,
|
|
related_name="ticket",
|
|
verbose_name="Owner"
|
|
)
|
|
ticket_type = models.ForeignKey(
|
|
TicketType,
|
|
on_delete=models.PROTECT,
|
|
verbose_name="Ticket Type",
|
|
)
|
|
description = models.TextField(max_length=255)
|
|
priority = models.CharField(choices = PRIORITY, max_length=100, default= "low")
|
|
created_date = models.DateField(auto_now_add=True)
|
|
resolved_date = models.DateField(blank=True, null=True)
|
|
assigning_type = models.CharField(choices = MANAGER_TYPES, max_length=100)
|
|
raised_on = models.CharField(max_length=100,verbose_name="Forward To")
|
|
assigned_to = models.ManyToManyField(Employee,blank=True,related_name="ticket_assigned_to")
|
|
deadline = models.DateField(null=True,blank=True)
|
|
tags = models.ManyToManyField(Tags,blank=True,related_name="ticket_tags")
|
|
status = models.CharField(choices=TICKET_STATUS,default="new" ,max_length=50)
|
|
is_active = models.BooleanField(default=True)
|
|
history = HorillaAuditLog(
|
|
related_name="history_set",
|
|
bases=[
|
|
HorillaAuditInfo,
|
|
],
|
|
)
|
|
objects = HorillaCompanyManager(
|
|
related_company_field="employee_id__employee__work_info__company_id"
|
|
)
|
|
def clean(self, *args, **kwargs):
|
|
super().clean(*args, **kwargs)
|
|
deadline = self.deadline
|
|
today = datetime.today().date()
|
|
if deadline < today:
|
|
raise ValidationError(_("Deadline should be greater than today"))
|
|
def get_raised_on(self):
|
|
obj_id = self.raised_on
|
|
if self.assigning_type == 'department':
|
|
raised_on = Department.objects.get(id=obj_id).department
|
|
elif self.assigning_type == 'job_position':
|
|
raised_on = JobPosition.objects.get(id=obj_id).job_position
|
|
elif self.assigning_type == 'individual':
|
|
raised_on = Employee.objects.get(id=obj_id).get_full_name()
|
|
return raised_on
|
|
|
|
def get_raised_on_object(self):
|
|
obj_id = self.raised_on
|
|
if self.assigning_type == 'department':
|
|
raised_on = Department.objects.get(id=obj_id)
|
|
elif self.assigning_type == 'job_position':
|
|
raised_on = JobPosition.objects.get(id=obj_id)
|
|
elif self.assigning_type == 'individual':
|
|
raised_on = Employee.objects.get(id=obj_id)
|
|
return raised_on
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
def tracking(self):
|
|
"""
|
|
This method is used to return the tracked history of the instance
|
|
"""
|
|
return get_diff(self)
|
|
|
|
|
|
class Comment(models.Model):
|
|
comment = models.TextField(null=True, blank=True,max_length=255)
|
|
ticket = models.ForeignKey(Ticket, on_delete=models.CASCADE, related_name="comment")
|
|
employee_id = models.ForeignKey(Employee, on_delete=models.DO_NOTHING, related_name="employee_comment")
|
|
date = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.comment
|
|
|
|
|
|
class Attachment(models.Model):
|
|
file = models.FileField(upload_to="Tickets/Attachment")
|
|
description = models.CharField(max_length=100, blank=True, null=True)
|
|
format = models.CharField(max_length=50, blank=True, null=True)
|
|
ticket = models.ForeignKey(
|
|
Ticket,
|
|
on_delete=models.CASCADE,
|
|
null=True, blank=True,
|
|
related_name = "ticket_attachment",
|
|
)
|
|
comment = models.ForeignKey(
|
|
Comment,
|
|
on_delete=models.CASCADE,
|
|
null=True, blank=True,
|
|
related_name = "comment_attachment",
|
|
)
|
|
def get_file_format(self):
|
|
image_format = ['.jpg','.jpeg','.png','.svg']
|
|
audio_format = ['.m4a','.mp3']
|
|
file_extension = os.path.splitext(self.file.url)[1].lower()
|
|
if file_extension in audio_format:
|
|
self.format = 'audio'
|
|
elif file_extension in image_format:
|
|
self.format = 'image'
|
|
else:
|
|
self.format = 'file'
|
|
|
|
|
|
def save(self,*args,**kwargs):
|
|
self.get_file_format()
|
|
|
|
super().save(self,*args,**kwargs)
|
|
|
|
def __str__(self):
|
|
return os.path.basename(self.file.name)
|
|
|
|
|
|
class FAQCategory(models.Model):
|
|
title = models.CharField(max_length=30)
|
|
description = models.TextField(blank=True, null=True,max_length=255)
|
|
|
|
def __str__(self):
|
|
return self.title
|
|
|
|
|
|
class FAQ(models.Model):
|
|
question = models.CharField(max_length=255)
|
|
answer=models.TextField(max_length=255)
|
|
tags = models.ManyToManyField(Tags)
|
|
category = models.ForeignKey(FAQCategory,on_delete=models.PROTECT)
|
|
is_active = models.BooleanField(default=True)
|
|
company_id = models.ForeignKey(Company,null=True, editable=False, on_delete=models.PROTECT)
|
|
objects = HorillaCompanyManager(
|
|
related_company_field="company_id"
|
|
)
|
|
|
|
def __str__(self):
|
|
return self.question
|
|
|
|
|
|
# updating the faq search index when a new faq is created or deleted
|
|
|
|
def update_index(sender, instance, **kwargs):
|
|
from .search_indexes import FAQIndex
|
|
index = FAQIndex()
|
|
index.update_object(instance)
|
|
|
|
def remove_from_index(sender, instance, **kwargs):
|
|
from .search_indexes import FAQIndex
|
|
index = FAQIndex()
|
|
index.remove_object(instance)
|
|
|
|
|
|
post_save.connect(update_index, sender=FAQ)
|
|
post_delete.connect(remove_from_index, sender=FAQ) |