U
|
¸ý°dë! ã @ sp d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ G d d
|
d
|
ejej eZ
|
G d d d eZ dS ) é Né )Ú _StringTypeé )Úexc)Úsql)Úutil)Úsqltypesc sD e Zd ZdZd ZdZdd Zedd Z fddZ 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 in
|
enums are not quoted, they will be escaped and surrounded by single
|
quotes when generating the schema. 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.
|
|
ÚstrictNÚlength)ÚpopZ
|
_enum_initr Ú__init__r )ÚselfÚenumsÚkw© r ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/mysql/enumerated.pyr s & z ENUM.__init__c K s4 | d|j¡ | d|j¡ | d|j¡ | f |S )zXProduce a MySQL native :class:`.mysql.ENUM` from plain
|
:class:`.Enum`.
|
|
Úvalidate_stringsÚvalues_callableZ omit_aliases)Ú
|
setdefaultr r Z _omit_aliases)ÚclsÚimplr r r r Úadapt_emulated_to_nativeD s z ENUM.adapt_emulated_to_nativec s |dkr |S t |¡S d S )NÚ )ÚsuperÚ_object_value_for_elem)r Úelem©Ú __class__r r r O s zENUM._object_value_for_elemc C s t j| tttjgdS )N)Ú
|
to_inspect)r Ú generic_reprr r r ÚEnum©r r r r Ú__repr__Y s
|
ÿz ENUM.__repr__) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú__visit_name__Z native_enumr Ú classmethodr r r# Ú __classcell__r r r r r s *
|
|
|
r c sT e Zd ZdZd Z fddZdd Z fddZ fdd Zd
|
d Z d d Z
|
Z S )ÚSETzMySQL SET type.c s | dd¡| _t|| _| js0d|kr0t d¡| jrhdd t| jD | _| j dd t| jD ¡ t d d
|
|D d g }|
|
d |¡ t j f | 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. The values
|
are not quoted, they will be escaped and surrounded by single
|
quotes when generating the schema.
|
|
: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 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.
|
|
Úretrieve_as_bitwiseFr zNCan't use the blank value '' in a SET without setting retrieve_as_bitwise=Truec S s i | ]\}}|d | qS )é r ©Ú.0ÚidxÚvaluer r r Ú
|
<dictcomp> s z SET.__init__.<locals>.<dictcomp>c s s | ]\}}d | |fV qdS )r- Nr r. r r r Ú <genexpr> s z SET.__init__.<locals>.<genexpr>c S s g | ] }t |qS r )Úlen)r/ Úvr r r Ú
|
<listcomp> s z SET.__init__.<locals>.<listcomp>r r N) r r, ÚtupleÚvaluesr Ú ArgumentErrorÚ enumerateÚ_bitmapÚupdateÚmaxr r r )r r8 r r r r r r d 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 s2 j r fdd}nt ||¡fdd}|S )Nc s, | d k r$t | } tt jj| ¡S d S d S )N)ÚintÚsetr Zmap_bitsr; Ú __getitem__©r1 r" r r Úprocess¯ s z%SET.result_processor.<locals>.processc s@ t | tr& r | } tt d| ¡S | d k r8| d¡ | S d S )Nz[^,]+r )Ú
|
isinstanceÚstrrA ÚreÚfindallÚdiscardrC ©Ú super_convertr r rD º s
|
|
)r, r Úresult_processor)r ÚdialectZcoltyperD r ©r rK r rL ¬ s
|
zSET.result_processorc s2 t |¡ jr" fdd}n fdd}|S )Nc sP | d kr d S t | ttfr,r&| S | S n d}| D ]}| j| O }q4|S d S r> )rE r@ rF r; )r1 Z int_valuer5 rN r r rD Í s z#SET.bind_processor.<locals>.processc s4 | d k r t | ttfs d | ¡} r, | S | S d S )Nú,)rE r@ rF ÚjoinrC rJ r r rD Ý s
|
|
)r Úbind_processorr, )r rM rD r rN r rQ É s
|
|
zSET.bind_processorc K s | j |d< tj| |f| j|S )Nr, )r, r Zconstructor_copyr8 )r Zimpltyper r r r Úadapté s
|
z SET.adaptc C s t j| ttgdgdS )N)r, F)r Z additional_kw)r r r+ r r" r r r r# í s ÿýz SET.__repr__) r$ r% r& r' r( r r? rL rQ rR r# r* r r r r r+ _ s @ r+ ) rG Útypesr r r r r r ZNativeForEmulatedr! r r+ r r r r Ú<module>
|
s L
|
