models.py
2.62 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import os
import time
from django.db import models
from django.core import validators
from django.core.exceptions import ValidationError
from django.utils.translation import ugettext_lazy as _
from topics.decorators import always_as_child
from categories.models import Category
from subjects.models import Subject
from topics.models import KnowsChild
from users.models import User
def validate_img_extension(value):
valid_formats = ['image/jpeg','image/x-citrix-jpeg','image/png','image/x-citrix-png','image/x-png','image/gif']
if hasattr(value.file, 'content_type'):
if not value.file.content_type in valid_formats:
raise ValidationError(_('File not supported.'))
def upload_filename(instance, filename):
path = "chat/"
filename = str(int(time.time())) + "_" + filename
return os.path.join(path, filename)
class Conversation(KnowsChild):
user_one = models.ForeignKey(User, verbose_name = _('User One'), related_name = 'talk_user_start')
user_two = models.ForeignKey(User, verbose_name = _('User Two'), related_name = 'talk_user_end')
class GeneralTalk(Conversation):
space = models.IntegerField(_('Space'), default = 0, blank = True)
class CategoryTalk(Conversation):
space = models.ForeignKey(Category, verbose_name = ('Category'), related_name = 'talk_category', null = True)
class SubjectTalk(Conversation):
space = models.ForeignKey(Subject, verbose_name = _('Subject'), related_name = 'talk_subject', null = True)
class TalkMessages(models.Model):
text = models.TextField(_('Comment'), blank = True)
image = models.ImageField(verbose_name = _('Image'), null = True, blank = True, upload_to = upload_filename, validators = [validate_img_extension])
talk = models.ForeignKey(Conversation, verbose_name = _('Conversation'), related_name = 'message_talk', null = True)
user = models.ForeignKey(User, verbose_name = _('User'), related_name = 'message_user', null = True)
create_date = models.DateTimeField(_('Create Date'), auto_now_add = True)
class ChatVisualizations(models.Model):
viewed = models.BooleanField(_('Viewed'), default = False)
message = models.ForeignKey(TalkMessages, verbose_name = _('Message'), related_name = 'chat_visualization_message', null = True)
user = models.ForeignKey(User, verbose_name = _('User'), related_name = "chat_visualization_user", null = True)
date_viewed = models.DateTimeField(_('Date/Time Viewed'), null = True, blank = True)
class ChatFavorites(models.Model):
message = models.ForeignKey(TalkMessages, verbose_name = _('Message'), related_name = 'chat_favorites_message', null = True)
user = models.ForeignKey(User, verbose_name = _('User'), related_name = "chat_favorites_user", null = True)