91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
|
from abc import ABCMeta
|
||
|
|
|
||
|
|
|
||
|
|
class UserService(object):
|
||
|
|
|
||
|
|
__metaclass__ = Singleton
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.users_by_id = {} # key: user id, value: User
|
||
|
|
|
||
|
|
def add_user(self, user_id, name, pass_hash): # ...
|
||
|
|
def remove_user(self, user_id): # ...
|
||
|
|
def add_friend_request(self, from_user_id, to_user_id): # ...
|
||
|
|
def approve_friend_request(self, from_user_id, to_user_id): # ...
|
||
|
|
def reject_friend_request(self, from_user_id, to_user_id): # ...
|
||
|
|
|
||
|
|
|
||
|
|
class User(object):
|
||
|
|
|
||
|
|
def __init__(self, user_id, name, pass_hash):
|
||
|
|
self.user_id = user_id
|
||
|
|
self.name = name
|
||
|
|
self.pass_hash = pass_hash
|
||
|
|
self.friends_by_id = {} # key: friend id, value: User
|
||
|
|
self.friend_ids_to_private_chats = {} # key: friend id, value: private chats
|
||
|
|
self.group_chats_by_id = {} # key: chat id, value: GroupChat
|
||
|
|
self.received_friend_requests_by_friend_id = {} # key: friend id, value: AddRequest
|
||
|
|
self.sent_friend_requests_by_friend_id = {} # key: friend id, value: AddRequest
|
||
|
|
|
||
|
|
def message_user(self, friend_id, message): # ...
|
||
|
|
def message_group(self, group_id, message): # ...
|
||
|
|
def send_friend_request(self, friend_id): # ...
|
||
|
|
def receive_friend_request(self, friend_id): # ...
|
||
|
|
def approve_friend_request(self, friend_id): # ...
|
||
|
|
def reject_friend_request(self, friend_id): # ...
|
||
|
|
|
||
|
|
|
||
|
|
class Chat(metaclass=ABCMeta):
|
||
|
|
|
||
|
|
def __init__(self, chat_id):
|
||
|
|
self.users = []
|
||
|
|
self.chat_id = chat_id
|
||
|
|
self.messages = []
|
||
|
|
|
||
|
|
|
||
|
|
class PrivateChat(Chat):
|
||
|
|
|
||
|
|
def __init__(self, first_user, second_user):
|
||
|
|
super(PrivateChat, self).__init__()
|
||
|
|
self.users.append(first_user)
|
||
|
|
self.users.append(second_user)
|
||
|
|
|
||
|
|
|
||
|
|
class GroupChat(Chat):
|
||
|
|
|
||
|
|
def add_user(self, user): # ...
|
||
|
|
def remove_user(self, user): # ...
|
||
|
|
|
||
|
|
|
||
|
|
class Message(object):
|
||
|
|
|
||
|
|
def __init__(self, message_id, message, timestamp):
|
||
|
|
self.message_id = message_id
|
||
|
|
self.message = message
|
||
|
|
self.timestamp = timestamp
|
||
|
|
|
||
|
|
|
||
|
|
class AddRequest(object):
|
||
|
|
|
||
|
|
def __init__(self, from_user_id, to_user_id, request_status, timestamp):
|
||
|
|
self.from_user_id = from_user_id
|
||
|
|
self.to_user_id = to_user_id
|
||
|
|
self.request_status = request_status
|
||
|
|
self.timestamp = timestamp
|
||
|
|
|
||
|
|
|
||
|
|
class RequestStatus(Enum):
|
||
|
|
|
||
|
|
UNREAD = 0
|
||
|
|
READ = 1
|
||
|
|
ACCEPTED = 2
|
||
|
|
REJECTED = 3
|
||
|
|
|
||
|
|
|
||
|
|
class Singleton(type):
|
||
|
|
|
||
|
|
_instances = {}
|
||
|
|
def __call__(cls, *args, **kwargs):
|
||
|
|
if cls not in cls._instances:
|
||
|
|
cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs)
|
||
|
|
return cls._instances[cls]
|