o
    1_7                     @   s  d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
mZmZmZmZmZmZ G dd deZd Zej Zedd dkrnd dlZdZd	Zd
ZdZdZdZdZdZdZdZ G dd deZ!n]edkr{G dd deZ!nPedd dkrd dlZdZ"G dd deZ!n9edd dksedd dksedd dksedd dkrG d d! d!eZ#G d"d deZ!nG d#d deZ!e$ed$d%Z%e$ed&d'Z&e$ed(d)Z'e$ed*d+Z(e$ed,dZ)e$ed-dZ*e$ed.d/Z+e$ed0d1Z,e$ed2d3Z-e$ed4d5Z.e$ed6e,Z/e$ed7e-Z0e1ed8rej2Z2ne$ed9d:Z2e$ed;d<Z3e4d=d Z5e4d=e*Z6e4d=e)Z7e$ed>d?Z8e$ed@dAZ9G dBdC dCee!Z:G dDdE dEe:Z;G dFdG dGe:Z<dS )H    )absolute_importN)
SerialBaseSerialExceptionto_bytesPortNotOpenErrorSerialTimeoutExceptionTimeoutc                   @   s0   e Zd Zi Zdd Zdd Zdd Zdd Zd	S )
PlatformSpecificBasec                 C      t d)Nz9non-standard baudrates are not supported on this platformNotImplementedError)selfbaudrate r   4/usr/lib/python3/dist-packages/serial/serialposix.py_set_special_baudrate0      z*PlatformSpecificBase._set_special_baudratec                 C   r
   )Nz$RS485 not supported on this platformr   )r   rs485_settingsr   r   r   _set_rs485_mode3   r   z$PlatformSpecificBase._set_rs485_modec                 C   r
   )Nz*Low latency not supported on this platformr   )r   low_latency_settingsr   r   r   set_low_latency_mode6   r   z)PlatformSpecificBase.set_low_latency_modec                 C   s*   | j rt| jt dS t| jt dS )zS        Set break: Controls TXD. When active, no transmitting is possible.
        N)_break_statefcntlioctlfdTIOCSBRKTIOCCBRKr   r   r   r   _update_break_state9   s   z(PlatformSpecificBase._update_break_stateN)__name__
__module____qualname__BAUDRATE_CONSTANTSr   r   r   r   r   r   r   r   r	   -   s    r	      Zlinuxi   @l   *TX  i+T,@i   i.T  i/T              c                   @   s   e Zd Zi dddddddddd	d
dddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0Zd1d2 Zd3d4 Zd5d6 Zd7S )8PlatformSpecificr   2   r$   K   r%   n         r&      r#         i,     iX     i  	   i  
   i`	     i     i%     i K     i      i   i  i    i    i      	  
            )i  i  i        @B   ` 逄 %& - ig5 i 	= c              
   C   s   t  ddgd }z(t| jtj| |r|d  dO  < n|d  dM  < t| jtj| W d S  tyE } ztd	||d }~ww )Nir       r&   i    iz1Failed to update ASYNC_LOW_LATENCY flag to {}: {})
arrayr   r   r   termiosZTIOCGSERIALZTIOCSSERIALIOError
ValueErrorformat)r   r   bufer   r   r   r      s   z%PlatformSpecific.set_low_latency_modec              
   C   s   t  ddgd }z-t| jt| |d  tj M  < |d  tO  < | |d< |d< t| jt| W d S  t	yJ } zt
d||d }~ww )NrN   r   @   r%   r3   r4   z'Failed to set custom baud rate ({}): {})rP   r   r   r   TCGETS2rQ   ZCBAUDBOTHERTCSETS2rR   rS   rT   )r   r   rU   rV   r   r   r   r      s   &PlatformSpecific._set_special_baudratec              
   C   s8  t  ddgd }zt| jt| |d  tO  < |d urz|jr*|d  tO  < n	|d  t M  < |jr?|d  t	O  < n	|d  t	 M  < |j
rT|d  tO  < n	|d  t M  < |jd urkt|jd |d< |jd uryt|jd |d< nd|d< t| jt| W d S  ty } ztd|d }~ww )NrN   r   r2     r$   r%   zFailed to set RS485 mode: {})rP   r   r   r   
TIOCGRS485SER_RS485_ENABLEDZloopbackSER_RS485_RX_DURING_TXZrts_level_for_txSER_RS485_RTS_ON_SENDZrts_level_for_rxSER_RS485_RTS_AFTER_SENDZdelay_before_txintZdelay_before_rx
TIOCSRS485rR   rS   rT   )r   r   rU   rV   r   r   r   r      s2   

