mirror of
https://github.com/partitioncloud/partitioncloud-server.git
synced 2025-01-24 01:36:25 +01:00
156 lines
4.3 KiB
Python
156 lines
4.3 KiB
Python
from flask import current_app
|
|
|
|
from ..db import get_db
|
|
from .album import Album
|
|
|
|
|
|
# Variables defined in the CSS
|
|
colors = [
|
|
"--color-rosewater",
|
|
"--color-flamingo",
|
|
"--color-pink",
|
|
"--color-mauve",
|
|
"--color-red",
|
|
"--color-maroon",
|
|
"--color-peach",
|
|
"--color-yellow",
|
|
"--color-green",
|
|
"--color-teal",
|
|
"--color-sky",
|
|
"--color-sapphire",
|
|
"--color-blue",
|
|
"--color-lavender"
|
|
]
|
|
|
|
|
|
class User():
|
|
def __init__(self, user_id=None, name=None):
|
|
self.id = user_id
|
|
self.username = name
|
|
self.albums = None
|
|
self.partitions = None
|
|
self.max_queries = 0
|
|
|
|
db = get_db()
|
|
if self.id is None and self.username is None:
|
|
self.username = ""
|
|
self.access_level = -1
|
|
|
|
else:
|
|
if self.id is not None:
|
|
data = db.execute(
|
|
"""
|
|
SELECT * FROM user
|
|
WHERE id = ?
|
|
""",
|
|
(self.id,)
|
|
).fetchone()
|
|
elif self.username is not None:
|
|
data = db.execute(
|
|
"""
|
|
SELECT * FROM user
|
|
WHERE username = ?
|
|
""",
|
|
(self.username,)
|
|
).fetchone()
|
|
|
|
self.id = data["id"]
|
|
self.username = data["username"]
|
|
self.access_level = data["access_level"]
|
|
self.color = self.get_color()
|
|
if self.access_level == 1:
|
|
self.max_queries = 10
|
|
else:
|
|
self.max_queries = current_app.config["MAX_ONLINE_QUERIES"]
|
|
|
|
|
|
def is_participant(self, album_uuid):
|
|
db = get_db()
|
|
|
|
return len(db.execute(
|
|
"""
|
|
SELECT album.id FROM album
|
|
JOIN contient_user ON album_id = album.id
|
|
JOIN user ON user_id = user.id
|
|
WHERE user.id = ? AND album.uuid = ?
|
|
""",
|
|
(self.id, album_uuid)
|
|
).fetchall()) == 1
|
|
|
|
|
|
def get_albums(self, force_reload=False):
|
|
if self.albums is None or force_reload:
|
|
db = get_db()
|
|
if self.access_level == 1:
|
|
self.albums = db.execute(
|
|
"""
|
|
SELECT * FROM album
|
|
"""
|
|
).fetchall()
|
|
else:
|
|
self.albums = db.execute(
|
|
"""
|
|
SELECT album.id, name, uuid FROM album
|
|
JOIN contient_user ON album_id = album.id
|
|
JOIN user ON user_id = user.id
|
|
WHERE user.id = ?
|
|
""",
|
|
(self.id,),
|
|
).fetchall()
|
|
return self.albums
|
|
|
|
|
|
def get_partitions(self, force_reload=False):
|
|
if self.partitions is None or force_reload:
|
|
db = get_db()
|
|
if self.access_level == 1:
|
|
self.partitions = db.execute(
|
|
"""
|
|
SELECT * FROM partition
|
|
"""
|
|
).fetchall()
|
|
else:
|
|
self.partitions = db.execute(
|
|
"""
|
|
SELECT * FROM partition
|
|
JOIN user ON user_id = user.id
|
|
WHERE user.id = ?
|
|
""",
|
|
(self.id,),
|
|
).fetchall()
|
|
return self.partitions
|
|
|
|
def join_album(self, album_uuid):
|
|
db = get_db()
|
|
album = Album(uuid=album_uuid)
|
|
|
|
db.execute(
|
|
"""
|
|
INSERT INTO contient_user (user_id, album_id)
|
|
VALUES (?, ?)
|
|
""",
|
|
(self.id, album.id)
|
|
)
|
|
db.commit()
|
|
|
|
def quit_album(self, album_uuid):
|
|
db = get_db()
|
|
album = Album(uuid=album_uuid)
|
|
|
|
db.execute(
|
|
"""
|
|
DELETE FROM contient_user
|
|
WHERE user_id = ?
|
|
AND album_id = ?
|
|
""",
|
|
(self.id, album.id)
|
|
)
|
|
db.commit()
|
|
|
|
|
|
def get_color(self):
|
|
if len(colors) == 0:
|
|
integer = hash(self.username) % 16777215
|
|
return "#" + str(hex(integer))[2:]
|
|
else:
|
|
return f"var({colors[hash(self.username) %len(colors)]})" |