-
Notifications
You must be signed in to change notification settings - Fork 641
/
models.py
123 lines (99 loc) · 4.35 KB
/
models.py
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# -*- coding: utf-8 -*-
"""
flaskbb.message.models
~~~~~~~~~~~~~~~~~~~~~~
The models for the conversations and messages are located here.
:copyright: (c) 2014 by the FlaskBB Team.
:license: BSD, see LICENSE for more details.
"""
import logging
from sqlalchemy_utils import UUIDType
from flaskbb.extensions import db
from flaskbb.utils.helpers import time_utcnow
from flaskbb.utils.database import CRUDMixin, UTCDateTime
logger = logging.getLogger(__name__)
class Conversation(db.Model, CRUDMixin):
__tablename__ = "conversations"
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer,
db.ForeignKey("users.id", ondelete="CASCADE"),
nullable=False)
from_user_id = db.Column(db.Integer, db.ForeignKey("users.id",
ondelete="SET NULL"),
nullable=True)
to_user_id = db.Column(db.Integer, db.ForeignKey("users.id",
ondelete="SET NULL"),
nullable=True)
shared_id = db.Column(UUIDType, nullable=False)
subject = db.Column(db.String(255), nullable=True)
date_created = db.Column(UTCDateTime(timezone=True), default=time_utcnow,
nullable=False)
date_modified = db.Column(UTCDateTime(timezone=True), default=time_utcnow,
nullable=False)
trash = db.Column(db.Boolean, default=False, nullable=False)
draft = db.Column(db.Boolean, default=False, nullable=False)
unread = db.Column(db.Boolean, default=False, nullable=False)
messages = db.relationship(
"Message", lazy="joined", backref="conversation",
primaryjoin="Message.conversation_id == Conversation.id",
order_by="asc(Message.id)", cascade="all, delete-orphan"
)
# this is actually the users message box
user = db.relationship("User", lazy="joined", foreign_keys=[user_id])
# the user to whom the conversation is addressed
to_user = db.relationship("User", lazy="joined", foreign_keys=[to_user_id])
# the user who sent the message
from_user = db.relationship("User", lazy="joined",
foreign_keys=[from_user_id])
@property
def first_message(self):
"""Returns the first message object."""
return self.messages[0]
@property
def last_message(self):
"""Returns the last message object."""
return self.messages[-1]
def save(self, message=None):
"""Saves a conversation and returns the saved conversation object.
:param message: If given, it will also save the message for the
conversation. It expects a Message object.
"""
if message is not None:
# create the conversation
self.date_created = time_utcnow()
db.session.add(self)
db.session.commit()
# create the actual message for the conversation
message.save(self)
return self
self.date_modified = time_utcnow()
db.session.add(self)
db.session.commit()
return self
class Message(db.Model, CRUDMixin):
__tablename__ = "messages"
id = db.Column(db.Integer, primary_key=True)
conversation_id = db.Column(db.Integer,
db.ForeignKey("conversations.id",
ondelete="CASCADE"),
nullable=False)
# the user who wrote the message
user_id = db.Column(db.Integer,
db.ForeignKey("users.id", ondelete="SET NULL"),
nullable=True)
message = db.Column(db.Text, nullable=False)
date_created = db.Column(UTCDateTime(timezone=True), default=time_utcnow,
nullable=False)
user = db.relationship("User", lazy="joined")
def save(self, conversation=None):
"""Saves a private message.
:param conversation: The conversation to which the message
belongs to.
"""
if conversation is not None:
self.conversation = conversation
conversation.date_modified = time_utcnow()
self.date_created = time_utcnow()
db.session.add(self)
db.session.commit()
return self