z PlatformSpecific._set_rs485_modeN)r   r    r!   r"   r   r   r   r   r   r   r   r(   _   sl    	
"r(   cygwinc                   @   s(   e Zd Zddddddddd	d
ddZdS )r(   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   )i  i  rE   rF   rG   rH   rI   rJ   rK   rL   rM   N)r   r    r!   r"   r   r   r   r   r(      s    
r0   darwinl   T  c                   @   sF   e Zd Ze d dZdZdZe	ed dkrdd Z
d	d
 ZdS )r(   r%   .{t  zt  r   r2   c                 C   s$   t  d|g}t| jt|d d S )NrN   r$   )rP   r   r   r   IOSSIOSPEED)r   r   rU   r   r   r   r      s   r[   c                 C   .   | j rt| jtj dS t| jtj dS z[            Set break: Controls TXD. When active, no transmitting is possible.
            Nr   r   r   r   r(   r   r   r   r   r   r   r         $PlatformSpecific._update_break_stateN)r   r    r!   osunamesplitZosx_versionr   r   rb   r   r   r   r   r   r   r(      s    r,   Zbsdr1   ZfreebsdZnetbsdZopenbsdc                   @   s   e Zd Zdd ZdS )ReturnBaudratec                 C   s   |S )Nr   )r   keyr   r   r   __getitem__   s   zReturnBaudrate.__getitem__N)r   r    r!   rt   r   r   r   r   rr      s    rr   c                   @   s"   e Zd Ze ZdZdZdd ZdS )r(   rg   rh   c                 C   rj   rk   rl   r   r   r   r   r      rm   rn   N)r   r    r!   rr   r"   r   r   r   r   r   r   r   r(      s
    c                   @   s   e Zd ZdS )r(   N)r   r    r!   r   r   r   r   r(   	  s    TIOCMGETiT  TIOCMBISiT  TIOCMBICiT  TIOCMSETiT  	TIOCM_DTR	TIOCM_RTS	TIOCM_CTSrO   	TIOCM_CARrW   	TIOCM_RNG   	TIOCM_DSR   TIOCM_CDTIOCM_RITIOCINQZFIONREADiT  TIOCOUTQiT  Ir   i'T  r   i(T  c                   @   s   e Zd ZdZdd Zd7ddZdd Zed	d
 Zd8ddZ	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zd9ddZdd  Zd!d" Zed#d$ Zed%d& Zed'd( Zed)d* Zed+d, Zd-d. Zd:d0d1Zd:d2d3Zd4d5 Zd6S );Serialz    Serial port class POSIX implementation. Serial port configuration is
    done with termios and fcntl. Runs on Linux and many other Un*x like
    systems.
    c                 C   s  | j du r	td| jrtdd| _zt| jtjtjB tj	B | _W n t
y> } zd| _t|jd| j |d}~ww d\| _| _d\| _| _z]| jdd z| jsZ|   | jsa|   W n ty~ } z|jtjtjfvrt W Y d}~nd}~ww |   t \| _| _t \| _| _t| jtjtj	 t| jtjtj	 W nU ty   zt| j W n	 ty   Y nw d| _| jdurt| j d| _| jdurt| j d| _| jdurt| j d| _| jdurt| j d| _ w d| _dS )zo        Open port with current settings. This may throw a SerialException
        if the port cannot be opened.Nz.Port must be configured before it can be used.zPort is already open.zcould not open port {}: {}NNT)force_update) _portr   is_openr   ro   openZportstrO_RDWRO_NOCTTY
O_NONBLOCKOSErrorerrnorT   pipe_abort_read_rpipe_abort_read_wpipe_abort_write_rpipe_abort_write_w_reconfigure_portZ_dsrdtr_update_dtr_state_rtscts_update_rts_staterR   EINVALZENOTTY_reset_input_bufferpiper   F_SETFLBaseExceptionclose	Exception)r   msgrV   r   r   r   r   7  sn   
"




