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
|
import sqlalchemy as sa
from szurubooru.model.base import Base
from szurubooru.model.comment import Comment
from szurubooru.model.post import Post, PostFavorite, PostScore
class User(Base):
__tablename__ = "user"
AVATAR_GRAVATAR = "gravatar"
AVATAR_MANUAL = "manual"
RANK_ANONYMOUS = "anonymous"
RANK_RESTRICTED = "restricted"
RANK_REGULAR = "regular"
RANK_POWER = "power"
RANK_MODERATOR = "moderator"
RANK_ADMINISTRATOR = "administrator"
RANK_NOBODY = "nobody" # unattainable, used for privileges
user_id = sa.Column("id", sa.Integer, primary_key=True)
creation_time = sa.Column("creation_time", sa.DateTime, nullable=False)
last_login_time = sa.Column("last_login_time", sa.DateTime)
version = sa.Column("version", sa.Integer, default=1, nullable=False)
name = sa.Column("name", sa.Unicode(50), nullable=False, unique=True)
password_hash = sa.Column("password_hash", sa.Unicode(128), nullable=False)
password_salt = sa.Column("password_salt", sa.Unicode(32))
password_revision = sa.Column(
"password_revision", sa.SmallInteger, default=0, nullable=False
)
email = sa.Column("email", sa.Unicode(64), nullable=True)
rank = sa.Column("rank", sa.Unicode(32), nullable=False)
avatar_style = sa.Column(
"avatar_style", sa.Unicode(32), nullable=False, default=AVATAR_GRAVATAR
)
comments = sa.orm.relationship("Comment")
@property
def post_count(self) -> int:
from szurubooru.db import session
return (
session.query(sa.sql.expression.func.sum(1))
.filter(Post.user_id == self.user_id)
.one()[0]
or 0
)
@property
def comment_count(self) -> int:
from szurubooru.db import session
return (
session.query(sa.sql.expression.func.sum(1))
.filter(Comment.user_id == self.user_id)
.one()[0]
or 0
)
@property
def favorite_post_count(self) -> int:
from szurubooru.db import session
return (
session.query(sa.sql.expression.func.sum(1))
.filter(PostFavorite.user_id == self.user_id)
.one()[0]
or 0
)
@property
def liked_post_count(self) -> int:
from szurubooru.db import session
return (
session.query(sa.sql.expression.func.sum(1))
.filter(PostScore.user_id == self.user_id)
.filter(PostScore.score == 1)
.one()[0]
or 0
)
@property
def disliked_post_count(self) -> int:
from szurubooru.db import session
return (
session.query(sa.sql.expression.func.sum(1))
.filter(PostScore.user_id == self.user_id)
.filter(PostScore.score == -1)
.one()[0]
or 0
)
__mapper_args__ = {
"version_id_col": version,
"version_id_generator": False,
}
class UserToken(Base):
__tablename__ = "user_token"
user_token_id = sa.Column("id", sa.Integer, primary_key=True)
user_id = sa.Column(
"user_id",
sa.Integer,
sa.ForeignKey("user.id", ondelete="CASCADE"),
nullable=False,
index=True,
)
token = sa.Column("token", sa.Unicode(36), nullable=False)
note = sa.Column("note", sa.Unicode(128), nullable=True)
enabled = sa.Column("enabled", sa.Boolean, nullable=False, default=True)
expiration_time = sa.Column("expiration_time", sa.DateTime, nullable=True)
creation_time = sa.Column("creation_time", sa.DateTime, nullable=False)
last_edit_time = sa.Column("last_edit_time", sa.DateTime)
last_usage_time = sa.Column("last_usage_time", sa.DateTime)
version = sa.Column("version", sa.Integer, default=1, nullable=False)
user = sa.orm.relationship("User")
|
