microblog.pub/core/remote.py
2019-08-09 00:08:22 +02:00

38 lines
872 B
Python

from urllib.parse import urlparse
from core.db import _Q
from core.db import update_one_remote
from utils import now
def server(url: str) -> str:
return urlparse(url).netloc
def _update(url: str, replace: _Q) -> None:
update_one_remote({"server": server(url)}, replace, upsert=True)
# TODO(tsileo): track receive (and the user agent to help debug issues)
def track_successful_send(url: str) -> None:
now_ = now()
_update(
url,
{
"$inc": {"successful_send": 1},
"$set": {
"last_successful_contact": now_,
"last_successful_send": now_,
"last_contact": now_,
},
},
)
return None
def track_failed_send(url: str) -> None:
now_ = now()
_update(url, {"$inc": {"failed_send": 1}, "$set": {"last_contact": now_}})
return None