zSerial.openFc                 C   s  | j du r	td| jdur>| jr6zt| j tjtjB  W n ty5 } zt|jd	| j
|d}~ww t| j tj d}d }}| jdurRd}t| jd }zt| j }|\}}}	}
}}}W n tjyw } ztd	|d}~ww |	tjtjB O }	|
tjtjB tjB tjB tjB tjB tjB  M }
dD ]}tt|r|
tt| M }
q|tjtjB tjB  M }|tjtj B tj!B tj"B  M }ttd	r|tj# M }ttd
r|tj$ M }zttd	| j% }}W n\ t&yD   z
| j'| j%  }}W nG t(yA   zt) }}W n t*y   ttd }}Y nw zt| j%}W n t+y1   t+d	| j%w |dk r?t+d	| j%Y nw Y nw |	tj, M }	| j-dkrW|	tj.O }	n,| j-dkrc|	tj/O }	n | j-dkro|	tj0O }	n| j-dkr{|	tj1O }	nt+d	| j-| j2t3j4kr|	tj5 M }	n"| j2t3j6kr|	tj5O }	n| j2t3j7kr|	tj5O }	nt+d	| j2|tj8tj9B  M }| j:t3j;kr|	tj<tj=B t>B  M }	n_| j:t3j?kr|	tj=t>B  M }	|	tj<O }	nJ| j:t3j@kr|	t> M }	|	tj<tj=B O }	n5| j:t3jAkrt>r|	tj<t>B tj=B O }	n | j:t3jBkr&t>r&|	tj<t>B O }	|	tj= M }	nt+d	| j:ttdrN| jCrA|tjDtjEB O }n#|tjDtjEB tjFB  M }n| jCr[|tjDtjEB O }n	|tjDtjEB  M }ttdr{| jGrt|	tjHO }	n|	tjH M }	nttdr| jGr|	tjIO }	n|	tjI M }	|dk s|dkrt+d	|||tjJ< |dk s|dkrt+d	|||tjK< |s|||	|
|||g|krtL| j tjM|||	|
|||g |dur| N| | jOdur| P| jO dS dS ),Set communication parameters on opened port.Nz+Can only operate on a valid file descriptorz&Could not exclusively lock port {}: {}r   r$   r4   Could not configure port: {})ZECHOCTLZECHOKEIUCLCPARMRKzB{}ZB38400zInvalid baud rate: {!r}r2   r1   r0   r#   zInvalid char len: {!r}z$Invalid stop bit specification: {!r}zInvalid parity: {!r}IXANYCRTSCTSCNEW_RTSCTS   zInvalid vmin: {!r}Invalid vtime: {!r})Qr   r   Z
_exclusiver   ZflockZLOCK_EXZLOCK_NBrR   r   rT   r   ZLOCK_UN_inter_byte_timeoutrb   rQ   	tcgetattrerrorZCLOCALZCREADZICANONZECHOZECHOEZECHOKZECHONLZISIGZIEXTENhasattrgetattrZOPOSTZONLCRZOCRNLZINLCRZIGNCRZICRNLZIGNBRKr   r   Z	_baudrateAttributeErrorr"   KeyErrorrY   	NameErrorrS   ZCSIZEZ	_bytesizeZCS8ZCS7ZCS6ZCS5Z	_stopbitsserialZSTOPBITS_ONEZCSTOPBZSTOPBITS_ONE_POINT_FIVEZSTOPBITS_TWOZINPCKZISTRIPZ_parityZPARITY_NONEZPARENBZPARODDCMSPARZPARITY_EVENZ
PARITY_ODDZPARITY_MARKZPARITY_SPACEZ_xonxoffZIXONZIXOFFr   r   r   r   VMINVTIME	tcsetattrTCSANOWr   Z_rs485_moder   )r   r   r   Zcustom_baudvminvtime	orig_attriflagoflagcflaglflagispeedospeedccflagr   r   r   r   x  s  











zSerial._reconfigure_portc                 C   sx   | j r:| jdur5t| j d| _t| j t| j t| j t| j d\| _| _d\| _| _d| _ dS dS )z
Close portNr   F)r   r   ro   r   r   r   r   r   r   r   r   r   r     s   

zSerial.closec                 C       t | jtt}td|d S )z9Return the number of bytes currently in the input buffer.r   r   )r   r   r   r   TIOCM_zero_strstructunpackr   sr   r   r   
in_waiting!     zSerial.in_waitingr$   c              
   C   st  | j st t }t| j}t||k rz6t| j| jgg g |	 \}}}| j|v r7t
