blob: f7f682ca950000922187675ad114cbaddde286e6 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
import asyncio
import logging
from .base import Connection, INITALIZED
from ..utils.decorators import memoize
class AioIRC(Connection):
def __init__(self, host, port):
self._loop = asyncio.get_event_loop()
self.log = logging.getLogger(self.__class__.__name__)
self.status = INITALIZED
self.transport = None
self.protocol = None
self.reader = None
self.writer = None
self.host = host
self.port = port
self.nick = 'warmachin49'
self.user = 'WarMachine'
self.server_info = {
'host': ''
}
async def connect(self):
self.log.info('Connecting to {}:{}'.format(self.host, self.port))
self.reader, self.writer = await asyncio.open_connection(
self.host, self.port)
self.writer.write('NICK {} r\n'.format(self.nick).encode())
self.writer.write('USER {} 8 * :War Machine\r\n'.format(
self.user).encode())
return True
@asyncio.coroutine
def read(self):
if self.reader.at_eof():
raise Exception('eof')
if self.reader:
message = yield from self.reader.readline()
# if not self.server_info['host']:
# self.server_info['host'] = message.split(' ')[0].replace(':', '')
# if message.startswith('PING'):
# yield from self.send_pong()
# return
return message.decode().strip()
async def send_pong(self):
msg = 'PONG :{}'.format(self.server_info['host'])
return self.writer.write(msg)
@property
@memoize
def id(self):
return 'asdfasdf'
|
