509318396 ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 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_tuplesdefaultZclausesZ simple_caseexprvaluer r r case s"   rcCstt|td|S)NzAS %s)fnZCASTrr)ZnodeZas_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)rrr r!r#)r_selectrZFieldaddrZ_aliastype_metaZdeclared_fields_datagetnameForeignKeyFieldgetattr model_to_dictcallable reverse_relitemsr)Zmodelrrr r!r"Z extra_attrsZfields_from_queryr#itemdata model_classfieldZ field_dataZrel_objZ attr_nameattr related_nameZ foreign_keyZ descriptorZaccumZ related_queryr r r r-Es                 r-c 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_modelr3).0Zrow_datar4ignore_unknownr r s z!dict_to_model..)r'r0Zfieldsr/setattrAttributeErrorrr+dictr*r7Z rel_modellisttupler6) r3r2r:instancemetakeyrZ is_backrefZis_foreign_keyZ instancesZ rel_instancer r9r r7s>        r7cseZdZdfdd ZZS)RetryOperationalErrorNTc sztt||||}Wndtk r||s8||2|}|||pVd|rn| rn| W5QRXYnX|S)Nr ) superrD execute_sqlZOperationalErrorZ is_closedcloseZexception_wrapperZ get_cursorexecuteZget_autocommitZcommit)selfZsqlZparamsZrequire_commitZcursor __class__r r rFs   z!RetryOperationalError.execute_sql)NT)__name__ __module__ __qualname__rF __classcell__r r rJr rDsrD)N)TFNNNNNN)F)sysZpeeweer version_info collectionsrr.rrrr-r7objectrDr r r r s"   1 a (