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.
code_spécifique_watcha
Richard van der Hoff 4 years ago committed by GitHub
parent 12528dc42f
commit a973bcb8a4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 1
      changelog.d/7870.misc
  2. 10
      synapse/handlers/_base.py
  3. 4
      synapse/server.py

@ -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…
Cancel
Save