blob: 433cc75141a64f0a749b459873c122fb529c0ad5 (
plain)
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
|
#
# Describe a connection over a socket.
#
# This class is intended to be apdapted into IEndpoint interface
import socket
from zope import interface
import errno
from interfaces import endpoint
from zope import component
from interfaces.message import Debug
@interface.implementer(endpoint.IConnection)
class SocketConnection(object):
""" Define a connected element (serial, network…)
"""
def __init__(self, configuration):
self.port = int(configuration["port"])
self.host = configuration["host"]
def connect(self) -> None:
""" Connect """
self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.s.connect((self.host, self.port))
self.s.settimeout(0.0)
component.handle(Debug("Connected to the socket"))
def read(self) -> str:
""" Read from the connection and return the bytes.
Return None if there is nothing to read (non-blocking)
Raise an exception if disconnected """
try:
return self.s.recv(1024)
except socket.error as e:
err = e.args[0]
if err == errno.EAGAIN or err == errno.EWOULDBLOCK:
return None
raise e
def write(self, content:str) -> None:
""" Write into the connection.
Raise an exception if disconnected """
self.s.sendall(content)
self.s.sendall(bytes("\n", "utf-8"))
|