Random Coffee alternative - random meetings for Telegram chats
https://t.me/ranteabot
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
2.1 KiB
58 lines
2.1 KiB
2 years ago
|
from __future__ import annotations
|
||
|
|
from dataclasses import dataclass, field, asdict
|
||
|
|
import typing
|
||
|
|
from datetime import date
|
||
|
|
from telebot.types import ChatMember, Chat
|
||
|
|
import time
|
||
|
|
from threading import Thread
|
||
|
|
|
||
|
|
if typing.TYPE_CHECKING:
|
||
|
|
from bot import Bot
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class Community:
|
||
|
|
chat_id: int # Telegram chat id
|
||
|
|
name: str = ''
|
||
|
|
members: typing.List[int] = field(default_factory=list)
|
||
|
|
pool: typing.List[typing.Tuple[int, int]] = field(default_factory=list) # (user_id, number_of_copies)
|
||
|
|
scheduled_meetings: typing.List[typing.Tuple[int, int, date]] = field(
|
||
|
|
default_factory=list) # (user_id_1, user_id_2, meeting date)
|
||
|
|
archived_meetings: typing.List[typing.Tuple[int, int, date]] = field(default_factory=list)
|
||
|
|
start_timestamp: int = field(default_factory=lambda: int(time.time()))
|
||
|
|
|
||
|
|
def dict(self) -> dict:
|
||
|
|
data = asdict(self)
|
||
|
|
return data
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, data: dict) -> Community:
|
||
|
|
data = getattr(data, '__dict__', data)
|
||
|
|
data_ = {key: data[key] for key in data.keys() if
|
||
|
|
key in ['chat_id', 'name', 'members', 'pool', 'scheduled_meetings', 'archived_meetings',
|
||
|
|
'start_timestamp']}
|
||
|
|
self = cls(**data_)
|
||
|
|
return self
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def by_id(cls, chat_id: int, bot: Bot, create_if_not_exists=True) -> typing.Optional[Community]:
|
||
|
|
data = bot.get_community_from_db({'chat_id': chat_id}, create_if_not_exists)
|
||
|
|
if data is None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def upd_data():
|
||
|
|
chat: Chat = bot.get_chat(chat_id)
|
||
|
|
data.name = chat.title
|
||
|
|
bot.save_community(data)
|
||
|
|
|
||
|
|
Thread(target=upd_data).start()
|
||
|
|
return data
|
||
|
|
|
||
|
|
def add_member(self, user_id: int, bot: Bot) -> bool:
|
||
|
|
chat_member_info: ChatMember = bot.get_chat_member(self.chat_id, user_id)
|
||
|
|
if chat_member_info.is_member or chat_member_info.status in ['creator', 'administrator', 'admin']:
|
||
|
|
self.members.append(user_id)
|
||
|
|
bot.save_community(self)
|
||
|
|
return True
|
||
|
|
return False
|