mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2025-12-14 08:06:23 +08:00
dc1ca344e5
- For Arcaea 5.10.4 - Add support for the Link Play notification setting. - Fix a bug that public rooms cannot be matched possibly if some players leave the room. - Continue to fix the bug that the player cannot complete one course twice. - Continue to fix the bug that users cannot change online settings.
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
from .config_manager import Config
|
|
from .user import User
|
|
from .sql import Connect
|
|
|
|
from time import time
|
|
|
|
|
|
class BaseNotification:
|
|
|
|
notification_type = None
|
|
|
|
def __init__(self, c_m=None) -> None:
|
|
self.receiver = None
|
|
self.sender = None
|
|
self.timestamp = None
|
|
self.content = None
|
|
|
|
self.c_m = c_m
|
|
|
|
@property
|
|
def is_expired(self) -> bool:
|
|
now = round(time() * 1000)
|
|
return now - self.timestamp > Config.NOTIFICATION_EXPIRE_TIME
|
|
|
|
def to_dict(self) -> dict:
|
|
raise NotImplementedError()
|
|
|
|
def insert(self):
|
|
self.receiver.select_user_one_column('mp_notification_enabled', True, bool)
|
|
if not self.receiver.mp_notification_enabled:
|
|
return
|
|
|
|
self.c_m.execute(
|
|
'''select max(id) from notification where user_id = ?''', (self.receiver.user_id,))
|
|
x = self.c_m.fetchone()
|
|
if x is None or x[0] is None:
|
|
x = 0
|
|
else:
|
|
x = x[0] + 1
|
|
|
|
self.c_m.execute(
|
|
'''insert into notification values (?, ?, ?, ?, ?, ?, ?)''',
|
|
(self.receiver.user_id, x, self.notification_type, self.content,
|
|
self.sender.user_id, self.sender.name, self.timestamp)
|
|
)
|
|
|
|
|
|
class RoomInviteNotification(BaseNotification):
|
|
|
|
notification_type = 'room_inv'
|
|
|
|
@classmethod
|
|
def from_list(cls, l: list, user: User = None) -> 'RoomInviteNotification':
|
|
x = cls()
|
|
x.sender = User()
|
|
x.sender.user_id = l[2]
|
|
x.sender.name = l[3]
|
|
x.content = l[1]
|
|
x.timestamp = l[4]
|
|
x.receiver = user
|
|
return x
|
|
|
|
@classmethod
|
|
def from_sender(cls, sender: User, receiver: User, share_token: str, c_m) -> 'RoomInviteNotification':
|
|
x = cls()
|
|
x.c_m = c_m
|
|
x.sender = sender
|
|
x.receiver = receiver
|
|
x.content = share_token
|
|
x.timestamp = round(time() * 1000)
|
|
return x
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
'sender': self.sender.name,
|
|
'type': self.notification_type,
|
|
'shareToken': self.content,
|
|
'sendTs': self.timestamp
|
|
}
|
|
|
|
|
|
class NotificationFactory:
|
|
def __init__(self, c_m: Connect, user=None):
|
|
self.c_m = c_m
|
|
self.user = user
|
|
|
|
def get_notification(self) -> 'list[BaseNotification]':
|
|
r = []
|
|
|
|
self.c_m.execute('''select type, content, sender_user_id, sender_name, timestamp from notification where user_id = ?''',
|
|
(self.user.user_id,))
|
|
for i in self.c_m.fetchall():
|
|
x = None
|
|
if i[0] == 'room_inv':
|
|
x = RoomInviteNotification.from_list(i, self.user)
|
|
|
|
if x is not None and not x.is_expired:
|
|
r.append(x)
|
|
self.c_m.execute(
|
|
'''delete from notification where user_id = ?''', (self.user.user_id,))
|
|
return r
|