Edit File: enumerated.cpython-37.pyc
B ��4]+, � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G d d � d e�ZG d d � d ej ej e�Z G d d� de�Z dS )� N� )� _StringType� )�exc)�sql)�util)�sqltypesc @ s e Zd Zdd� Zedd� �ZdS )�_EnumeratedValuesc C s� |� dd�| _| jdkr�t|�r�d }xd|D ]V}t|�dkrDd| _P n|d krT|d }t|�dksx|d |ksx|d |kr*d| _P q*W d| _| jdkr�t�d| jj � | �|�}|| _t d d � |D �dg �}||fS ) N�quoting�autor Zunquotedr ���Zquotedz|Manually quoting %s value literals is deprecated. Supply unquoted values and use the quoting= option in cases of ambiguity.c S s g | ] }t |��qS � )�len)�.0�vr r �W/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/dialects/mysql/enumerated.py� <listcomp>/ s z2_EnumeratedValues._init_values.<locals>.<listcomp>) �popr r r Zwarn_deprecated� __class__�__name__� _strip_valuesZ_enumerated_values�max)�self�values�kw�q�e�lengthr r r � _init_values s* $ z _EnumeratedValues._init_valuesc C s` g }xV|D ]N}|dd� dks.|dd� dkrN|dd� � |d d |d �}|�|� q W |S )Nr r �"�'r � )�replace�append)�clsr Z strip_values�ar r r r 2 s z _EnumeratedValues._strip_valuesN)r � __module__� __qualname__r � classmethodr r r r r r s r c sP e Zd ZdZd ZdZdd� Zedd� �Z� fdd�Z � fd d �Z d d � Z � Z S ) �ENUMzMySQL ENUM type.Tc O s4 |� dd� | �||� tj| fd| ji|�� dS )a~ Construct an ENUM. E.g.:: Column('myenum', ENUM("foo", "bar", "baz")) :param enums: The range of valid values for this ENUM. Values will be quoted when generating the schema according to the quoting flag (see below). This object may also be a PEP-435-compliant enumerated type. .. versionadded: 1.1 added support for PEP-435-compliant enumerated types. :param strict: This flag has no effect. .. versionchanged:: The MySQL ENUM type as well as the base Enum type now validates all Python data values. :param charset: Optional, a column-level character set for this string value. Takes precedence to 'ascii' or 'unicode' short-hand. :param collation: Optional, a column-level collation for this string value. Takes precedence to 'binary' short-hand. :param ascii: Defaults to False: short-hand for the ``latin1`` character set, generates ASCII in schema. :param unicode: Defaults to False: short-hand for the ``ucs2`` character set, generates UNICODE in schema. :param binary: Defaults to False: short-hand, pick the binary collation type that matches the column's character set. Generates BINARY in schema. This does not affect the type of data stored, only the collation of character data. :param quoting: Defaults to 'auto': automatically determine enum value quoting. If all enum values are surrounded by the same quoting character, then use 'quoted' mode. Otherwise, use 'unquoted' mode. 'quoted': values in enums are already quoted, they will be used directly when generating the schema - this usage is deprecated. 'unquoted': values in enums are not quoted, they will be escaped and surrounded by single quotes when generating the schema. Previous versions of this type always required manually quoted values to be supplied; future versions will always quote the string literals for you. This is a transitional option. �strictNr )r Z _enum_initr �__init__r )r Zenumsr r r r r+ D s 5 z ENUM.__init__c K s& |� d|j� |� d|j� | f |�S )zXProduce a MySQL native :class:`.mysql.ENUM` from plain :class:`.Enum`. �validate_strings�values_callable)� setdefaultr, r- )r$ �implr r r r �adapt_emulated_to_native} s z ENUM.adapt_emulated_to_nativec s$ | � ||�\}}tt| ��|||�S )N)r �superr) �_setup_for_values)r r �objectsr r )r r r r2 � s zENUM._setup_for_valuesc s |dkr |S t t| ��|�S d S )N� )r1 r) �_object_value_for_elem)r �elem)r r r r5 � s zENUM._object_value_for_elemc C s t j| tttjgd�S )N)Z to_inspect)r Z generic_reprr) r r �Enum)r r r r �__repr__� s z ENUM.__repr__) r r& r' �__doc__�__visit_name__Z native_enumr+ r( r0 r2 r5 r8 � __classcell__r r )r r r) = s 9 r) c sL e Zd ZdZd Z� fdd�Zdd� Z� fdd�Z� fdd �Zd d � Z � Z S ) �SETzMySQL SET type.c s� |� dd�| _| �||�\}}t|�| _| js@d|kr@t�d��| jr|tdd� t| j�D ��| _ | j � dd� t| j�D �� |� d|� t t | �jf |� d S ) a� Construct a SET. E.g.:: Column('myset', SET("foo", "bar", "baz")) The list of potential values is required in the case that this set will be used to generate DDL for a table, or if the :paramref:`.SET.retrieve_as_bitwise` flag is set to True. :param values: The range of valid values for this SET. :param convert_unicode: Same flag as that of :paramref:`.String.convert_unicode`. :param collation: same as that of :paramref:`.String.collation` :param charset: same as that of :paramref:`.VARCHAR.charset`. :param ascii: same as that of :paramref:`.VARCHAR.ascii`. :param unicode: same as that of :paramref:`.VARCHAR.unicode`. :param binary: same as that of :paramref:`.VARCHAR.binary`. :param quoting: Defaults to 'auto': automatically determine set value quoting. If all values are surrounded by the same quoting character, then use 'quoted' mode. Otherwise, use 'unquoted' mode. 'quoted': values in enums are already quoted, they will be used directly when generating the schema - this usage is deprecated. 'unquoted': values in enums are not quoted, they will be escaped and surrounded by single quotes when generating the schema. Previous versions of this type always required manually quoted values to be supplied; future versions will always quote the string literals for you. This is a transitional option. .. versionadded:: 0.9.0 :param retrieve_as_bitwise: if True, the data for the set type will be persisted and selected using an integer value, where a set is coerced into a bitwise mask for persistence. MySQL allows this mode which has the advantage of being able to store values unambiguously, such as the blank string ``''``. The datatype will appear as the expression ``col + 0`` in a SELECT statement, so that the value is coerced into an integer value in result sets. This flag is required if one wishes to persist a set that can store the blank string ``''`` as a value. .. warning:: When using :paramref:`.mysql.SET.retrieve_as_bitwise`, it is essential that the list of set values is expressed in the **exact same order** as exists on the MySQL database. .. versionadded:: 1.0.0 �retrieve_as_bitwiseFr4 zNCan't use the blank value '' in a SET without setting retrieve_as_bitwise=Truec s s | ]\}}|d | fV qdS )r! Nr )r �idx�valuer r r � <genexpr>� s z SET.__init__.<locals>.<genexpr>c s s | ]\}}d | |fV qdS )r! Nr )r r> r? r r r r@ � s r N)r r= r �tupler r � ArgumentError�dict� enumerate�_bitmap�updater. r1 r< r+ )r r r r )r r r r+ � s ? z SET.__init__c C s( | j r t�t�|tj�d | �S |S d S )Nr )r= r Z type_coercer ZInteger)r Zcolexprr r r �column_expression� s zSET.column_expressionc s6 � j r� fdd�}n tt� ��||���fdd�}|S )Nc s, | d k r$t | �} tt�� jj| ��S d S d S )N)�int�setr Zmap_bitsrE � __getitem__)r? )r r r �process� s z%SET.result_processor.<locals>.processc sB t | tj�r(� r� | �} tt�d| ��S | d k r:| �d� | S d S )Nz[^,]+r4 )� isinstancer � string_typesrI �re�findall�discard)r? )� super_convertr r rK s )r= r1 r< �result_processor)r �dialectZcoltyperK )r )r rQ r rR � s zSET.result_processorc s6 t t� ��|��� jr&� �fdd�}n �fdd�}|S )Nc sX | d kr d S t | tjtj �r0�r*�| �S | S n$d}x| D ]}|� j| O }q:W |S d S )Nr )rL r � int_typesrM rE )r? Z int_valuer )r rQ r r rK s z#SET.bind_processor.<locals>.processc s8 | d k r$t | tjtj �s$d�| �} � r0� | �S | S d S )N�,)rL r rT rM �join)r? )rQ r r rK * s )r1 r< �bind_processorr= )r rS rK )r )r rQ r rW s zSET.bind_processorc K s | j |d< tj| |f| j�|�S )Nr= )r= r Zconstructor_copyr )r Zimpltyper r r r �adapt8 s z SET.adapt) r r& r' r9 r: r+ rG rR rW rX r; r r )r r r< � s Q "r< ) rN �typesr r4 r r r r r ZNativeForEmulatedr7 r) r< r r r r �<module> s ,^