Appease pep8

This commit is contained in:
Paul "LeoNerd" Evans 2015-03-05 20:53:33 +00:00
parent 5eab2549ab
commit d79d91a4a7

View File

@ -102,7 +102,8 @@ class TransportLayerServer(object):
Args: Args:
handler (TransportReceivedHandler) handler (TransportReceivedHandler)
""" """
FederationSendServlet(handler, FederationSendServlet(
handler,
authenticator=self, authenticator=self,
ratelimiter=self.ratelimiter, ratelimiter=self.ratelimiter,
server_name=self.server_name, server_name=self.server_name,
@ -115,20 +116,9 @@ class TransportLayerServer(object):
Args: Args:
handler (TransportRequestHandler) handler (TransportRequestHandler)
""" """
for servletclass in ( for servletclass in SERVLET_CLASSES:
FederationPullServlet, servletclass(
FederationEventServlet, handler,
FederationStateServlet,
FederationBackfillServlet,
FederationQueryServlet,
FederationMakeJoinServlet,
FederationEventServlet,
FederationSendJoinServlet,
FederationInviteServlet,
FederationQueryAuthServlet,
FederationGetMissingEventsServlet,
):
servletclass(handler,
authenticator=self, authenticator=self,
ratelimiter=self.ratelimiter, ratelimiter=self.ratelimiter,
).register(self.server) ).register(self.server)
@ -249,7 +239,9 @@ class FederationStateServlet(BaseFederationServlet):
# This is when someone asks for all data for a given context. # This is when someone asks for all data for a given context.
def on_GET(self, origin, content, query, context): def on_GET(self, origin, content, query, context):
return self.handler.on_context_state_request(origin, context, return self.handler.on_context_state_request(
origin,
context,
query.get("event_id", [None])[0], query.get("event_id", [None])[0],
) )
@ -274,7 +266,8 @@ class FederationQueryServlet(BaseFederationServlet):
# This is when we receive a server-server Query # This is when we receive a server-server Query
def on_GET(self, origin, content, query, query_type): def on_GET(self, origin, content, query, query_type):
return self.handler.on_query_request(query_type, return self.handler.on_query_request(
query_type,
{k: v[0].decode("utf-8") for k, v in query.items()} {k: v[0].decode("utf-8") for k, v in query.items()}
) )
@ -350,3 +343,18 @@ class FederationGetMissingEventsServlet(BaseFederationServlet):
) )
defer.returnValue((200, content)) defer.returnValue((200, content))
SERVLET_CLASSES = (
FederationPullServlet,
FederationEventServlet,
FederationStateServlet,
FederationBackfillServlet,
FederationQueryServlet,
FederationMakeJoinServlet,
FederationEventServlet,
FederationSendJoinServlet,
FederationInviteServlet,
FederationQueryAuthServlet,
FederationGetMissingEventsServlet,
)