MiniFaceBook/models.py at main · Michatec/MiniFaceBook · GitHub

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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin
from datetime import datetime
from hashlib import md5

db = SQLAlchemy()

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(150), unique=True, nullable=False)
    email = db.Column(db.String(150), unique=True, nullable=False)
    password = db.Column(db.String(150), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)
    is_admin = db.Column(db.Boolean, default=False)
    is_owner = db.Column(db.Boolean, default=False)
    profile_pic = db.Column(db.String(200), default='default.png')
    discord_id = db.Column(db.String(50), unique=True)
    discord_linked = db.Column(db.Boolean, default=False)
    posts = db.relationship('Post', backref='user', cascade="all, delete", passive_deletes=True)
    comments = db.relationship('Comment', backref='user', cascade="all, delete", passive_deletes=True)
    likes = db.relationship('Like', backref='user', cascade="all, delete", passive_deletes=True)
    friendships_sent = db.relationship(
        'Friendship',
        foreign_keys='Friendship.requester_id',
        backref='requester',
        cascade="all, delete",
        passive_deletes=True
    )
    friendships_received = db.relationship(
        'Friendship',
        foreign_keys='Friendship.receiver_id',
        backref='receiver',
        cascade="all, delete",
        passive_deletes=True
    )
    uploads = db.relationship('Upload', backref='user', cascade="all, delete", passive_deletes=True)
    rewards = db.relationship('Reward', backref='user', cascade="all, delete", passive_deletes=True)
    shop_items = db.relationship('UserShopItem', backref='user', cascade="all, delete", passive_deletes=True)

    def reward_points(self):
        return sum(r.points for r in self.rewards)

    def avatar(self):
        digest = md5(self.email.lower().encode('utf-8')).hexdigest()
        return f'https://www.gravatar.com/avatar/{digest}?d=identicon&s=120'

class Friendship(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    requester_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    receiver_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    status = db.Column(db.String(20), default='pending')

class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    content = db.Column(db.Text, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)
    visibility = db.Column(db.String(20), default='public')
    likes = db.relationship('Like', backref='post', lazy='dynamic', cascade="all, delete-orphan", passive_deletes=True)
    comments = db.relationship('Comment', backref='post', cascade="all, delete-orphan", passive_deletes=True)
    uploads = db.relationship('Upload', backref='post', cascade="all, delete-orphan", passive_deletes=True)

class Comment(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    content = db.Column(db.Text, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)

class Like(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    post_id = db.Column(db.Integer, db.ForeignKey('post.id', ondelete='CASCADE'))
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))

class PasswordResetRequest(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    requested_at = db.Column(db.DateTime, default=datetime.now)
    status = db.Column(db.String(20), default='pending')
    user = db.relationship('User', backref='reset_requests')

class Upload(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
    filename = db.Column(db.String(255))
    filetype = db.Column(db.String(50))
    uploaded_at = db.Column(db.DateTime, default=datetime.now)

class Event(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    timestamp = db.Column(db.DateTime, default=datetime.now)
    message = db.Column(db.String(255))

class Notification(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    message = db.Column(db.String(255))
    created_at = db.Column(db.DateTime, default=datetime.now)
    read = db.Column(db.Boolean, default=False)

class Reward(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    type = db.Column(db.String(50))
    points = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.now)

class ShopItem(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), nullable=False)
    description = db.Column(db.String(255))
    price = db.Column(db.Integer, nullable=False)
    icon = db.Column(db.String(50), default="bi-gift")

class UserShopItem(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    item_id = db.Column(db.Integer, db.ForeignKey('shop_item.id'))
    bought_at = db.Column(db.DateTime, default=datetime.now)
    item = db.relationship('ShopItem')

    __table_args__ = (db.UniqueConstraint('user_id', 'item_id', name='unique_user_item_purchase'),)

class SupportRequest(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    title = db.Column(db.String(100), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)
    status = db.Column(db.String(20), default='open')
    comments = db.relationship('SupportComment', backref='request', cascade="all, delete", passive_deletes=True)

class SupportComment(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    request_id = db.Column(db.Integer, db.ForeignKey('support_request.id'))
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    message = db.Column(db.Text, nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.now)
    user = db.relationship('User')

SHOPITEM_ID_GOLDRAHMEN = 2
SHOPITEM_ID_PREMIUM = 1
SHOPITEM_ID_EXTRA_UPLOAD = 3
SHOPITEM_ID_EXTRA_TYPES = 4