hng_boilerplate_python_fastapi_web/api/v1/models/user.py at 577cdfaebff3f13369e14c78c26209b54da7e2f7 · hngprojects/hng_boilerplate_python_fastapi_web · 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
""" User data model
"""

from sqlalchemy import Column, String, text, Boolean
from sqlalchemy.orm import relationship
from api.v1.models.associations import user_organisation_association
from api.v1.models.permissions.user_org_role import user_organisation_roles
from api.v1.models.base_model import BaseTableModel


class User(BaseTableModel):
    __tablename__ = "users"

    email = Column(String, unique=True, nullable=False)
    password = Column(String, nullable=True)
    first_name = Column(String, nullable=True)
    last_name = Column(String, nullable=True)
    avatar_url = Column(String, nullable=True)
    is_active = Column(Boolean, server_default=text("true"))
    is_superadmin = Column(Boolean, server_default=text("false"))
    is_deleted = Column(Boolean, server_default=text("false"))
    is_verified = Column(Boolean, server_default=text("false"))
    username = Column(String, nullable=True)

    profile = relationship(
        "Profile", uselist=False, back_populates="user", cascade="all, delete-orphan"
    )
    organisations = relationship(
        "Organisation", secondary=user_organisation_roles, back_populates="users"
    )
    notifications = relationship(
        "Notification", back_populates="user", cascade="all, delete-orphan"
    )
    activity_logs = relationship(
        "ActivityLog", back_populates="user", cascade="all, delete-orphan"
    )
    jobs = relationship("Job", back_populates="author", cascade="all, delete-orphan")
    token_login = relationship(
        "TokenLogin", back_populates="user", uselist=False, cascade="all, delete-orphan"
    )
    oauth = relationship(
        "OAuth", back_populates="user", uselist=False, cascade="all, delete-orphan"
    )
    testimonials = relationship(
        "Testimonial", back_populates="author", cascade="all, delete-orphan"
    )
    payments = relationship(
        "Payment", back_populates="user", cascade="all, delete-orphan"
    )
    blogs = relationship("Blog", back_populates="author", cascade="all, delete-orphan")
    comments = relationship(
        "Comment", back_populates="user", cascade="all, delete-orphan"
    )
    invitations = relationship(
        "Invitation", back_populates="user", cascade="all, delete-orphan"
    )
    messages = relationship(
        "Message", back_populates="user", cascade="all, delete-orphan"
    )
    blog_likes = relationship(
        "BlogLike", back_populates="user", cascade="all, delete-orphan"
    )
    blog_dislikes = relationship(
        "BlogDislike", back_populates="user", cascade="all, delete-orphan"
    )
    comment_likes = relationship(
        "CommentLike", back_populates="user", cascade="all, delete-orphan"
    )
    comment_dislikes = relationship(
        "CommentDislike", back_populates="user", cascade="all, delete-orphan"
    )
    notification_setting = relationship(
        "NotificationSetting",
        back_populates="user",
        cascade="all, delete-orphan",
        uselist=False,
    )
    region = relationship("Region", back_populates="user", cascade="all, delete-orphan")
    squeeze = relationship(
        "Squeeze", back_populates="user", cascade="all, delete-orphan"
    )
    data_privacy_setting = relationship(
        "DataPrivacySetting",
        uselist=False,
        back_populates="user",
        cascade="all, delete-orphan",
    )
    product_comments = relationship("ProductComment", back_populates="user", cascade="all, delete-orphan")

    subscriptions = relationship(
        "UserSubscription", back_populates="user", cascade="all, delete-orphan"
    )

    reset_password_token = relationship("ResetPasswordToken",
                                        back_populates="user",
                                        cascade="all, delete-orphan")

    def to_dict(self):
        obj_dict = super().to_dict()
        obj_dict.pop("password")
        return obj_dict

    def __str__(self):
        return self.email