import asyncio
from pywebio import start_server
from pywebio.input import input, input_group, actions
from pywebio.output import put_markdown, put_scrollable, output, put_buttons, toast
from pywebio.session import run_async
chat_msgs = []
online_users = set()
MAX_MESSAGES_COUNT = 100
async def main():
global chat_msgs
put_markdown("## Добро пожаловать в онлайн чат!\nПриветствую всех либеральщиков, остальным сболезную")
msg_box = output()
put_scrollable(msg_box, height=300, keep_bottom=True)
nickname = await input("Войти в чат", required=True,
placeholder="Имя либерала-центриста",
validate=lambda n: "Такой либерал уже существует в чате!" if n in online_users or n == '' else None)
online_users.add(nickname)
chat_msgs.append(('', f'Новый либерал-центрист {nickname} присоединился к чату!'))
msg_box.append(put_markdown(f' Новый либерал-центрист {nickname} присоединился к чату'))
refresh_task = run_async(refresh_msg(nickname, msg_box))
while True:
data = await input_group(" Новое сообщение либерала", [
input(placeholder="Текст сообщения сообщения либерала...", name="msg"),
actions(name="cmd", buttons=["Отправить", {'label': "Выйти из чата либералов (карается расстрелом)", 'type': 'cancel'}])
],
validate=lambda m: ('msg', "Введите текст сообщения, далбаебина безграмотная!") if m["cmd"] == "Отправить" and not m['msg'] else None)
if data is None:
break
msg_box.append(put_markdown(f"{nickname}: {data['msg']}"))
chat_msgs.append((nickname, data['msg']))
refresh_task.close()
online_users.remove(nickname)
toast("Вы вышли из чата! За вами выехали)))")
msg_box.append(put_markdown(f' Пользователь {nickname} покинул чат! За ним уже выехали)'))
chat_msgs.append(('', f'Пользователь {nickname} покинул чат!'))
put_buttons(['Перезайти (расстрел отменят)'], onclick=lambda btn: run_js('window.location.reload()'))
async def refresh_msg(nickname, msg_box):
global chat_msgs
last_idx = len(chat_msgs)
while True:
await asyncio.sleep(1)
for m in chat_msgs[last_idx:]:
if m[0] != nickname: # if not a message from current user
msg_box.append(put_markdown(f"{m[0]}: {m[1]}"))
# remove expired messages
if len(chat_msgs) > MAX_MESSAGES_COUNT:
chat_msgs = chat_msgs[len(chat_msgs) // 2:]
last_idx = len(chat_msgs)
if __name__ == "__main__":
start_server(main, debug=True, port=8080, cdn=False)
