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 .server import ConnectionParams, Server
from .transport import TCPTransport from .transport import TCPTransport
from .interface import IBot, IServer from .interface import IBot, IServer, ITCPTransport
class Bot(IBot): class Bot(IBot):
def __init__(self): def __init__(self):
@ -38,10 +38,13 @@ class Bot(IBot):
del self.servers[server.name] del self.servers[server.name]
await server.disconnect() 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) server = self.create_server(name)
self.servers[name] = server self.servers[name] = server
await server.connect(TCPTransport(), params) await server.connect(transport, params)
await self._server_queue.put(server) await self._server_queue.put(server)
return server return server