GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-15 12:50:47 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix type hints for conection manager (#1076)

Browse source
This commit is contained in:
Jonathan Zernik 2021-08-27 06:36:19 -07:00 • committed by GitHub
parent 87e5b0b079
commit 1a4b908f00
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
squeaknode/network/connection_manager.py
View file

@ -1,6 +1,8 @@
import logging
import threading
from typing import Dict
from typing import List
from typing import Optional
from squeaknode.core.peer_address import PeerAddress
from squeaknode.network.peer import Peer
@ -24,7 +26,7 @@ class ConnectionManager(object):
self.peers_changed_callbacks = {}
@property
def peers(self):
def peers(self) -> List[Peer]:
return list(self._peers.values())
def has_connection(self, address):
@ -76,7 +78,7 @@ class ConnectionManager(object):
logger.debug('Removed peer {}'.format(peer))
self.on_peers_changed()
def get_peer(self, address):
def get_peer(self, address) -> Optional[Peer]:
"""Get a peer info by address.
"""
return self._peers.get(address)

5
squeaknode/network/network_manager.py
View file

@ -1,6 +1,7 @@
import logging
import socket
from typing import List
from typing import Optional
import squeak.params
from squeak.messages import MsgSerializable
@ -66,10 +67,10 @@ class NetworkManager(object):
def disconnect_peer(self, peer_address: PeerAddress) -> None:
self.connection_manager.stop_connection(peer_address)
def get_connected_peer(self, peer_address: PeerAddress) -> List[Peer]:
def get_connected_peer(self, peer_address: PeerAddress) -> Optional[Peer]:
return self.connection_manager.get_peer(peer_address)
def get_connected_peers(self):
def get_connected_peers(self) -> List[Peer]:
return self.connection_manager.peers
def broadcast_msg(self, msg: MsgSerializable) -> None: