import os
from django.core.files.storage import FileSystemStorage
from django.db import models
from django.contrib.auth.models import User
import tagging
from tagging.fields import TagField
from pytask.taskapp.utilities.helper import get_key
GENDER_CHOICES = (( 'M', 'Male'), ('F', 'Female'))
RIGHTS_CHOICES = (
("AD", "Admin"),
("MG", "Manager"),
("DV", "Developer"),
("CT", "Contributor"),)
STATUS_CHOICES = (
("UP", "Unpublished"),
("OP", "Open"),
("LO", "Locked"),
("WR", "Working"),
("CD", "Closed"),
("DL", "Deleted"),
("CM", "Completed"))
NOTIFY_CHOICES = (
("MT", "Add Reviewer"),
("DV", "Developer"),
("MG", "Manager"),
("AD", "Admin"),
("PY", "Assign pynts"),
("CM", "Task completed"),
("CD", "Task closed"),
("DL", "Task deleted"),
("NU", "New User"),
("NT", "New Reviewer"),
("ND", "New Developer"),
("NG", "New Manager"),
("NA", "New Admin"),
("AU", "Assign user"), ## i mean assign the task
("RU", "Remove user"), ## remove from working users list in task
)
IMAGES_DIR = "./images"
UPLOADS_DIR = "./uploads"
class CustomImageStorage(FileSystemStorage):
def path(self, name):
""" we return images directory path.
"""
return os.path.join(IMAGES_DIR, name)
def get_available_name(self, name):
""" here we are going with username as the name of image.
"""
root, ext = os.path.splitext(name)
file_name = get_key() + ext
while self.exists(file_name):
file_name = get_key() + ext
return file_name
class Profile(models.Model):
user = models.ForeignKey(User, unique = True)
dob = models.DateField(verbose_name = u"Date of Birth", help_text = "YYYY-MM-DD")
gender = models.CharField(max_length = 1, choices = GENDER_CHOICES)
rights = models.CharField(max_length = 2, choices = RIGHTS_CHOICES, default = u"CT")
pynts = models.PositiveSmallIntegerField(default = 0)
aboutme = models.TextField(blank = True)
foss_comm = TagField(verbose_name="FOSS Communities")
phonenum = models.CharField(max_length = 15, blank = True, verbose_name = u"Phone Number")
homepage = models.URLField(blank = True, verbose_name = u"Homepage/Blog")
street = models.CharField(max_length = 80, blank = True)
city = models.CharField(max_length = 25, blank = True)
country = models.CharField(max_length = 25, blank = True)
nick = models.CharField(max_length = 20, blank = True)
photo = models.ImageField(storage = CustomImageStorage(),upload_to = IMAGES_DIR, blank = True)
def __unicode__(self):
return unicode(self.user.username)
class Task(models.Model):
prim_key = models.AutoField(primary_key = True)
id = models.CharField(max_length = 10, unique = True)
title = models.CharField(max_length = 100, verbose_name = u"Title", help_text = u"Keep it simple and below 100 chars.")
desc = models.TextField(verbose_name = u"Description")
status = models.CharField(max_length = 2, choices = STATUS_CHOICES, default = "UP")
tags_field = TagField(verbose_name = u"Tags", help_text = u"Give comma seperated tags")
pynts = models.PositiveSmallIntegerField(help_text = u"No.of pynts a user gets on completing the task")
progress = models.PositiveSmallIntegerField(default = 0)
reviewers = models.ManyToManyField(User, related_name = "%(class)s_reviewers")
created_by = models.ForeignKey(User, related_name = "%(class)s_created_by")
claimed_users = models.ManyToManyField(User, blank = True, related_name = "%(class)s_claimed_users")
assigned_users = models.ManyToManyField(User, blank = True, related_name = "%(class)s_assigned_users")
creation_datetime = models.DateTimeField()
published_datetime = models.DateTimeField()
sub_type = models.CharField(max_length=1, choices = (('D','Dependency'),('S','Subtask')), default = 'D')
def __unicode__(self):
return unicode(self.title)
class Map(models.Model):
main = models.ForeignKey('Task', related_name = "%(class)s_main")
subs = models.ManyToManyField('Task', blank = True, null = True, related_name = "%(class)s_subs")
class Comment(models.Model):
task = models.ForeignKey('Task')
data = models.TextField()
created_by = models.ForeignKey(User, related_name = "%(class)s_created_by")
creation_datetime = models.DateTimeField()
deleted_by = models.ForeignKey(User, null = True, blank = True, related_name = "%(class)s_deleted_by")
is_deleted = models.BooleanField()
attachment = models.FileField(upload_to = UPLOADS_DIR, blank = True)
def __unicode__(self):
return unicode(self.task.title)
class Request(models.Model):
sent_to = models.ManyToManyField(User, related_name = "%(class)s_sent_to", blank = False)
sent_by = models.ForeignKey(User, related_name = "%(class)s_sent_by", blank = False)
role = models.CharField(max_length = 2, blank = False)
reply = models.BooleanField(default = False, blank = False)
remarks = models.TextField(default = "",blank = True)
is_read = models.BooleanField(default = False, blank = False)
is_valid = models.BooleanField(default = True, blank = False)
creation_date = models.DateTimeField()
reply_date = models.DateTimeField()
is_replied = models.BooleanField(default = False)
replied_by = models.ForeignKey(User, related_name = "%(class)s_replied_by", blank = True, null = True)
task = models.ForeignKey(Task,related_name = "%(class)s_task", blank = True, null = True)
receiving_user = models.ForeignKey(User, related_name = "%(class)s_receiving_user", blank = True, null = True)
pynts = models.PositiveIntegerField(default=0)
def __unicode__(self):
return u"Request %s %s"%(self.sent_by.username, self.role)
class Notification(models.Model):
role = models.CharField(max_length = 2, choices = NOTIFY_CHOICES, blank = False)
sent_to = models.ForeignKey(User, related_name = "%(class)s_sent_to", blank = False)
sent_from = models.ForeignKey(User, related_name = "%(class)s_sent_from", null = True, blank = True)
task = models.ForeignKey(Task, related_name = "%(class)s_task", null = True, blank = True)
sub = models.CharField(max_length = 100)
message = models.TextField()
remarks = models.CharField(max_length = 100)
sent_date = models.DateTimeField()
is_read = models.BooleanField(default = False)
is_deleted = models.BooleanField(default = False)
def __unicode__(self):
return u"%s %s %s"%(self.sent_to, self.message, self.sent_date.ctime())
class WorkReport(models.Model):
attachment = models.FileField(upload_to = UPLOADS_DIR, blank = False)
remarks = models.TextField()
revision = models.PositiveIntegerField(default=0)
task = models.ForeignKey(Task, related_name = "%(class)s_report")
submitted_by = models.ForeignKey(Task, related_name = "%(class)s_submitted_by")
created_at = models.DateTimeField()
#tagging.register(Profile)
#tagging.register(Task)