| jd W t|S |s>W t|S t
| j|t| }W nQ tyr } z|jtjtjtjtjtjfvrhtd|W Y d}~n9d}~w tjy } z|d tjtjtjtjtjfvrtd|W Y d}~nd}~ww |std|| | r	 t|S t||k st|S )        Read size bytes from the serial port. If a timeout is set it may
        return less characters as requested. With no timeout it will block
        until the requested number of bytes is read.
        r\   zread failed: {}Nr   zgdevice reports readiness to read but returned no data (device disconnected or multiple access on port?))r   r   	bytearrayr   _timeoutlenselectr   r   	time_leftro   readr   r   EAGAINEALREADYEWOULDBLOCKEINPROGRESSEINTRr   rT   r   extendexpiredbytes)r   sizer   timeoutready_rU   rV   r   r   r   r   )  sH   
"
" 	
'zSerial.readc                 C      | j rt| jd d S d S N   x)r   ro   writer   r   r   r   r   cancel_read\     zSerial.cancel_readc                 C   r   r   )r   ro   r   r   r   r   r   r   cancel_write`  r   zSerial.cancel_writec              
   C   s  | j st t|}t| }}t| j}|dkrzot| j|}|j	r'|W S |j
sU| r2tdt| jg| jgg | \}}}	|rNt| jd W n|sTtdn(| du s]J t| jg| jgg d\}}}	|rwt| jd W ny|s}td||d }||8 }W nX ty     ty }
 z|
jtjtjtjtjtjfvrtd|
W Y d}
~
n.d}
~
w tjy }
 z|
d tjtjtjtjtjfvrtd|
W Y d}
~
nd}
~
ww |j	s| rtd|dks|t| S )z2Output the given byte string over the serial port.r   zWrite timeoutr\   Nr$   zwrite failed (select)zwrite failed: {})r   r   r   r   r   Z_write_timeoutro   r   r   Zis_non_blockingis_infiniter   r   r   r   r   r   r   r   r   r   r   r   r   r   rT   r   )r   datadZtx_lenlengthr   nabortr   r   rV   r   r   r   r   d  s\   
" -zSerial.writec                 C   s   | j st t| j dS )zb        Flush of file like objects. In this case, wait until all data
        is written.
        N)r   r   rQ   Ztcdrainr   r   r   r   r   flush  s   zSerial.flushc                 C   s   t | jt j dS z9Clear input buffer, discarding all that is in the buffer.N)rQ   tcflushr   ZTCIFLUSHr   r   r   r   r     s   zSerial._reset_input_bufferc                 C   s   | j st |   dS r   )r   r   r   r   r   r   r   reset_input_buffer  s   zSerial.reset_input_bufferc                 C   s    | j st t| jtj dS )zs        Clear output buffer, aborting the current output and discarding all
        that is in the buffer.
        N)r   r   rQ   r   r   ZTCOFLUSHr   r   r   r   reset_output_buffer  s   zSerial.reset_output_buffer      ?c                 C   s&   | j st t| jt|d  dS )za        Send break condition. Timed, returns to idle state after given
        duration.
        r   N)r   r   rQ   Ztcsendbreakr   rb   )r   Zdurationr   r   r   
