Skip to content
Snippets Groups Projects
Unverified Commit a973bcb8 authored by Richard van der Hoff's avatar Richard van der Hoff Committed by GitHub
Browse files

Add some tiny type annotations (#7870)

I found these made pycharm have more of a clue as to what was going on in other places.
parent 12528dc4
No related branches found
No related tags found
No related merge requests found
Add some type annotations to `HomeServer` and `BaseHandler`.
......@@ -17,6 +17,8 @@ import logging
from twisted.internet import defer
import synapse.state
import synapse.storage
import synapse.types
from synapse.api.constants import EventTypes, Membership
from synapse.api.ratelimiting import Ratelimiter
......@@ -28,10 +30,6 @@ logger = logging.getLogger(__name__)
class BaseHandler(object):
"""
Common base class for the event handlers.
Attributes:
store (synapse.storage.DataStore):
state_handler (synapse.state.StateHandler):
"""
def __init__(self, hs):
......@@ -39,10 +37,10 @@ class BaseHandler(object):
Args:
hs (synapse.server.HomeServer):
"""
self.store = hs.get_datastore()
self.store = hs.get_datastore() # type: synapse.storage.DataStore
self.auth = hs.get_auth()
self.notifier = hs.get_notifier()
self.state_handler = hs.get_state_handler()
self.state_handler = hs.get_state_handler() # type: synapse.state.StateHandler
self.distributor = hs.get_distributor()
self.clock = hs.get_clock()
self.hs = hs
......
......@@ -106,7 +106,7 @@ from synapse.server_notices.worker_server_notices_sender import (
WorkerServerNoticesSender,
)
from synapse.state import StateHandler, StateResolutionHandler
from synapse.storage import DataStores, Storage
from synapse.storage import DataStore, DataStores, Storage
from synapse.streams.events import EventSources
from synapse.util import Clock
from synapse.util.distributor import Distributor
......@@ -312,7 +312,7 @@ class HomeServer(object):
def get_clock(self):
return self.clock
def get_datastore(self):
def get_datastore(self) -> DataStore:
return self.datastores.main
def get_datastores(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment