o
    Rc                     @   s   d Z ddlZddlmZ ddlmZ ddlmZ ddlmZ dd	lmZ dd
lm	Z	 G dd deZ
G dd deZG dd deZG dd deZeZdS )a  
.. dialect:: mysql+mysqlconnector
    :name: MySQL Connector/Python
    :dbapi: myconnpy
    :connectstring: mysql+mysqlconnector://<user>:<password>@<host>[:<port>]/<dbname>
    :url: https://pypi.org/project/mysql-connector-python/

.. note::

    The MySQL Connector/Python DBAPI has had many issues since its release,
    some of which may remain unresolved, and the mysqlconnector dialect is
    **not tested as part of SQLAlchemy's continuous integration**.
    The recommended MySQL dialects are mysqlclient and PyMySQL.

    N   )BIT)MySQLCompiler)MySQLDialect)MySQLIdentifierPreparer   )
processors)utilc                   @   s$   e Zd Zdd Zdd Zdd ZdS )MySQLCompiler_mysqlconnectorc                 K   s`   | j jr| j|jfi |d | j|jfi | S | j|jfi |d | j|jfi | S )Nz %% z % )dialect_mysqlconnector_double_percentsprocessleftright)selfbinaryoperatorkw r   JD:\Flask\env\Lib\site-packages\sqlalchemy/dialects/mysql/mysqlconnector.pyvisit_mod_binary#   s   z-MySQLCompiler_mysqlconnector.visit_mod_binaryc                 C      | j jr
|ddS |S N%z%%r   r   replacer   textr   r   r   post_process_text1      z.MySQLCompiler_mysqlconnector.post_process_textc                 C   r   r   r   r   r   r   r   escape_literal_column7   r   z2MySQLCompiler_mysqlconnector.escape_literal_columnN)__name__
__module____qualname__r   r   r    r   r   r   r   r
   "   s    r
   c                   @   s.   e Zd Zedd Zejdd Zdd ZdS )&MySQLIdentifierPreparer_mysqlconnectorc                 C   s   | j jS N)r   r   r   r   r   r   _double_percents?   s   z7MySQLIdentifierPreparer_mysqlconnector._double_percentsc                 C   s   d S r%   r   r   valuer   r   r   r'   C   s   c                 C   s(   | | j| j}| jjr| ddS |S r   )r   Zescape_quoteZescape_to_quoter   r   r(   r   r   r   _escape_identifierG   s   z9MySQLIdentifierPreparer_mysqlconnector._escape_identifierN)r!   r"   r#   propertyr'   setterr*   r   r   r   r   r$   >   s    

r$   c                   @   s   e Zd Zdd ZdS )_myconnpyBITc                 C      dS )z0MySQL-connector already converts mysql bits, so.Nr   )r   r   Zcoltyper   r   r   result_processorP      z_myconnpyBIT.result_processorN)r!   r"   r#   r/   r   r   r   r   r-   O   s    r-   c                       s   e Zd ZdZdZdZdZdZdZdZ	e
ZeZeejeeiZ fddZdd Zedd	 Zejd
d Zedd Zdd Zdd Zejdd Zejdd Zdd Z dd Z!dd Z"d$ddZ#d$dd Z$e%g d!Z& fd"d#Z'  Z(S )%MySQLDialect_mysqlconnectorZmysqlconnectorTformatc                    s(   t t| j|i | t| j| _d S r%   )superr1   __init__r   Z(to_conditional_unicode_processor_factorydescription_encodingZ_description_decoder)r   argr   	__class__r   r   r4   h   s
   z$MySQLDialect_mysqlconnector.__init__c                 C   r.   )NFr   r   
connectionr   r   r   _check_unicode_descriptionq   r0   z6MySQLDialect_mysqlconnector._check_unicode_descriptionc                 C   r.   )Nzlatin-1r   r&   r   r   r   r5   v   r0   z0MySQLDialect_mysqlconnector.description_encodingc                 C   s   t jp| jdkS N)   r   r	   Zpy3k_mysqlconnector_version_infor&   r   r   r   supports_unicode_statements{   s   z7MySQLDialect_mysqlconnector.supports_unicode_statementsc                 C   s   ddl m} |S )Nr   )	connector)mysqlrA   )clsrA   r   r   r   dbapi   s   z!MySQLDialect_mysqlconnector.dbapic              
   C   sL   z| d W dS  | jjy% } z| ||d r W Y d }~dS  d }~ww )NFT)ZpingrD   Erroris_disconnect)r   Zdbapi_connectionerrr   r   r   do_ping   s   z#MySQLDialect_mysqlconnector.do_pingc                 C   sh  |j dd}||j t|dt t|dt t|dt t|dt t|dt t|dt t|d	t t|d
t t|dt t|dt t|dt t|dt t|dt t|dt t|dt t|dt |dd | jd urzddl	m
} |d| }||jO }||d< W g |gS  ty   Y g |gS w g |gS )Nuser)usernameZallow_local_infile
autocommitbufferedcompressZconnection_timeoutconnect_timeoutZconsume_resultsZ
force_ipv6Zget_warningsZpool_reset_sessionZ	pool_sizeZraise_on_warningsrawZssl_verify_certZuse_pureZuse_unicodeTr   )
ClientFlagclient_flags)Ztranslate_connect_argsupdatequeryr	   Zcoerce_kw_typeboolint
setdefaultrD   Zmysql.connector.constantsrP   getZget_defaultZ
FOUND_ROWS	Exception)r   urloptsrP   rQ   r   r   r   create_connect_args   sB   


z/MySQLDialect_mysqlconnector.create_connect_argsc                 C   sN   | j r!t| j dr#td| j j}|r%tdd |dddD S d S d S d S )N__version__z(\d+)\.(\d+)(?:\.(\d+))?c                 s   s     | ]}|d urt |V  qd S r%   )rU   ).0xr   r   r   	<genexpr>   s    zKMySQLDialect_mysqlconnector._mysqlconnector_version_info.<locals>.<genexpr>r   r=   r   )rD   hasattrrematchr\   tuplegroup)r   mr   r   r   r?      s   z8MySQLDialect_mysqlconnector._mysqlconnector_version_infoc                 C   s   t j o| jdk S r<   r>   r&   r   r   r   r      s   z;MySQLDialect_mysqlconnector._mysqlconnector_double_percentsc                 C   s   |j jS r%   )r:   charsetr9   r   r   r   _detect_charset      z+MySQLDialect_mysqlconnector._detect_charsetc                 C   s   |j S r%   )errno)r   	exceptionr   r   r   _extract_error_code   s   z/MySQLDialect_mysqlconnector._extract_error_codec                 C   sD   d}| j j| j jf}t||r |j|v pdt|v pdt|v S dS )N)i  i  i  i  i  i   zMySQL Connection not available.z$Connection to MySQL is not availableF)rD   ZOperationalErrorZInterfaceError
isinstanceri   str)r   er:   cursorZerrnos
exceptionsr   r   r   rF      s   



z)MySQLDialect_mysqlconnector.is_disconnectNc                 C      |  S r%   )Zfetchallr   rprf   r   r   r   _compat_fetchall   rh   z,MySQLDialect_mysqlconnector._compat_fetchallc                 C   rq   r%   )Zfetchonerr   r   r   r   _compat_fetchone   rh   z,MySQLDialect_mysqlconnector._compat_fetchone)ZSERIALIZABLEzREAD UNCOMMITTEDzREAD COMMITTEDzREPEATABLE READ
AUTOCOMMITc                    s.   |dkr	d|_ d S d|_ tt| || d S )Nrv   TF)rK   r3   r1   _set_isolation_level)r   r:   levelr7   r   r   rw      s   

z0MySQLDialect_mysqlconnector._set_isolation_levelr%   ))r!   r"   r#   ZdriverZsupports_statement_cacheZsupports_unicode_bindsZsupports_sane_rowcountZsupports_sane_multi_rowcountZsupports_native_decimalZdefault_paramstyler
   Zstatement_compilerr$   preparerr	   Zupdate_copyr   Zcolspecsr   r-   r4   r;   r+   r5   Zmemoized_propertyr@   classmethodrD   rH   r[   r?   r   rg   rk   rF   rt   ru   setZ_isolation_lookuprw   __classcell__r   r   r7   r   r1   V   sD    	


)




r1   )__doc__ra   baser   r   r   r    r   r	   r
   r$   r-   r1   r   r   r   r   r   <module>   s    