­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U SW#@sddlZddlTddlmZejddkr>ddlmZddZddd Zd d Zd d Z dddZ dddZ Gddde Z dS)N)*)Node)CallablecCs t|tSN) isinstancer)cr F/opt/alt/python38/lib64/python3.8/site-packages/playhouse/shortcuts.pyr cCsxtdg}|dk }|r |||D]"\}}|td|td|fq$|dk rb|td|f|tdt|S)a& CASE statement builder. Example CASE statements: SELECT foo, CASE WHEN foo = 1 THEN "one" WHEN foo = 2 THEN "two" ELSE "?" END -- will be in column named "case" in postgres -- FROM bar; -- equivalent to above -- SELECT foo, CASE foo WHEN 1 THEN "one" WHEN 2 THEN "two" ELSE "?" END Corresponding peewee: # No predicate, use expressions. Bar.select(Bar.foo, case(None, ( (Bar.foo == 1, "one"), (Bar.foo == 2, "two")), "?")) # Predicate, will test for equality. Bar.select(Bar.foo, case(Bar.foo, ( (1, "one"), (2, "two")), "?")) ZCASENZWHENZTHENELSEZEND)SQLappendextendClause) predicateZexpression_tuplesdefaultclausesZ simple_caseexprvaluer r r case s"   rcCstt|td|S)NzAS %s)fnZCASTrr)nodeZas_typer r r cast<srcCs|r t|StSr)set)sr r r _clone_set@srTFc  Cs|dkr dn|}|dkrd}t|}t|}|dk rr|jD]6} t| trT|| q:t| tr:| jr:|| jq:i} t|}t|}||O}t|} |jj D]} | |ks|r| |krq|j | j } t| t r|r| r|| t|| j }t|||||||dd} ni} | | | j <q|rZ|D].}t||}t|rN|| |<n|| |<q*|r|r|jjD]\}}t| |}||ksr||krqr|r||kr||krqrg}||t||dt||}|D]$}|t||||||ddq|| |<qr| S) a Convert a model instance (and any related objects) to a dictionary. :param bool recurse: Whether foreign-keys should be recursed. :param bool backrefs: Whether lists of related objects should be recursed. :param only: A list (or set) of field instances indicating which fields should be included. :param exclude: A list (or set) of field instances that should be excluded from the dictionary. :param list extra_attrs: Names of model instance attributes or methods that should be included. :param SelectQuery fields_from_query: Query that was source of model. Take fields explicitly selected by the query and serialize them. :param int max_depth: Maximum depth to recurse, value <= 0 means no max. NrF)recursebackrefsonlyexcludeseen max_depthZ _prefetch)r r!r"r#r%)r_selectrFieldaddr_aliastype_metadeclared_fields_datagetnameForeignKeyFieldgetattr model_to_dictcallable reverse_relitemsr)modelr r!r"r#r$Z extra_attrsZfields_from_queryr%itemdata model_classfieldZ field_dataZrel_obj attr_nameattr related_name foreign_keyZ descriptoraccumZ related_queryr r r r2Es                 r2c s|}|j}|D]\}}||jkr6|j|d}n@||jjkrR|j|d}n$rft|||qntd||ftt}|s|rt|trt|j t j |q|rt|t t frfdd|D} | D]} t| j |qt|j| qt|j |q|S)NFTz/Unrecognized attribute "%s" for model class %s.csg|]}tj|qSr ) dict_to_modelr9).0Zrow_datar:ignore_unknownr r s z!dict_to_model..)r+r5fieldsr4setattrAttributeErrorrr0dictr/r@ rel_modellisttupler=) r9r8rCinstancemetakeyr is_backrefZis_foreign_key instancesZ rel_instancer rBr r@s>        r@cseZdZdfdd ZZS)RetryOperationalErrorNTc sztt||||}Wndtk r||s8||2|}|||pVd|rn| rn| W5QRXYnX|S)Nr ) superrQ execute_sqlOperationalError is_closedcloseexception_wrapper get_cursorexecuteget_autocommitcommit)selfsqlparamsrequire_commitcursor __class__r r rSs   z!RetryOperationalError.execute_sql)NT)__name__ __module__ __qualname__rS __classcell__r r rar rQsrQ)N)TFNNNNNN)F)syspeeweer version_info collectionsrr3rrrr2r@objectrQr r r r s"   1 a (