4713a00016
* Use websocket events to sync usernames/avatars instead of periodic syncing. * Use caches for fetched room_ids, room state (for usernames and avatars). Also switch to using a single cache with locks. * Don't store full message objects in cache, just store the relation of matrix event IDs to discord message IDs and vice-versa. Content can be fetched from the server instead. * Don't rely on websocket events for mentioning Discord users, mentions are now done by mentioning the dummy matrix user. The ID to be mentioned is extracted from the MXID instead. * General clean-ups.
85 lines
1.8 KiB
Python
85 lines
1.8 KiB
Python
import json
|
|
from dataclasses import fields
|
|
from typing import Any
|
|
|
|
import urllib3
|
|
|
|
from errors import RequestError
|
|
|
|
|
|
def dict_cls(d: dict, cls: Any) -> Any:
|
|
"""
|
|
Create a dataclass from a dictionary.
|
|
"""
|
|
|
|
field_names = set(f.name for f in fields(cls))
|
|
filtered_dict = {k: v for k, v in d.items() if k in field_names}
|
|
|
|
return cls(**filtered_dict)
|
|
|
|
|
|
def log_except(fn):
|
|
"""
|
|
Log unhandled exceptions to a logger instead of `stderr`.
|
|
"""
|
|
|
|
def wrapper(self, *args, **kwargs):
|
|
try:
|
|
return fn(self, *args, **kwargs)
|
|
except Exception:
|
|
self.logger.exception(f"Exception in '{fn.__name__}':")
|
|
raise
|
|
|
|
return wrapper
|
|
|
|
|
|
def request(fn):
|
|
"""
|
|
Either return json data or raise a `RequestError` if the request was
|
|
unsuccessful.
|
|
"""
|
|
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
resp = fn(*args, **kwargs)
|
|
except urllib3.exceptions.HTTPError as e:
|
|
raise RequestError(None, f"Failed to connect: {e}") from None
|
|
|
|
if resp.status < 200 or resp.status >= 300:
|
|
raise RequestError(
|
|
resp.status,
|
|
f"Failed to get response from '{resp.geturl()}':\n{resp.data}",
|
|
)
|
|
|
|
return {} if resp.status == 204 else json.loads(resp.data)
|
|
|
|
return wrapper
|
|
|
|
|
|
def except_deleted(fn):
|
|
"""
|
|
Ignore the `RequestError` on 404s, the content might have been removed.
|
|
"""
|
|
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
return fn(*args, **kwargs)
|
|
except RequestError as e:
|
|
if e.status != 404:
|
|
raise
|
|
|
|
return wrapper
|
|
|
|
|
|
def hash_str(string: str) -> int:
|
|
"""
|
|
Create the hash for a string (poorly).
|
|
"""
|
|
|
|
hashed = 0
|
|
results = map(ord, string)
|
|
|
|
for result in results:
|
|
hashed += result
|
|
|
|
return hashed
|