3
M�]WZJ � @ sT d dl Z d dlZd dlmZ d dlZd dlmZmZmZmZm Z G dd� de�Z
dS )� N)�win32)�
SerialBase�SerialException�to_bytes�portNotOpenError�writeTimeoutErrorc s e Zd ZdZdIZ� fdd�Zdd� Zdd� Zdd� Zdd � Z e
d d � �Z dJd d!�Z d"d#� Z
d$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Ze
d2d3� �Ze
d4d5� �Ze
d6d7� �Ze
d8d9� �ZdKd<d=�ZdLd?d@�Ze
dAdB� �ZdCdD� Z dEdF� Z dGdH� Z � Z S )M�Serialz5Serial port implementation for Win32 based on ctypes.�2 �K �n � � �� �, �X � � �` � �% � K � � � � � � c s( d | _ d | _d | _tt| �j||� d S )N)� _port_handle�_overlapped_read�_overlapped_write�superr �__init__)�self�args�kwargs)� __class__� �!/usr/lib/python3.6/serialwin32.pyr s zSerial.__init__c C s� | j dkrtd��| jr td��| j}y.|j� jd�rRt|dd� �dkrRd| }W n tk
rh Y nX tj |tj
tj B ddtj tj
tjB d�| _| jtjkr�d| _td j| jtj� ���y�tj� | _tjdd
dd�| j_tj� | _tjdddd�| j_tj| jd d � tj� | _ tj | jtj | j �� | j � tj | jtj!tj"B tj#B tj$B � W n. y | j%� W n Y nX d| _� Y nX d | _dS )
zx 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.zPort is already open.ZCOM� � z\\.\r z could not open port {!r}: {!r}� i T)&Z_portr �is_open�name�upper�
startswith�int�
ValueErrorr Z
CreateFileZ GENERIC_READZ
GENERIC_WRITEZ
OPEN_EXISTING�FILE_ATTRIBUTE_NORMALZFILE_FLAG_OVERLAPPEDr ZINVALID_HANDLE_VALUE�formatZportstr�ctypes�WinErrorZ
OVERLAPPEDr Z CreateEvent�hEventr � SetupComm� COMMTIMEOUTS� _orgTimeoutsZGetCommTimeouts�byref�_reconfigure_port� PurgeComm�
PURGE_TXCLEAR�
PURGE_TXABORT�
PURGE_RXCLEAR�
PURGE_RXABORT�_close)r Zportr# r# r$ �open! sR
"
z Serial.openc C s� | j std��tj� }| jdkr"n*| jdkr6tj|_ntt| jd �d�|_ | jdkrv| j
dk rvtt| j
d �d�|_| j dkr�n*| j dkr�tj|_ ntt| j d �d�|_ tj
| j tj|�� tj| j tj� tj� }tj| j tj|�� | j|_| jtjk�r
d|_nR| jtjk�r d|_n<| jtjk�r6d|_n&| jtj k�rLd |_nt d
j | j���| j tj k�rztj!|_"d|_#n�| j tj$k�r�tj%|_"d|_#nj| j tj&k�r�tj'|_"d|_#nL| j tj(k�r�tj)|_"d|_#n.| j tj*k�r�tj+|_"d|_#nt d j | j ���| j,tj-k�rtj.|_/n@| j,tj0k�r2tj1|_/n(| j,tj2k�rJtj3|_/nt d j | j,���d|_4| j5dk�r�| j6�r~tj7|_8n| j9�r�tj:ntj;|_8| j6|_<n�| j5j=�s�t d
j | j5j=���| j5j>�r�t dj | j5j>���| j5j?dk �r�t dj | j5j?���| j5j@dk �rt dj | j5j@���| j5jA�r2t dj | j5jA���tjB|_8d|_<| jC�rRtjD|_En| jF�r`tjGntjH|_E| jC|_I| jJ|_K| jJ|_Ld|_Md|_Nd|_OtjP|_QtjR|_StjT| j tj|���s�tdj tjU� ���dS )z,Set communication parameters on opened port.z'Can only operate on a valid port handleNr i� r' � � � r& z%Unsupported number of data bits: {!r}z Unsupported parity mode: {!r}z%Unsupported number of stop bits: {!r}z:Unsupported value for RS485Settings.rts_level_for_tx: {!r}z:Unsupported value for RS485Settings.rts_level_for_rx: {!r}z9Unsupported value for RS485Settings.delay_before_tx: {!r}z9Unsupported value for RS485Settings.delay_before_rx: {!r}z2Unsupported value for RS485Settings.loopback: {!r}zCCannot configure port, something went wrong. Original message: {!r})Vr r r r4 Z_timeoutZMAXDWORDZReadIntervalTimeout�maxr, ZReadTotalTimeoutConstantZ_inter_byte_timeout�_write_timeoutZWriteTotalTimeoutConstant�SetCommTimeoutsr0 r6 Z SetCommMaskZEV_ERRZDCBZ GetCommStateZ _baudrateZBaudRateZ _bytesize�serialZFIVEBITSZByteSizeZSIXBITSZ SEVENBITSZ EIGHTBITSr- r/ Z_parityZ PARITY_NONEZNOPARITYZParityZfParityZ PARITY_EVENZ
EVENPARITYZ
PARITY_ODDZ ODDPARITYZ PARITY_MARKZ
MARKPARITYZ PARITY_SPACEZ SPACEPARITYZ _stopbitsZ STOPBITS_ONEZ
ONESTOPBITZStopBitsZSTOPBITS_ONE_POINT_FIVEZ ONE5STOPBITSZ STOPBITS_TWOZ TWOSTOPBITSZfBinaryZ _rs485_modeZ_rtsctsZRTS_CONTROL_HANDSHAKEZ fRtsControl�
_rts_stateZRTS_CONTROL_ENABLEZRTS_CONTROL_DISABLEZ fOutxCtsFlowZrts_level_for_txZrts_level_for_rxZdelay_before_txZdelay_before_rxZloopbackZRTS_CONTROL_TOGGLEZ_dsrdtrZDTR_CONTROL_HANDSHAKEZ fDtrControl�
_dtr_stateZDTR_CONTROL_ENABLEZDTR_CONTROL_DISABLEZ fOutxDsrFlowZ_xonxoffZfOutXZfInXZfNullZ
fErrorCharZ
fAbortOnErrorZXONZXonCharZXOFFZXoffCharZ SetCommStater1 )r ZtimeoutsZcomDCBr# r# r$ r7 b s�
zSerial._reconfigure_portc C s| | j dk rxtj| j | j� | jdk r@| j� tj| jj� d| _| jdk rf| j � tj| jj� d| _tj| j � d| _ dS )zinternal close port helperN)
r r rD r5 r � cancel_readZ CloseHandler2 r � cancel_write)r r# r# r$ r= � s
z
Serial._closec C s | j r| j� d| _ dS )z
Close portFN)r( r= )r r# r# r$ �close� s z Serial.closec C s: t j� }t j� }t j| jtj|�tj|��s4td��|jS )z9Return the number of bytes currently in the input buffer.z call to ClearCommError failed) r �DWORD�COMSTAT�ClearCommErrorr r0 r6 r �cbInQue)r �flags�comstatr# r# r$ �
in_waiting� s
zSerial.in_waitingr' c
C sT | j s
t�|dk�rFtj| jj� tj� }tj� }tj| j t
j |�t
j |��s`t dj
t
j� ���| jdkrvt|j|�n|}|dk�r>t
j|�}tj� }tj| j ||t
j |�t
j | j��}| r�tj� tjtjfkr�t dj
t
j� ���tj| j t
j | j�t
j |�d�}|�s,tj� tjk�r,t dj
t
j� ���|jd|j� } nt� } nt� } t| �S )z� 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 ClearCommError failed ({!r})zReadFile failed ({!r})Tz!GetOverlappedResult failed ({!r})N) r( r r Z
ResetEventr r2 rK rL rM r r0 r6 r r/ r1 Ztimeout�minrN Zcreate_string_bufferZReadFile� GetLastErrorZ
ERROR_SUCCESS�ERROR_IO_PENDING�GetOverlappedResult�ERROR_OPERATION_ABORTED�raw�value�bytes)
r �sizerO rP �nZbuf�rcZread_okZ result_ok�readr# r# r$ r] s@
z Serial.readc C s� | j s
t�t|�}|r�tj� }tj| j|t|�tj |�| j
�}| rdtj � tj krdt
djtj� ���| jdkr�tj| j| j
tj |�d�}tj � tjkr�|jS |jt|�kr�t�|jS dS dS )z2Output the given byte string over the serial port.zWriteFile failed ({!r})r TN)r( r r r rK Z WriteFiler �lenr0 r6 r rS rT r r/ r1 rC rU rV rX r )r �datar[ �errr# r# r$ �write- s
z Serial.writec C s x| j rtjd� qW dS )zb Flush of file like objects. In this case, wait until all data
is written.
g�������?N)� out_waiting�timeZsleep)r r# r# r$ �flushG s z Serial.flushc C s$ | j s
t�tj| jtjtjB � dS )z9Clear input buffer, discarding all that is in the buffer.N)r( r r r8 r r; r< )r r# r# r$ �reset_input_bufferR s zSerial.reset_input_bufferc C s$ | j s
t�tj| jtjtjB � dS )zs Clear output buffer, aborting the current output and discarding all
that is in the buffer.
N)r( r r r8 r r9 r: )r r# r# r$ �reset_output_bufferX s zSerial.reset_output_bufferc C s. | j s
t�| jr tj| j� n tj| j� dS )zBSet break: Controls TXD. When active, to transmitting is possible.N)r( r Z _break_stater Z SetCommBreakr ZClearCommBreak)r r# r# r$ �_update_break_statea s
zSerial._update_break_statec C s, | j rtj| jtj� ntj| jtj� dS )z)Set terminal status line: Request To SendN)rF r �EscapeCommFunctionr ZSETRTSZCLRRTS)r r# r# r$ �_update_rts_statej s zSerial._update_rts_statec C s, | j rtj| jtj� ntj| jtj� dS )z-Set terminal status line: Data Terminal ReadyN)rG r rh r ZSETDTRZCLRDTR)r r# r# r$ �_update_dtr_stateq s zSerial._update_dtr_statec C s, | j s
t�tj� }tj| jtj|�� |jS )N) r( r r rK ZGetCommModemStatusr r0 r6 rX )r �statr# r# r$ �_GetCommModemStatusx s
zSerial._GetCommModemStatusc C s t j| j� @ dkS )z(Read terminal status line: Clear To Sendr )r Z MS_CTS_ONrl )r r# r# r$ �cts s z
Serial.ctsc C s t j| j� @ dkS )z)Read terminal status line: Data Set Readyr )r Z MS_DSR_ONrl )r r# r# r$ �dsr� s z
Serial.dsrc C s t j| j� @ dkS )z)Read terminal status line: Ring Indicatorr )r Z
MS_RING_ONrl )r r# r# r$ �ri� s z Serial.ric C s t j| j� @ dkS )z)Read terminal status line: Carrier Detectr )r Z
MS_RLSD_ONrl )r r# r# r$ �cd� s z Serial.cd� Nc C s |dkr |}t j| j||� dS )z� Recommend a buffer size to the driver (device driver can ignore this
value). Must be called before the port is opended.
N)r r3 r )r Zrx_sizeZtx_sizer# r# r$ �set_buffer_size� s zSerial.set_buffer_sizeTc C s4 | j s
t�|r tj| jtj� ntj| jtj� dS )a* Manually control flow - when software flow control is enabled.
This will do the same as if XON (true) or XOFF (false) are received
from the other device and control the transmission accordingly.
WARNING: this function is not portable to different platforms!
N)r( r r rh r ZSETXONZSETXOFF)r �enabler# r# r$ �set_output_flow_control� s
z Serial.set_output_flow_controlc C s: t j� }t j� }t j| jtj|�tj|��s4td��|jS )z0Return how many bytes the in the outgoing bufferz call to ClearCommError failed) r rK rL rM r r0 r6 r ZcbOutQue)r rO rP r# r# r$ rb � s
zSerial.out_waitingc C sR t j� }t j| jtj|�tj|�d�}| rNt j� t jt jfkrNt j | j|� dS )zACancel a blocking read operation, may be called from other threadFN)
r rK rU r r0 r6 rS rT ZERROR_IO_INCOMPLETEZ
CancelIoEx)r Z
overlappedr\ r` r# r# r$ �_cancel_overlapped_io� s z Serial._cancel_overlapped_ioc C s | j | j� dS )zACancel a blocking read operation, may be called from other threadN)ru r )r r# r# r$ rH � s zSerial.cancel_readc C s | j | j� dS )zBCancel a blocking write operation, may be called from other threadN)ru r )r r# r# r$ rI � s zSerial.cancel_write)r r
r r r
r r r r r r r r r r r r )r' )rq N)T) �__name__�
__module__� __qualname__�__doc__Z BAUDRATESr r> r7 r= rJ �propertyrQ r] ra rd re rf rg ri rj rl rm rn ro rp rr rt rb ru rH rI �
__classcell__r# r# )r" r$ r s: A
)
r ) r0 rc rE r Zserial.serialutilr r r r r r r# r# r# r$ �<module>
s