o ѷ6i@sdZddlmZddlZddlZddlmZddlmZdZ e dZ e dej Z d"d d ZegdZeeejeZdddeDZe deej ejBZd#ddZdZdZd$d d!ZdS)%z Implements a function for hoisting specially constructed comments in SQL queries and using them to rewrite that query. This is done to support query hints whilst obviating patching Django's ORM in complex ways. ) annotationsN) OrderedDict)reducez/*QueryRewrite':z/\*QueryRewrite':(.*?)\*/a index= (?P`[^`]+`) \ # space (?PUSE|IGNORE|FORCE) \ # space ( FOR \ # space (?PJOIN|ORDER\ BY|GROUP\ BY) \ # space )? (?P(`[^`]+`(,`[^`]+`)*)|NONE) sqlstrreturnc Csg}g}g}t|D]<}|tvr||q |dr&||ddq |drGt|}|rG||d|d|d|dfq td|}|sT|sT|r[t ||||}|S) Nzlabel=zindex= table_namerule index_namesfor_what) query_rewrite_refindallSELECT_HINT_TOKENSappend startswith index_rule_rematchgroupsub modify_sql)rcommentshints index_hintsr index_matchr_/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/django_mysql/rewrite_query.py rewrite_query*s.     r)) distinctness)ALLDISTINCT DISTINCTROW)priority) HIGH_PRIORITY) join_order) STRAIGHT_JOIN) result_size)SQL_SMALL_RESULTSQL_BIG_RESULT) buffer_result)SQL_BUFFER_RESULT) query_cache) SQL_CACHE SQL_NO_CACHE) found_rows)SQL_CALC_FOUND_ROWS ccs(|]\}}dj|d|dVqdS)z (?P<{group_name}>({tokens})\s+)?|) group_nametokensN)formatjoin).0r3 token_setrrr ]s  r9z ^ \s* (?PSELECT|UPDATE|DELETE) # comments - N times /*a*/whitespace (?P(\s*/\*.*?\*/\s*)+|\s+) add_comments list[str] add_hintsadd_index_hintslist[tuple[str, str, str, str]]c st|}|s |S|dg}|d}|r|||D] }|d|dq|ddkretD]/\}zfdd|Dd } || Wq5tyd||} | d urb|| Yq5w|| d } |ddkr|r|D] } t | g| R} qw|| d |S) a Parse the start of the SQL, injecting each string in add_comments in individual SQL comments after the first keyword, and adding the named SELECT hints from add_hints, taking the latest in the list in cases of multiple mutually exclusive hints being given keywordrz/*z*/rSELECTcsg|]}|vr|qSrr)r7hinthint_setrr szmodify_sql..N ) query_start_rerrstripr SELECT_HINTSitems IndexErrorrstripendmodify_sql_index_hintsr6) rr:r<r=rr4rcommentr3to_addexisting remainder index_hintrrBrrws4        rzm \b(?PFROM|JOIN) \s+ (?P{table_name}(\s+[A-Z]+[0-9]+)?) \s+ zQ\g \g {rule} INDEX {for_section}({index_names}) r r r r cCsTtj|d}|rd|d}nd}tj|||dkrdn|d}tj|||dtjdS) N)r zFOR rFr NONE)r for_sectionr )countflags)table_spec_re_templater5replacement_templatererVERBOSE)rr r r r table_spec_rerU replacementrrrrNs rN)rrrr) rrr:r;r<r;r=r>rr) rrr rr rr rr rrr)__doc__ __future__roperatorr[ collectionsr functoolsrREWRITE_MARKERcompilerr\rrrI frozensetaddvaluesrr6rJhints_re_piece IGNORECASErGrrYrZrNrrrrs>     #    4