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.
This commit is contained in:
parent
12528dc42f
commit
a973bcb8a4
|
@ -0,0 +1 @@
|
|||
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):
|
||||
|
|
Loading…
Reference in New Issue