49 行
1008 B
Python
49 行
1008 B
Python
import base64
|
|
import hashlib
|
|
import secrets
|
|
from collections import OrderedDict
|
|
from enum import Enum
|
|
from math import inf
|
|
|
|
from quart import escape, Markup
|
|
|
|
Presence = Enum(
|
|
'Presence',
|
|
names=(
|
|
'WATCHING',
|
|
'NOTWATCHING',
|
|
'TENTATIVE',
|
|
'ABSENT',
|
|
)
|
|
)
|
|
|
|
def generate_token():
|
|
return secrets.token_hex(16)
|
|
|
|
def concatenate_for_notice(string, *tuples):
|
|
if not tuples:
|
|
return string
|
|
markup = Markup(
|
|
''.join(
|
|
f' <mark>{escape(x)}</mark>{escape(y)}'
|
|
for x, y in tuples
|
|
)
|
|
)
|
|
return string + markup
|
|
|
|
def trilean(presence):
|
|
match presence:
|
|
case Presence.WATCHING:
|
|
return True
|
|
case Presence.NOTWATCHING:
|
|
return False
|
|
case _:
|
|
return None
|
|
|
|
def get_user_for_websocket(user):
|
|
keys = ('broadcaster', 'name', 'color', 'tripcode', 'tag')
|
|
return {
|
|
**{key: user[key] for key in keys},
|
|
'watching': trilean(user['presence']),
|
|
}
|