pytask/taskapp/models.py
branchbuildout
changeset 227 3c8f3b0e5b00
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pytask/taskapp/models.py	Thu Jan 06 19:07:04 2011 +0530
@@ -0,0 +1,173 @@
+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 Mentor"),
+    ("DV", "Developer"),
+    ("MG", "Manager"),
+    ("AD", "Admin"),
+    ("PY", "Assign credits"),
+    ("CM", "Task completed"),
+    ("CD", "Task closed"),
+    ("DL", "Task deleted"),
+    ("NU", "New User"),
+    ("NT", "New Mentor"),
+    ("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")
+    credits = 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") 
+    
+    credits = models.PositiveSmallIntegerField(help_text = u"No.of credits a user gets on completing the task")
+    progress = models.PositiveSmallIntegerField(default = 0)
+        
+    mentors = models.ManyToManyField(User, related_name = "%(class)s_mentors")
+    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())
+    
+tagging.register(Profile)
+tagging.register(Task)