send_break  s   zSerial.send_breakc                 C   .   | j rt| jtt dS t| jtt dS )z)Set terminal status line: Request To SendN)Z
_rts_stater   r   r   rv   TIOCM_RTS_strrw   r   r   r   r   r        zSerial._update_rts_statec                 C   r   )z-Set terminal status line: Data Terminal ReadyN)Z
_dtr_stater   r   r   rv   TIOCM_DTR_strrw   r   r   r   r   r     r   zSerial._update_dtr_statec                 C   4   | j st t| jtt}td|d t	@ dkS )z(Read terminal status line: Clear To Sendr   r   )
r   r   r   r   r   ru   r   r   r   r{   r   r   r   r   cts     z
Serial.ctsc                 C   r   )z)Read terminal status line: Data Set Readyr   r   )
r   r   r   r   r   ru   r   r   r   r   r   r   r   r   dsr  r   z
Serial.dsrc                 C   r   )z)Read terminal status line: Ring Indicatorr   r   )
r   r   r   r   r   ru   r   r   r   r   r   r   r   r   ri  r   z	Serial.ric                 C   r   )z)Read terminal status line: Carrier Detectr   r   )
r   r   r   r   r   ru   r   r   r   r   r   r   r   r   cd  r   z	Serial.cdc                 C   r   )z:Return the number of bytes currently in the output buffer.r   r   )r   r   r   r   r   r   r   r   r   r   r   out_waiting  r   zSerial.out_waitingc                 C   s   | j st | jS )z        For easier use of the serial port instance with select.
        WARNING: this function is not portable to different platforms!
        )r   r   r   r   r   r   r   fileno  s   zSerial.filenoTc                 C   8   | j st |rt| jtj dS t| jtj dS )z        Manually control flow - when software flow control is enabled.
        This will send XON (true) or XOFF (false) to the other device.
        WARNING: this function is not portable to different platforms!
        N)r   r   rQ   tcflowr   ZTCIONZTCIOFFr   Zenabler   r   r   set_input_flow_control  
   zSerial.set_input_flow_controlc                 C   r   )z        Manually control flow of outgoing data - when hardware or software flow
        control is enabled.
        WARNING: this function is not portable to different platforms!
        N)r   r   rQ   r   r   ZTCOONZTCOOFFr   r   r   r   set_output_flow_control  r  zSerial.set_output_flow_controlc                 C   s   ddl }|dt dS )zDEPRECATED - has no user   Nz0nonblocking() has no effect, already nonblocking)warningswarnDeprecationWarning)r   r  r   r   r   nonblocking  s   zSerial.nonblockingN)Fr$   )r   T)r   r    r!   __doc__r   r   r   propertyr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r   r   r   r   r   0  sB    
A 

36	
		



	

	
r   c                   @   s   e Zd ZdZdddZdS )PosixPollSerialz    Poll based read implementation. Not all systems support poll properly.
    However this one has better handling of errors, such as a device
    disconnecting while it's in use (e.g. USB-serial unplugged).
    r$   c                 C   sN  | j st t }t| j}t }|| jtj	tj
B tjB tjB  || jtj	tj
B tjB tjB  |dkrt||k r||jrEdn| d D ]\}}|| jkrW n|tj
tjB tjB @ rftdqL|| jkrxt| jd 	 t|S t| j|t| }|| | s| jdur| jdkr|s	 t|S t||k s>t|S )r   r   Nr\   zdevice reports error (poll))r   r   r   r   r   r   pollregisterr   ZPOLLINZPOLLERRZPOLLHUPZPOLLNVALr   r   r   r   r   ro   r   r   r   r   r   )r   r   r   r   r  r   ZeventrU   r   r   r   r   &  s@   
""$


zPosixPollSerial.readNr  )r   r    r!   r
  r   r   r   r   r   r    s    r  c                       s2   e Zd ZdZd fdd	Zd	ddZe Z  ZS )
VTIMESerialab      Implement timeout using vtime of tty device instead of using select.
    This means that no inter character timeout can be specified and that
    the error handling is degraded.

    Overall timeout is disabled when inter-character timeout is used.

    Note that this implementation does NOT support cancel_read(), it will
    just ignore that.
    Tc                    s  t t|   t| jtjd | jdurd}t| jd }n| jdu r)d}d}n	d}t| jd }zt	
| j}|\}}}}}	}
}W n t	jyX } ztd|d}~ww |dk sa|dkrhtd|||t	j< ||t	j< t	| jt	j|||||	|
|g dS )r   r   Nr$   r4   r   r   r   )superr  r   r   r   r   r   rb   r   rQ   r   r   r   r   rT   rS   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   	__class__r   r   r   U  s4   



zVTIMESerial._reconfigure_portr$   c                 C   s`   | j st t }t||k r,t| j|t| }|s!	 t|S || t||k st|S )r   )	r   r   r   r   ro   r   r   r   r   )r   r   r   rU   r   r   r   r   s  s   
zVTIMESerial.readr	  r  )	r   r    r!   r
  r   r   r  r   __classcell__r   r   r  r   r  I  s
    
r  )=Z
__future__r   r   r   ro   r   r   sysrQ   r   Zserial.serialutilr   r   r   r   r   r   objectr	   r   platformlowerZplatrP   rX   rZ   rY   r]   rc   r^   r`   ra   r_   r(   ri   rr   r   ru   rv   rw   rx   ry   rz   r{   r|   r}   r   r   r   r   r   r   Zpackr   r   r   r   r   r   r  r  r   r   r   r   <module>   s    
b   r*