zedfrigg/bookwyrm
1
0
Fork 0
Code Activity

black formatting

Browse source
This commit is contained in:
Hugh Rundle 2023-01-20 16:31:27 +11:00
parent 0c614e828f
commit 0da5473b0c
3 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
bookwyrm/signatures.py
View file

@ -28,7 +28,7 @@ def make_signature(method, sender, destination, date, digest=None):
signature_headers = [ signature_headers = [
f"(request-target): {method} {inbox_parts.path}", f"(request-target): {method} {inbox_parts.path}",
f"host: {inbox_parts.netloc}", f"host: {inbox_parts.netloc}",
f"date: {date}" f"date: {date}",
] ]
headers = "(request-target) host date" headers = "(request-target) host date"
if digest is not None: if digest is not None:
@ -46,6 +46,7 @@ def make_signature(method, sender, destination, date, digest=None):
} }
return ",".join(f'{k}="{v}"' for (k, v) in signature.items()) return ",".join(f'{k}="{v}"' for (k, v) in signature.items())
def make_digest(data): def make_digest(data):
"""creates a message digest for signing""" """creates a message digest for signing"""
return "SHA-256=" + b64encode(hashlib.sha256(data.encode("utf-8")).digest()).decode( return "SHA-256=" + b64encode(hashlib.sha256(data.encode("utf-8")).digest()).decode(

2
bookwyrm/tests/activitypub/test_base_activity.py
View file

@ -14,7 +14,7 @@ from bookwyrm.activitypub.base_activity import (
ActivityObject, ActivityObject,
resolve_remote_id, resolve_remote_id,
set_related_field, set_related_field,
get_representative get_representative,
) )
from bookwyrm.activitypub import ActivitySerializerError from bookwyrm.activitypub import ActivitySerializerError
from bookwyrm import models from bookwyrm import models

4
bookwyrm/tests/test_signing.py
View file

@ -85,7 +85,9 @@ class Signature(TestCase):
now = date or http_date() now = date or http_date()
data = json.dumps(get_follow_activity(sender, self.rat)) data = json.dumps(get_follow_activity(sender, self.rat))
digest = digest or make_digest(data) digest = digest or make_digest(data)
signature = make_signature("post", signer or sender, self.rat.inbox, now, digest) signature = make_signature(
"post", signer or sender, self.rat.inbox, now, digest
)
with patch("bookwyrm.views.inbox.activity_task.delay"): with patch("bookwyrm.views.inbox.activity_task.delay"):
with patch("bookwyrm.models.user.set_remote_server.delay"): with patch("bookwyrm.models.user.set_remote_server.delay"):
return self.send(signature, now, send_data or data, digest) return self.send(signature, now, send_data or data, digest)