irctokens
/
ircrobots
mirror of https://github.com/jesopo/ircrobots
4
0
Fork 0
Code Issues Releases Wiki Activity

add `transport` (ITCPTransport) param to bot.add_server

This commit is contained in:
jesopo 2021-12-18 16:48:01 +00:00
parent 025fde97ee
commit 9ba5b2b90f
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ircrobots/bot.py
View File

@ -6,7 +6,7 @@ from ircstates.server import ServerDisconnectedException
from .server import ConnectionParams, Server
from .transport import TCPTransport
from .interface import IBot, IServer
from .interface import IBot, IServer, ITCPTransport
class Bot(IBot):
def __init__(self):
@ -38,10 +38,13 @@ class Bot(IBot):
del self.servers[server.name]
await server.disconnect()
async def add_server(self, name: str, params: ConnectionParams) -> Server:
async def add_server(self,
name: str,
params: ConnectionParams,
transport: ITCPTransport = TCPTransport()) -> Server:
server = self.create_server(name)
self.servers[name] = server
await server.connect(TCPTransport(), params)
await server.connect(transport, params)
await self._server_queue.put(server)
return server