3
M]W$ @ s$ d Z ddlZddlZG dd deZG dd deZG dd deZG d d
d
eZG d d d ejZ e
d
kr ddl Z ddl Z ddl
Z
dZG dd deZejedddZe eeZejd e jd W dQ R X ejedddZe eeZej ej \ZZejd e jd ej dS )z%Support threading with serial ports.
Nc @ s( e Zd ZdZdd Zdd Zdd ZdS ) Protocolzq Protocol as used by the ReaderThread. This base class provides empty
implementations of all methods.
c C s dS )z$Called when reader thread is startedN )self transportr r /usr/lib/python3.6/__init__.pyconnection_made s zProtocol.connection_madec C s dS )z2Called with snippets received from the serial portNr )r datar r r
data_received s zProtocol.data_receivedc C s t |tr|dS )zg Called when the serial port is closed or the reader loop terminated
otherwise.
N)
isinstance Exception)r excr r r connection_lost s
zProtocol.connection_lostN)__name__
__module__ __qualname____doc__r r r
r r r r r s r c sD e Zd ZdZdZdd Zdd Z fddZd d
Zd d Z Z
S )
Packetizerz
Read binary packets from serial port. Packets are expected to be terminated
with a TERMINATOR byte (null byte by default).
The class also keeps track of the transport.
c C s t | _d | _d S )N) bytearraybufferr )r r r r __init__/ s zPacketizer.__init__c C s
|| _ dS )zStore transportN)r )r r r r r r 3 s zPacketizer.connection_madec s d| _ tt| j| dS )zForget transportN)r superr r
)r r ) __class__r r r
7 s zPacketizer.connection_lostc C sB | j j| x0| j| j kr<| j j| jd\}| _ | j| qW dS )z9Buffer received data, find TERMINATOR, call handle_packet N)r extend
TERMINATORsplit
handle_packet)r r packetr r r r < s zPacketizer.data_receivedc C s t ddS )z1Process packets - to be overridden by subclassingz/please implement functionality in handle_packetN)NotImplementedError)r r r r r r C s zPacketizer.handle_packet) r r r r r r r r
r r
__classcell__r r )r r r % s r c sP e Zd ZdZdZdZdd Zdd Z fdd Zd
d Z d d
Z
dd Z Z S ) FramedPacketz
Read binary packets. Packets are expected to have a start and stop marker.
The class also keeps track of the transport.
( )c C s t | _d| _d | _d S )NF)r r in_packetr )r r r r r R s zFramedPacket.__init__c C s
|| _ dS )zStore transportN)r )r r r r r r W s z FramedPacket.connection_madec s, d| _ d| _| jdd= tt| j| dS )zForget transportNF)r r$ r r r! r
)r r )r r r r
[ s z FramedPacket.connection_lostc C sr xlt j|D ]^}|| jkr"d| _q || jkrLd| _| j| j | jdd= q | jr`| jj| q | j| q W dS )z4Find data enclosed in START/STOP, call handle_packetTFN) serialZ iterbytesSTARTr$ STOPr r appendhandle_out_of_packet_data)r r Zbyter r r r b s
zFramedPacket.data_receivedc C s t ddS )z1Process packets - to be overridden by subclassingz/please implement functionality in handle_packetN)r )r r r r r r p s zFramedPacket.handle_packetc C s dS )z0Process data that is received outside of packetsNr )r r r r r r) t s z&FramedPacket.handle_out_of_packet_data)
r r r r r& r' r r r
r r r) r r r )r r r! H s r! c @ s4 e Zd ZdZdZdZdZdd Zdd Zd d
Z d S )
LineReaderzZ
Read and write (Unicode) lines from/to serial port.
The encoding is applied.
s
zutf-8replacec C s | j |j| j| j d S )N) handle_linedecodeENCODINGUNICODE_HANDLING)r r r r r r s zLineReader.handle_packetc C s t ddS )z2Process one line - to be overridden by subclassingz-please implement functionality in handle_lineN)r )r liner r r r, s zLineReader.handle_linec C s" | j j|j| j| j| j dS )z
Write text to the transport. ``text`` is a Unicode string and the encoding
is applied before sending ans also the newline is append.
N)r writeencoder. r/ r )r textr r r
write_line s zLineReader.write_lineN)
r r r r r r. r/ r r, r4 r r r r r* y s r* c sX e Zd ZdZ fddZdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z Z S ) ReaderThreada Implement a serial port read loop and dispatch to a Protocol instance (like
the asyncio.Protocol) but do it with threads.
Calls to close() will close the serial port but it is also possible to just
stop() this thread and continue the serial port instance otherwise.
c sD t t| j d| _|| _|| _d| _tj | _ tj
| _ d| _ dS )z Initialize thread.
Note that the serial_instance' timeout is set to one second!
Other settings are not changed.
TN)
r r5 r Zdaemonr% protocol_factoryalive threadingZLock_lockZEvent_connection_madeprotocol)r Zserial_instancer6 )r r r r s
zReaderThread.__init__c C s* d| _ t| jdr | jj | jd dS )zStop the reader threadF cancel_read N)r7 hasattrr% r< join)r r r r stop s
zReaderThread.stopc & C s0 t | jdsd| j_| j | _y| jj| W n< tk
rj } z d| _| jj| | j j
dS d}~X nX d}| j j
x| jo| jj ry| jj | jj
pd}W n* tjk
r } z |}P W Y dd}~X q|X |r|y| jj| W q| tk
r } z |}P W Y dd}~X q|X q|W d| _| jj| d| _dS )z Reader loopr< r FN)r> r% timeoutr6 r; r r r7 r
r: setZis_openreadZ
in_waitingZSerialExceptionr )r eerrorr r r r run s6
zReaderThread.runc C s" | j | jj| W dQ R X dS )z Thread safe writing (uses lock)N)r9 r% r1 )r r r r r r1 s zReaderThread.writec C s( | j | j | jj W dQ R X dS )z8Close the serial port and exit reader thread (uses lock)N)r9 r@ r% close)r r r r rG s zReaderThread.closec C s4 | j r(| jj | j s td| | jfS tddS )zj
Wait until connection is set up and return the transport and protocol
instances.
z connection_lost already calledzalready stoppedN)r7 r: wait RuntimeErrorr; )r r r r connect s
zReaderThread.connectc C s& | j | jj | js td| jS )zs Enter context handler. May raise RuntimeError in case the connection
could not be created.
z connection_lost already called)startr: rH r7 rI r; )r r r r __enter__ s
zReaderThread.__enter__c C s | j dS )zLeave context: close portN)rG )r exc_typeZexc_valZexc_tbr r r __exit__ s zReaderThread.__exit__)
r r r r r r@ rF r1 rG rJ rL rN r r r )r r r5 s # r5 __main__zloop://c s, e Zd Z fddZdd Zdd Z ZS )
PrintLinesc s* t t| j| tjjd | jd d S )Nz port opened
z hello world)r rP r sysstdoutr1 r4 )r r )r r r r s zPrintLines.connection_madec C s t jjdjt| d S )Nzline received: {}
)rQ rR r1 formatrepr)r r r r r r, s zPrintLines.handle_linec C s |rt j| tjjd d S )Nz port closed
) traceback print_excrQ rR r1 )r r r r r r
s
zPrintLines.connection_lost)r r r r r, r
r r r )r r rP
s rP i r )ZbaudraterA Zhellor= )r r% r8 objectr r r! r* ZThreadr5 r rQ ZtimerU ZPORTrP Zserial_for_urlZserr; r4 ZsleeptrK rJ r rG r r r r s0 #1q