U
    !Jj                     @  s   d 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 ddlm	Z	 d	d
l
mZ ddlmZ ddlmZ erd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 dddddZG dd deZeZdS )a  

.. dialect:: mysql+pymysql
    :name: PyMySQL
    :dbapi: pymysql
    :connectstring: mysql+pymysql://<username>:<password>@<host>/<dbname>[?<options>]
    :url: https://pymysql.readthedocs.io/

Unicode
-------

Please see :ref:`mysql_unicode` for current recommendations on unicode
handling.

.. _pymysql_ssl:

SSL Connections
------------------

The PyMySQL DBAPI accepts the same SSL arguments as that of MySQLdb,
described at :ref:`mysqldb_ssl`.   See that section for additional examples.

If the server uses an automatically-generated certificate that is self-signed
or does not match the host name (as seen from the client), it may also be
necessary to indicate ``ssl_check_hostname=false`` in PyMySQL::

    connection_uri = (
        "mysql+pymysql://scott:tiger@192.168.0.134/test"
        "?ssl_ca=/home/gord/client-ssl/ca.pem"
        "&ssl_cert=/home/gord/client-ssl/client-cert.pem"
        "&ssl_key=/home/gord/client-ssl/client-key.pem"
        "&ssl_check_hostname=false"
    )

MySQL-Python Compatibility
--------------------------

The pymysql DBAPI is a pure Python port of the MySQL-python (MySQLdb) driver,
and targets 100% compatibility.   Most behavioral notes for MySQL-python apply
to the pymysql driver as well.

    )annotations)Any)Dict)Optional)Type)TYPE_CHECKING)Union   )MySQLDialect_mysqldb   )langhelpers)Literal)ConnectArgsType)DBAPIConnection)DBAPICursor)DBAPIModule)PoolProxiedConnection)URLz	Type[Any]bool)connection_clsreturnc                 C  sT   t | j}z|jd }W n tk
r0   Y dS X |dkoN|j pN|jd dk	S dS )ax  Given a Connection class like pymysql.Connection, aiomysql.Connection,
    asyncmy.Connection, inspect the ping() method and determine if it
    has a "reconnect" parameter that either defaults to True, or is positional.

    a return value of True here means that when we call ``connection.ping()``,
    we **must** pass `reconnect=False`.  a return value of False means that
    we should call ``connection.ping()`` with **no arguments**.

    This routine originates from issue #10492 for pymysql, however arg
    signature mismatches in aiomysql/asyncmy tracked by issue #13306
    necessitated a more open ended function.

    r	   FZ	reconnectr   N)r   Zget_callable_argspecpingargs
IndexErrordefaults)r   ZinspZreconnect_arg r   /var/www/api.dyntland.com/public_html/python_social_comments/venv/lib/python3.8/site-packages/sqlalchemy/dialects/mysql/pymysql.py _connection_ping_reconnects_trueJ   s    r   c                      s   e Zd ZdZdZdZejddddZe	ddd	d
Z
ejddddZdddddZd#dddd fddZddddd fddZddd d!d"Z  ZS )$MySQLDialect_pymysqlpymysqlTNr   )r   c              	   C  s8   zt dj}|j| _W dS  ttfk
r2   Y dS X d S )Nzpymysql.cursorsTF)
__import__cursorsZSSCursorZ	_sscursorImportErrorAttributeError)selfr!   r   r   r   supports_server_side_cursorsi   s    
z1MySQLDialect_pymysql.supports_server_side_cursorsr   c                 C  s   t dS )Nr   )r    )clsr   r   r   import_dbapir   s    z!MySQLDialect_pymysql.import_dbapic              	   C  s8   zt djj}W n ttfk
r*   Y dS X t|S dS )a4  determine if pymysql has deprecated, changed the default of,
        or removed the 'reconnect' argument of connection.ping().

        See #10492 and
        https://github.com/PyMySQL/mysqlclient/discussions/651#discussioncomment-7308971
        for background.

        Revised as part of #13306

        zpymysql.connectionsTN)r    connections
Connectionr"   r#   r   )r$   r)   r   r   r   _send_false_to_pingv   s    z(MySQLDialect_pymysql._send_false_to_pingr   zLiteral[True])dbapi_connectionr   c                 C  s   | j r|d n|  dS )NFT)r*   r   )r$   r+   r   r   r   do_ping   s    zMySQLDialect_pymysql.do_pingr   zOptional[Dict[str, Any]]r   )url_translate_argsr   c                   s"   |d krt dd}t j||dS )Nuser)username)r.   )dictsupercreate_connect_args)r$   r-   r.   	__class__r   r   r3      s    
 z(MySQLDialect_pymysql.create_connect_argszDBAPIModule.Errorz7Optional[Union[PoolProxiedConnection, DBAPIConnection]]zOptional[DBAPICursor])e
connectioncursorr   c                   sF   t  |||rdS t|| jjr>t| }d|kp<d|kS dS d S )NTzalready closedzconnection was killedF)r2   is_disconnect
isinstanceZloaded_dbapiErrorstrlower)r$   r6   r7   r8   Zstr_er4   r   r   r9      s    z"MySQLDialect_pymysql.is_disconnectBaseExceptionr   )	exceptionr   c                 C  s$   t |jd tr|jd }|jd S )Nr   )r:   r   	Exception)r$   r?   r   r   r   _extract_error_code   s    
z(MySQLDialect_pymysql._extract_error_code)N)__name__
__module____qualname__ZdriverZsupports_statement_cacheZdescription_encodingr   Zmemoized_propertyr%   classmethodr'   r*   r,   r3   r9   rA   __classcell__r   r   r4   r   r   c   s   	 	r   N)__doc__
__future__r   typingr   r   r   r   r   r   Zmysqldbr
   utilr   Zutil.typingr   Zengine.interfacesr   r   r   r   r   Z
engine.urlr   r   r   dialectr   r   r   r   <module>   s(   +P