relay: only import urlsplit from urllib.parse

This commit is contained in:
kaniini 2018-11-20 00:41:52 +00:00
parent 4623aaf5b6
commit b607e382cb

View file

@ -4,9 +4,9 @@ import asyncio
import logging import logging
import uuid import uuid
import re import re
import urllib.parse
import simplejson as json import simplejson as json
import cgi import cgi
from urllib.parse import urlsplit
from Crypto.PublicKey import RSA from Crypto.PublicKey import RSA
from .database import DATABASE from .database import DATABASE
from .http_debug import http_debug from .http_debug import http_debug
@ -78,7 +78,7 @@ get_actor_inbox = lambda actor: actor.get('endpoints', {}).get('sharedInbox', ac
async def push_message_to_actor(actor, message, our_key_id): async def push_message_to_actor(actor, message, our_key_id):
inbox = get_actor_inbox(actor) inbox = get_actor_inbox(actor)
url = urllib.parse.urlsplit(inbox) url = urlsplit(inbox)
# XXX: Digest # XXX: Digest
data = json.dumps(message) data = json.dumps(message)
@ -208,7 +208,7 @@ async def handle_follow(actor, data, request):
following = DATABASE.get('relay-list', []) following = DATABASE.get('relay-list', [])
inbox = get_actor_inbox(actor) inbox = get_actor_inbox(actor)
if urllib.parse.urlsplit(inbox).hostname in AP_CONFIG['blocked_instances']: if urlsplit(inbox).hostname in AP_CONFIG['blocked_instances']:
return return
if inbox not in following: if inbox not in following: