diff options
Diffstat (limited to '')
| -rw-r--r-- | docs/sqlglot/dialects/dialect.html | 1410 |
1 files changed, 675 insertions, 735 deletions
diff --git a/docs/sqlglot/dialects/dialect.html b/docs/sqlglot/dialects/dialect.html index 7b67d8a..60f2619 100644 --- a/docs/sqlglot/dialects/dialect.html +++ b/docs/sqlglot/dialects/dialect.html @@ -379,9 +379,6 @@ <a class="function" href="#time_format">time_format</a> </li> <li> - <a class="function" href="#create_with_partitions_sql">create_with_partitions_sql</a> - </li> - <li> <a class="function" href="#parse_date_delta">parse_date_delta</a> </li> <li> @@ -1174,417 +1171,398 @@ </span><span id="L-654"><a href="#L-654"><span class="linenos"> 654</span></a> <span class="k">return</span> <span class="n">_time_format</span> </span><span id="L-655"><a href="#L-655"><span class="linenos"> 655</span></a> </span><span id="L-656"><a href="#L-656"><span class="linenos"> 656</span></a> -</span><span id="L-657"><a href="#L-657"><span class="linenos"> 657</span></a><span class="k">def</span> <span class="nf">create_with_partitions_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Create</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-658"><a href="#L-658"><span class="linenos"> 658</span></a><span class="w"> </span><span class="sd">"""</span> -</span><span id="L-659"><a href="#L-659"><span class="linenos"> 659</span></a><span class="sd"> In Hive and Spark, the PARTITIONED BY property acts as an extension of a table's schema. When the</span> -</span><span id="L-660"><a href="#L-660"><span class="linenos"> 660</span></a><span class="sd"> PARTITIONED BY value is an array of column names, they are transformed into a schema. The corresponding</span> -</span><span id="L-661"><a href="#L-661"><span class="linenos"> 661</span></a><span class="sd"> columns are removed from the create statement.</span> -</span><span id="L-662"><a href="#L-662"><span class="linenos"> 662</span></a><span class="sd"> """</span> -</span><span id="L-663"><a href="#L-663"><span class="linenos"> 663</span></a> <span class="n">has_schema</span> <span class="o">=</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">)</span> -</span><span id="L-664"><a href="#L-664"><span class="linenos"> 664</span></a> <span class="n">is_partitionable</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"kind"</span><span class="p">)</span> <span class="ow">in</span> <span class="p">(</span><span class="s2">"TABLE"</span><span class="p">,</span> <span class="s2">"VIEW"</span><span class="p">)</span> -</span><span id="L-665"><a href="#L-665"><span class="linenos"> 665</span></a> -</span><span id="L-666"><a href="#L-666"><span class="linenos"> 666</span></a> <span class="k">if</span> <span class="n">has_schema</span> <span class="ow">and</span> <span class="n">is_partitionable</span><span class="p">:</span> -</span><span id="L-667"><a href="#L-667"><span class="linenos"> 667</span></a> <span class="n">prop</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">PartitionedByProperty</span><span class="p">)</span> -</span><span id="L-668"><a href="#L-668"><span class="linenos"> 668</span></a> <span class="k">if</span> <span class="n">prop</span> <span class="ow">and</span> <span class="n">prop</span><span class="o">.</span><span class="n">this</span> <span class="ow">and</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">prop</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">):</span> -</span><span id="L-669"><a href="#L-669"><span class="linenos"> 669</span></a> <span class="n">schema</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> -</span><span id="L-670"><a href="#L-670"><span class="linenos"> 670</span></a> <span class="n">columns</span> <span class="o">=</span> <span class="p">{</span><span class="n">v</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">for</span> <span class="n">v</span> <span class="ow">in</span> <span class="n">prop</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">}</span> -</span><span id="L-671"><a href="#L-671"><span class="linenos"> 671</span></a> <span class="n">partitions</span> <span class="o">=</span> <span class="p">[</span><span class="n">col</span> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">schema</span><span class="o">.</span><span class="n">expressions</span> <span class="k">if</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">in</span> <span class="n">columns</span><span class="p">]</span> -</span><span id="L-672"><a href="#L-672"><span class="linenos"> 672</span></a> <span class="n">schema</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s2">"expressions"</span><span class="p">,</span> <span class="p">[</span><span class="n">e</span> <span class="k">for</span> <span class="n">e</span> <span class="ow">in</span> <span class="n">schema</span><span class="o">.</span><span class="n">expressions</span> <span class="k">if</span> <span class="n">e</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">partitions</span><span class="p">])</span> -</span><span id="L-673"><a href="#L-673"><span class="linenos"> 673</span></a> <span class="n">prop</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">PartitionedByProperty</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">partitions</span><span class="p">)))</span> -</span><span id="L-674"><a href="#L-674"><span class="linenos"> 674</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s2">"this"</span><span class="p">,</span> <span class="n">schema</span><span class="p">)</span> -</span><span id="L-675"><a href="#L-675"><span class="linenos"> 675</span></a> -</span><span id="L-676"><a href="#L-676"><span class="linenos"> 676</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-677"><a href="#L-677"><span class="linenos"> 677</span></a> +</span><span id="L-657"><a href="#L-657"><span class="linenos"> 657</span></a><span class="k">def</span> <span class="nf">parse_date_delta</span><span class="p">(</span> +</span><span id="L-658"><a href="#L-658"><span class="linenos"> 658</span></a> <span class="n">exp_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> <span class="n">unit_mapping</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">t</span><span class="o">.</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span> +</span><span id="L-659"><a href="#L-659"><span class="linenos"> 659</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">E</span><span class="p">]:</span> +</span><span id="L-660"><a href="#L-660"><span class="linenos"> 660</span></a> <span class="k">def</span> <span class="nf">inner_func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-661"><a href="#L-661"><span class="linenos"> 661</span></a> <span class="n">unit_based</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o">==</span> <span class="mi">3</span> +</span><span id="L-662"><a href="#L-662"><span class="linenos"> 662</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> +</span><span id="L-663"><a href="#L-663"><span class="linenos"> 663</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="s2">"DAY"</span><span class="p">)</span> +</span><span id="L-664"><a href="#L-664"><span class="linenos"> 664</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit_mapping</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">(),</span> <span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="p">))</span> <span class="k">if</span> <span class="n">unit_mapping</span> <span class="k">else</span> <span class="n">unit</span> +</span><span id="L-665"><a href="#L-665"><span class="linenos"> 665</span></a> <span class="k">return</span> <span class="n">exp_class</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> +</span><span id="L-666"><a href="#L-666"><span class="linenos"> 666</span></a> +</span><span id="L-667"><a href="#L-667"><span class="linenos"> 667</span></a> <span class="k">return</span> <span class="n">inner_func</span> +</span><span id="L-668"><a href="#L-668"><span class="linenos"> 668</span></a> +</span><span id="L-669"><a href="#L-669"><span class="linenos"> 669</span></a> +</span><span id="L-670"><a href="#L-670"><span class="linenos"> 670</span></a><span class="k">def</span> <span class="nf">parse_date_delta_with_interval</span><span class="p">(</span> +</span><span id="L-671"><a href="#L-671"><span class="linenos"> 671</span></a> <span class="n">expression_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> +</span><span id="L-672"><a href="#L-672"><span class="linenos"> 672</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]]:</span> +</span><span id="L-673"><a href="#L-673"><span class="linenos"> 673</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]:</span> +</span><span id="L-674"><a href="#L-674"><span class="linenos"> 674</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o"><</span> <span class="mi">2</span><span class="p">:</span> +</span><span id="L-675"><a href="#L-675"><span class="linenos"> 675</span></a> <span class="k">return</span> <span class="kc">None</span> +</span><span id="L-676"><a href="#L-676"><span class="linenos"> 676</span></a> +</span><span id="L-677"><a href="#L-677"><span class="linenos"> 677</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> </span><span id="L-678"><a href="#L-678"><span class="linenos"> 678</span></a> -</span><span id="L-679"><a href="#L-679"><span class="linenos"> 679</span></a><span class="k">def</span> <span class="nf">parse_date_delta</span><span class="p">(</span> -</span><span id="L-680"><a href="#L-680"><span class="linenos"> 680</span></a> <span class="n">exp_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> <span class="n">unit_mapping</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">t</span><span class="o">.</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span> -</span><span id="L-681"><a href="#L-681"><span class="linenos"> 681</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">E</span><span class="p">]:</span> -</span><span id="L-682"><a href="#L-682"><span class="linenos"> 682</span></a> <span class="k">def</span> <span class="nf">inner_func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-683"><a href="#L-683"><span class="linenos"> 683</span></a> <span class="n">unit_based</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o">==</span> <span class="mi">3</span> -</span><span id="L-684"><a href="#L-684"><span class="linenos"> 684</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> -</span><span id="L-685"><a href="#L-685"><span class="linenos"> 685</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="s2">"DAY"</span><span class="p">)</span> -</span><span id="L-686"><a href="#L-686"><span class="linenos"> 686</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit_mapping</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">(),</span> <span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="p">))</span> <span class="k">if</span> <span class="n">unit_mapping</span> <span class="k">else</span> <span class="n">unit</span> -</span><span id="L-687"><a href="#L-687"><span class="linenos"> 687</span></a> <span class="k">return</span> <span class="n">exp_class</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> -</span><span id="L-688"><a href="#L-688"><span class="linenos"> 688</span></a> -</span><span id="L-689"><a href="#L-689"><span class="linenos"> 689</span></a> <span class="k">return</span> <span class="n">inner_func</span> -</span><span id="L-690"><a href="#L-690"><span class="linenos"> 690</span></a> +</span><span id="L-679"><a href="#L-679"><span class="linenos"> 679</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interval</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">):</span> +</span><span id="L-680"><a href="#L-680"><span class="linenos"> 680</span></a> <span class="k">raise</span> <span class="n">ParseError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"INTERVAL expression expected but got '</span><span class="si">{</span><span class="n">interval</span><span class="si">}</span><span class="s2">'"</span><span class="p">)</span> +</span><span id="L-681"><a href="#L-681"><span class="linenos"> 681</span></a> +</span><span id="L-682"><a href="#L-682"><span class="linenos"> 682</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">interval</span><span class="o">.</span><span class="n">this</span> +</span><span id="L-683"><a href="#L-683"><span class="linenos"> 683</span></a> <span class="k">if</span> <span class="n">expression</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> +</span><span id="L-684"><a href="#L-684"><span class="linenos"> 684</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="L-685"><a href="#L-685"><span class="linenos"> 685</span></a> +</span><span id="L-686"><a href="#L-686"><span class="linenos"> 686</span></a> <span class="k">return</span> <span class="n">expression_class</span><span class="p">(</span> +</span><span id="L-687"><a href="#L-687"><span class="linenos"> 687</span></a> <span class="n">this</span><span class="o">=</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">expression</span><span class="o">=</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">interval</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">))</span> +</span><span id="L-688"><a href="#L-688"><span class="linenos"> 688</span></a> <span class="p">)</span> +</span><span id="L-689"><a href="#L-689"><span class="linenos"> 689</span></a> +</span><span id="L-690"><a href="#L-690"><span class="linenos"> 690</span></a> <span class="k">return</span> <span class="n">func</span> </span><span id="L-691"><a href="#L-691"><span class="linenos"> 691</span></a> -</span><span id="L-692"><a href="#L-692"><span class="linenos"> 692</span></a><span class="k">def</span> <span class="nf">parse_date_delta_with_interval</span><span class="p">(</span> -</span><span id="L-693"><a href="#L-693"><span class="linenos"> 693</span></a> <span class="n">expression_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> -</span><span id="L-694"><a href="#L-694"><span class="linenos"> 694</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]]:</span> -</span><span id="L-695"><a href="#L-695"><span class="linenos"> 695</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]:</span> -</span><span id="L-696"><a href="#L-696"><span class="linenos"> 696</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o"><</span> <span class="mi">2</span><span class="p">:</span> -</span><span id="L-697"><a href="#L-697"><span class="linenos"> 697</span></a> <span class="k">return</span> <span class="kc">None</span> -</span><span id="L-698"><a href="#L-698"><span class="linenos"> 698</span></a> -</span><span id="L-699"><a href="#L-699"><span class="linenos"> 699</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> +</span><span id="L-692"><a href="#L-692"><span class="linenos"> 692</span></a> +</span><span id="L-693"><a href="#L-693"><span class="linenos"> 693</span></a><span class="k">def</span> <span class="nf">date_trunc_to_time</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> +</span><span id="L-694"><a href="#L-694"><span class="linenos"> 694</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> +</span><span id="L-695"><a href="#L-695"><span class="linenos"> 695</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span> +</span><span id="L-696"><a href="#L-696"><span class="linenos"> 696</span></a> +</span><span id="L-697"><a href="#L-697"><span class="linenos"> 697</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">this</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"date"</span><span class="p">):</span> +</span><span id="L-698"><a href="#L-698"><span class="linenos"> 698</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">(</span><span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">,</span> <span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">)</span> +</span><span id="L-699"><a href="#L-699"><span class="linenos"> 699</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> </span><span id="L-700"><a href="#L-700"><span class="linenos"> 700</span></a> -</span><span id="L-701"><a href="#L-701"><span class="linenos"> 701</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interval</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">):</span> -</span><span id="L-702"><a href="#L-702"><span class="linenos"> 702</span></a> <span class="k">raise</span> <span class="n">ParseError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"INTERVAL expression expected but got '</span><span class="si">{</span><span class="n">interval</span><span class="si">}</span><span class="s2">'"</span><span class="p">)</span> -</span><span id="L-703"><a href="#L-703"><span class="linenos"> 703</span></a> -</span><span id="L-704"><a href="#L-704"><span class="linenos"> 704</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">interval</span><span class="o">.</span><span class="n">this</span> -</span><span id="L-705"><a href="#L-705"><span class="linenos"> 705</span></a> <span class="k">if</span> <span class="n">expression</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> -</span><span id="L-706"><a href="#L-706"><span class="linenos"> 706</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="L-707"><a href="#L-707"><span class="linenos"> 707</span></a> -</span><span id="L-708"><a href="#L-708"><span class="linenos"> 708</span></a> <span class="k">return</span> <span class="n">expression_class</span><span class="p">(</span> -</span><span id="L-709"><a href="#L-709"><span class="linenos"> 709</span></a> <span class="n">this</span><span class="o">=</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">expression</span><span class="o">=</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">interval</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">))</span> -</span><span id="L-710"><a href="#L-710"><span class="linenos"> 710</span></a> <span class="p">)</span> +</span><span id="L-701"><a href="#L-701"><span class="linenos"> 701</span></a> +</span><span id="L-702"><a href="#L-702"><span class="linenos"> 702</span></a><span class="k">def</span> <span class="nf">date_add_interval_sql</span><span class="p">(</span> +</span><span id="L-703"><a href="#L-703"><span class="linenos"> 703</span></a> <span class="n">data_type</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">kind</span><span class="p">:</span> <span class="nb">str</span> +</span><span id="L-704"><a href="#L-704"><span class="linenos"> 704</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="L-705"><a href="#L-705"><span class="linenos"> 705</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-706"><a href="#L-706"><span class="linenos"> 706</span></a> <span class="n">this</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> +</span><span id="L-707"><a href="#L-707"><span class="linenos"> 707</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span> +</span><span id="L-708"><a href="#L-708"><span class="linenos"> 708</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">if</span> <span class="n">unit</span> <span class="k">else</span> <span class="s2">"DAY"</span><span class="p">)</span> +</span><span id="L-709"><a href="#L-709"><span class="linenos"> 709</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> +</span><span id="L-710"><a href="#L-710"><span class="linenos"> 710</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">data_type</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">kind</span><span class="si">}</span><span class="s2">(</span><span class="si">{</span><span class="n">this</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">interval</span><span class="p">)</span><span class="si">}</span><span class="s2">)"</span> </span><span id="L-711"><a href="#L-711"><span class="linenos"> 711</span></a> </span><span id="L-712"><a href="#L-712"><span class="linenos"> 712</span></a> <span class="k">return</span> <span class="n">func</span> </span><span id="L-713"><a href="#L-713"><span class="linenos"> 713</span></a> </span><span id="L-714"><a href="#L-714"><span class="linenos"> 714</span></a> -</span><span id="L-715"><a href="#L-715"><span class="linenos"> 715</span></a><span class="k">def</span> <span class="nf">date_trunc_to_time</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> -</span><span id="L-716"><a href="#L-716"><span class="linenos"> 716</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> -</span><span id="L-717"><a href="#L-717"><span class="linenos"> 717</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span> -</span><span id="L-718"><a href="#L-718"><span class="linenos"> 718</span></a> -</span><span id="L-719"><a href="#L-719"><span class="linenos"> 719</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">this</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"date"</span><span class="p">):</span> -</span><span id="L-720"><a href="#L-720"><span class="linenos"> 720</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">(</span><span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">,</span> <span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">)</span> -</span><span id="L-721"><a href="#L-721"><span class="linenos"> 721</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> -</span><span id="L-722"><a href="#L-722"><span class="linenos"> 722</span></a> -</span><span id="L-723"><a href="#L-723"><span class="linenos"> 723</span></a> -</span><span id="L-724"><a href="#L-724"><span class="linenos"> 724</span></a><span class="k">def</span> <span class="nf">date_add_interval_sql</span><span class="p">(</span> -</span><span id="L-725"><a href="#L-725"><span class="linenos"> 725</span></a> <span class="n">data_type</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">kind</span><span class="p">:</span> <span class="nb">str</span> -</span><span id="L-726"><a href="#L-726"><span class="linenos"> 726</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="L-727"><a href="#L-727"><span class="linenos"> 727</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-728"><a href="#L-728"><span class="linenos"> 728</span></a> <span class="n">this</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> -</span><span id="L-729"><a href="#L-729"><span class="linenos"> 729</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span> -</span><span id="L-730"><a href="#L-730"><span class="linenos"> 730</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">if</span> <span class="n">unit</span> <span class="k">else</span> <span class="s2">"DAY"</span><span class="p">)</span> -</span><span id="L-731"><a href="#L-731"><span class="linenos"> 731</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> -</span><span id="L-732"><a href="#L-732"><span class="linenos"> 732</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">data_type</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">kind</span><span class="si">}</span><span class="s2">(</span><span class="si">{</span><span class="n">this</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">interval</span><span class="p">)</span><span class="si">}</span><span class="s2">)"</span> -</span><span id="L-733"><a href="#L-733"><span class="linenos"> 733</span></a> -</span><span id="L-734"><a href="#L-734"><span class="linenos"> 734</span></a> <span class="k">return</span> <span class="n">func</span> +</span><span id="L-715"><a href="#L-715"><span class="linenos"> 715</span></a><span class="k">def</span> <span class="nf">timestamptrunc_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-716"><a href="#L-716"><span class="linenos"> 716</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="L-717"><a href="#L-717"><span class="linenos"> 717</span></a> <span class="s2">"DATE_TRUNC"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> +</span><span id="L-718"><a href="#L-718"><span class="linenos"> 718</span></a> <span class="p">)</span> +</span><span id="L-719"><a href="#L-719"><span class="linenos"> 719</span></a> +</span><span id="L-720"><a href="#L-720"><span class="linenos"> 720</span></a> +</span><span id="L-721"><a href="#L-721"><span class="linenos"> 721</span></a><span class="k">def</span> <span class="nf">no_timestamp_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Timestamp</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-722"><a href="#L-722"><span class="linenos"> 722</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> +</span><span id="L-723"><a href="#L-723"><span class="linenos"> 723</span></a> <span class="kn">from</span> <span class="nn">sqlglot.optimizer.annotate_types</span> <span class="kn">import</span> <span class="n">annotate_types</span> +</span><span id="L-724"><a href="#L-724"><span class="linenos"> 724</span></a> +</span><span id="L-725"><a href="#L-725"><span class="linenos"> 725</span></a> <span class="n">target_type</span> <span class="o">=</span> <span class="n">annotate_types</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span><span class="o">.</span><span class="n">type</span> <span class="ow">or</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span> +</span><span id="L-726"><a href="#L-726"><span class="linenos"> 726</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">target_type</span><span class="p">))</span> +</span><span id="L-727"><a href="#L-727"><span class="linenos"> 727</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"expression"</span><span class="p">)</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="ow">in</span> <span class="n">TIMEZONES</span><span class="p">:</span> +</span><span id="L-728"><a href="#L-728"><span class="linenos"> 728</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="L-729"><a href="#L-729"><span class="linenos"> 729</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">AtTimeZone</span><span class="p">(</span> +</span><span id="L-730"><a href="#L-730"><span class="linenos"> 730</span></a> <span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">),</span> +</span><span id="L-731"><a href="#L-731"><span class="linenos"> 731</span></a> <span class="n">zone</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> +</span><span id="L-732"><a href="#L-732"><span class="linenos"> 732</span></a> <span class="p">)</span> +</span><span id="L-733"><a href="#L-733"><span class="linenos"> 733</span></a> <span class="p">)</span> +</span><span id="L-734"><a href="#L-734"><span class="linenos"> 734</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"TIMESTAMP"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> </span><span id="L-735"><a href="#L-735"><span class="linenos"> 735</span></a> </span><span id="L-736"><a href="#L-736"><span class="linenos"> 736</span></a> -</span><span id="L-737"><a href="#L-737"><span class="linenos"> 737</span></a><span class="k">def</span> <span class="nf">timestamptrunc_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-738"><a href="#L-738"><span class="linenos"> 738</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="L-739"><a href="#L-739"><span class="linenos"> 739</span></a> <span class="s2">"DATE_TRUNC"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> +</span><span id="L-737"><a href="#L-737"><span class="linenos"> 737</span></a><span class="k">def</span> <span class="nf">locate_to_strposition</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> +</span><span id="L-738"><a href="#L-738"><span class="linenos"> 738</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">(</span> +</span><span id="L-739"><a href="#L-739"><span class="linenos"> 739</span></a> <span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">substr</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">position</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> </span><span id="L-740"><a href="#L-740"><span class="linenos"> 740</span></a> <span class="p">)</span> </span><span id="L-741"><a href="#L-741"><span class="linenos"> 741</span></a> </span><span id="L-742"><a href="#L-742"><span class="linenos"> 742</span></a> -</span><span id="L-743"><a href="#L-743"><span class="linenos"> 743</span></a><span class="k">def</span> <span class="nf">no_timestamp_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Timestamp</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-744"><a href="#L-744"><span class="linenos"> 744</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> -</span><span id="L-745"><a href="#L-745"><span class="linenos"> 745</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">))</span> -</span><span id="L-746"><a href="#L-746"><span class="linenos"> 746</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"expression"</span><span class="p">)</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="ow">in</span> <span class="n">TIMEZONES</span><span class="p">:</span> -</span><span id="L-747"><a href="#L-747"><span class="linenos"> 747</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="L-748"><a href="#L-748"><span class="linenos"> 748</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">AtTimeZone</span><span class="p">(</span> -</span><span id="L-749"><a href="#L-749"><span class="linenos"> 749</span></a> <span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">),</span> -</span><span id="L-750"><a href="#L-750"><span class="linenos"> 750</span></a> <span class="n">zone</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> -</span><span id="L-751"><a href="#L-751"><span class="linenos"> 751</span></a> <span class="p">)</span> -</span><span id="L-752"><a href="#L-752"><span class="linenos"> 752</span></a> <span class="p">)</span> -</span><span id="L-753"><a href="#L-753"><span class="linenos"> 753</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">function_fallback_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-754"><a href="#L-754"><span class="linenos"> 754</span></a> +</span><span id="L-743"><a href="#L-743"><span class="linenos"> 743</span></a><span class="k">def</span> <span class="nf">strposition_to_locate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-744"><a href="#L-744"><span class="linenos"> 744</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="L-745"><a href="#L-745"><span class="linenos"> 745</span></a> <span class="s2">"LOCATE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"substr"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"position"</span><span class="p">)</span> +</span><span id="L-746"><a href="#L-746"><span class="linenos"> 746</span></a> <span class="p">)</span> +</span><span id="L-747"><a href="#L-747"><span class="linenos"> 747</span></a> +</span><span id="L-748"><a href="#L-748"><span class="linenos"> 748</span></a> +</span><span id="L-749"><a href="#L-749"><span class="linenos"> 749</span></a><span class="k">def</span> <span class="nf">left_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-750"><a href="#L-750"><span class="linenos"> 750</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="L-751"><a href="#L-751"><span class="linenos"> 751</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> +</span><span id="L-752"><a href="#L-752"><span class="linenos"> 752</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="mi">1</span><span class="p">),</span> <span class="n">length</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> +</span><span id="L-753"><a href="#L-753"><span class="linenos"> 753</span></a> <span class="p">)</span> +</span><span id="L-754"><a href="#L-754"><span class="linenos"> 754</span></a> <span class="p">)</span> </span><span id="L-755"><a href="#L-755"><span class="linenos"> 755</span></a> -</span><span id="L-756"><a href="#L-756"><span class="linenos"> 756</span></a><span class="k">def</span> <span class="nf">locate_to_strposition</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> -</span><span id="L-757"><a href="#L-757"><span class="linenos"> 757</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">(</span> -</span><span id="L-758"><a href="#L-758"><span class="linenos"> 758</span></a> <span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">substr</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">position</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> -</span><span id="L-759"><a href="#L-759"><span class="linenos"> 759</span></a> <span class="p">)</span> -</span><span id="L-760"><a href="#L-760"><span class="linenos"> 760</span></a> -</span><span id="L-761"><a href="#L-761"><span class="linenos"> 761</span></a> -</span><span id="L-762"><a href="#L-762"><span class="linenos"> 762</span></a><span class="k">def</span> <span class="nf">strposition_to_locate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-763"><a href="#L-763"><span class="linenos"> 763</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="L-764"><a href="#L-764"><span class="linenos"> 764</span></a> <span class="s2">"LOCATE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"substr"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"position"</span><span class="p">)</span> -</span><span id="L-765"><a href="#L-765"><span class="linenos"> 765</span></a> <span class="p">)</span> -</span><span id="L-766"><a href="#L-766"><span class="linenos"> 766</span></a> -</span><span id="L-767"><a href="#L-767"><span class="linenos"> 767</span></a> -</span><span id="L-768"><a href="#L-768"><span class="linenos"> 768</span></a><span class="k">def</span> <span class="nf">left_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-769"><a href="#L-769"><span class="linenos"> 769</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="L-770"><a href="#L-770"><span class="linenos"> 770</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> -</span><span id="L-771"><a href="#L-771"><span class="linenos"> 771</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="mi">1</span><span class="p">),</span> <span class="n">length</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> -</span><span id="L-772"><a href="#L-772"><span class="linenos"> 772</span></a> <span class="p">)</span> -</span><span id="L-773"><a href="#L-773"><span class="linenos"> 773</span></a> <span class="p">)</span> -</span><span id="L-774"><a href="#L-774"><span class="linenos"> 774</span></a> -</span><span id="L-775"><a href="#L-775"><span class="linenos"> 775</span></a> -</span><span id="L-776"><a href="#L-776"><span class="linenos"> 776</span></a><span class="k">def</span> <span class="nf">right_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-777"><a href="#L-777"><span class="linenos"> 777</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="L-778"><a href="#L-778"><span class="linenos"> 778</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> -</span><span id="L-779"><a href="#L-779"><span class="linenos"> 779</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> -</span><span id="L-780"><a href="#L-780"><span class="linenos"> 780</span></a> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Length</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> <span class="o">-</span> <span class="n">exp</span><span class="o">.</span><span class="n">paren</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> <span class="o">-</span> <span class="mi">1</span><span class="p">),</span> -</span><span id="L-781"><a href="#L-781"><span class="linenos"> 781</span></a> <span class="p">)</span> -</span><span id="L-782"><a href="#L-782"><span class="linenos"> 782</span></a> <span class="p">)</span> +</span><span id="L-756"><a href="#L-756"><span class="linenos"> 756</span></a> +</span><span id="L-757"><a href="#L-757"><span class="linenos"> 757</span></a><span class="k">def</span> <span class="nf">right_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-758"><a href="#L-758"><span class="linenos"> 758</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="L-759"><a href="#L-759"><span class="linenos"> 759</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> +</span><span id="L-760"><a href="#L-760"><span class="linenos"> 760</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> +</span><span id="L-761"><a href="#L-761"><span class="linenos"> 761</span></a> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Length</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> <span class="o">-</span> <span class="n">exp</span><span class="o">.</span><span class="n">paren</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> <span class="o">-</span> <span class="mi">1</span><span class="p">),</span> +</span><span id="L-762"><a href="#L-762"><span class="linenos"> 762</span></a> <span class="p">)</span> +</span><span id="L-763"><a href="#L-763"><span class="linenos"> 763</span></a> <span class="p">)</span> +</span><span id="L-764"><a href="#L-764"><span class="linenos"> 764</span></a> +</span><span id="L-765"><a href="#L-765"><span class="linenos"> 765</span></a> +</span><span id="L-766"><a href="#L-766"><span class="linenos"> 766</span></a><span class="k">def</span> <span class="nf">timestrtotime_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeStrToTime</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-767"><a href="#L-767"><span class="linenos"> 767</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"timestamp"</span><span class="p">))</span> +</span><span id="L-768"><a href="#L-768"><span class="linenos"> 768</span></a> +</span><span id="L-769"><a href="#L-769"><span class="linenos"> 769</span></a> +</span><span id="L-770"><a href="#L-770"><span class="linenos"> 770</span></a><span class="k">def</span> <span class="nf">datestrtodate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateStrToDate</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-771"><a href="#L-771"><span class="linenos"> 771</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> +</span><span id="L-772"><a href="#L-772"><span class="linenos"> 772</span></a> +</span><span id="L-773"><a href="#L-773"><span class="linenos"> 773</span></a> +</span><span id="L-774"><a href="#L-774"><span class="linenos"> 774</span></a><span class="c1"># Used for Presto and Duckdb which use functions that don't support charset, and assume utf-8</span> +</span><span id="L-775"><a href="#L-775"><span class="linenos"> 775</span></a><span class="k">def</span> <span class="nf">encode_decode_sql</span><span class="p">(</span> +</span><span id="L-776"><a href="#L-776"><span class="linenos"> 776</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">replace</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="L-777"><a href="#L-777"><span class="linenos"> 777</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-778"><a href="#L-778"><span class="linenos"> 778</span></a> <span class="n">charset</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"charset"</span><span class="p">)</span> +</span><span id="L-779"><a href="#L-779"><span class="linenos"> 779</span></a> <span class="k">if</span> <span class="n">charset</span> <span class="ow">and</span> <span class="n">charset</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="o">!=</span> <span class="s2">"utf-8"</span><span class="p">:</span> +</span><span id="L-780"><a href="#L-780"><span class="linenos"> 780</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Expected utf-8 character set, got </span><span class="si">{</span><span class="n">charset</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> +</span><span id="L-781"><a href="#L-781"><span class="linenos"> 781</span></a> +</span><span id="L-782"><a href="#L-782"><span class="linenos"> 782</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"replace"</span><span class="p">)</span> <span class="k">if</span> <span class="n">replace</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> </span><span id="L-783"><a href="#L-783"><span class="linenos"> 783</span></a> </span><span id="L-784"><a href="#L-784"><span class="linenos"> 784</span></a> -</span><span id="L-785"><a href="#L-785"><span class="linenos"> 785</span></a><span class="k">def</span> <span class="nf">timestrtotime_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeStrToTime</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-786"><a href="#L-786"><span class="linenos"> 786</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"timestamp"</span><span class="p">))</span> -</span><span id="L-787"><a href="#L-787"><span class="linenos"> 787</span></a> +</span><span id="L-785"><a href="#L-785"><span class="linenos"> 785</span></a><span class="k">def</span> <span class="nf">min_or_least</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-786"><a href="#L-786"><span class="linenos"> 786</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"LEAST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MIN"</span> +</span><span id="L-787"><a href="#L-787"><span class="linenos"> 787</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> </span><span id="L-788"><a href="#L-788"><span class="linenos"> 788</span></a> -</span><span id="L-789"><a href="#L-789"><span class="linenos"> 789</span></a><span class="k">def</span> <span class="nf">datestrtodate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateStrToDate</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-790"><a href="#L-790"><span class="linenos"> 790</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> -</span><span id="L-791"><a href="#L-791"><span class="linenos"> 791</span></a> -</span><span id="L-792"><a href="#L-792"><span class="linenos"> 792</span></a> -</span><span id="L-793"><a href="#L-793"><span class="linenos"> 793</span></a><span class="c1"># Used for Presto and Duckdb which use functions that don't support charset, and assume utf-8</span> -</span><span id="L-794"><a href="#L-794"><span class="linenos"> 794</span></a><span class="k">def</span> <span class="nf">encode_decode_sql</span><span class="p">(</span> -</span><span id="L-795"><a href="#L-795"><span class="linenos"> 795</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">replace</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="L-796"><a href="#L-796"><span class="linenos"> 796</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-797"><a href="#L-797"><span class="linenos"> 797</span></a> <span class="n">charset</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"charset"</span><span class="p">)</span> -</span><span id="L-798"><a href="#L-798"><span class="linenos"> 798</span></a> <span class="k">if</span> <span class="n">charset</span> <span class="ow">and</span> <span class="n">charset</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="o">!=</span> <span class="s2">"utf-8"</span><span class="p">:</span> -</span><span id="L-799"><a href="#L-799"><span class="linenos"> 799</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Expected utf-8 character set, got </span><span class="si">{</span><span class="n">charset</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> -</span><span id="L-800"><a href="#L-800"><span class="linenos"> 800</span></a> -</span><span id="L-801"><a href="#L-801"><span class="linenos"> 801</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"replace"</span><span class="p">)</span> <span class="k">if</span> <span class="n">replace</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> -</span><span id="L-802"><a href="#L-802"><span class="linenos"> 802</span></a> +</span><span id="L-789"><a href="#L-789"><span class="linenos"> 789</span></a> +</span><span id="L-790"><a href="#L-790"><span class="linenos"> 790</span></a><span class="k">def</span> <span class="nf">max_or_greatest</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-791"><a href="#L-791"><span class="linenos"> 791</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"GREATEST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MAX"</span> +</span><span id="L-792"><a href="#L-792"><span class="linenos"> 792</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> +</span><span id="L-793"><a href="#L-793"><span class="linenos"> 793</span></a> +</span><span id="L-794"><a href="#L-794"><span class="linenos"> 794</span></a> +</span><span id="L-795"><a href="#L-795"><span class="linenos"> 795</span></a><span class="k">def</span> <span class="nf">count_if_to_sum</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">CountIf</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-796"><a href="#L-796"><span class="linenos"> 796</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> +</span><span id="L-797"><a href="#L-797"><span class="linenos"> 797</span></a> +</span><span id="L-798"><a href="#L-798"><span class="linenos"> 798</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">):</span> +</span><span id="L-799"><a href="#L-799"><span class="linenos"> 799</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> +</span><span id="L-800"><a href="#L-800"><span class="linenos"> 800</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"DISTINCT is not supported when converting COUNT_IF to SUM"</span><span class="p">)</span> +</span><span id="L-801"><a href="#L-801"><span class="linenos"> 801</span></a> +</span><span id="L-802"><a href="#L-802"><span class="linenos"> 802</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"sum"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"if"</span><span class="p">,</span> <span class="n">cond</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> </span><span id="L-803"><a href="#L-803"><span class="linenos"> 803</span></a> -</span><span id="L-804"><a href="#L-804"><span class="linenos"> 804</span></a><span class="k">def</span> <span class="nf">min_or_least</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-805"><a href="#L-805"><span class="linenos"> 805</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"LEAST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MIN"</span> -</span><span id="L-806"><a href="#L-806"><span class="linenos"> 806</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> -</span><span id="L-807"><a href="#L-807"><span class="linenos"> 807</span></a> -</span><span id="L-808"><a href="#L-808"><span class="linenos"> 808</span></a> -</span><span id="L-809"><a href="#L-809"><span class="linenos"> 809</span></a><span class="k">def</span> <span class="nf">max_or_greatest</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-810"><a href="#L-810"><span class="linenos"> 810</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"GREATEST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MAX"</span> -</span><span id="L-811"><a href="#L-811"><span class="linenos"> 811</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> -</span><span id="L-812"><a href="#L-812"><span class="linenos"> 812</span></a> -</span><span id="L-813"><a href="#L-813"><span class="linenos"> 813</span></a> -</span><span id="L-814"><a href="#L-814"><span class="linenos"> 814</span></a><span class="k">def</span> <span class="nf">count_if_to_sum</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">CountIf</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-815"><a href="#L-815"><span class="linenos"> 815</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> -</span><span id="L-816"><a href="#L-816"><span class="linenos"> 816</span></a> -</span><span id="L-817"><a href="#L-817"><span class="linenos"> 817</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">):</span> -</span><span id="L-818"><a href="#L-818"><span class="linenos"> 818</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> -</span><span id="L-819"><a href="#L-819"><span class="linenos"> 819</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"DISTINCT is not supported when converting COUNT_IF to SUM"</span><span class="p">)</span> +</span><span id="L-804"><a href="#L-804"><span class="linenos"> 804</span></a> +</span><span id="L-805"><a href="#L-805"><span class="linenos"> 805</span></a><span class="k">def</span> <span class="nf">trim_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Trim</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-806"><a href="#L-806"><span class="linenos"> 806</span></a> <span class="n">target</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> +</span><span id="L-807"><a href="#L-807"><span class="linenos"> 807</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"position"</span><span class="p">)</span> +</span><span id="L-808"><a href="#L-808"><span class="linenos"> 808</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">)</span> +</span><span id="L-809"><a href="#L-809"><span class="linenos"> 809</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"collation"</span><span class="p">)</span> +</span><span id="L-810"><a href="#L-810"><span class="linenos"> 810</span></a> +</span><span id="L-811"><a href="#L-811"><span class="linenos"> 811</span></a> <span class="c1"># Use TRIM/LTRIM/RTRIM syntax if the expression isn't database-specific</span> +</span><span id="L-812"><a href="#L-812"><span class="linenos"> 812</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">remove_chars</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">collation</span><span class="p">:</span> +</span><span id="L-813"><a href="#L-813"><span class="linenos"> 813</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">trim_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-814"><a href="#L-814"><span class="linenos"> 814</span></a> +</span><span id="L-815"><a href="#L-815"><span class="linenos"> 815</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="L-816"><a href="#L-816"><span class="linenos"> 816</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">remove_chars</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="L-817"><a href="#L-817"><span class="linenos"> 817</span></a> <span class="n">from_part</span> <span class="o">=</span> <span class="s2">"FROM "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="ow">or</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="L-818"><a href="#L-818"><span class="linenos"> 818</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">" COLLATE </span><span class="si">{</span><span class="n">collation</span><span class="si">}</span><span class="s2">"</span> <span class="k">if</span> <span class="n">collation</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="L-819"><a href="#L-819"><span class="linenos"> 819</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"TRIM(</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}{</span><span class="n">remove_chars</span><span class="si">}{</span><span class="n">from_part</span><span class="si">}{</span><span class="n">target</span><span class="si">}{</span><span class="n">collation</span><span class="si">}</span><span class="s2">)"</span> </span><span id="L-820"><a href="#L-820"><span class="linenos"> 820</span></a> -</span><span id="L-821"><a href="#L-821"><span class="linenos"> 821</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"sum"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"if"</span><span class="p">,</span> <span class="n">cond</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> -</span><span id="L-822"><a href="#L-822"><span class="linenos"> 822</span></a> -</span><span id="L-823"><a href="#L-823"><span class="linenos"> 823</span></a> -</span><span id="L-824"><a href="#L-824"><span class="linenos"> 824</span></a><span class="k">def</span> <span class="nf">trim_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Trim</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-825"><a href="#L-825"><span class="linenos"> 825</span></a> <span class="n">target</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> -</span><span id="L-826"><a href="#L-826"><span class="linenos"> 826</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"position"</span><span class="p">)</span> -</span><span id="L-827"><a href="#L-827"><span class="linenos"> 827</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">)</span> -</span><span id="L-828"><a href="#L-828"><span class="linenos"> 828</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"collation"</span><span class="p">)</span> +</span><span id="L-821"><a href="#L-821"><span class="linenos"> 821</span></a> +</span><span id="L-822"><a href="#L-822"><span class="linenos"> 822</span></a><span class="k">def</span> <span class="nf">str_to_time_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-823"><a href="#L-823"><span class="linenos"> 823</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"STRPTIME"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">format_time</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> +</span><span id="L-824"><a href="#L-824"><span class="linenos"> 824</span></a> +</span><span id="L-825"><a href="#L-825"><span class="linenos"> 825</span></a> +</span><span id="L-826"><a href="#L-826"><span class="linenos"> 826</span></a><span class="k">def</span> <span class="nf">concat_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Concat</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-827"><a href="#L-827"><span class="linenos"> 827</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">reduce</span><span class="p">(</span><span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">))</span> +</span><span id="L-828"><a href="#L-828"><span class="linenos"> 828</span></a> </span><span id="L-829"><a href="#L-829"><span class="linenos"> 829</span></a> -</span><span id="L-830"><a href="#L-830"><span class="linenos"> 830</span></a> <span class="c1"># Use TRIM/LTRIM/RTRIM syntax if the expression isn't database-specific</span> -</span><span id="L-831"><a href="#L-831"><span class="linenos"> 831</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">remove_chars</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">collation</span><span class="p">:</span> -</span><span id="L-832"><a href="#L-832"><span class="linenos"> 832</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">trim_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-833"><a href="#L-833"><span class="linenos"> 833</span></a> -</span><span id="L-834"><a href="#L-834"><span class="linenos"> 834</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="L-835"><a href="#L-835"><span class="linenos"> 835</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">remove_chars</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="L-836"><a href="#L-836"><span class="linenos"> 836</span></a> <span class="n">from_part</span> <span class="o">=</span> <span class="s2">"FROM "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="ow">or</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="L-837"><a href="#L-837"><span class="linenos"> 837</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">" COLLATE </span><span class="si">{</span><span class="n">collation</span><span class="si">}</span><span class="s2">"</span> <span class="k">if</span> <span class="n">collation</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="L-838"><a href="#L-838"><span class="linenos"> 838</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"TRIM(</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}{</span><span class="n">remove_chars</span><span class="si">}{</span><span class="n">from_part</span><span class="si">}{</span><span class="n">target</span><span class="si">}{</span><span class="n">collation</span><span class="si">}</span><span class="s2">)"</span> +</span><span id="L-830"><a href="#L-830"><span class="linenos"> 830</span></a><span class="k">def</span> <span class="nf">concat_ws_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ConcatWs</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-831"><a href="#L-831"><span class="linenos"> 831</span></a> <span class="n">delim</span><span class="p">,</span> <span class="o">*</span><span class="n">rest_args</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> +</span><span id="L-832"><a href="#L-832"><span class="linenos"> 832</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="L-833"><a href="#L-833"><span class="linenos"> 833</span></a> <span class="n">reduce</span><span class="p">(</span> +</span><span id="L-834"><a href="#L-834"><span class="linenos"> 834</span></a> <span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">delim</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">)),</span> +</span><span id="L-835"><a href="#L-835"><span class="linenos"> 835</span></a> <span class="n">rest_args</span><span class="p">,</span> +</span><span id="L-836"><a href="#L-836"><span class="linenos"> 836</span></a> <span class="p">)</span> +</span><span id="L-837"><a href="#L-837"><span class="linenos"> 837</span></a> <span class="p">)</span> +</span><span id="L-838"><a href="#L-838"><span class="linenos"> 838</span></a> </span><span id="L-839"><a href="#L-839"><span class="linenos"> 839</span></a> -</span><span id="L-840"><a href="#L-840"><span class="linenos"> 840</span></a> -</span><span id="L-841"><a href="#L-841"><span class="linenos"> 841</span></a><span class="k">def</span> <span class="nf">str_to_time_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-842"><a href="#L-842"><span class="linenos"> 842</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"STRPTIME"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">format_time</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> -</span><span id="L-843"><a href="#L-843"><span class="linenos"> 843</span></a> +</span><span id="L-840"><a href="#L-840"><span class="linenos"> 840</span></a><span class="k">def</span> <span class="nf">regexp_extract_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpExtract</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-841"><a href="#L-841"><span class="linenos"> 841</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">)))</span> +</span><span id="L-842"><a href="#L-842"><span class="linenos"> 842</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> +</span><span id="L-843"><a href="#L-843"><span class="linenos"> 843</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_EXTRACT does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> </span><span id="L-844"><a href="#L-844"><span class="linenos"> 844</span></a> -</span><span id="L-845"><a href="#L-845"><span class="linenos"> 845</span></a><span class="k">def</span> <span class="nf">concat_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Concat</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-846"><a href="#L-846"><span class="linenos"> 846</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">reduce</span><span class="p">(</span><span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">))</span> -</span><span id="L-847"><a href="#L-847"><span class="linenos"> 847</span></a> +</span><span id="L-845"><a href="#L-845"><span class="linenos"> 845</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="L-846"><a href="#L-846"><span class="linenos"> 846</span></a> <span class="s2">"REGEXP_EXTRACT"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"group"</span><span class="p">)</span> +</span><span id="L-847"><a href="#L-847"><span class="linenos"> 847</span></a> <span class="p">)</span> </span><span id="L-848"><a href="#L-848"><span class="linenos"> 848</span></a> -</span><span id="L-849"><a href="#L-849"><span class="linenos"> 849</span></a><span class="k">def</span> <span class="nf">concat_ws_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ConcatWs</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-850"><a href="#L-850"><span class="linenos"> 850</span></a> <span class="n">delim</span><span class="p">,</span> <span class="o">*</span><span class="n">rest_args</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> -</span><span id="L-851"><a href="#L-851"><span class="linenos"> 851</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="L-852"><a href="#L-852"><span class="linenos"> 852</span></a> <span class="n">reduce</span><span class="p">(</span> -</span><span id="L-853"><a href="#L-853"><span class="linenos"> 853</span></a> <span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">delim</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">)),</span> -</span><span id="L-854"><a href="#L-854"><span class="linenos"> 854</span></a> <span class="n">rest_args</span><span class="p">,</span> -</span><span id="L-855"><a href="#L-855"><span class="linenos"> 855</span></a> <span class="p">)</span> -</span><span id="L-856"><a href="#L-856"><span class="linenos"> 856</span></a> <span class="p">)</span> -</span><span id="L-857"><a href="#L-857"><span class="linenos"> 857</span></a> -</span><span id="L-858"><a href="#L-858"><span class="linenos"> 858</span></a> -</span><span id="L-859"><a href="#L-859"><span class="linenos"> 859</span></a><span class="k">def</span> <span class="nf">regexp_extract_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpExtract</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-860"><a href="#L-860"><span class="linenos"> 860</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">)))</span> -</span><span id="L-861"><a href="#L-861"><span class="linenos"> 861</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> -</span><span id="L-862"><a href="#L-862"><span class="linenos"> 862</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_EXTRACT does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> -</span><span id="L-863"><a href="#L-863"><span class="linenos"> 863</span></a> -</span><span id="L-864"><a href="#L-864"><span class="linenos"> 864</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="L-865"><a href="#L-865"><span class="linenos"> 865</span></a> <span class="s2">"REGEXP_EXTRACT"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"group"</span><span class="p">)</span> -</span><span id="L-866"><a href="#L-866"><span class="linenos"> 866</span></a> <span class="p">)</span> -</span><span id="L-867"><a href="#L-867"><span class="linenos"> 867</span></a> -</span><span id="L-868"><a href="#L-868"><span class="linenos"> 868</span></a> -</span><span id="L-869"><a href="#L-869"><span class="linenos"> 869</span></a><span class="k">def</span> <span class="nf">regexp_replace_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpReplace</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-870"><a href="#L-870"><span class="linenos"> 870</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span> -</span><span id="L-871"><a href="#L-871"><span class="linenos"> 871</span></a> <span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">,</span> <span class="s2">"modifiers"</span><span class="p">))</span> -</span><span id="L-872"><a href="#L-872"><span class="linenos"> 872</span></a> <span class="p">)</span> -</span><span id="L-873"><a href="#L-873"><span class="linenos"> 873</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> -</span><span id="L-874"><a href="#L-874"><span class="linenos"> 874</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_REPLACE does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> -</span><span id="L-875"><a href="#L-875"><span class="linenos"> 875</span></a> -</span><span id="L-876"><a href="#L-876"><span class="linenos"> 876</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="L-877"><a href="#L-877"><span class="linenos"> 877</span></a> <span class="s2">"REGEXP_REPLACE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"replacement"</span><span class="p">]</span> -</span><span id="L-878"><a href="#L-878"><span class="linenos"> 878</span></a> <span class="p">)</span> -</span><span id="L-879"><a href="#L-879"><span class="linenos"> 879</span></a> -</span><span id="L-880"><a href="#L-880"><span class="linenos"> 880</span></a> -</span><span id="L-881"><a href="#L-881"><span class="linenos"> 881</span></a><span class="k">def</span> <span class="nf">pivot_column_names</span><span class="p">(</span><span class="n">aggregations</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="n">dialect</span><span class="p">:</span> <span class="n">DialectType</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> -</span><span id="L-882"><a href="#L-882"><span class="linenos"> 882</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="L-883"><a href="#L-883"><span class="linenos"> 883</span></a> <span class="k">for</span> <span class="n">agg</span> <span class="ow">in</span> <span class="n">aggregations</span><span class="p">:</span> -</span><span id="L-884"><a href="#L-884"><span class="linenos"> 884</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">agg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Alias</span><span class="p">):</span> -</span><span id="L-885"><a href="#L-885"><span class="linenos"> 885</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg</span><span class="o">.</span><span class="n">alias</span><span class="p">)</span> -</span><span id="L-886"><a href="#L-886"><span class="linenos"> 886</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-887"><a href="#L-887"><span class="linenos"> 887</span></a><span class="w"> </span><span class="sd">"""</span> -</span><span id="L-888"><a href="#L-888"><span class="linenos"> 888</span></a><span class="sd"> This case corresponds to aggregations without aliases being used as suffixes</span> -</span><span id="L-889"><a href="#L-889"><span class="linenos"> 889</span></a><span class="sd"> (e.g. col_avg(foo)). We need to unquote identifiers because they're going to</span> -</span><span id="L-890"><a href="#L-890"><span class="linenos"> 890</span></a><span class="sd"> be quoted in the base parser's `_parse_pivot` method, due to `to_identifier`.</span> -</span><span id="L-891"><a href="#L-891"><span class="linenos"> 891</span></a><span class="sd"> Otherwise, we'd end up with `col_avg(`foo`)` (notice the double quotes).</span> -</span><span id="L-892"><a href="#L-892"><span class="linenos"> 892</span></a><span class="sd"> """</span> -</span><span id="L-893"><a href="#L-893"><span class="linenos"> 893</span></a> <span class="n">agg_all_unquoted</span> <span class="o">=</span> <span class="n">agg</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> -</span><span id="L-894"><a href="#L-894"><span class="linenos"> 894</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> -</span><span id="L-895"><a href="#L-895"><span class="linenos"> 895</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">node</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">quoted</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span> -</span><span id="L-896"><a href="#L-896"><span class="linenos"> 896</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">)</span> -</span><span id="L-897"><a href="#L-897"><span class="linenos"> 897</span></a> <span class="k">else</span> <span class="n">node</span> -</span><span id="L-898"><a href="#L-898"><span class="linenos"> 898</span></a> <span class="p">)</span> -</span><span id="L-899"><a href="#L-899"><span class="linenos"> 899</span></a> <span class="p">)</span> -</span><span id="L-900"><a href="#L-900"><span class="linenos"> 900</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg_all_unquoted</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">dialect</span><span class="o">=</span><span class="n">dialect</span><span class="p">,</span> <span class="n">normalize_functions</span><span class="o">=</span><span class="s2">"lower"</span><span class="p">))</span> -</span><span id="L-901"><a href="#L-901"><span class="linenos"> 901</span></a> -</span><span id="L-902"><a href="#L-902"><span class="linenos"> 902</span></a> <span class="k">return</span> <span class="n">names</span> +</span><span id="L-849"><a href="#L-849"><span class="linenos"> 849</span></a> +</span><span id="L-850"><a href="#L-850"><span class="linenos"> 850</span></a><span class="k">def</span> <span class="nf">regexp_replace_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpReplace</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-851"><a href="#L-851"><span class="linenos"> 851</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span> +</span><span id="L-852"><a href="#L-852"><span class="linenos"> 852</span></a> <span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">,</span> <span class="s2">"modifiers"</span><span class="p">))</span> +</span><span id="L-853"><a href="#L-853"><span class="linenos"> 853</span></a> <span class="p">)</span> +</span><span id="L-854"><a href="#L-854"><span class="linenos"> 854</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> +</span><span id="L-855"><a href="#L-855"><span class="linenos"> 855</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_REPLACE does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> +</span><span id="L-856"><a href="#L-856"><span class="linenos"> 856</span></a> +</span><span id="L-857"><a href="#L-857"><span class="linenos"> 857</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="L-858"><a href="#L-858"><span class="linenos"> 858</span></a> <span class="s2">"REGEXP_REPLACE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"replacement"</span><span class="p">]</span> +</span><span id="L-859"><a href="#L-859"><span class="linenos"> 859</span></a> <span class="p">)</span> +</span><span id="L-860"><a href="#L-860"><span class="linenos"> 860</span></a> +</span><span id="L-861"><a href="#L-861"><span class="linenos"> 861</span></a> +</span><span id="L-862"><a href="#L-862"><span class="linenos"> 862</span></a><span class="k">def</span> <span class="nf">pivot_column_names</span><span class="p">(</span><span class="n">aggregations</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="n">dialect</span><span class="p">:</span> <span class="n">DialectType</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> +</span><span id="L-863"><a href="#L-863"><span class="linenos"> 863</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="L-864"><a href="#L-864"><span class="linenos"> 864</span></a> <span class="k">for</span> <span class="n">agg</span> <span class="ow">in</span> <span class="n">aggregations</span><span class="p">:</span> +</span><span id="L-865"><a href="#L-865"><span class="linenos"> 865</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">agg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Alias</span><span class="p">):</span> +</span><span id="L-866"><a href="#L-866"><span class="linenos"> 866</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg</span><span class="o">.</span><span class="n">alias</span><span class="p">)</span> +</span><span id="L-867"><a href="#L-867"><span class="linenos"> 867</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-868"><a href="#L-868"><span class="linenos"> 868</span></a><span class="w"> </span><span class="sd">"""</span> +</span><span id="L-869"><a href="#L-869"><span class="linenos"> 869</span></a><span class="sd"> This case corresponds to aggregations without aliases being used as suffixes</span> +</span><span id="L-870"><a href="#L-870"><span class="linenos"> 870</span></a><span class="sd"> (e.g. col_avg(foo)). We need to unquote identifiers because they're going to</span> +</span><span id="L-871"><a href="#L-871"><span class="linenos"> 871</span></a><span class="sd"> be quoted in the base parser's `_parse_pivot` method, due to `to_identifier`.</span> +</span><span id="L-872"><a href="#L-872"><span class="linenos"> 872</span></a><span class="sd"> Otherwise, we'd end up with `col_avg(`foo`)` (notice the double quotes).</span> +</span><span id="L-873"><a href="#L-873"><span class="linenos"> 873</span></a><span class="sd"> """</span> +</span><span id="L-874"><a href="#L-874"><span class="linenos"> 874</span></a> <span class="n">agg_all_unquoted</span> <span class="o">=</span> <span class="n">agg</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> +</span><span id="L-875"><a href="#L-875"><span class="linenos"> 875</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> +</span><span id="L-876"><a href="#L-876"><span class="linenos"> 876</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">node</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">quoted</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span> +</span><span id="L-877"><a href="#L-877"><span class="linenos"> 877</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">)</span> +</span><span id="L-878"><a href="#L-878"><span class="linenos"> 878</span></a> <span class="k">else</span> <span class="n">node</span> +</span><span id="L-879"><a href="#L-879"><span class="linenos"> 879</span></a> <span class="p">)</span> +</span><span id="L-880"><a href="#L-880"><span class="linenos"> 880</span></a> <span class="p">)</span> +</span><span id="L-881"><a href="#L-881"><span class="linenos"> 881</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg_all_unquoted</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">dialect</span><span class="o">=</span><span class="n">dialect</span><span class="p">,</span> <span class="n">normalize_functions</span><span class="o">=</span><span class="s2">"lower"</span><span class="p">))</span> +</span><span id="L-882"><a href="#L-882"><span class="linenos"> 882</span></a> +</span><span id="L-883"><a href="#L-883"><span class="linenos"> 883</span></a> <span class="k">return</span> <span class="n">names</span> +</span><span id="L-884"><a href="#L-884"><span class="linenos"> 884</span></a> +</span><span id="L-885"><a href="#L-885"><span class="linenos"> 885</span></a> +</span><span id="L-886"><a href="#L-886"><span class="linenos"> 886</span></a><span class="k">def</span> <span class="nf">binary_from_function</span><span class="p">(</span><span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">B</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">B</span><span class="p">]:</span> +</span><span id="L-887"><a href="#L-887"><span class="linenos"> 887</span></a> <span class="k">return</span> <span class="k">lambda</span> <span class="n">args</span><span class="p">:</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span> +</span><span id="L-888"><a href="#L-888"><span class="linenos"> 888</span></a> +</span><span id="L-889"><a href="#L-889"><span class="linenos"> 889</span></a> +</span><span id="L-890"><a href="#L-890"><span class="linenos"> 890</span></a><span class="c1"># Used to represent DATE_TRUNC in Doris, Postgres and Starrocks dialects</span> +</span><span id="L-891"><a href="#L-891"><span class="linenos"> 891</span></a><span class="k">def</span> <span class="nf">parse_timestamp_trunc</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> +</span><span id="L-892"><a href="#L-892"><span class="linenos"> 892</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> +</span><span id="L-893"><a href="#L-893"><span class="linenos"> 893</span></a> +</span><span id="L-894"><a href="#L-894"><span class="linenos"> 894</span></a> +</span><span id="L-895"><a href="#L-895"><span class="linenos"> 895</span></a><span class="k">def</span> <span class="nf">any_value_to_max_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">AnyValue</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-896"><a href="#L-896"><span class="linenos"> 896</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"MAX"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="L-897"><a href="#L-897"><span class="linenos"> 897</span></a> +</span><span id="L-898"><a href="#L-898"><span class="linenos"> 898</span></a> +</span><span id="L-899"><a href="#L-899"><span class="linenos"> 899</span></a><span class="k">def</span> <span class="nf">bool_xor_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Xor</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-900"><a href="#L-900"><span class="linenos"> 900</span></a> <span class="n">a</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">)</span> +</span><span id="L-901"><a href="#L-901"><span class="linenos"> 901</span></a> <span class="n">b</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">right</span><span class="p">)</span> +</span><span id="L-902"><a href="#L-902"><span class="linenos"> 902</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"(</span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2"> AND (NOT </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)) OR ((NOT </span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2">) AND </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)"</span> </span><span id="L-903"><a href="#L-903"><span class="linenos"> 903</span></a> </span><span id="L-904"><a href="#L-904"><span class="linenos"> 904</span></a> -</span><span id="L-905"><a href="#L-905"><span class="linenos"> 905</span></a><span class="k">def</span> <span class="nf">binary_from_function</span><span class="p">(</span><span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">B</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">B</span><span class="p">]:</span> -</span><span id="L-906"><a href="#L-906"><span class="linenos"> 906</span></a> <span class="k">return</span> <span class="k">lambda</span> <span class="n">args</span><span class="p">:</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span> -</span><span id="L-907"><a href="#L-907"><span class="linenos"> 907</span></a> -</span><span id="L-908"><a href="#L-908"><span class="linenos"> 908</span></a> -</span><span id="L-909"><a href="#L-909"><span class="linenos"> 909</span></a><span class="c1"># Used to represent DATE_TRUNC in Doris, Postgres and Starrocks dialects</span> -</span><span id="L-910"><a href="#L-910"><span class="linenos"> 910</span></a><span class="k">def</span> <span class="nf">parse_timestamp_trunc</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> -</span><span id="L-911"><a href="#L-911"><span class="linenos"> 911</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> -</span><span id="L-912"><a href="#L-912"><span class="linenos"> 912</span></a> +</span><span id="L-905"><a href="#L-905"><span class="linenos"> 905</span></a><span class="k">def</span> <span class="nf">is_parse_json</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span> +</span><span id="L-906"><a href="#L-906"><span class="linenos"> 906</span></a> <span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ParseJSON</span><span class="p">)</span> <span class="ow">or</span> <span class="p">(</span> +</span><span id="L-907"><a href="#L-907"><span class="linenos"> 907</span></a> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"json"</span><span class="p">)</span> +</span><span id="L-908"><a href="#L-908"><span class="linenos"> 908</span></a> <span class="p">)</span> +</span><span id="L-909"><a href="#L-909"><span class="linenos"> 909</span></a> +</span><span id="L-910"><a href="#L-910"><span class="linenos"> 910</span></a> +</span><span id="L-911"><a href="#L-911"><span class="linenos"> 911</span></a><span class="k">def</span> <span class="nf">isnull_to_is_null</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> +</span><span id="L-912"><a href="#L-912"><span class="linenos"> 912</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">Paren</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Is</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">null</span><span class="p">()))</span> </span><span id="L-913"><a href="#L-913"><span class="linenos"> 913</span></a> -</span><span id="L-914"><a href="#L-914"><span class="linenos"> 914</span></a><span class="k">def</span> <span class="nf">any_value_to_max_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">AnyValue</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-915"><a href="#L-915"><span class="linenos"> 915</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"MAX"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="L-916"><a href="#L-916"><span class="linenos"> 916</span></a> -</span><span id="L-917"><a href="#L-917"><span class="linenos"> 917</span></a> -</span><span id="L-918"><a href="#L-918"><span class="linenos"> 918</span></a><span class="k">def</span> <span class="nf">bool_xor_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Xor</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-919"><a href="#L-919"><span class="linenos"> 919</span></a> <span class="n">a</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">)</span> -</span><span id="L-920"><a href="#L-920"><span class="linenos"> 920</span></a> <span class="n">b</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">right</span><span class="p">)</span> -</span><span id="L-921"><a href="#L-921"><span class="linenos"> 921</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"(</span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2"> AND (NOT </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)) OR ((NOT </span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2">) AND </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)"</span> +</span><span id="L-914"><a href="#L-914"><span class="linenos"> 914</span></a> +</span><span id="L-915"><a href="#L-915"><span class="linenos"> 915</span></a><span class="k">def</span> <span class="nf">generatedasidentitycolumnconstraint_sql</span><span class="p">(</span> +</span><span id="L-916"><a href="#L-916"><span class="linenos"> 916</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">GeneratedAsIdentityColumnConstraint</span> +</span><span id="L-917"><a href="#L-917"><span class="linenos"> 917</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-918"><a href="#L-918"><span class="linenos"> 918</span></a> <span class="n">start</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"start"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> +</span><span id="L-919"><a href="#L-919"><span class="linenos"> 919</span></a> <span class="n">increment</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"increment"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> +</span><span id="L-920"><a href="#L-920"><span class="linenos"> 920</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"IDENTITY(</span><span class="si">{</span><span class="n">start</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="n">increment</span><span class="si">}</span><span class="s2">)"</span> +</span><span id="L-921"><a href="#L-921"><span class="linenos"> 921</span></a> </span><span id="L-922"><a href="#L-922"><span class="linenos"> 922</span></a> -</span><span id="L-923"><a href="#L-923"><span class="linenos"> 923</span></a> -</span><span id="L-924"><a href="#L-924"><span class="linenos"> 924</span></a><span class="k">def</span> <span class="nf">is_parse_json</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span> -</span><span id="L-925"><a href="#L-925"><span class="linenos"> 925</span></a> <span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ParseJSON</span><span class="p">)</span> <span class="ow">or</span> <span class="p">(</span> -</span><span id="L-926"><a href="#L-926"><span class="linenos"> 926</span></a> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"json"</span><span class="p">)</span> -</span><span id="L-927"><a href="#L-927"><span class="linenos"> 927</span></a> <span class="p">)</span> -</span><span id="L-928"><a href="#L-928"><span class="linenos"> 928</span></a> +</span><span id="L-923"><a href="#L-923"><span class="linenos"> 923</span></a><span class="k">def</span> <span class="nf">arg_max_or_min_no_count</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="L-924"><a href="#L-924"><span class="linenos"> 924</span></a> <span class="k">def</span> <span class="nf">_arg_max_or_min_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-925"><a href="#L-925"><span class="linenos"> 925</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"count"</span><span class="p">):</span> +</span><span id="L-926"><a href="#L-926"><span class="linenos"> 926</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Only two arguments are supported in function </span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> +</span><span id="L-927"><a href="#L-927"><span class="linenos"> 927</span></a> +</span><span id="L-928"><a href="#L-928"><span class="linenos"> 928</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> </span><span id="L-929"><a href="#L-929"><span class="linenos"> 929</span></a> -</span><span id="L-930"><a href="#L-930"><span class="linenos"> 930</span></a><span class="k">def</span> <span class="nf">isnull_to_is_null</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> -</span><span id="L-931"><a href="#L-931"><span class="linenos"> 931</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">Paren</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Is</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">null</span><span class="p">()))</span> +</span><span id="L-930"><a href="#L-930"><span class="linenos"> 930</span></a> <span class="k">return</span> <span class="n">_arg_max_or_min_sql</span> +</span><span id="L-931"><a href="#L-931"><span class="linenos"> 931</span></a> </span><span id="L-932"><a href="#L-932"><span class="linenos"> 932</span></a> -</span><span id="L-933"><a href="#L-933"><span class="linenos"> 933</span></a> -</span><span id="L-934"><a href="#L-934"><span class="linenos"> 934</span></a><span class="k">def</span> <span class="nf">generatedasidentitycolumnconstraint_sql</span><span class="p">(</span> -</span><span id="L-935"><a href="#L-935"><span class="linenos"> 935</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">GeneratedAsIdentityColumnConstraint</span> -</span><span id="L-936"><a href="#L-936"><span class="linenos"> 936</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-937"><a href="#L-937"><span class="linenos"> 937</span></a> <span class="n">start</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"start"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> -</span><span id="L-938"><a href="#L-938"><span class="linenos"> 938</span></a> <span class="n">increment</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"increment"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> -</span><span id="L-939"><a href="#L-939"><span class="linenos"> 939</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"IDENTITY(</span><span class="si">{</span><span class="n">start</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="n">increment</span><span class="si">}</span><span class="s2">)"</span> -</span><span id="L-940"><a href="#L-940"><span class="linenos"> 940</span></a> +</span><span id="L-933"><a href="#L-933"><span class="linenos"> 933</span></a><span class="k">def</span> <span class="nf">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">:</span> +</span><span id="L-934"><a href="#L-934"><span class="linenos"> 934</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span> +</span><span id="L-935"><a href="#L-935"><span class="linenos"> 935</span></a> +</span><span id="L-936"><a href="#L-936"><span class="linenos"> 936</span></a> <span class="n">return_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">return_type</span> +</span><span id="L-937"><a href="#L-937"><span class="linenos"> 937</span></a> <span class="k">if</span> <span class="n">return_type</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DATE</span><span class="p">):</span> +</span><span id="L-938"><a href="#L-938"><span class="linenos"> 938</span></a> <span class="c1"># If we need to cast to a DATE, we cast to TIMESTAMP first to make sure we</span> +</span><span id="L-939"><a href="#L-939"><span class="linenos"> 939</span></a> <span class="c1"># can truncate timestamp strings, because some dialects can't cast them to DATE</span> +</span><span id="L-940"><a href="#L-940"><span class="linenos"> 940</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">)</span> </span><span id="L-941"><a href="#L-941"><span class="linenos"> 941</span></a> -</span><span id="L-942"><a href="#L-942"><span class="linenos"> 942</span></a><span class="k">def</span> <span class="nf">arg_max_or_min_no_count</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="L-943"><a href="#L-943"><span class="linenos"> 943</span></a> <span class="k">def</span> <span class="nf">_arg_max_or_min_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-944"><a href="#L-944"><span class="linenos"> 944</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"count"</span><span class="p">):</span> -</span><span id="L-945"><a href="#L-945"><span class="linenos"> 945</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Only two arguments are supported in function </span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> -</span><span id="L-946"><a href="#L-946"><span class="linenos"> 946</span></a> -</span><span id="L-947"><a href="#L-947"><span class="linenos"> 947</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-948"><a href="#L-948"><span class="linenos"> 948</span></a> -</span><span id="L-949"><a href="#L-949"><span class="linenos"> 949</span></a> <span class="k">return</span> <span class="n">_arg_max_or_min_sql</span> +</span><span id="L-942"><a href="#L-942"><span class="linenos"> 942</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">return_type</span><span class="p">))</span> +</span><span id="L-943"><a href="#L-943"><span class="linenos"> 943</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-944"><a href="#L-944"><span class="linenos"> 944</span></a> +</span><span id="L-945"><a href="#L-945"><span class="linenos"> 945</span></a> +</span><span id="L-946"><a href="#L-946"><span class="linenos"> 946</span></a><span class="k">def</span> <span class="nf">date_delta_sql</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">cast</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="L-947"><a href="#L-947"><span class="linenos"> 947</span></a> <span class="k">def</span> <span class="nf">_delta_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-948"><a href="#L-948"><span class="linenos"> 948</span></a> <span class="k">if</span> <span class="n">cast</span> <span class="ow">and</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">):</span> +</span><span id="L-949"><a href="#L-949"><span class="linenos"> 949</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> </span><span id="L-950"><a href="#L-950"><span class="linenos"> 950</span></a> -</span><span id="L-951"><a href="#L-951"><span class="linenos"> 951</span></a> -</span><span id="L-952"><a href="#L-952"><span class="linenos"> 952</span></a><span class="k">def</span> <span class="nf">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">:</span> -</span><span id="L-953"><a href="#L-953"><span class="linenos"> 953</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span> -</span><span id="L-954"><a href="#L-954"><span class="linenos"> 954</span></a> -</span><span id="L-955"><a href="#L-955"><span class="linenos"> 955</span></a> <span class="n">return_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">return_type</span> -</span><span id="L-956"><a href="#L-956"><span class="linenos"> 956</span></a> <span class="k">if</span> <span class="n">return_type</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DATE</span><span class="p">):</span> -</span><span id="L-957"><a href="#L-957"><span class="linenos"> 957</span></a> <span class="c1"># If we need to cast to a DATE, we cast to TIMESTAMP first to make sure we</span> -</span><span id="L-958"><a href="#L-958"><span class="linenos"> 958</span></a> <span class="c1"># can truncate timestamp strings, because some dialects can't cast them to DATE</span> -</span><span id="L-959"><a href="#L-959"><span class="linenos"> 959</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">)</span> +</span><span id="L-951"><a href="#L-951"><span class="linenos"> 951</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="L-952"><a href="#L-952"><span class="linenos"> 952</span></a> <span class="n">name</span><span class="p">,</span> +</span><span id="L-953"><a href="#L-953"><span class="linenos"> 953</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> +</span><span id="L-954"><a href="#L-954"><span class="linenos"> 954</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> +</span><span id="L-955"><a href="#L-955"><span class="linenos"> 955</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> +</span><span id="L-956"><a href="#L-956"><span class="linenos"> 956</span></a> <span class="p">)</span> +</span><span id="L-957"><a href="#L-957"><span class="linenos"> 957</span></a> +</span><span id="L-958"><a href="#L-958"><span class="linenos"> 958</span></a> <span class="k">return</span> <span class="n">_delta_sql</span> +</span><span id="L-959"><a href="#L-959"><span class="linenos"> 959</span></a> </span><span id="L-960"><a href="#L-960"><span class="linenos"> 960</span></a> -</span><span id="L-961"><a href="#L-961"><span class="linenos"> 961</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">return_type</span><span class="p">))</span> -</span><span id="L-962"><a href="#L-962"><span class="linenos"> 962</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-963"><a href="#L-963"><span class="linenos"> 963</span></a> -</span><span id="L-964"><a href="#L-964"><span class="linenos"> 964</span></a> -</span><span id="L-965"><a href="#L-965"><span class="linenos"> 965</span></a><span class="k">def</span> <span class="nf">date_delta_sql</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">cast</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="L-966"><a href="#L-966"><span class="linenos"> 966</span></a> <span class="k">def</span> <span class="nf">_delta_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-967"><a href="#L-967"><span class="linenos"> 967</span></a> <span class="k">if</span> <span class="n">cast</span> <span class="ow">and</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">):</span> -</span><span id="L-968"><a href="#L-968"><span class="linenos"> 968</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-969"><a href="#L-969"><span class="linenos"> 969</span></a> -</span><span id="L-970"><a href="#L-970"><span class="linenos"> 970</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="L-971"><a href="#L-971"><span class="linenos"> 971</span></a> <span class="n">name</span><span class="p">,</span> -</span><span id="L-972"><a href="#L-972"><span class="linenos"> 972</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> -</span><span id="L-973"><a href="#L-973"><span class="linenos"> 973</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> -</span><span id="L-974"><a href="#L-974"><span class="linenos"> 974</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> -</span><span id="L-975"><a href="#L-975"><span class="linenos"> 975</span></a> <span class="p">)</span> -</span><span id="L-976"><a href="#L-976"><span class="linenos"> 976</span></a> -</span><span id="L-977"><a href="#L-977"><span class="linenos"> 977</span></a> <span class="k">return</span> <span class="n">_delta_sql</span> -</span><span id="L-978"><a href="#L-978"><span class="linenos"> 978</span></a> -</span><span id="L-979"><a href="#L-979"><span class="linenos"> 979</span></a> -</span><span id="L-980"><a href="#L-980"><span class="linenos"> 980</span></a><span class="k">def</span> <span class="nf">no_last_day_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">LastDay</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-981"><a href="#L-981"><span class="linenos"> 981</span></a> <span class="n">trunc_curr_date</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_trunc"</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="L-982"><a href="#L-982"><span class="linenos"> 982</span></a> <span class="n">plus_one_month</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_add"</span><span class="p">,</span> <span class="n">trunc_curr_date</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">)</span> -</span><span id="L-983"><a href="#L-983"><span class="linenos"> 983</span></a> <span class="n">minus_one_day</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_sub"</span><span class="p">,</span> <span class="n">plus_one_month</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"day"</span><span class="p">)</span> -</span><span id="L-984"><a href="#L-984"><span class="linenos"> 984</span></a> -</span><span id="L-985"><a href="#L-985"><span class="linenos"> 985</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">minus_one_day</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> -</span><span id="L-986"><a href="#L-986"><span class="linenos"> 986</span></a> -</span><span id="L-987"><a href="#L-987"><span class="linenos"> 987</span></a> -</span><span id="L-988"><a href="#L-988"><span class="linenos"> 988</span></a><span class="k">def</span> <span class="nf">merge_without_target_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Merge</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-989"><a href="#L-989"><span class="linenos"> 989</span></a><span class="w"> </span><span class="sd">"""Remove table refs from columns in when statements."""</span> -</span><span id="L-990"><a href="#L-990"><span class="linenos"> 990</span></a> <span class="n">alias</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"alias"</span><span class="p">)</span> -</span><span id="L-991"><a href="#L-991"><span class="linenos"> 991</span></a> -</span><span id="L-992"><a href="#L-992"><span class="linenos"> 992</span></a> <span class="k">def</span> <span class="nf">normalize</span><span class="p">(</span><span class="n">identifier</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> -</span><span id="L-993"><a href="#L-993"><span class="linenos"> 993</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">normalize_identifier</span><span class="p">(</span><span class="n">identifier</span><span class="p">)</span><span class="o">.</span><span class="n">name</span> <span class="k">if</span> <span class="n">identifier</span> <span class="k">else</span> <span class="kc">None</span> -</span><span id="L-994"><a href="#L-994"><span class="linenos"> 994</span></a> -</span><span id="L-995"><a href="#L-995"><span class="linenos"> 995</span></a> <span class="n">targets</span> <span class="o">=</span> <span class="p">{</span><span class="n">normalize</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">this</span><span class="p">)}</span> -</span><span id="L-996"><a href="#L-996"><span class="linenos"> 996</span></a> -</span><span id="L-997"><a href="#L-997"><span class="linenos"> 997</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> -</span><span id="L-998"><a href="#L-998"><span class="linenos"> 998</span></a> <span class="n">targets</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="n">normalize</span><span class="p">(</span><span class="n">alias</span><span class="o">.</span><span class="n">this</span><span class="p">))</span> -</span><span id="L-999"><a href="#L-999"><span class="linenos"> 999</span></a> -</span><span id="L-1000"><a href="#L-1000"><span class="linenos">1000</span></a> <span class="k">for</span> <span class="n">when</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> -</span><span id="L-1001"><a href="#L-1001"><span class="linenos">1001</span></a> <span class="n">when</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> -</span><span id="L-1002"><a href="#L-1002"><span class="linenos">1002</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> -</span><span id="L-1003"><a href="#L-1003"><span class="linenos">1003</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">column</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="L-1004"><a href="#L-1004"><span class="linenos">1004</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Column</span><span class="p">)</span> <span class="ow">and</span> <span class="n">normalize</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"table"</span><span class="p">))</span> <span class="ow">in</span> <span class="n">targets</span> -</span><span id="L-1005"><a href="#L-1005"><span class="linenos">1005</span></a> <span class="k">else</span> <span class="n">node</span> -</span><span id="L-1006"><a href="#L-1006"><span class="linenos">1006</span></a> <span class="p">),</span> -</span><span id="L-1007"><a href="#L-1007"><span class="linenos">1007</span></a> <span class="n">copy</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> -</span><span id="L-1008"><a href="#L-1008"><span class="linenos">1008</span></a> <span class="p">)</span> -</span><span id="L-1009"><a href="#L-1009"><span class="linenos">1009</span></a> -</span><span id="L-1010"><a href="#L-1010"><span class="linenos">1010</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">merge_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-1011"><a href="#L-1011"><span class="linenos">1011</span></a> +</span><span id="L-961"><a href="#L-961"><span class="linenos"> 961</span></a><span class="k">def</span> <span class="nf">no_last_day_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">LastDay</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-962"><a href="#L-962"><span class="linenos"> 962</span></a> <span class="n">trunc_curr_date</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_trunc"</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="L-963"><a href="#L-963"><span class="linenos"> 963</span></a> <span class="n">plus_one_month</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_add"</span><span class="p">,</span> <span class="n">trunc_curr_date</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">)</span> +</span><span id="L-964"><a href="#L-964"><span class="linenos"> 964</span></a> <span class="n">minus_one_day</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_sub"</span><span class="p">,</span> <span class="n">plus_one_month</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"day"</span><span class="p">)</span> +</span><span id="L-965"><a href="#L-965"><span class="linenos"> 965</span></a> +</span><span id="L-966"><a href="#L-966"><span class="linenos"> 966</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">minus_one_day</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> +</span><span id="L-967"><a href="#L-967"><span class="linenos"> 967</span></a> +</span><span id="L-968"><a href="#L-968"><span class="linenos"> 968</span></a> +</span><span id="L-969"><a href="#L-969"><span class="linenos"> 969</span></a><span class="k">def</span> <span class="nf">merge_without_target_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Merge</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-970"><a href="#L-970"><span class="linenos"> 970</span></a><span class="w"> </span><span class="sd">"""Remove table refs from columns in when statements."""</span> +</span><span id="L-971"><a href="#L-971"><span class="linenos"> 971</span></a> <span class="n">alias</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"alias"</span><span class="p">)</span> +</span><span id="L-972"><a href="#L-972"><span class="linenos"> 972</span></a> +</span><span id="L-973"><a href="#L-973"><span class="linenos"> 973</span></a> <span class="k">def</span> <span class="nf">normalize</span><span class="p">(</span><span class="n">identifier</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> +</span><span id="L-974"><a href="#L-974"><span class="linenos"> 974</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">normalize_identifier</span><span class="p">(</span><span class="n">identifier</span><span class="p">)</span><span class="o">.</span><span class="n">name</span> <span class="k">if</span> <span class="n">identifier</span> <span class="k">else</span> <span class="kc">None</span> +</span><span id="L-975"><a href="#L-975"><span class="linenos"> 975</span></a> +</span><span id="L-976"><a href="#L-976"><span class="linenos"> 976</span></a> <span class="n">targets</span> <span class="o">=</span> <span class="p">{</span><span class="n">normalize</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">this</span><span class="p">)}</span> +</span><span id="L-977"><a href="#L-977"><span class="linenos"> 977</span></a> +</span><span id="L-978"><a href="#L-978"><span class="linenos"> 978</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> +</span><span id="L-979"><a href="#L-979"><span class="linenos"> 979</span></a> <span class="n">targets</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="n">normalize</span><span class="p">(</span><span class="n">alias</span><span class="o">.</span><span class="n">this</span><span class="p">))</span> +</span><span id="L-980"><a href="#L-980"><span class="linenos"> 980</span></a> +</span><span id="L-981"><a href="#L-981"><span class="linenos"> 981</span></a> <span class="k">for</span> <span class="n">when</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> +</span><span id="L-982"><a href="#L-982"><span class="linenos"> 982</span></a> <span class="n">when</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> +</span><span id="L-983"><a href="#L-983"><span class="linenos"> 983</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> +</span><span id="L-984"><a href="#L-984"><span class="linenos"> 984</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">column</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="L-985"><a href="#L-985"><span class="linenos"> 985</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Column</span><span class="p">)</span> <span class="ow">and</span> <span class="n">normalize</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"table"</span><span class="p">))</span> <span class="ow">in</span> <span class="n">targets</span> +</span><span id="L-986"><a href="#L-986"><span class="linenos"> 986</span></a> <span class="k">else</span> <span class="n">node</span> +</span><span id="L-987"><a href="#L-987"><span class="linenos"> 987</span></a> <span class="p">),</span> +</span><span id="L-988"><a href="#L-988"><span class="linenos"> 988</span></a> <span class="n">copy</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> +</span><span id="L-989"><a href="#L-989"><span class="linenos"> 989</span></a> <span class="p">)</span> +</span><span id="L-990"><a href="#L-990"><span class="linenos"> 990</span></a> +</span><span id="L-991"><a href="#L-991"><span class="linenos"> 991</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">merge_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-992"><a href="#L-992"><span class="linenos"> 992</span></a> +</span><span id="L-993"><a href="#L-993"><span class="linenos"> 993</span></a> +</span><span id="L-994"><a href="#L-994"><span class="linenos"> 994</span></a><span class="k">def</span> <span class="nf">parse_json_extract_path</span><span class="p">(</span> +</span><span id="L-995"><a href="#L-995"><span class="linenos"> 995</span></a> <span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">F</span><span class="p">],</span> <span class="n">zero_based_indexing</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="L-996"><a href="#L-996"><span class="linenos"> 996</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">F</span><span class="p">]:</span> +</span><span id="L-997"><a href="#L-997"><span class="linenos"> 997</span></a> <span class="k">def</span> <span class="nf">_parse_json_extract_path</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">F</span><span class="p">:</span> +</span><span id="L-998"><a href="#L-998"><span class="linenos"> 998</span></a> <span class="n">segments</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathRoot</span><span class="p">()]</span> +</span><span id="L-999"><a href="#L-999"><span class="linenos"> 999</span></a> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">:]:</span> +</span><span id="L-1000"><a href="#L-1000"><span class="linenos">1000</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">):</span> +</span><span id="L-1001"><a href="#L-1001"><span class="linenos">1001</span></a> <span class="c1"># We use the fallback parser because we can't really transpile non-literals safely</span> +</span><span id="L-1002"><a href="#L-1002"><span class="linenos">1002</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="o">.</span><span class="n">from_arg_list</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> +</span><span id="L-1003"><a href="#L-1003"><span class="linenos">1003</span></a> +</span><span id="L-1004"><a href="#L-1004"><span class="linenos">1004</span></a> <span class="n">text</span> <span class="o">=</span> <span class="n">arg</span><span class="o">.</span><span class="n">name</span> +</span><span id="L-1005"><a href="#L-1005"><span class="linenos">1005</span></a> <span class="k">if</span> <span class="n">is_int</span><span class="p">(</span><span class="n">text</span><span class="p">):</span> +</span><span id="L-1006"><a href="#L-1006"><span class="linenos">1006</span></a> <span class="n">index</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">text</span><span class="p">)</span> +</span><span id="L-1007"><a href="#L-1007"><span class="linenos">1007</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span> +</span><span id="L-1008"><a href="#L-1008"><span class="linenos">1008</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">index</span> <span class="k">if</span> <span class="n">zero_based_indexing</span> <span class="k">else</span> <span class="n">index</span> <span class="o">-</span> <span class="mi">1</span><span class="p">)</span> +</span><span id="L-1009"><a href="#L-1009"><span class="linenos">1009</span></a> <span class="p">)</span> +</span><span id="L-1010"><a href="#L-1010"><span class="linenos">1010</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-1011"><a href="#L-1011"><span class="linenos">1011</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">text</span><span class="p">))</span> </span><span id="L-1012"><a href="#L-1012"><span class="linenos">1012</span></a> -</span><span id="L-1013"><a href="#L-1013"><span class="linenos">1013</span></a><span class="k">def</span> <span class="nf">parse_json_extract_path</span><span class="p">(</span> -</span><span id="L-1014"><a href="#L-1014"><span class="linenos">1014</span></a> <span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">F</span><span class="p">],</span> <span class="n">zero_based_indexing</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="L-1015"><a href="#L-1015"><span class="linenos">1015</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">F</span><span class="p">]:</span> -</span><span id="L-1016"><a href="#L-1016"><span class="linenos">1016</span></a> <span class="k">def</span> <span class="nf">_parse_json_extract_path</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">F</span><span class="p">:</span> -</span><span id="L-1017"><a href="#L-1017"><span class="linenos">1017</span></a> <span class="n">segments</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathRoot</span><span class="p">()]</span> -</span><span id="L-1018"><a href="#L-1018"><span class="linenos">1018</span></a> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">:]:</span> -</span><span id="L-1019"><a href="#L-1019"><span class="linenos">1019</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">):</span> -</span><span id="L-1020"><a href="#L-1020"><span class="linenos">1020</span></a> <span class="c1"># We use the fallback parser because we can't really transpile non-literals safely</span> -</span><span id="L-1021"><a href="#L-1021"><span class="linenos">1021</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="o">.</span><span class="n">from_arg_list</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> -</span><span id="L-1022"><a href="#L-1022"><span class="linenos">1022</span></a> -</span><span id="L-1023"><a href="#L-1023"><span class="linenos">1023</span></a> <span class="n">text</span> <span class="o">=</span> <span class="n">arg</span><span class="o">.</span><span class="n">name</span> -</span><span id="L-1024"><a href="#L-1024"><span class="linenos">1024</span></a> <span class="k">if</span> <span class="n">is_int</span><span class="p">(</span><span class="n">text</span><span class="p">):</span> -</span><span id="L-1025"><a href="#L-1025"><span class="linenos">1025</span></a> <span class="n">index</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">text</span><span class="p">)</span> -</span><span id="L-1026"><a href="#L-1026"><span class="linenos">1026</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span> -</span><span id="L-1027"><a href="#L-1027"><span class="linenos">1027</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">index</span> <span class="k">if</span> <span class="n">zero_based_indexing</span> <span class="k">else</span> <span class="n">index</span> <span class="o">-</span> <span class="mi">1</span><span class="p">)</span> -</span><span id="L-1028"><a href="#L-1028"><span class="linenos">1028</span></a> <span class="p">)</span> -</span><span id="L-1029"><a href="#L-1029"><span class="linenos">1029</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-1030"><a href="#L-1030"><span class="linenos">1030</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">text</span><span class="p">))</span> -</span><span id="L-1031"><a href="#L-1031"><span class="linenos">1031</span></a> -</span><span id="L-1032"><a href="#L-1032"><span class="linenos">1032</span></a> <span class="c1"># This is done to avoid failing in the expression validator due to the arg count</span> -</span><span id="L-1033"><a href="#L-1033"><span class="linenos">1033</span></a> <span class="k">del</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">:]</span> -</span><span id="L-1034"><a href="#L-1034"><span class="linenos">1034</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">segments</span><span class="p">))</span> -</span><span id="L-1035"><a href="#L-1035"><span class="linenos">1035</span></a> -</span><span id="L-1036"><a href="#L-1036"><span class="linenos">1036</span></a> <span class="k">return</span> <span class="n">_parse_json_extract_path</span> -</span><span id="L-1037"><a href="#L-1037"><span class="linenos">1037</span></a> +</span><span id="L-1013"><a href="#L-1013"><span class="linenos">1013</span></a> <span class="c1"># This is done to avoid failing in the expression validator due to the arg count</span> +</span><span id="L-1014"><a href="#L-1014"><span class="linenos">1014</span></a> <span class="k">del</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">:]</span> +</span><span id="L-1015"><a href="#L-1015"><span class="linenos">1015</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">segments</span><span class="p">))</span> +</span><span id="L-1016"><a href="#L-1016"><span class="linenos">1016</span></a> +</span><span id="L-1017"><a href="#L-1017"><span class="linenos">1017</span></a> <span class="k">return</span> <span class="n">_parse_json_extract_path</span> +</span><span id="L-1018"><a href="#L-1018"><span class="linenos">1018</span></a> +</span><span id="L-1019"><a href="#L-1019"><span class="linenos">1019</span></a> +</span><span id="L-1020"><a href="#L-1020"><span class="linenos">1020</span></a><span class="k">def</span> <span class="nf">json_extract_segments</span><span class="p">(</span> +</span><span id="L-1021"><a href="#L-1021"><span class="linenos">1021</span></a> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">quoted_index</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="L-1022"><a href="#L-1022"><span class="linenos">1022</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="L-1023"><a href="#L-1023"><span class="linenos">1023</span></a> <span class="k">def</span> <span class="nf">_json_extract_segments</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-1024"><a href="#L-1024"><span class="linenos">1024</span></a> <span class="n">path</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span> +</span><span id="L-1025"><a href="#L-1025"><span class="linenos">1025</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">):</span> +</span><span id="L-1026"><a href="#L-1026"><span class="linenos">1026</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> +</span><span id="L-1027"><a href="#L-1027"><span class="linenos">1027</span></a> +</span><span id="L-1028"><a href="#L-1028"><span class="linenos">1028</span></a> <span class="n">segments</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="L-1029"><a href="#L-1029"><span class="linenos">1029</span></a> <span class="k">for</span> <span class="n">segment</span> <span class="ow">in</span> <span class="n">path</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> +</span><span id="L-1030"><a href="#L-1030"><span class="linenos">1030</span></a> <span class="n">path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">segment</span><span class="p">)</span> +</span><span id="L-1031"><a href="#L-1031"><span class="linenos">1031</span></a> <span class="k">if</span> <span class="n">path</span><span class="p">:</span> +</span><span id="L-1032"><a href="#L-1032"><span class="linenos">1032</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span> +</span><span id="L-1033"><a href="#L-1033"><span class="linenos">1033</span></a> <span class="n">quoted_index</span> <span class="ow">or</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">)</span> +</span><span id="L-1034"><a href="#L-1034"><span class="linenos">1034</span></a> <span class="p">):</span> +</span><span id="L-1035"><a href="#L-1035"><span class="linenos">1035</span></a> <span class="n">path</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_START</span><span class="si">}{</span><span class="n">path</span><span class="si">}{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_END</span><span class="si">}</span><span class="s2">"</span> +</span><span id="L-1036"><a href="#L-1036"><span class="linenos">1036</span></a> +</span><span id="L-1037"><a href="#L-1037"><span class="linenos">1037</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">path</span><span class="p">)</span> </span><span id="L-1038"><a href="#L-1038"><span class="linenos">1038</span></a> -</span><span id="L-1039"><a href="#L-1039"><span class="linenos">1039</span></a><span class="k">def</span> <span class="nf">json_extract_segments</span><span class="p">(</span> -</span><span id="L-1040"><a href="#L-1040"><span class="linenos">1040</span></a> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">quoted_index</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="L-1041"><a href="#L-1041"><span class="linenos">1041</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="L-1042"><a href="#L-1042"><span class="linenos">1042</span></a> <span class="k">def</span> <span class="nf">_json_extract_segments</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-1043"><a href="#L-1043"><span class="linenos">1043</span></a> <span class="n">path</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span> -</span><span id="L-1044"><a href="#L-1044"><span class="linenos">1044</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">):</span> -</span><span id="L-1045"><a href="#L-1045"><span class="linenos">1045</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> -</span><span id="L-1046"><a href="#L-1046"><span class="linenos">1046</span></a> -</span><span id="L-1047"><a href="#L-1047"><span class="linenos">1047</span></a> <span class="n">segments</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="L-1048"><a href="#L-1048"><span class="linenos">1048</span></a> <span class="k">for</span> <span class="n">segment</span> <span class="ow">in</span> <span class="n">path</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> -</span><span id="L-1049"><a href="#L-1049"><span class="linenos">1049</span></a> <span class="n">path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">segment</span><span class="p">)</span> -</span><span id="L-1050"><a href="#L-1050"><span class="linenos">1050</span></a> <span class="k">if</span> <span class="n">path</span><span class="p">:</span> -</span><span id="L-1051"><a href="#L-1051"><span class="linenos">1051</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span> -</span><span id="L-1052"><a href="#L-1052"><span class="linenos">1052</span></a> <span class="n">quoted_index</span> <span class="ow">or</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">)</span> -</span><span id="L-1053"><a href="#L-1053"><span class="linenos">1053</span></a> <span class="p">):</span> -</span><span id="L-1054"><a href="#L-1054"><span class="linenos">1054</span></a> <span class="n">path</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_START</span><span class="si">}{</span><span class="n">path</span><span class="si">}{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_END</span><span class="si">}</span><span class="s2">"</span> -</span><span id="L-1055"><a href="#L-1055"><span class="linenos">1055</span></a> -</span><span id="L-1056"><a href="#L-1056"><span class="linenos">1056</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">path</span><span class="p">)</span> -</span><span id="L-1057"><a href="#L-1057"><span class="linenos">1057</span></a> -</span><span id="L-1058"><a href="#L-1058"><span class="linenos">1058</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="o">*</span><span class="n">segments</span><span class="p">)</span> -</span><span id="L-1059"><a href="#L-1059"><span class="linenos">1059</span></a> -</span><span id="L-1060"><a href="#L-1060"><span class="linenos">1060</span></a> <span class="k">return</span> <span class="n">_json_extract_segments</span> -</span><span id="L-1061"><a href="#L-1061"><span class="linenos">1061</span></a> -</span><span id="L-1062"><a href="#L-1062"><span class="linenos">1062</span></a> -</span><span id="L-1063"><a href="#L-1063"><span class="linenos">1063</span></a><span class="k">def</span> <span class="nf">json_path_key_only_name</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="L-1064"><a href="#L-1064"><span class="linenos">1064</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathWildcard</span><span class="p">):</span> -</span><span id="L-1065"><a href="#L-1065"><span class="linenos">1065</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"Unsupported wildcard in JSONPathKey expression"</span><span class="p">)</span> -</span><span id="L-1066"><a href="#L-1066"><span class="linenos">1066</span></a> -</span><span id="L-1067"><a href="#L-1067"><span class="linenos">1067</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">name</span> +</span><span id="L-1039"><a href="#L-1039"><span class="linenos">1039</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="o">*</span><span class="n">segments</span><span class="p">)</span> +</span><span id="L-1040"><a href="#L-1040"><span class="linenos">1040</span></a> +</span><span id="L-1041"><a href="#L-1041"><span class="linenos">1041</span></a> <span class="k">return</span> <span class="n">_json_extract_segments</span> +</span><span id="L-1042"><a href="#L-1042"><span class="linenos">1042</span></a> +</span><span id="L-1043"><a href="#L-1043"><span class="linenos">1043</span></a> +</span><span id="L-1044"><a href="#L-1044"><span class="linenos">1044</span></a><span class="k">def</span> <span class="nf">json_path_key_only_name</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="L-1045"><a href="#L-1045"><span class="linenos">1045</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathWildcard</span><span class="p">):</span> +</span><span id="L-1046"><a href="#L-1046"><span class="linenos">1046</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"Unsupported wildcard in JSONPathKey expression"</span><span class="p">)</span> +</span><span id="L-1047"><a href="#L-1047"><span class="linenos">1047</span></a> +</span><span id="L-1048"><a href="#L-1048"><span class="linenos">1048</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">name</span> </span></pre></div> @@ -4115,47 +4093,6 @@ that it can analyze queries in the optimizer and successfully capture their sema </section> - <section id="create_with_partitions_sql"> - <input id="create_with_partitions_sql-view-source" class="view-source-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> -<div class="attr function"> - - <span class="def">def</span> - <span class="name">create_with_partitions_sql</span><span class="signature pdoc-code multiline">(<span class="param"> <span class="bp">self</span><span class="p">:</span> <span class="n"><a href="../generator.html#Generator">sqlglot.generator.Generator</a></span>,</span><span class="param"> <span class="n">expression</span><span class="p">:</span> <span class="n"><a href="../expressions.html#Create">sqlglot.expressions.Create</a></span></span><span class="return-annotation">) -> <span class="nb">str</span>:</span></span> - - <label class="view-source-button" for="create_with_partitions_sql-view-source"><span>View Source</span></label> - - </div> - <a class="headerlink" href="#create_with_partitions_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="create_with_partitions_sql-658"><a href="#create_with_partitions_sql-658"><span class="linenos">658</span></a><span class="k">def</span> <span class="nf">create_with_partitions_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Create</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="create_with_partitions_sql-659"><a href="#create_with_partitions_sql-659"><span class="linenos">659</span></a><span class="w"> </span><span class="sd">"""</span> -</span><span id="create_with_partitions_sql-660"><a href="#create_with_partitions_sql-660"><span class="linenos">660</span></a><span class="sd"> In Hive and Spark, the PARTITIONED BY property acts as an extension of a table's schema. When the</span> -</span><span id="create_with_partitions_sql-661"><a href="#create_with_partitions_sql-661"><span class="linenos">661</span></a><span class="sd"> PARTITIONED BY value is an array of column names, they are transformed into a schema. The corresponding</span> -</span><span id="create_with_partitions_sql-662"><a href="#create_with_partitions_sql-662"><span class="linenos">662</span></a><span class="sd"> columns are removed from the create statement.</span> -</span><span id="create_with_partitions_sql-663"><a href="#create_with_partitions_sql-663"><span class="linenos">663</span></a><span class="sd"> """</span> -</span><span id="create_with_partitions_sql-664"><a href="#create_with_partitions_sql-664"><span class="linenos">664</span></a> <span class="n">has_schema</span> <span class="o">=</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">)</span> -</span><span id="create_with_partitions_sql-665"><a href="#create_with_partitions_sql-665"><span class="linenos">665</span></a> <span class="n">is_partitionable</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"kind"</span><span class="p">)</span> <span class="ow">in</span> <span class="p">(</span><span class="s2">"TABLE"</span><span class="p">,</span> <span class="s2">"VIEW"</span><span class="p">)</span> -</span><span id="create_with_partitions_sql-666"><a href="#create_with_partitions_sql-666"><span class="linenos">666</span></a> -</span><span id="create_with_partitions_sql-667"><a href="#create_with_partitions_sql-667"><span class="linenos">667</span></a> <span class="k">if</span> <span class="n">has_schema</span> <span class="ow">and</span> <span class="n">is_partitionable</span><span class="p">:</span> -</span><span id="create_with_partitions_sql-668"><a href="#create_with_partitions_sql-668"><span class="linenos">668</span></a> <span class="n">prop</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">find</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">PartitionedByProperty</span><span class="p">)</span> -</span><span id="create_with_partitions_sql-669"><a href="#create_with_partitions_sql-669"><span class="linenos">669</span></a> <span class="k">if</span> <span class="n">prop</span> <span class="ow">and</span> <span class="n">prop</span><span class="o">.</span><span class="n">this</span> <span class="ow">and</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">prop</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">):</span> -</span><span id="create_with_partitions_sql-670"><a href="#create_with_partitions_sql-670"><span class="linenos">670</span></a> <span class="n">schema</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> -</span><span id="create_with_partitions_sql-671"><a href="#create_with_partitions_sql-671"><span class="linenos">671</span></a> <span class="n">columns</span> <span class="o">=</span> <span class="p">{</span><span class="n">v</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">for</span> <span class="n">v</span> <span class="ow">in</span> <span class="n">prop</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">}</span> -</span><span id="create_with_partitions_sql-672"><a href="#create_with_partitions_sql-672"><span class="linenos">672</span></a> <span class="n">partitions</span> <span class="o">=</span> <span class="p">[</span><span class="n">col</span> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">schema</span><span class="o">.</span><span class="n">expressions</span> <span class="k">if</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">in</span> <span class="n">columns</span><span class="p">]</span> -</span><span id="create_with_partitions_sql-673"><a href="#create_with_partitions_sql-673"><span class="linenos">673</span></a> <span class="n">schema</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s2">"expressions"</span><span class="p">,</span> <span class="p">[</span><span class="n">e</span> <span class="k">for</span> <span class="n">e</span> <span class="ow">in</span> <span class="n">schema</span><span class="o">.</span><span class="n">expressions</span> <span class="k">if</span> <span class="n">e</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">partitions</span><span class="p">])</span> -</span><span id="create_with_partitions_sql-674"><a href="#create_with_partitions_sql-674"><span class="linenos">674</span></a> <span class="n">prop</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">PartitionedByProperty</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Schema</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">partitions</span><span class="p">)))</span> -</span><span id="create_with_partitions_sql-675"><a href="#create_with_partitions_sql-675"><span class="linenos">675</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">set</span><span class="p">(</span><span class="s2">"this"</span><span class="p">,</span> <span class="n">schema</span><span class="p">)</span> -</span><span id="create_with_partitions_sql-676"><a href="#create_with_partitions_sql-676"><span class="linenos">676</span></a> -</span><span id="create_with_partitions_sql-677"><a href="#create_with_partitions_sql-677"><span class="linenos">677</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span></pre></div> - - - <div class="docstring"><p>In Hive and Spark, the PARTITIONED BY property acts as an extension of a table's schema. When the -PARTITIONED BY value is an array of column names, they are transformed into a schema. The corresponding -columns are removed from the create statement.</p> -</div> - - - </section> <section id="parse_date_delta"> <input id="parse_date_delta-view-source" class="view-source-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> <div class="attr function"> @@ -4167,17 +4104,17 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#parse_date_delta"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="parse_date_delta-680"><a href="#parse_date_delta-680"><span class="linenos">680</span></a><span class="k">def</span> <span class="nf">parse_date_delta</span><span class="p">(</span> -</span><span id="parse_date_delta-681"><a href="#parse_date_delta-681"><span class="linenos">681</span></a> <span class="n">exp_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> <span class="n">unit_mapping</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">t</span><span class="o">.</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span> -</span><span id="parse_date_delta-682"><a href="#parse_date_delta-682"><span class="linenos">682</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">E</span><span class="p">]:</span> -</span><span id="parse_date_delta-683"><a href="#parse_date_delta-683"><span class="linenos">683</span></a> <span class="k">def</span> <span class="nf">inner_func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="parse_date_delta-684"><a href="#parse_date_delta-684"><span class="linenos">684</span></a> <span class="n">unit_based</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o">==</span> <span class="mi">3</span> -</span><span id="parse_date_delta-685"><a href="#parse_date_delta-685"><span class="linenos">685</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> -</span><span id="parse_date_delta-686"><a href="#parse_date_delta-686"><span class="linenos">686</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="s2">"DAY"</span><span class="p">)</span> -</span><span id="parse_date_delta-687"><a href="#parse_date_delta-687"><span class="linenos">687</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit_mapping</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">(),</span> <span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="p">))</span> <span class="k">if</span> <span class="n">unit_mapping</span> <span class="k">else</span> <span class="n">unit</span> -</span><span id="parse_date_delta-688"><a href="#parse_date_delta-688"><span class="linenos">688</span></a> <span class="k">return</span> <span class="n">exp_class</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> -</span><span id="parse_date_delta-689"><a href="#parse_date_delta-689"><span class="linenos">689</span></a> -</span><span id="parse_date_delta-690"><a href="#parse_date_delta-690"><span class="linenos">690</span></a> <span class="k">return</span> <span class="n">inner_func</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="parse_date_delta-658"><a href="#parse_date_delta-658"><span class="linenos">658</span></a><span class="k">def</span> <span class="nf">parse_date_delta</span><span class="p">(</span> +</span><span id="parse_date_delta-659"><a href="#parse_date_delta-659"><span class="linenos">659</span></a> <span class="n">exp_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> <span class="n">unit_mapping</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">t</span><span class="o">.</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="nb">str</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span> +</span><span id="parse_date_delta-660"><a href="#parse_date_delta-660"><span class="linenos">660</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">E</span><span class="p">]:</span> +</span><span id="parse_date_delta-661"><a href="#parse_date_delta-661"><span class="linenos">661</span></a> <span class="k">def</span> <span class="nf">inner_func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="parse_date_delta-662"><a href="#parse_date_delta-662"><span class="linenos">662</span></a> <span class="n">unit_based</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o">==</span> <span class="mi">3</span> +</span><span id="parse_date_delta-663"><a href="#parse_date_delta-663"><span class="linenos">663</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> +</span><span id="parse_date_delta-664"><a href="#parse_date_delta-664"><span class="linenos">664</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="k">if</span> <span class="n">unit_based</span> <span class="k">else</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="s2">"DAY"</span><span class="p">)</span> +</span><span id="parse_date_delta-665"><a href="#parse_date_delta-665"><span class="linenos">665</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit_mapping</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">(),</span> <span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="p">))</span> <span class="k">if</span> <span class="n">unit_mapping</span> <span class="k">else</span> <span class="n">unit</span> +</span><span id="parse_date_delta-666"><a href="#parse_date_delta-666"><span class="linenos">666</span></a> <span class="k">return</span> <span class="n">exp_class</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> +</span><span id="parse_date_delta-667"><a href="#parse_date_delta-667"><span class="linenos">667</span></a> +</span><span id="parse_date_delta-668"><a href="#parse_date_delta-668"><span class="linenos">668</span></a> <span class="k">return</span> <span class="n">inner_func</span> </span></pre></div> @@ -4195,27 +4132,27 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#parse_date_delta_with_interval"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="parse_date_delta_with_interval-693"><a href="#parse_date_delta_with_interval-693"><span class="linenos">693</span></a><span class="k">def</span> <span class="nf">parse_date_delta_with_interval</span><span class="p">(</span> -</span><span id="parse_date_delta_with_interval-694"><a href="#parse_date_delta_with_interval-694"><span class="linenos">694</span></a> <span class="n">expression_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> -</span><span id="parse_date_delta_with_interval-695"><a href="#parse_date_delta_with_interval-695"><span class="linenos">695</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]]:</span> -</span><span id="parse_date_delta_with_interval-696"><a href="#parse_date_delta_with_interval-696"><span class="linenos">696</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]:</span> -</span><span id="parse_date_delta_with_interval-697"><a href="#parse_date_delta_with_interval-697"><span class="linenos">697</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o"><</span> <span class="mi">2</span><span class="p">:</span> -</span><span id="parse_date_delta_with_interval-698"><a href="#parse_date_delta_with_interval-698"><span class="linenos">698</span></a> <span class="k">return</span> <span class="kc">None</span> -</span><span id="parse_date_delta_with_interval-699"><a href="#parse_date_delta_with_interval-699"><span class="linenos">699</span></a> -</span><span id="parse_date_delta_with_interval-700"><a href="#parse_date_delta_with_interval-700"><span class="linenos">700</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> -</span><span id="parse_date_delta_with_interval-701"><a href="#parse_date_delta_with_interval-701"><span class="linenos">701</span></a> -</span><span id="parse_date_delta_with_interval-702"><a href="#parse_date_delta_with_interval-702"><span class="linenos">702</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interval</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">):</span> -</span><span id="parse_date_delta_with_interval-703"><a href="#parse_date_delta_with_interval-703"><span class="linenos">703</span></a> <span class="k">raise</span> <span class="n">ParseError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"INTERVAL expression expected but got '</span><span class="si">{</span><span class="n">interval</span><span class="si">}</span><span class="s2">'"</span><span class="p">)</span> -</span><span id="parse_date_delta_with_interval-704"><a href="#parse_date_delta_with_interval-704"><span class="linenos">704</span></a> -</span><span id="parse_date_delta_with_interval-705"><a href="#parse_date_delta_with_interval-705"><span class="linenos">705</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">interval</span><span class="o">.</span><span class="n">this</span> -</span><span id="parse_date_delta_with_interval-706"><a href="#parse_date_delta_with_interval-706"><span class="linenos">706</span></a> <span class="k">if</span> <span class="n">expression</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> -</span><span id="parse_date_delta_with_interval-707"><a href="#parse_date_delta_with_interval-707"><span class="linenos">707</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="parse_date_delta_with_interval-708"><a href="#parse_date_delta_with_interval-708"><span class="linenos">708</span></a> -</span><span id="parse_date_delta_with_interval-709"><a href="#parse_date_delta_with_interval-709"><span class="linenos">709</span></a> <span class="k">return</span> <span class="n">expression_class</span><span class="p">(</span> -</span><span id="parse_date_delta_with_interval-710"><a href="#parse_date_delta_with_interval-710"><span class="linenos">710</span></a> <span class="n">this</span><span class="o">=</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">expression</span><span class="o">=</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">interval</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">))</span> -</span><span id="parse_date_delta_with_interval-711"><a href="#parse_date_delta_with_interval-711"><span class="linenos">711</span></a> <span class="p">)</span> -</span><span id="parse_date_delta_with_interval-712"><a href="#parse_date_delta_with_interval-712"><span class="linenos">712</span></a> -</span><span id="parse_date_delta_with_interval-713"><a href="#parse_date_delta_with_interval-713"><span class="linenos">713</span></a> <span class="k">return</span> <span class="n">func</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="parse_date_delta_with_interval-671"><a href="#parse_date_delta_with_interval-671"><span class="linenos">671</span></a><span class="k">def</span> <span class="nf">parse_date_delta_with_interval</span><span class="p">(</span> +</span><span id="parse_date_delta_with_interval-672"><a href="#parse_date_delta_with_interval-672"><span class="linenos">672</span></a> <span class="n">expression_class</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">E</span><span class="p">],</span> +</span><span id="parse_date_delta_with_interval-673"><a href="#parse_date_delta_with_interval-673"><span class="linenos">673</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]]:</span> +</span><span id="parse_date_delta_with_interval-674"><a href="#parse_date_delta_with_interval-674"><span class="linenos">674</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">E</span><span class="p">]:</span> +</span><span id="parse_date_delta_with_interval-675"><a href="#parse_date_delta_with_interval-675"><span class="linenos">675</span></a> <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> <span class="o"><</span> <span class="mi">2</span><span class="p">:</span> +</span><span id="parse_date_delta_with_interval-676"><a href="#parse_date_delta_with_interval-676"><span class="linenos">676</span></a> <span class="k">return</span> <span class="kc">None</span> +</span><span id="parse_date_delta_with_interval-677"><a href="#parse_date_delta_with_interval-677"><span class="linenos">677</span></a> +</span><span id="parse_date_delta_with_interval-678"><a href="#parse_date_delta_with_interval-678"><span class="linenos">678</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> +</span><span id="parse_date_delta_with_interval-679"><a href="#parse_date_delta_with_interval-679"><span class="linenos">679</span></a> +</span><span id="parse_date_delta_with_interval-680"><a href="#parse_date_delta_with_interval-680"><span class="linenos">680</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">interval</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">):</span> +</span><span id="parse_date_delta_with_interval-681"><a href="#parse_date_delta_with_interval-681"><span class="linenos">681</span></a> <span class="k">raise</span> <span class="n">ParseError</span><span class="p">(</span><span class="sa">f</span><span class="s2">"INTERVAL expression expected but got '</span><span class="si">{</span><span class="n">interval</span><span class="si">}</span><span class="s2">'"</span><span class="p">)</span> +</span><span id="parse_date_delta_with_interval-682"><a href="#parse_date_delta_with_interval-682"><span class="linenos">682</span></a> +</span><span id="parse_date_delta_with_interval-683"><a href="#parse_date_delta_with_interval-683"><span class="linenos">683</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">interval</span><span class="o">.</span><span class="n">this</span> +</span><span id="parse_date_delta_with_interval-684"><a href="#parse_date_delta_with_interval-684"><span class="linenos">684</span></a> <span class="k">if</span> <span class="n">expression</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> +</span><span id="parse_date_delta_with_interval-685"><a href="#parse_date_delta_with_interval-685"><span class="linenos">685</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="parse_date_delta_with_interval-686"><a href="#parse_date_delta_with_interval-686"><span class="linenos">686</span></a> +</span><span id="parse_date_delta_with_interval-687"><a href="#parse_date_delta_with_interval-687"><span class="linenos">687</span></a> <span class="k">return</span> <span class="n">expression_class</span><span class="p">(</span> +</span><span id="parse_date_delta_with_interval-688"><a href="#parse_date_delta_with_interval-688"><span class="linenos">688</span></a> <span class="n">this</span><span class="o">=</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">expression</span><span class="o">=</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">interval</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">))</span> +</span><span id="parse_date_delta_with_interval-689"><a href="#parse_date_delta_with_interval-689"><span class="linenos">689</span></a> <span class="p">)</span> +</span><span id="parse_date_delta_with_interval-690"><a href="#parse_date_delta_with_interval-690"><span class="linenos">690</span></a> +</span><span id="parse_date_delta_with_interval-691"><a href="#parse_date_delta_with_interval-691"><span class="linenos">691</span></a> <span class="k">return</span> <span class="n">func</span> </span></pre></div> @@ -4233,13 +4170,13 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#date_trunc_to_time"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="date_trunc_to_time-716"><a href="#date_trunc_to_time-716"><span class="linenos">716</span></a><span class="k">def</span> <span class="nf">date_trunc_to_time</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> -</span><span id="date_trunc_to_time-717"><a href="#date_trunc_to_time-717"><span class="linenos">717</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> -</span><span id="date_trunc_to_time-718"><a href="#date_trunc_to_time-718"><span class="linenos">718</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span> -</span><span id="date_trunc_to_time-719"><a href="#date_trunc_to_time-719"><span class="linenos">719</span></a> -</span><span id="date_trunc_to_time-720"><a href="#date_trunc_to_time-720"><span class="linenos">720</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">this</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"date"</span><span class="p">):</span> -</span><span id="date_trunc_to_time-721"><a href="#date_trunc_to_time-721"><span class="linenos">721</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">(</span><span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">,</span> <span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">)</span> -</span><span id="date_trunc_to_time-722"><a href="#date_trunc_to_time-722"><span class="linenos">722</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="date_trunc_to_time-694"><a href="#date_trunc_to_time-694"><span class="linenos">694</span></a><span class="k">def</span> <span class="nf">date_trunc_to_time</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> +</span><span id="date_trunc_to_time-695"><a href="#date_trunc_to_time-695"><span class="linenos">695</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span> +</span><span id="date_trunc_to_time-696"><a href="#date_trunc_to_time-696"><span class="linenos">696</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span> +</span><span id="date_trunc_to_time-697"><a href="#date_trunc_to_time-697"><span class="linenos">697</span></a> +</span><span id="date_trunc_to_time-698"><a href="#date_trunc_to_time-698"><span class="linenos">698</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">this</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"date"</span><span class="p">):</span> +</span><span id="date_trunc_to_time-699"><a href="#date_trunc_to_time-699"><span class="linenos">699</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">(</span><span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">,</span> <span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">)</span> +</span><span id="date_trunc_to_time-700"><a href="#date_trunc_to_time-700"><span class="linenos">700</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">this</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> </span></pre></div> @@ -4257,17 +4194,17 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#date_add_interval_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="date_add_interval_sql-725"><a href="#date_add_interval_sql-725"><span class="linenos">725</span></a><span class="k">def</span> <span class="nf">date_add_interval_sql</span><span class="p">(</span> -</span><span id="date_add_interval_sql-726"><a href="#date_add_interval_sql-726"><span class="linenos">726</span></a> <span class="n">data_type</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">kind</span><span class="p">:</span> <span class="nb">str</span> -</span><span id="date_add_interval_sql-727"><a href="#date_add_interval_sql-727"><span class="linenos">727</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="date_add_interval_sql-728"><a href="#date_add_interval_sql-728"><span class="linenos">728</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="date_add_interval_sql-729"><a href="#date_add_interval_sql-729"><span class="linenos">729</span></a> <span class="n">this</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> -</span><span id="date_add_interval_sql-730"><a href="#date_add_interval_sql-730"><span class="linenos">730</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span> -</span><span id="date_add_interval_sql-731"><a href="#date_add_interval_sql-731"><span class="linenos">731</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">if</span> <span class="n">unit</span> <span class="k">else</span> <span class="s2">"DAY"</span><span class="p">)</span> -</span><span id="date_add_interval_sql-732"><a href="#date_add_interval_sql-732"><span class="linenos">732</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> -</span><span id="date_add_interval_sql-733"><a href="#date_add_interval_sql-733"><span class="linenos">733</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">data_type</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">kind</span><span class="si">}</span><span class="s2">(</span><span class="si">{</span><span class="n">this</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">interval</span><span class="p">)</span><span class="si">}</span><span class="s2">)"</span> -</span><span id="date_add_interval_sql-734"><a href="#date_add_interval_sql-734"><span class="linenos">734</span></a> -</span><span id="date_add_interval_sql-735"><a href="#date_add_interval_sql-735"><span class="linenos">735</span></a> <span class="k">return</span> <span class="n">func</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="date_add_interval_sql-703"><a href="#date_add_interval_sql-703"><span class="linenos">703</span></a><span class="k">def</span> <span class="nf">date_add_interval_sql</span><span class="p">(</span> +</span><span id="date_add_interval_sql-704"><a href="#date_add_interval_sql-704"><span class="linenos">704</span></a> <span class="n">data_type</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">kind</span><span class="p">:</span> <span class="nb">str</span> +</span><span id="date_add_interval_sql-705"><a href="#date_add_interval_sql-705"><span class="linenos">705</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="date_add_interval_sql-706"><a href="#date_add_interval_sql-706"><span class="linenos">706</span></a> <span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="date_add_interval_sql-707"><a href="#date_add_interval_sql-707"><span class="linenos">707</span></a> <span class="n">this</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> +</span><span id="date_add_interval_sql-708"><a href="#date_add_interval_sql-708"><span class="linenos">708</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span> +</span><span id="date_add_interval_sql-709"><a href="#date_add_interval_sql-709"><span class="linenos">709</span></a> <span class="n">unit</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">unit</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="k">if</span> <span class="n">unit</span> <span class="k">else</span> <span class="s2">"DAY"</span><span class="p">)</span> +</span><span id="date_add_interval_sql-710"><a href="#date_add_interval_sql-710"><span class="linenos">710</span></a> <span class="n">interval</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">unit</span><span class="o">=</span><span class="n">unit</span><span class="p">)</span> +</span><span id="date_add_interval_sql-711"><a href="#date_add_interval_sql-711"><span class="linenos">711</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">data_type</span><span class="si">}</span><span class="s2">_</span><span class="si">{</span><span class="n">kind</span><span class="si">}</span><span class="s2">(</span><span class="si">{</span><span class="n">this</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">interval</span><span class="p">)</span><span class="si">}</span><span class="s2">)"</span> +</span><span id="date_add_interval_sql-712"><a href="#date_add_interval_sql-712"><span class="linenos">712</span></a> +</span><span id="date_add_interval_sql-713"><a href="#date_add_interval_sql-713"><span class="linenos">713</span></a> <span class="k">return</span> <span class="n">func</span> </span></pre></div> @@ -4285,10 +4222,10 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#timestamptrunc_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="timestamptrunc_sql-738"><a href="#timestamptrunc_sql-738"><span class="linenos">738</span></a><span class="k">def</span> <span class="nf">timestamptrunc_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="timestamptrunc_sql-739"><a href="#timestamptrunc_sql-739"><span class="linenos">739</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="timestamptrunc_sql-740"><a href="#timestamptrunc_sql-740"><span class="linenos">740</span></a> <span class="s2">"DATE_TRUNC"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> -</span><span id="timestamptrunc_sql-741"><a href="#timestamptrunc_sql-741"><span class="linenos">741</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="timestamptrunc_sql-716"><a href="#timestamptrunc_sql-716"><span class="linenos">716</span></a><span class="k">def</span> <span class="nf">timestamptrunc_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="timestamptrunc_sql-717"><a href="#timestamptrunc_sql-717"><span class="linenos">717</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="timestamptrunc_sql-718"><a href="#timestamptrunc_sql-718"><span class="linenos">718</span></a> <span class="s2">"DATE_TRUNC"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">string</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> +</span><span id="timestamptrunc_sql-719"><a href="#timestamptrunc_sql-719"><span class="linenos">719</span></a> <span class="p">)</span> </span></pre></div> @@ -4306,17 +4243,20 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#no_timestamp_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="no_timestamp_sql-744"><a href="#no_timestamp_sql-744"><span class="linenos">744</span></a><span class="k">def</span> <span class="nf">no_timestamp_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Timestamp</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="no_timestamp_sql-745"><a href="#no_timestamp_sql-745"><span class="linenos">745</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> -</span><span id="no_timestamp_sql-746"><a href="#no_timestamp_sql-746"><span class="linenos">746</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">))</span> -</span><span id="no_timestamp_sql-747"><a href="#no_timestamp_sql-747"><span class="linenos">747</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"expression"</span><span class="p">)</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="ow">in</span> <span class="n">TIMEZONES</span><span class="p">:</span> -</span><span id="no_timestamp_sql-748"><a href="#no_timestamp_sql-748"><span class="linenos">748</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="no_timestamp_sql-749"><a href="#no_timestamp_sql-749"><span class="linenos">749</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">AtTimeZone</span><span class="p">(</span> -</span><span id="no_timestamp_sql-750"><a href="#no_timestamp_sql-750"><span class="linenos">750</span></a> <span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">),</span> -</span><span id="no_timestamp_sql-751"><a href="#no_timestamp_sql-751"><span class="linenos">751</span></a> <span class="n">zone</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> -</span><span id="no_timestamp_sql-752"><a href="#no_timestamp_sql-752"><span class="linenos">752</span></a> <span class="p">)</span> -</span><span id="no_timestamp_sql-753"><a href="#no_timestamp_sql-753"><span class="linenos">753</span></a> <span class="p">)</span> -</span><span id="no_timestamp_sql-754"><a href="#no_timestamp_sql-754"><span class="linenos">754</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">function_fallback_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="no_timestamp_sql-722"><a href="#no_timestamp_sql-722"><span class="linenos">722</span></a><span class="k">def</span> <span class="nf">no_timestamp_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Timestamp</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="no_timestamp_sql-723"><a href="#no_timestamp_sql-723"><span class="linenos">723</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> +</span><span id="no_timestamp_sql-724"><a href="#no_timestamp_sql-724"><span class="linenos">724</span></a> <span class="kn">from</span> <span class="nn">sqlglot.optimizer.annotate_types</span> <span class="kn">import</span> <span class="n">annotate_types</span> +</span><span id="no_timestamp_sql-725"><a href="#no_timestamp_sql-725"><span class="linenos">725</span></a> +</span><span id="no_timestamp_sql-726"><a href="#no_timestamp_sql-726"><span class="linenos">726</span></a> <span class="n">target_type</span> <span class="o">=</span> <span class="n">annotate_types</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span><span class="o">.</span><span class="n">type</span> <span class="ow">or</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span> +</span><span id="no_timestamp_sql-727"><a href="#no_timestamp_sql-727"><span class="linenos">727</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">target_type</span><span class="p">))</span> +</span><span id="no_timestamp_sql-728"><a href="#no_timestamp_sql-728"><span class="linenos">728</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"expression"</span><span class="p">)</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="ow">in</span> <span class="n">TIMEZONES</span><span class="p">:</span> +</span><span id="no_timestamp_sql-729"><a href="#no_timestamp_sql-729"><span class="linenos">729</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="no_timestamp_sql-730"><a href="#no_timestamp_sql-730"><span class="linenos">730</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">AtTimeZone</span><span class="p">(</span> +</span><span id="no_timestamp_sql-731"><a href="#no_timestamp_sql-731"><span class="linenos">731</span></a> <span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">to</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">),</span> +</span><span id="no_timestamp_sql-732"><a href="#no_timestamp_sql-732"><span class="linenos">732</span></a> <span class="n">zone</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> +</span><span id="no_timestamp_sql-733"><a href="#no_timestamp_sql-733"><span class="linenos">733</span></a> <span class="p">)</span> +</span><span id="no_timestamp_sql-734"><a href="#no_timestamp_sql-734"><span class="linenos">734</span></a> <span class="p">)</span> +</span><span id="no_timestamp_sql-735"><a href="#no_timestamp_sql-735"><span class="linenos">735</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"TIMESTAMP"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> </span></pre></div> @@ -4334,10 +4274,10 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#locate_to_strposition"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="locate_to_strposition-757"><a href="#locate_to_strposition-757"><span class="linenos">757</span></a><span class="k">def</span> <span class="nf">locate_to_strposition</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> -</span><span id="locate_to_strposition-758"><a href="#locate_to_strposition-758"><span class="linenos">758</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">(</span> -</span><span id="locate_to_strposition-759"><a href="#locate_to_strposition-759"><span class="linenos">759</span></a> <span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">substr</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">position</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> -</span><span id="locate_to_strposition-760"><a href="#locate_to_strposition-760"><span class="linenos">760</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="locate_to_strposition-738"><a href="#locate_to_strposition-738"><span class="linenos">738</span></a><span class="k">def</span> <span class="nf">locate_to_strposition</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> +</span><span id="locate_to_strposition-739"><a href="#locate_to_strposition-739"><span class="linenos">739</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">(</span> +</span><span id="locate_to_strposition-740"><a href="#locate_to_strposition-740"><span class="linenos">740</span></a> <span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">substr</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">position</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> +</span><span id="locate_to_strposition-741"><a href="#locate_to_strposition-741"><span class="linenos">741</span></a> <span class="p">)</span> </span></pre></div> @@ -4355,10 +4295,10 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#strposition_to_locate_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="strposition_to_locate_sql-763"><a href="#strposition_to_locate_sql-763"><span class="linenos">763</span></a><span class="k">def</span> <span class="nf">strposition_to_locate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="strposition_to_locate_sql-764"><a href="#strposition_to_locate_sql-764"><span class="linenos">764</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="strposition_to_locate_sql-765"><a href="#strposition_to_locate_sql-765"><span class="linenos">765</span></a> <span class="s2">"LOCATE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"substr"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"position"</span><span class="p">)</span> -</span><span id="strposition_to_locate_sql-766"><a href="#strposition_to_locate_sql-766"><span class="linenos">766</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="strposition_to_locate_sql-744"><a href="#strposition_to_locate_sql-744"><span class="linenos">744</span></a><span class="k">def</span> <span class="nf">strposition_to_locate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">StrPosition</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="strposition_to_locate_sql-745"><a href="#strposition_to_locate_sql-745"><span class="linenos">745</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="strposition_to_locate_sql-746"><a href="#strposition_to_locate_sql-746"><span class="linenos">746</span></a> <span class="s2">"LOCATE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"substr"</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"position"</span><span class="p">)</span> +</span><span id="strposition_to_locate_sql-747"><a href="#strposition_to_locate_sql-747"><span class="linenos">747</span></a> <span class="p">)</span> </span></pre></div> @@ -4376,12 +4316,12 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#left_to_substring_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="left_to_substring_sql-769"><a href="#left_to_substring_sql-769"><span class="linenos">769</span></a><span class="k">def</span> <span class="nf">left_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="left_to_substring_sql-770"><a href="#left_to_substring_sql-770"><span class="linenos">770</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="left_to_substring_sql-771"><a href="#left_to_substring_sql-771"><span class="linenos">771</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> -</span><span id="left_to_substring_sql-772"><a href="#left_to_substring_sql-772"><span class="linenos">772</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="mi">1</span><span class="p">),</span> <span class="n">length</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> -</span><span id="left_to_substring_sql-773"><a href="#left_to_substring_sql-773"><span class="linenos">773</span></a> <span class="p">)</span> -</span><span id="left_to_substring_sql-774"><a href="#left_to_substring_sql-774"><span class="linenos">774</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="left_to_substring_sql-750"><a href="#left_to_substring_sql-750"><span class="linenos">750</span></a><span class="k">def</span> <span class="nf">left_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="left_to_substring_sql-751"><a href="#left_to_substring_sql-751"><span class="linenos">751</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="left_to_substring_sql-752"><a href="#left_to_substring_sql-752"><span class="linenos">752</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> +</span><span id="left_to_substring_sql-753"><a href="#left_to_substring_sql-753"><span class="linenos">753</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="o">.</span><span class="n">number</span><span class="p">(</span><span class="mi">1</span><span class="p">),</span> <span class="n">length</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> +</span><span id="left_to_substring_sql-754"><a href="#left_to_substring_sql-754"><span class="linenos">754</span></a> <span class="p">)</span> +</span><span id="left_to_substring_sql-755"><a href="#left_to_substring_sql-755"><span class="linenos">755</span></a> <span class="p">)</span> </span></pre></div> @@ -4399,13 +4339,13 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#right_to_substring_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="right_to_substring_sql-777"><a href="#right_to_substring_sql-777"><span class="linenos">777</span></a><span class="k">def</span> <span class="nf">right_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="right_to_substring_sql-778"><a href="#right_to_substring_sql-778"><span class="linenos">778</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="right_to_substring_sql-779"><a href="#right_to_substring_sql-779"><span class="linenos">779</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> -</span><span id="right_to_substring_sql-780"><a href="#right_to_substring_sql-780"><span class="linenos">780</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> -</span><span id="right_to_substring_sql-781"><a href="#right_to_substring_sql-781"><span class="linenos">781</span></a> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Length</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> <span class="o">-</span> <span class="n">exp</span><span class="o">.</span><span class="n">paren</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> <span class="o">-</span> <span class="mi">1</span><span class="p">),</span> -</span><span id="right_to_substring_sql-782"><a href="#right_to_substring_sql-782"><span class="linenos">782</span></a> <span class="p">)</span> -</span><span id="right_to_substring_sql-783"><a href="#right_to_substring_sql-783"><span class="linenos">783</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="right_to_substring_sql-758"><a href="#right_to_substring_sql-758"><span class="linenos">758</span></a><span class="k">def</span> <span class="nf">right_to_substring_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Left</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="right_to_substring_sql-759"><a href="#right_to_substring_sql-759"><span class="linenos">759</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="right_to_substring_sql-760"><a href="#right_to_substring_sql-760"><span class="linenos">760</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Substring</span><span class="p">(</span> +</span><span id="right_to_substring_sql-761"><a href="#right_to_substring_sql-761"><span class="linenos">761</span></a> <span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> +</span><span id="right_to_substring_sql-762"><a href="#right_to_substring_sql-762"><span class="linenos">762</span></a> <span class="n">start</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Length</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> <span class="o">-</span> <span class="n">exp</span><span class="o">.</span><span class="n">paren</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span> <span class="o">-</span> <span class="mi">1</span><span class="p">),</span> +</span><span id="right_to_substring_sql-763"><a href="#right_to_substring_sql-763"><span class="linenos">763</span></a> <span class="p">)</span> +</span><span id="right_to_substring_sql-764"><a href="#right_to_substring_sql-764"><span class="linenos">764</span></a> <span class="p">)</span> </span></pre></div> @@ -4423,8 +4363,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#timestrtotime_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="timestrtotime_sql-786"><a href="#timestrtotime_sql-786"><span class="linenos">786</span></a><span class="k">def</span> <span class="nf">timestrtotime_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeStrToTime</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="timestrtotime_sql-787"><a href="#timestrtotime_sql-787"><span class="linenos">787</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"timestamp"</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="timestrtotime_sql-767"><a href="#timestrtotime_sql-767"><span class="linenos">767</span></a><span class="k">def</span> <span class="nf">timestrtotime_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeStrToTime</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="timestrtotime_sql-768"><a href="#timestrtotime_sql-768"><span class="linenos">768</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"timestamp"</span><span class="p">))</span> </span></pre></div> @@ -4442,8 +4382,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#datestrtodate_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="datestrtodate_sql-790"><a href="#datestrtodate_sql-790"><span class="linenos">790</span></a><span class="k">def</span> <span class="nf">datestrtodate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateStrToDate</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="datestrtodate_sql-791"><a href="#datestrtodate_sql-791"><span class="linenos">791</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="datestrtodate_sql-771"><a href="#datestrtodate_sql-771"><span class="linenos">771</span></a><span class="k">def</span> <span class="nf">datestrtodate_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateStrToDate</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="datestrtodate_sql-772"><a href="#datestrtodate_sql-772"><span class="linenos">772</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> </span></pre></div> @@ -4461,14 +4401,14 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#encode_decode_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="encode_decode_sql-795"><a href="#encode_decode_sql-795"><span class="linenos">795</span></a><span class="k">def</span> <span class="nf">encode_decode_sql</span><span class="p">(</span> -</span><span id="encode_decode_sql-796"><a href="#encode_decode_sql-796"><span class="linenos">796</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">replace</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="encode_decode_sql-797"><a href="#encode_decode_sql-797"><span class="linenos">797</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="encode_decode_sql-798"><a href="#encode_decode_sql-798"><span class="linenos">798</span></a> <span class="n">charset</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"charset"</span><span class="p">)</span> -</span><span id="encode_decode_sql-799"><a href="#encode_decode_sql-799"><span class="linenos">799</span></a> <span class="k">if</span> <span class="n">charset</span> <span class="ow">and</span> <span class="n">charset</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="o">!=</span> <span class="s2">"utf-8"</span><span class="p">:</span> -</span><span id="encode_decode_sql-800"><a href="#encode_decode_sql-800"><span class="linenos">800</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Expected utf-8 character set, got </span><span class="si">{</span><span class="n">charset</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> -</span><span id="encode_decode_sql-801"><a href="#encode_decode_sql-801"><span class="linenos">801</span></a> -</span><span id="encode_decode_sql-802"><a href="#encode_decode_sql-802"><span class="linenos">802</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"replace"</span><span class="p">)</span> <span class="k">if</span> <span class="n">replace</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="encode_decode_sql-776"><a href="#encode_decode_sql-776"><span class="linenos">776</span></a><span class="k">def</span> <span class="nf">encode_decode_sql</span><span class="p">(</span> +</span><span id="encode_decode_sql-777"><a href="#encode_decode_sql-777"><span class="linenos">777</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">replace</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="encode_decode_sql-778"><a href="#encode_decode_sql-778"><span class="linenos">778</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="encode_decode_sql-779"><a href="#encode_decode_sql-779"><span class="linenos">779</span></a> <span class="n">charset</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"charset"</span><span class="p">)</span> +</span><span id="encode_decode_sql-780"><a href="#encode_decode_sql-780"><span class="linenos">780</span></a> <span class="k">if</span> <span class="n">charset</span> <span class="ow">and</span> <span class="n">charset</span><span class="o">.</span><span class="n">name</span><span class="o">.</span><span class="n">lower</span><span class="p">()</span> <span class="o">!=</span> <span class="s2">"utf-8"</span><span class="p">:</span> +</span><span id="encode_decode_sql-781"><a href="#encode_decode_sql-781"><span class="linenos">781</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Expected utf-8 character set, got </span><span class="si">{</span><span class="n">charset</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> +</span><span id="encode_decode_sql-782"><a href="#encode_decode_sql-782"><span class="linenos">782</span></a> +</span><span id="encode_decode_sql-783"><a href="#encode_decode_sql-783"><span class="linenos">783</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"replace"</span><span class="p">)</span> <span class="k">if</span> <span class="n">replace</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> </span></pre></div> @@ -4486,9 +4426,9 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#min_or_least"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="min_or_least-805"><a href="#min_or_least-805"><span class="linenos">805</span></a><span class="k">def</span> <span class="nf">min_or_least</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="min_or_least-806"><a href="#min_or_least-806"><span class="linenos">806</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"LEAST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MIN"</span> -</span><span id="min_or_least-807"><a href="#min_or_least-807"><span class="linenos">807</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="min_or_least-786"><a href="#min_or_least-786"><span class="linenos">786</span></a><span class="k">def</span> <span class="nf">min_or_least</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="min_or_least-787"><a href="#min_or_least-787"><span class="linenos">787</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"LEAST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MIN"</span> +</span><span id="min_or_least-788"><a href="#min_or_least-788"><span class="linenos">788</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> </span></pre></div> @@ -4506,9 +4446,9 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#max_or_greatest"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="max_or_greatest-810"><a href="#max_or_greatest-810"><span class="linenos">810</span></a><span class="k">def</span> <span class="nf">max_or_greatest</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="max_or_greatest-811"><a href="#max_or_greatest-811"><span class="linenos">811</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"GREATEST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MAX"</span> -</span><span id="max_or_greatest-812"><a href="#max_or_greatest-812"><span class="linenos">812</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="max_or_greatest-791"><a href="#max_or_greatest-791"><span class="linenos">791</span></a><span class="k">def</span> <span class="nf">max_or_greatest</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="max_or_greatest-792"><a href="#max_or_greatest-792"><span class="linenos">792</span></a> <span class="n">name</span> <span class="o">=</span> <span class="s2">"GREATEST"</span> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> <span class="k">else</span> <span class="s2">"MAX"</span> +</span><span id="max_or_greatest-793"><a href="#max_or_greatest-793"><span class="linenos">793</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> </span></pre></div> @@ -4526,14 +4466,14 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#count_if_to_sum"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="count_if_to_sum-815"><a href="#count_if_to_sum-815"><span class="linenos">815</span></a><span class="k">def</span> <span class="nf">count_if_to_sum</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">CountIf</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="count_if_to_sum-816"><a href="#count_if_to_sum-816"><span class="linenos">816</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> -</span><span id="count_if_to_sum-817"><a href="#count_if_to_sum-817"><span class="linenos">817</span></a> -</span><span id="count_if_to_sum-818"><a href="#count_if_to_sum-818"><span class="linenos">818</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">):</span> -</span><span id="count_if_to_sum-819"><a href="#count_if_to_sum-819"><span class="linenos">819</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> -</span><span id="count_if_to_sum-820"><a href="#count_if_to_sum-820"><span class="linenos">820</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"DISTINCT is not supported when converting COUNT_IF to SUM"</span><span class="p">)</span> -</span><span id="count_if_to_sum-821"><a href="#count_if_to_sum-821"><span class="linenos">821</span></a> -</span><span id="count_if_to_sum-822"><a href="#count_if_to_sum-822"><span class="linenos">822</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"sum"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"if"</span><span class="p">,</span> <span class="n">cond</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="count_if_to_sum-796"><a href="#count_if_to_sum-796"><span class="linenos">796</span></a><span class="k">def</span> <span class="nf">count_if_to_sum</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">CountIf</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="count_if_to_sum-797"><a href="#count_if_to_sum-797"><span class="linenos">797</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span> +</span><span id="count_if_to_sum-798"><a href="#count_if_to_sum-798"><span class="linenos">798</span></a> +</span><span id="count_if_to_sum-799"><a href="#count_if_to_sum-799"><span class="linenos">799</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">):</span> +</span><span id="count_if_to_sum-800"><a href="#count_if_to_sum-800"><span class="linenos">800</span></a> <span class="n">cond</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> +</span><span id="count_if_to_sum-801"><a href="#count_if_to_sum-801"><span class="linenos">801</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"DISTINCT is not supported when converting COUNT_IF to SUM"</span><span class="p">)</span> +</span><span id="count_if_to_sum-802"><a href="#count_if_to_sum-802"><span class="linenos">802</span></a> +</span><span id="count_if_to_sum-803"><a href="#count_if_to_sum-803"><span class="linenos">803</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"sum"</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"if"</span><span class="p">,</span> <span class="n">cond</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> </span></pre></div> @@ -4551,21 +4491,21 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#trim_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="trim_sql-825"><a href="#trim_sql-825"><span class="linenos">825</span></a><span class="k">def</span> <span class="nf">trim_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Trim</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="trim_sql-826"><a href="#trim_sql-826"><span class="linenos">826</span></a> <span class="n">target</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> -</span><span id="trim_sql-827"><a href="#trim_sql-827"><span class="linenos">827</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"position"</span><span class="p">)</span> -</span><span id="trim_sql-828"><a href="#trim_sql-828"><span class="linenos">828</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">)</span> -</span><span id="trim_sql-829"><a href="#trim_sql-829"><span class="linenos">829</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"collation"</span><span class="p">)</span> -</span><span id="trim_sql-830"><a href="#trim_sql-830"><span class="linenos">830</span></a> -</span><span id="trim_sql-831"><a href="#trim_sql-831"><span class="linenos">831</span></a> <span class="c1"># Use TRIM/LTRIM/RTRIM syntax if the expression isn't database-specific</span> -</span><span id="trim_sql-832"><a href="#trim_sql-832"><span class="linenos">832</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">remove_chars</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">collation</span><span class="p">:</span> -</span><span id="trim_sql-833"><a href="#trim_sql-833"><span class="linenos">833</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">trim_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="trim_sql-834"><a href="#trim_sql-834"><span class="linenos">834</span></a> -</span><span id="trim_sql-835"><a href="#trim_sql-835"><span class="linenos">835</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="trim_sql-836"><a href="#trim_sql-836"><span class="linenos">836</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">remove_chars</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="trim_sql-837"><a href="#trim_sql-837"><span class="linenos">837</span></a> <span class="n">from_part</span> <span class="o">=</span> <span class="s2">"FROM "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="ow">or</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="trim_sql-838"><a href="#trim_sql-838"><span class="linenos">838</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">" COLLATE </span><span class="si">{</span><span class="n">collation</span><span class="si">}</span><span class="s2">"</span> <span class="k">if</span> <span class="n">collation</span> <span class="k">else</span> <span class="s2">""</span> -</span><span id="trim_sql-839"><a href="#trim_sql-839"><span class="linenos">839</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"TRIM(</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}{</span><span class="n">remove_chars</span><span class="si">}{</span><span class="n">from_part</span><span class="si">}{</span><span class="n">target</span><span class="si">}{</span><span class="n">collation</span><span class="si">}</span><span class="s2">)"</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="trim_sql-806"><a href="#trim_sql-806"><span class="linenos">806</span></a><span class="k">def</span> <span class="nf">trim_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Trim</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="trim_sql-807"><a href="#trim_sql-807"><span class="linenos">807</span></a> <span class="n">target</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">)</span> +</span><span id="trim_sql-808"><a href="#trim_sql-808"><span class="linenos">808</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"position"</span><span class="p">)</span> +</span><span id="trim_sql-809"><a href="#trim_sql-809"><span class="linenos">809</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">)</span> +</span><span id="trim_sql-810"><a href="#trim_sql-810"><span class="linenos">810</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"collation"</span><span class="p">)</span> +</span><span id="trim_sql-811"><a href="#trim_sql-811"><span class="linenos">811</span></a> +</span><span id="trim_sql-812"><a href="#trim_sql-812"><span class="linenos">812</span></a> <span class="c1"># Use TRIM/LTRIM/RTRIM syntax if the expression isn't database-specific</span> +</span><span id="trim_sql-813"><a href="#trim_sql-813"><span class="linenos">813</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">remove_chars</span> <span class="ow">and</span> <span class="ow">not</span> <span class="n">collation</span><span class="p">:</span> +</span><span id="trim_sql-814"><a href="#trim_sql-814"><span class="linenos">814</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">trim_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="trim_sql-815"><a href="#trim_sql-815"><span class="linenos">815</span></a> +</span><span id="trim_sql-816"><a href="#trim_sql-816"><span class="linenos">816</span></a> <span class="n">trim_type</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="trim_sql-817"><a href="#trim_sql-817"><span class="linenos">817</span></a> <span class="n">remove_chars</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="n">remove_chars</span><span class="si">}</span><span class="s2"> "</span> <span class="k">if</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="trim_sql-818"><a href="#trim_sql-818"><span class="linenos">818</span></a> <span class="n">from_part</span> <span class="o">=</span> <span class="s2">"FROM "</span> <span class="k">if</span> <span class="n">trim_type</span> <span class="ow">or</span> <span class="n">remove_chars</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="trim_sql-819"><a href="#trim_sql-819"><span class="linenos">819</span></a> <span class="n">collation</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">" COLLATE </span><span class="si">{</span><span class="n">collation</span><span class="si">}</span><span class="s2">"</span> <span class="k">if</span> <span class="n">collation</span> <span class="k">else</span> <span class="s2">""</span> +</span><span id="trim_sql-820"><a href="#trim_sql-820"><span class="linenos">820</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"TRIM(</span><span class="si">{</span><span class="n">trim_type</span><span class="si">}{</span><span class="n">remove_chars</span><span class="si">}{</span><span class="n">from_part</span><span class="si">}{</span><span class="n">target</span><span class="si">}{</span><span class="n">collation</span><span class="si">}</span><span class="s2">)"</span> </span></pre></div> @@ -4583,8 +4523,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#str_to_time_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="str_to_time_sql-842"><a href="#str_to_time_sql-842"><span class="linenos">842</span></a><span class="k">def</span> <span class="nf">str_to_time_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="str_to_time_sql-843"><a href="#str_to_time_sql-843"><span class="linenos">843</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"STRPTIME"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">format_time</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="str_to_time_sql-823"><a href="#str_to_time_sql-823"><span class="linenos">823</span></a><span class="k">def</span> <span class="nf">str_to_time_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="str_to_time_sql-824"><a href="#str_to_time_sql-824"><span class="linenos">824</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"STRPTIME"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">format_time</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> </span></pre></div> @@ -4602,8 +4542,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#concat_to_dpipe_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="concat_to_dpipe_sql-846"><a href="#concat_to_dpipe_sql-846"><span class="linenos">846</span></a><span class="k">def</span> <span class="nf">concat_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Concat</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="concat_to_dpipe_sql-847"><a href="#concat_to_dpipe_sql-847"><span class="linenos">847</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">reduce</span><span class="p">(</span><span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="concat_to_dpipe_sql-827"><a href="#concat_to_dpipe_sql-827"><span class="linenos">827</span></a><span class="k">def</span> <span class="nf">concat_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Concat</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="concat_to_dpipe_sql-828"><a href="#concat_to_dpipe_sql-828"><span class="linenos">828</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">reduce</span><span class="p">(</span><span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">),</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">))</span> </span></pre></div> @@ -4621,14 +4561,14 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#concat_ws_to_dpipe_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="concat_ws_to_dpipe_sql-850"><a href="#concat_ws_to_dpipe_sql-850"><span class="linenos">850</span></a><span class="k">def</span> <span class="nf">concat_ws_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ConcatWs</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="concat_ws_to_dpipe_sql-851"><a href="#concat_ws_to_dpipe_sql-851"><span class="linenos">851</span></a> <span class="n">delim</span><span class="p">,</span> <span class="o">*</span><span class="n">rest_args</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> -</span><span id="concat_ws_to_dpipe_sql-852"><a href="#concat_ws_to_dpipe_sql-852"><span class="linenos">852</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> -</span><span id="concat_ws_to_dpipe_sql-853"><a href="#concat_ws_to_dpipe_sql-853"><span class="linenos">853</span></a> <span class="n">reduce</span><span class="p">(</span> -</span><span id="concat_ws_to_dpipe_sql-854"><a href="#concat_ws_to_dpipe_sql-854"><span class="linenos">854</span></a> <span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">delim</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">)),</span> -</span><span id="concat_ws_to_dpipe_sql-855"><a href="#concat_ws_to_dpipe_sql-855"><span class="linenos">855</span></a> <span class="n">rest_args</span><span class="p">,</span> -</span><span id="concat_ws_to_dpipe_sql-856"><a href="#concat_ws_to_dpipe_sql-856"><span class="linenos">856</span></a> <span class="p">)</span> -</span><span id="concat_ws_to_dpipe_sql-857"><a href="#concat_ws_to_dpipe_sql-857"><span class="linenos">857</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="concat_ws_to_dpipe_sql-831"><a href="#concat_ws_to_dpipe_sql-831"><span class="linenos">831</span></a><span class="k">def</span> <span class="nf">concat_ws_to_dpipe_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ConcatWs</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="concat_ws_to_dpipe_sql-832"><a href="#concat_ws_to_dpipe_sql-832"><span class="linenos">832</span></a> <span class="n">delim</span><span class="p">,</span> <span class="o">*</span><span class="n">rest_args</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span> +</span><span id="concat_ws_to_dpipe_sql-833"><a href="#concat_ws_to_dpipe_sql-833"><span class="linenos">833</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span> +</span><span id="concat_ws_to_dpipe_sql-834"><a href="#concat_ws_to_dpipe_sql-834"><span class="linenos">834</span></a> <span class="n">reduce</span><span class="p">(</span> +</span><span id="concat_ws_to_dpipe_sql-835"><a href="#concat_ws_to_dpipe_sql-835"><span class="linenos">835</span></a> <span class="k">lambda</span> <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">x</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">DPipe</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">delim</span><span class="p">,</span> <span class="n">expression</span><span class="o">=</span><span class="n">y</span><span class="p">)),</span> +</span><span id="concat_ws_to_dpipe_sql-836"><a href="#concat_ws_to_dpipe_sql-836"><span class="linenos">836</span></a> <span class="n">rest_args</span><span class="p">,</span> +</span><span id="concat_ws_to_dpipe_sql-837"><a href="#concat_ws_to_dpipe_sql-837"><span class="linenos">837</span></a> <span class="p">)</span> +</span><span id="concat_ws_to_dpipe_sql-838"><a href="#concat_ws_to_dpipe_sql-838"><span class="linenos">838</span></a> <span class="p">)</span> </span></pre></div> @@ -4646,14 +4586,14 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#regexp_extract_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="regexp_extract_sql-860"><a href="#regexp_extract_sql-860"><span class="linenos">860</span></a><span class="k">def</span> <span class="nf">regexp_extract_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpExtract</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="regexp_extract_sql-861"><a href="#regexp_extract_sql-861"><span class="linenos">861</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">)))</span> -</span><span id="regexp_extract_sql-862"><a href="#regexp_extract_sql-862"><span class="linenos">862</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> -</span><span id="regexp_extract_sql-863"><a href="#regexp_extract_sql-863"><span class="linenos">863</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_EXTRACT does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> -</span><span id="regexp_extract_sql-864"><a href="#regexp_extract_sql-864"><span class="linenos">864</span></a> -</span><span id="regexp_extract_sql-865"><a href="#regexp_extract_sql-865"><span class="linenos">865</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="regexp_extract_sql-866"><a href="#regexp_extract_sql-866"><span class="linenos">866</span></a> <span class="s2">"REGEXP_EXTRACT"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"group"</span><span class="p">)</span> -</span><span id="regexp_extract_sql-867"><a href="#regexp_extract_sql-867"><span class="linenos">867</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="regexp_extract_sql-841"><a href="#regexp_extract_sql-841"><span class="linenos">841</span></a><span class="k">def</span> <span class="nf">regexp_extract_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpExtract</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="regexp_extract_sql-842"><a href="#regexp_extract_sql-842"><span class="linenos">842</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span><span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">)))</span> +</span><span id="regexp_extract_sql-843"><a href="#regexp_extract_sql-843"><span class="linenos">843</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> +</span><span id="regexp_extract_sql-844"><a href="#regexp_extract_sql-844"><span class="linenos">844</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_EXTRACT does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> +</span><span id="regexp_extract_sql-845"><a href="#regexp_extract_sql-845"><span class="linenos">845</span></a> +</span><span id="regexp_extract_sql-846"><a href="#regexp_extract_sql-846"><span class="linenos">846</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="regexp_extract_sql-847"><a href="#regexp_extract_sql-847"><span class="linenos">847</span></a> <span class="s2">"REGEXP_EXTRACT"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"group"</span><span class="p">)</span> +</span><span id="regexp_extract_sql-848"><a href="#regexp_extract_sql-848"><span class="linenos">848</span></a> <span class="p">)</span> </span></pre></div> @@ -4671,16 +4611,16 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#regexp_replace_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="regexp_replace_sql-870"><a href="#regexp_replace_sql-870"><span class="linenos">870</span></a><span class="k">def</span> <span class="nf">regexp_replace_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpReplace</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="regexp_replace_sql-871"><a href="#regexp_replace_sql-871"><span class="linenos">871</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span> -</span><span id="regexp_replace_sql-872"><a href="#regexp_replace_sql-872"><span class="linenos">872</span></a> <span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">,</span> <span class="s2">"modifiers"</span><span class="p">))</span> -</span><span id="regexp_replace_sql-873"><a href="#regexp_replace_sql-873"><span class="linenos">873</span></a> <span class="p">)</span> -</span><span id="regexp_replace_sql-874"><a href="#regexp_replace_sql-874"><span class="linenos">874</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> -</span><span id="regexp_replace_sql-875"><a href="#regexp_replace_sql-875"><span class="linenos">875</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_REPLACE does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> -</span><span id="regexp_replace_sql-876"><a href="#regexp_replace_sql-876"><span class="linenos">876</span></a> -</span><span id="regexp_replace_sql-877"><a href="#regexp_replace_sql-877"><span class="linenos">877</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="regexp_replace_sql-878"><a href="#regexp_replace_sql-878"><span class="linenos">878</span></a> <span class="s2">"REGEXP_REPLACE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"replacement"</span><span class="p">]</span> -</span><span id="regexp_replace_sql-879"><a href="#regexp_replace_sql-879"><span class="linenos">879</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="regexp_replace_sql-851"><a href="#regexp_replace_sql-851"><span class="linenos">851</span></a><span class="k">def</span> <span class="nf">regexp_replace_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">RegexpReplace</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="regexp_replace_sql-852"><a href="#regexp_replace_sql-852"><span class="linenos">852</span></a> <span class="n">bad_args</span> <span class="o">=</span> <span class="nb">list</span><span class="p">(</span> +</span><span id="regexp_replace_sql-853"><a href="#regexp_replace_sql-853"><span class="linenos">853</span></a> <span class="nb">filter</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">,</span> <span class="p">(</span><span class="s2">"position"</span><span class="p">,</span> <span class="s2">"occurrence"</span><span class="p">,</span> <span class="s2">"parameters"</span><span class="p">,</span> <span class="s2">"modifiers"</span><span class="p">))</span> +</span><span id="regexp_replace_sql-854"><a href="#regexp_replace_sql-854"><span class="linenos">854</span></a> <span class="p">)</span> +</span><span id="regexp_replace_sql-855"><a href="#regexp_replace_sql-855"><span class="linenos">855</span></a> <span class="k">if</span> <span class="n">bad_args</span><span class="p">:</span> +</span><span id="regexp_replace_sql-856"><a href="#regexp_replace_sql-856"><span class="linenos">856</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"REGEXP_REPLACE does not support the following arg(s): </span><span class="si">{</span><span class="n">bad_args</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> +</span><span id="regexp_replace_sql-857"><a href="#regexp_replace_sql-857"><span class="linenos">857</span></a> +</span><span id="regexp_replace_sql-858"><a href="#regexp_replace_sql-858"><span class="linenos">858</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="regexp_replace_sql-859"><a href="#regexp_replace_sql-859"><span class="linenos">859</span></a> <span class="s2">"REGEXP_REPLACE"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"replacement"</span><span class="p">]</span> +</span><span id="regexp_replace_sql-860"><a href="#regexp_replace_sql-860"><span class="linenos">860</span></a> <span class="p">)</span> </span></pre></div> @@ -4698,28 +4638,28 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#pivot_column_names"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="pivot_column_names-882"><a href="#pivot_column_names-882"><span class="linenos">882</span></a><span class="k">def</span> <span class="nf">pivot_column_names</span><span class="p">(</span><span class="n">aggregations</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="n">dialect</span><span class="p">:</span> <span class="n">DialectType</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> -</span><span id="pivot_column_names-883"><a href="#pivot_column_names-883"><span class="linenos">883</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="pivot_column_names-884"><a href="#pivot_column_names-884"><span class="linenos">884</span></a> <span class="k">for</span> <span class="n">agg</span> <span class="ow">in</span> <span class="n">aggregations</span><span class="p">:</span> -</span><span id="pivot_column_names-885"><a href="#pivot_column_names-885"><span class="linenos">885</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">agg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Alias</span><span class="p">):</span> -</span><span id="pivot_column_names-886"><a href="#pivot_column_names-886"><span class="linenos">886</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg</span><span class="o">.</span><span class="n">alias</span><span class="p">)</span> -</span><span id="pivot_column_names-887"><a href="#pivot_column_names-887"><span class="linenos">887</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="pivot_column_names-888"><a href="#pivot_column_names-888"><span class="linenos">888</span></a><span class="w"> </span><span class="sd">"""</span> -</span><span id="pivot_column_names-889"><a href="#pivot_column_names-889"><span class="linenos">889</span></a><span class="sd"> This case corresponds to aggregations without aliases being used as suffixes</span> -</span><span id="pivot_column_names-890"><a href="#pivot_column_names-890"><span class="linenos">890</span></a><span class="sd"> (e.g. col_avg(foo)). We need to unquote identifiers because they're going to</span> -</span><span id="pivot_column_names-891"><a href="#pivot_column_names-891"><span class="linenos">891</span></a><span class="sd"> be quoted in the base parser's `_parse_pivot` method, due to `to_identifier`.</span> -</span><span id="pivot_column_names-892"><a href="#pivot_column_names-892"><span class="linenos">892</span></a><span class="sd"> Otherwise, we'd end up with `col_avg(`foo`)` (notice the double quotes).</span> -</span><span id="pivot_column_names-893"><a href="#pivot_column_names-893"><span class="linenos">893</span></a><span class="sd"> """</span> -</span><span id="pivot_column_names-894"><a href="#pivot_column_names-894"><span class="linenos">894</span></a> <span class="n">agg_all_unquoted</span> <span class="o">=</span> <span class="n">agg</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> -</span><span id="pivot_column_names-895"><a href="#pivot_column_names-895"><span class="linenos">895</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> -</span><span id="pivot_column_names-896"><a href="#pivot_column_names-896"><span class="linenos">896</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">node</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">quoted</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span> -</span><span id="pivot_column_names-897"><a href="#pivot_column_names-897"><span class="linenos">897</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">)</span> -</span><span id="pivot_column_names-898"><a href="#pivot_column_names-898"><span class="linenos">898</span></a> <span class="k">else</span> <span class="n">node</span> -</span><span id="pivot_column_names-899"><a href="#pivot_column_names-899"><span class="linenos">899</span></a> <span class="p">)</span> -</span><span id="pivot_column_names-900"><a href="#pivot_column_names-900"><span class="linenos">900</span></a> <span class="p">)</span> -</span><span id="pivot_column_names-901"><a href="#pivot_column_names-901"><span class="linenos">901</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg_all_unquoted</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">dialect</span><span class="o">=</span><span class="n">dialect</span><span class="p">,</span> <span class="n">normalize_functions</span><span class="o">=</span><span class="s2">"lower"</span><span class="p">))</span> -</span><span id="pivot_column_names-902"><a href="#pivot_column_names-902"><span class="linenos">902</span></a> -</span><span id="pivot_column_names-903"><a href="#pivot_column_names-903"><span class="linenos">903</span></a> <span class="k">return</span> <span class="n">names</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="pivot_column_names-863"><a href="#pivot_column_names-863"><span class="linenos">863</span></a><span class="k">def</span> <span class="nf">pivot_column_names</span><span class="p">(</span><span class="n">aggregations</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">],</span> <span class="n">dialect</span><span class="p">:</span> <span class="n">DialectType</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> +</span><span id="pivot_column_names-864"><a href="#pivot_column_names-864"><span class="linenos">864</span></a> <span class="n">names</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="pivot_column_names-865"><a href="#pivot_column_names-865"><span class="linenos">865</span></a> <span class="k">for</span> <span class="n">agg</span> <span class="ow">in</span> <span class="n">aggregations</span><span class="p">:</span> +</span><span id="pivot_column_names-866"><a href="#pivot_column_names-866"><span class="linenos">866</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">agg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Alias</span><span class="p">):</span> +</span><span id="pivot_column_names-867"><a href="#pivot_column_names-867"><span class="linenos">867</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg</span><span class="o">.</span><span class="n">alias</span><span class="p">)</span> +</span><span id="pivot_column_names-868"><a href="#pivot_column_names-868"><span class="linenos">868</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="pivot_column_names-869"><a href="#pivot_column_names-869"><span class="linenos">869</span></a><span class="w"> </span><span class="sd">"""</span> +</span><span id="pivot_column_names-870"><a href="#pivot_column_names-870"><span class="linenos">870</span></a><span class="sd"> This case corresponds to aggregations without aliases being used as suffixes</span> +</span><span id="pivot_column_names-871"><a href="#pivot_column_names-871"><span class="linenos">871</span></a><span class="sd"> (e.g. col_avg(foo)). We need to unquote identifiers because they're going to</span> +</span><span id="pivot_column_names-872"><a href="#pivot_column_names-872"><span class="linenos">872</span></a><span class="sd"> be quoted in the base parser's `_parse_pivot` method, due to `to_identifier`.</span> +</span><span id="pivot_column_names-873"><a href="#pivot_column_names-873"><span class="linenos">873</span></a><span class="sd"> Otherwise, we'd end up with `col_avg(`foo`)` (notice the double quotes).</span> +</span><span id="pivot_column_names-874"><a href="#pivot_column_names-874"><span class="linenos">874</span></a><span class="sd"> """</span> +</span><span id="pivot_column_names-875"><a href="#pivot_column_names-875"><span class="linenos">875</span></a> <span class="n">agg_all_unquoted</span> <span class="o">=</span> <span class="n">agg</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> +</span><span id="pivot_column_names-876"><a href="#pivot_column_names-876"><span class="linenos">876</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> +</span><span id="pivot_column_names-877"><a href="#pivot_column_names-877"><span class="linenos">877</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">node</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">quoted</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span> +</span><span id="pivot_column_names-878"><a href="#pivot_column_names-878"><span class="linenos">878</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">)</span> +</span><span id="pivot_column_names-879"><a href="#pivot_column_names-879"><span class="linenos">879</span></a> <span class="k">else</span> <span class="n">node</span> +</span><span id="pivot_column_names-880"><a href="#pivot_column_names-880"><span class="linenos">880</span></a> <span class="p">)</span> +</span><span id="pivot_column_names-881"><a href="#pivot_column_names-881"><span class="linenos">881</span></a> <span class="p">)</span> +</span><span id="pivot_column_names-882"><a href="#pivot_column_names-882"><span class="linenos">882</span></a> <span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">agg_all_unquoted</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">dialect</span><span class="o">=</span><span class="n">dialect</span><span class="p">,</span> <span class="n">normalize_functions</span><span class="o">=</span><span class="s2">"lower"</span><span class="p">))</span> +</span><span id="pivot_column_names-883"><a href="#pivot_column_names-883"><span class="linenos">883</span></a> +</span><span id="pivot_column_names-884"><a href="#pivot_column_names-884"><span class="linenos">884</span></a> <span class="k">return</span> <span class="n">names</span> </span></pre></div> @@ -4737,8 +4677,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#binary_from_function"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="binary_from_function-906"><a href="#binary_from_function-906"><span class="linenos">906</span></a><span class="k">def</span> <span class="nf">binary_from_function</span><span class="p">(</span><span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">B</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">B</span><span class="p">]:</span> -</span><span id="binary_from_function-907"><a href="#binary_from_function-907"><span class="linenos">907</span></a> <span class="k">return</span> <span class="k">lambda</span> <span class="n">args</span><span class="p">:</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="binary_from_function-887"><a href="#binary_from_function-887"><span class="linenos">887</span></a><span class="k">def</span> <span class="nf">binary_from_function</span><span class="p">(</span><span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">B</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">B</span><span class="p">]:</span> +</span><span id="binary_from_function-888"><a href="#binary_from_function-888"><span class="linenos">888</span></a> <span class="k">return</span> <span class="k">lambda</span> <span class="n">args</span><span class="p">:</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span> </span></pre></div> @@ -4756,8 +4696,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#parse_timestamp_trunc"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="parse_timestamp_trunc-911"><a href="#parse_timestamp_trunc-911"><span class="linenos">911</span></a><span class="k">def</span> <span class="nf">parse_timestamp_trunc</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> -</span><span id="parse_timestamp_trunc-912"><a href="#parse_timestamp_trunc-912"><span class="linenos">912</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="parse_timestamp_trunc-892"><a href="#parse_timestamp_trunc-892"><span class="linenos">892</span></a><span class="k">def</span> <span class="nf">parse_timestamp_trunc</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">:</span> +</span><span id="parse_timestamp_trunc-893"><a href="#parse_timestamp_trunc-893"><span class="linenos">893</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">TimestampTrunc</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">1</span><span class="p">),</span> <span class="n">unit</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">))</span> </span></pre></div> @@ -4775,8 +4715,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#any_value_to_max_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="any_value_to_max_sql-915"><a href="#any_value_to_max_sql-915"><span class="linenos">915</span></a><span class="k">def</span> <span class="nf">any_value_to_max_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">AnyValue</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="any_value_to_max_sql-916"><a href="#any_value_to_max_sql-916"><span class="linenos">916</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"MAX"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="any_value_to_max_sql-896"><a href="#any_value_to_max_sql-896"><span class="linenos">896</span></a><span class="k">def</span> <span class="nf">any_value_to_max_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">AnyValue</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="any_value_to_max_sql-897"><a href="#any_value_to_max_sql-897"><span class="linenos">897</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"MAX"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> </span></pre></div> @@ -4794,10 +4734,10 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#bool_xor_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="bool_xor_sql-919"><a href="#bool_xor_sql-919"><span class="linenos">919</span></a><span class="k">def</span> <span class="nf">bool_xor_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Xor</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="bool_xor_sql-920"><a href="#bool_xor_sql-920"><span class="linenos">920</span></a> <span class="n">a</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">)</span> -</span><span id="bool_xor_sql-921"><a href="#bool_xor_sql-921"><span class="linenos">921</span></a> <span class="n">b</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">right</span><span class="p">)</span> -</span><span id="bool_xor_sql-922"><a href="#bool_xor_sql-922"><span class="linenos">922</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"(</span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2"> AND (NOT </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)) OR ((NOT </span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2">) AND </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)"</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="bool_xor_sql-900"><a href="#bool_xor_sql-900"><span class="linenos">900</span></a><span class="k">def</span> <span class="nf">bool_xor_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Xor</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="bool_xor_sql-901"><a href="#bool_xor_sql-901"><span class="linenos">901</span></a> <span class="n">a</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">)</span> +</span><span id="bool_xor_sql-902"><a href="#bool_xor_sql-902"><span class="linenos">902</span></a> <span class="n">b</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">right</span><span class="p">)</span> +</span><span id="bool_xor_sql-903"><a href="#bool_xor_sql-903"><span class="linenos">903</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"(</span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2"> AND (NOT </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)) OR ((NOT </span><span class="si">{</span><span class="n">a</span><span class="si">}</span><span class="s2">) AND </span><span class="si">{</span><span class="n">b</span><span class="si">}</span><span class="s2">)"</span> </span></pre></div> @@ -4815,10 +4755,10 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#is_parse_json"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="is_parse_json-925"><a href="#is_parse_json-925"><span class="linenos">925</span></a><span class="k">def</span> <span class="nf">is_parse_json</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span> -</span><span id="is_parse_json-926"><a href="#is_parse_json-926"><span class="linenos">926</span></a> <span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ParseJSON</span><span class="p">)</span> <span class="ow">or</span> <span class="p">(</span> -</span><span id="is_parse_json-927"><a href="#is_parse_json-927"><span class="linenos">927</span></a> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"json"</span><span class="p">)</span> -</span><span id="is_parse_json-928"><a href="#is_parse_json-928"><span class="linenos">928</span></a> <span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="is_parse_json-906"><a href="#is_parse_json-906"><span class="linenos">906</span></a><span class="k">def</span> <span class="nf">is_parse_json</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">)</span> <span class="o">-></span> <span class="nb">bool</span><span class="p">:</span> +</span><span id="is_parse_json-907"><a href="#is_parse_json-907"><span class="linenos">907</span></a> <span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ParseJSON</span><span class="p">)</span> <span class="ow">or</span> <span class="p">(</span> +</span><span id="is_parse_json-908"><a href="#is_parse_json-908"><span class="linenos">908</span></a> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Cast</span><span class="p">)</span> <span class="ow">and</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="s2">"json"</span><span class="p">)</span> +</span><span id="is_parse_json-909"><a href="#is_parse_json-909"><span class="linenos">909</span></a> <span class="p">)</span> </span></pre></div> @@ -4836,8 +4776,8 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#isnull_to_is_null"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="isnull_to_is_null-931"><a href="#isnull_to_is_null-931"><span class="linenos">931</span></a><span class="k">def</span> <span class="nf">isnull_to_is_null</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> -</span><span id="isnull_to_is_null-932"><a href="#isnull_to_is_null-932"><span class="linenos">932</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">Paren</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Is</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">null</span><span class="p">()))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="isnull_to_is_null-912"><a href="#isnull_to_is_null-912"><span class="linenos">912</span></a><span class="k">def</span> <span class="nf">isnull_to_is_null</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Expression</span><span class="p">:</span> +</span><span id="isnull_to_is_null-913"><a href="#isnull_to_is_null-913"><span class="linenos">913</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">Paren</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">Is</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">null</span><span class="p">()))</span> </span></pre></div> @@ -4855,12 +4795,12 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#generatedasidentitycolumnconstraint_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="generatedasidentitycolumnconstraint_sql-935"><a href="#generatedasidentitycolumnconstraint_sql-935"><span class="linenos">935</span></a><span class="k">def</span> <span class="nf">generatedasidentitycolumnconstraint_sql</span><span class="p">(</span> -</span><span id="generatedasidentitycolumnconstraint_sql-936"><a href="#generatedasidentitycolumnconstraint_sql-936"><span class="linenos">936</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">GeneratedAsIdentityColumnConstraint</span> -</span><span id="generatedasidentitycolumnconstraint_sql-937"><a href="#generatedasidentitycolumnconstraint_sql-937"><span class="linenos">937</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="generatedasidentitycolumnconstraint_sql-938"><a href="#generatedasidentitycolumnconstraint_sql-938"><span class="linenos">938</span></a> <span class="n">start</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"start"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> -</span><span id="generatedasidentitycolumnconstraint_sql-939"><a href="#generatedasidentitycolumnconstraint_sql-939"><span class="linenos">939</span></a> <span class="n">increment</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"increment"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> -</span><span id="generatedasidentitycolumnconstraint_sql-940"><a href="#generatedasidentitycolumnconstraint_sql-940"><span class="linenos">940</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"IDENTITY(</span><span class="si">{</span><span class="n">start</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="n">increment</span><span class="si">}</span><span class="s2">)"</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="generatedasidentitycolumnconstraint_sql-916"><a href="#generatedasidentitycolumnconstraint_sql-916"><span class="linenos">916</span></a><span class="k">def</span> <span class="nf">generatedasidentitycolumnconstraint_sql</span><span class="p">(</span> +</span><span id="generatedasidentitycolumnconstraint_sql-917"><a href="#generatedasidentitycolumnconstraint_sql-917"><span class="linenos">917</span></a> <span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">GeneratedAsIdentityColumnConstraint</span> +</span><span id="generatedasidentitycolumnconstraint_sql-918"><a href="#generatedasidentitycolumnconstraint_sql-918"><span class="linenos">918</span></a><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="generatedasidentitycolumnconstraint_sql-919"><a href="#generatedasidentitycolumnconstraint_sql-919"><span class="linenos">919</span></a> <span class="n">start</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"start"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> +</span><span id="generatedasidentitycolumnconstraint_sql-920"><a href="#generatedasidentitycolumnconstraint_sql-920"><span class="linenos">920</span></a> <span class="n">increment</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="s2">"increment"</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">"1"</span> +</span><span id="generatedasidentitycolumnconstraint_sql-921"><a href="#generatedasidentitycolumnconstraint_sql-921"><span class="linenos">921</span></a> <span class="k">return</span> <span class="sa">f</span><span class="s2">"IDENTITY(</span><span class="si">{</span><span class="n">start</span><span class="si">}</span><span class="s2">, </span><span class="si">{</span><span class="n">increment</span><span class="si">}</span><span class="s2">)"</span> </span></pre></div> @@ -4878,14 +4818,14 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#arg_max_or_min_no_count"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="arg_max_or_min_no_count-943"><a href="#arg_max_or_min_no_count-943"><span class="linenos">943</span></a><span class="k">def</span> <span class="nf">arg_max_or_min_no_count</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="arg_max_or_min_no_count-944"><a href="#arg_max_or_min_no_count-944"><span class="linenos">944</span></a> <span class="k">def</span> <span class="nf">_arg_max_or_min_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="arg_max_or_min_no_count-945"><a href="#arg_max_or_min_no_count-945"><span class="linenos">945</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"count"</span><span class="p">):</span> -</span><span id="arg_max_or_min_no_count-946"><a href="#arg_max_or_min_no_count-946"><span class="linenos">946</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Only two arguments are supported in function </span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> -</span><span id="arg_max_or_min_no_count-947"><a href="#arg_max_or_min_no_count-947"><span class="linenos">947</span></a> -</span><span id="arg_max_or_min_no_count-948"><a href="#arg_max_or_min_no_count-948"><span class="linenos">948</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> -</span><span id="arg_max_or_min_no_count-949"><a href="#arg_max_or_min_no_count-949"><span class="linenos">949</span></a> -</span><span id="arg_max_or_min_no_count-950"><a href="#arg_max_or_min_no_count-950"><span class="linenos">950</span></a> <span class="k">return</span> <span class="n">_arg_max_or_min_sql</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="arg_max_or_min_no_count-924"><a href="#arg_max_or_min_no_count-924"><span class="linenos">924</span></a><span class="k">def</span> <span class="nf">arg_max_or_min_no_count</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="arg_max_or_min_no_count-925"><a href="#arg_max_or_min_no_count-925"><span class="linenos">925</span></a> <span class="k">def</span> <span class="nf">_arg_max_or_min_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMax</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ArgMin</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="arg_max_or_min_no_count-926"><a href="#arg_max_or_min_no_count-926"><span class="linenos">926</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"count"</span><span class="p">):</span> +</span><span id="arg_max_or_min_no_count-927"><a href="#arg_max_or_min_no_count-927"><span class="linenos">927</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="sa">f</span><span class="s2">"Only two arguments are supported in function </span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">."</span><span class="p">)</span> +</span><span id="arg_max_or_min_no_count-928"><a href="#arg_max_or_min_no_count-928"><span class="linenos">928</span></a> +</span><span id="arg_max_or_min_no_count-929"><a href="#arg_max_or_min_no_count-929"><span class="linenos">929</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> +</span><span id="arg_max_or_min_no_count-930"><a href="#arg_max_or_min_no_count-930"><span class="linenos">930</span></a> +</span><span id="arg_max_or_min_no_count-931"><a href="#arg_max_or_min_no_count-931"><span class="linenos">931</span></a> <span class="k">return</span> <span class="n">_arg_max_or_min_sql</span> </span></pre></div> @@ -4903,17 +4843,17 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#ts_or_ds_add_cast"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="ts_or_ds_add_cast-953"><a href="#ts_or_ds_add_cast-953"><span class="linenos">953</span></a><span class="k">def</span> <span class="nf">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">:</span> -</span><span id="ts_or_ds_add_cast-954"><a href="#ts_or_ds_add_cast-954"><span class="linenos">954</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span> -</span><span id="ts_or_ds_add_cast-955"><a href="#ts_or_ds_add_cast-955"><span class="linenos">955</span></a> -</span><span id="ts_or_ds_add_cast-956"><a href="#ts_or_ds_add_cast-956"><span class="linenos">956</span></a> <span class="n">return_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">return_type</span> -</span><span id="ts_or_ds_add_cast-957"><a href="#ts_or_ds_add_cast-957"><span class="linenos">957</span></a> <span class="k">if</span> <span class="n">return_type</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DATE</span><span class="p">):</span> -</span><span id="ts_or_ds_add_cast-958"><a href="#ts_or_ds_add_cast-958"><span class="linenos">958</span></a> <span class="c1"># If we need to cast to a DATE, we cast to TIMESTAMP first to make sure we</span> -</span><span id="ts_or_ds_add_cast-959"><a href="#ts_or_ds_add_cast-959"><span class="linenos">959</span></a> <span class="c1"># can truncate timestamp strings, because some dialects can't cast them to DATE</span> -</span><span id="ts_or_ds_add_cast-960"><a href="#ts_or_ds_add_cast-960"><span class="linenos">960</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">)</span> -</span><span id="ts_or_ds_add_cast-961"><a href="#ts_or_ds_add_cast-961"><span class="linenos">961</span></a> -</span><span id="ts_or_ds_add_cast-962"><a href="#ts_or_ds_add_cast-962"><span class="linenos">962</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">return_type</span><span class="p">))</span> -</span><span id="ts_or_ds_add_cast-963"><a href="#ts_or_ds_add_cast-963"><span class="linenos">963</span></a> <span class="k">return</span> <span class="n">expression</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="ts_or_ds_add_cast-934"><a href="#ts_or_ds_add_cast-934"><span class="linenos">934</span></a><span class="k">def</span> <span class="nf">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">:</span> +</span><span id="ts_or_ds_add_cast-935"><a href="#ts_or_ds_add_cast-935"><span class="linenos">935</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span> +</span><span id="ts_or_ds_add_cast-936"><a href="#ts_or_ds_add_cast-936"><span class="linenos">936</span></a> +</span><span id="ts_or_ds_add_cast-937"><a href="#ts_or_ds_add_cast-937"><span class="linenos">937</span></a> <span class="n">return_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">return_type</span> +</span><span id="ts_or_ds_add_cast-938"><a href="#ts_or_ds_add_cast-938"><span class="linenos">938</span></a> <span class="k">if</span> <span class="n">return_type</span><span class="o">.</span><span class="n">is_type</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DATE</span><span class="p">):</span> +</span><span id="ts_or_ds_add_cast-939"><a href="#ts_or_ds_add_cast-939"><span class="linenos">939</span></a> <span class="c1"># If we need to cast to a DATE, we cast to TIMESTAMP first to make sure we</span> +</span><span id="ts_or_ds_add_cast-940"><a href="#ts_or_ds_add_cast-940"><span class="linenos">940</span></a> <span class="c1"># can truncate timestamp strings, because some dialects can't cast them to DATE</span> +</span><span id="ts_or_ds_add_cast-941"><a href="#ts_or_ds_add_cast-941"><span class="linenos">941</span></a> <span class="n">this</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">TIMESTAMP</span><span class="p">)</span> +</span><span id="ts_or_ds_add_cast-942"><a href="#ts_or_ds_add_cast-942"><span class="linenos">942</span></a> +</span><span id="ts_or_ds_add_cast-943"><a href="#ts_or_ds_add_cast-943"><span class="linenos">943</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="n">return_type</span><span class="p">))</span> +</span><span id="ts_or_ds_add_cast-944"><a href="#ts_or_ds_add_cast-944"><span class="linenos">944</span></a> <span class="k">return</span> <span class="n">expression</span> </span></pre></div> @@ -4931,19 +4871,19 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#date_delta_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="date_delta_sql-966"><a href="#date_delta_sql-966"><span class="linenos">966</span></a><span class="k">def</span> <span class="nf">date_delta_sql</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">cast</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="date_delta_sql-967"><a href="#date_delta_sql-967"><span class="linenos">967</span></a> <span class="k">def</span> <span class="nf">_delta_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="date_delta_sql-968"><a href="#date_delta_sql-968"><span class="linenos">968</span></a> <span class="k">if</span> <span class="n">cast</span> <span class="ow">and</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">):</span> -</span><span id="date_delta_sql-969"><a href="#date_delta_sql-969"><span class="linenos">969</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="date_delta_sql-970"><a href="#date_delta_sql-970"><span class="linenos">970</span></a> -</span><span id="date_delta_sql-971"><a href="#date_delta_sql-971"><span class="linenos">971</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> -</span><span id="date_delta_sql-972"><a href="#date_delta_sql-972"><span class="linenos">972</span></a> <span class="n">name</span><span class="p">,</span> -</span><span id="date_delta_sql-973"><a href="#date_delta_sql-973"><span class="linenos">973</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> -</span><span id="date_delta_sql-974"><a href="#date_delta_sql-974"><span class="linenos">974</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> -</span><span id="date_delta_sql-975"><a href="#date_delta_sql-975"><span class="linenos">975</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> -</span><span id="date_delta_sql-976"><a href="#date_delta_sql-976"><span class="linenos">976</span></a> <span class="p">)</span> -</span><span id="date_delta_sql-977"><a href="#date_delta_sql-977"><span class="linenos">977</span></a> -</span><span id="date_delta_sql-978"><a href="#date_delta_sql-978"><span class="linenos">978</span></a> <span class="k">return</span> <span class="n">_delta_sql</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="date_delta_sql-947"><a href="#date_delta_sql-947"><span class="linenos">947</span></a><span class="k">def</span> <span class="nf">date_delta_sql</span><span class="p">(</span><span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">cast</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="date_delta_sql-948"><a href="#date_delta_sql-948"><span class="linenos">948</span></a> <span class="k">def</span> <span class="nf">_delta_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">DATE_ADD_OR_DIFF</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="date_delta_sql-949"><a href="#date_delta_sql-949"><span class="linenos">949</span></a> <span class="k">if</span> <span class="n">cast</span> <span class="ow">and</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">TsOrDsAdd</span><span class="p">):</span> +</span><span id="date_delta_sql-950"><a href="#date_delta_sql-950"><span class="linenos">950</span></a> <span class="n">expression</span> <span class="o">=</span> <span class="n">ts_or_ds_add_cast</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="date_delta_sql-951"><a href="#date_delta_sql-951"><span class="linenos">951</span></a> +</span><span id="date_delta_sql-952"><a href="#date_delta_sql-952"><span class="linenos">952</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span> +</span><span id="date_delta_sql-953"><a href="#date_delta_sql-953"><span class="linenos">953</span></a> <span class="n">name</span><span class="p">,</span> +</span><span id="date_delta_sql-954"><a href="#date_delta_sql-954"><span class="linenos">954</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">var</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">text</span><span class="p">(</span><span class="s2">"unit"</span><span class="p">)</span><span class="o">.</span><span class="n">upper</span><span class="p">()</span> <span class="ow">or</span> <span class="s2">"DAY"</span><span class="p">),</span> +</span><span id="date_delta_sql-955"><a href="#date_delta_sql-955"><span class="linenos">955</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> +</span><span id="date_delta_sql-956"><a href="#date_delta_sql-956"><span class="linenos">956</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> +</span><span id="date_delta_sql-957"><a href="#date_delta_sql-957"><span class="linenos">957</span></a> <span class="p">)</span> +</span><span id="date_delta_sql-958"><a href="#date_delta_sql-958"><span class="linenos">958</span></a> +</span><span id="date_delta_sql-959"><a href="#date_delta_sql-959"><span class="linenos">959</span></a> <span class="k">return</span> <span class="n">_delta_sql</span> </span></pre></div> @@ -4961,12 +4901,12 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#no_last_day_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="no_last_day_sql-981"><a href="#no_last_day_sql-981"><span class="linenos">981</span></a><span class="k">def</span> <span class="nf">no_last_day_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">LastDay</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="no_last_day_sql-982"><a href="#no_last_day_sql-982"><span class="linenos">982</span></a> <span class="n">trunc_curr_date</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_trunc"</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="no_last_day_sql-983"><a href="#no_last_day_sql-983"><span class="linenos">983</span></a> <span class="n">plus_one_month</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_add"</span><span class="p">,</span> <span class="n">trunc_curr_date</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">)</span> -</span><span id="no_last_day_sql-984"><a href="#no_last_day_sql-984"><span class="linenos">984</span></a> <span class="n">minus_one_day</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_sub"</span><span class="p">,</span> <span class="n">plus_one_month</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"day"</span><span class="p">)</span> -</span><span id="no_last_day_sql-985"><a href="#no_last_day_sql-985"><span class="linenos">985</span></a> -</span><span id="no_last_day_sql-986"><a href="#no_last_day_sql-986"><span class="linenos">986</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">minus_one_day</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="no_last_day_sql-962"><a href="#no_last_day_sql-962"><span class="linenos">962</span></a><span class="k">def</span> <span class="nf">no_last_day_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">LastDay</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="no_last_day_sql-963"><a href="#no_last_day_sql-963"><span class="linenos">963</span></a> <span class="n">trunc_curr_date</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_trunc"</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="no_last_day_sql-964"><a href="#no_last_day_sql-964"><span class="linenos">964</span></a> <span class="n">plus_one_month</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_add"</span><span class="p">,</span> <span class="n">trunc_curr_date</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"month"</span><span class="p">)</span> +</span><span id="no_last_day_sql-965"><a href="#no_last_day_sql-965"><span class="linenos">965</span></a> <span class="n">minus_one_day</span> <span class="o">=</span> <span class="n">exp</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="s2">"date_sub"</span><span class="p">,</span> <span class="n">plus_one_month</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="s2">"day"</span><span class="p">)</span> +</span><span id="no_last_day_sql-966"><a href="#no_last_day_sql-966"><span class="linenos">966</span></a> +</span><span id="no_last_day_sql-967"><a href="#no_last_day_sql-967"><span class="linenos">967</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">minus_one_day</span><span class="p">,</span> <span class="s2">"date"</span><span class="p">))</span> </span></pre></div> @@ -4984,29 +4924,29 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#merge_without_target_sql"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="merge_without_target_sql-989"><a href="#merge_without_target_sql-989"><span class="linenos"> 989</span></a><span class="k">def</span> <span class="nf">merge_without_target_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Merge</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="merge_without_target_sql-990"><a href="#merge_without_target_sql-990"><span class="linenos"> 990</span></a><span class="w"> </span><span class="sd">"""Remove table refs from columns in when statements."""</span> -</span><span id="merge_without_target_sql-991"><a href="#merge_without_target_sql-991"><span class="linenos"> 991</span></a> <span class="n">alias</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"alias"</span><span class="p">)</span> -</span><span id="merge_without_target_sql-992"><a href="#merge_without_target_sql-992"><span class="linenos"> 992</span></a> -</span><span id="merge_without_target_sql-993"><a href="#merge_without_target_sql-993"><span class="linenos"> 993</span></a> <span class="k">def</span> <span class="nf">normalize</span><span class="p">(</span><span class="n">identifier</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> -</span><span id="merge_without_target_sql-994"><a href="#merge_without_target_sql-994"><span class="linenos"> 994</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">normalize_identifier</span><span class="p">(</span><span class="n">identifier</span><span class="p">)</span><span class="o">.</span><span class="n">name</span> <span class="k">if</span> <span class="n">identifier</span> <span class="k">else</span> <span class="kc">None</span> -</span><span id="merge_without_target_sql-995"><a href="#merge_without_target_sql-995"><span class="linenos"> 995</span></a> -</span><span id="merge_without_target_sql-996"><a href="#merge_without_target_sql-996"><span class="linenos"> 996</span></a> <span class="n">targets</span> <span class="o">=</span> <span class="p">{</span><span class="n">normalize</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">this</span><span class="p">)}</span> -</span><span id="merge_without_target_sql-997"><a href="#merge_without_target_sql-997"><span class="linenos"> 997</span></a> -</span><span id="merge_without_target_sql-998"><a href="#merge_without_target_sql-998"><span class="linenos"> 998</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> -</span><span id="merge_without_target_sql-999"><a href="#merge_without_target_sql-999"><span class="linenos"> 999</span></a> <span class="n">targets</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="n">normalize</span><span class="p">(</span><span class="n">alias</span><span class="o">.</span><span class="n">this</span><span class="p">))</span> -</span><span id="merge_without_target_sql-1000"><a href="#merge_without_target_sql-1000"><span class="linenos">1000</span></a> -</span><span id="merge_without_target_sql-1001"><a href="#merge_without_target_sql-1001"><span class="linenos">1001</span></a> <span class="k">for</span> <span class="n">when</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> -</span><span id="merge_without_target_sql-1002"><a href="#merge_without_target_sql-1002"><span class="linenos">1002</span></a> <span class="n">when</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> -</span><span id="merge_without_target_sql-1003"><a href="#merge_without_target_sql-1003"><span class="linenos">1003</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> -</span><span id="merge_without_target_sql-1004"><a href="#merge_without_target_sql-1004"><span class="linenos">1004</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">column</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> -</span><span id="merge_without_target_sql-1005"><a href="#merge_without_target_sql-1005"><span class="linenos">1005</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Column</span><span class="p">)</span> <span class="ow">and</span> <span class="n">normalize</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"table"</span><span class="p">))</span> <span class="ow">in</span> <span class="n">targets</span> -</span><span id="merge_without_target_sql-1006"><a href="#merge_without_target_sql-1006"><span class="linenos">1006</span></a> <span class="k">else</span> <span class="n">node</span> -</span><span id="merge_without_target_sql-1007"><a href="#merge_without_target_sql-1007"><span class="linenos">1007</span></a> <span class="p">),</span> -</span><span id="merge_without_target_sql-1008"><a href="#merge_without_target_sql-1008"><span class="linenos">1008</span></a> <span class="n">copy</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> -</span><span id="merge_without_target_sql-1009"><a href="#merge_without_target_sql-1009"><span class="linenos">1009</span></a> <span class="p">)</span> -</span><span id="merge_without_target_sql-1010"><a href="#merge_without_target_sql-1010"><span class="linenos">1010</span></a> -</span><span id="merge_without_target_sql-1011"><a href="#merge_without_target_sql-1011"><span class="linenos">1011</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">merge_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="merge_without_target_sql-970"><a href="#merge_without_target_sql-970"><span class="linenos">970</span></a><span class="k">def</span> <span class="nf">merge_without_target_sql</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">Merge</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="merge_without_target_sql-971"><a href="#merge_without_target_sql-971"><span class="linenos">971</span></a><span class="w"> </span><span class="sd">"""Remove table refs from columns in when statements."""</span> +</span><span id="merge_without_target_sql-972"><a href="#merge_without_target_sql-972"><span class="linenos">972</span></a> <span class="n">alias</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"alias"</span><span class="p">)</span> +</span><span id="merge_without_target_sql-973"><a href="#merge_without_target_sql-973"><span class="linenos">973</span></a> +</span><span id="merge_without_target_sql-974"><a href="#merge_without_target_sql-974"><span class="linenos">974</span></a> <span class="k">def</span> <span class="nf">normalize</span><span class="p">(</span><span class="n">identifier</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">Identifier</span><span class="p">])</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span> +</span><span id="merge_without_target_sql-975"><a href="#merge_without_target_sql-975"><span class="linenos">975</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">normalize_identifier</span><span class="p">(</span><span class="n">identifier</span><span class="p">)</span><span class="o">.</span><span class="n">name</span> <span class="k">if</span> <span class="n">identifier</span> <span class="k">else</span> <span class="kc">None</span> +</span><span id="merge_without_target_sql-976"><a href="#merge_without_target_sql-976"><span class="linenos">976</span></a> +</span><span id="merge_without_target_sql-977"><a href="#merge_without_target_sql-977"><span class="linenos">977</span></a> <span class="n">targets</span> <span class="o">=</span> <span class="p">{</span><span class="n">normalize</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">this</span><span class="p">)}</span> +</span><span id="merge_without_target_sql-978"><a href="#merge_without_target_sql-978"><span class="linenos">978</span></a> +</span><span id="merge_without_target_sql-979"><a href="#merge_without_target_sql-979"><span class="linenos">979</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> +</span><span id="merge_without_target_sql-980"><a href="#merge_without_target_sql-980"><span class="linenos">980</span></a> <span class="n">targets</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="n">normalize</span><span class="p">(</span><span class="n">alias</span><span class="o">.</span><span class="n">this</span><span class="p">))</span> +</span><span id="merge_without_target_sql-981"><a href="#merge_without_target_sql-981"><span class="linenos">981</span></a> +</span><span id="merge_without_target_sql-982"><a href="#merge_without_target_sql-982"><span class="linenos">982</span></a> <span class="k">for</span> <span class="n">when</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> +</span><span id="merge_without_target_sql-983"><a href="#merge_without_target_sql-983"><span class="linenos">983</span></a> <span class="n">when</span><span class="o">.</span><span class="n">transform</span><span class="p">(</span> +</span><span id="merge_without_target_sql-984"><a href="#merge_without_target_sql-984"><span class="linenos">984</span></a> <span class="k">lambda</span> <span class="n">node</span><span class="p">:</span> <span class="p">(</span> +</span><span id="merge_without_target_sql-985"><a href="#merge_without_target_sql-985"><span class="linenos">985</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">column</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">this</span><span class="p">)</span> +</span><span id="merge_without_target_sql-986"><a href="#merge_without_target_sql-986"><span class="linenos">986</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">node</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Column</span><span class="p">)</span> <span class="ow">and</span> <span class="n">normalize</span><span class="p">(</span><span class="n">node</span><span class="o">.</span><span class="n">args</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"table"</span><span class="p">))</span> <span class="ow">in</span> <span class="n">targets</span> +</span><span id="merge_without_target_sql-987"><a href="#merge_without_target_sql-987"><span class="linenos">987</span></a> <span class="k">else</span> <span class="n">node</span> +</span><span id="merge_without_target_sql-988"><a href="#merge_without_target_sql-988"><span class="linenos">988</span></a> <span class="p">),</span> +</span><span id="merge_without_target_sql-989"><a href="#merge_without_target_sql-989"><span class="linenos">989</span></a> <span class="n">copy</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> +</span><span id="merge_without_target_sql-990"><a href="#merge_without_target_sql-990"><span class="linenos">990</span></a> <span class="p">)</span> +</span><span id="merge_without_target_sql-991"><a href="#merge_without_target_sql-991"><span class="linenos">991</span></a> +</span><span id="merge_without_target_sql-992"><a href="#merge_without_target_sql-992"><span class="linenos">992</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">merge_sql</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> </span></pre></div> @@ -5026,30 +4966,30 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#parse_json_extract_path"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="parse_json_extract_path-1014"><a href="#parse_json_extract_path-1014"><span class="linenos">1014</span></a><span class="k">def</span> <span class="nf">parse_json_extract_path</span><span class="p">(</span> -</span><span id="parse_json_extract_path-1015"><a href="#parse_json_extract_path-1015"><span class="linenos">1015</span></a> <span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">F</span><span class="p">],</span> <span class="n">zero_based_indexing</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="parse_json_extract_path-1016"><a href="#parse_json_extract_path-1016"><span class="linenos">1016</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">F</span><span class="p">]:</span> -</span><span id="parse_json_extract_path-1017"><a href="#parse_json_extract_path-1017"><span class="linenos">1017</span></a> <span class="k">def</span> <span class="nf">_parse_json_extract_path</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">F</span><span class="p">:</span> -</span><span id="parse_json_extract_path-1018"><a href="#parse_json_extract_path-1018"><span class="linenos">1018</span></a> <span class="n">segments</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathRoot</span><span class="p">()]</span> -</span><span id="parse_json_extract_path-1019"><a href="#parse_json_extract_path-1019"><span class="linenos">1019</span></a> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">:]:</span> -</span><span id="parse_json_extract_path-1020"><a href="#parse_json_extract_path-1020"><span class="linenos">1020</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">):</span> -</span><span id="parse_json_extract_path-1021"><a href="#parse_json_extract_path-1021"><span class="linenos">1021</span></a> <span class="c1"># We use the fallback parser because we can't really transpile non-literals safely</span> -</span><span id="parse_json_extract_path-1022"><a href="#parse_json_extract_path-1022"><span class="linenos">1022</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="o">.</span><span class="n">from_arg_list</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> -</span><span id="parse_json_extract_path-1023"><a href="#parse_json_extract_path-1023"><span class="linenos">1023</span></a> -</span><span id="parse_json_extract_path-1024"><a href="#parse_json_extract_path-1024"><span class="linenos">1024</span></a> <span class="n">text</span> <span class="o">=</span> <span class="n">arg</span><span class="o">.</span><span class="n">name</span> -</span><span id="parse_json_extract_path-1025"><a href="#parse_json_extract_path-1025"><span class="linenos">1025</span></a> <span class="k">if</span> <span class="n">is_int</span><span class="p">(</span><span class="n">text</span><span class="p">):</span> -</span><span id="parse_json_extract_path-1026"><a href="#parse_json_extract_path-1026"><span class="linenos">1026</span></a> <span class="n">index</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">text</span><span class="p">)</span> -</span><span id="parse_json_extract_path-1027"><a href="#parse_json_extract_path-1027"><span class="linenos">1027</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span> -</span><span id="parse_json_extract_path-1028"><a href="#parse_json_extract_path-1028"><span class="linenos">1028</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">index</span> <span class="k">if</span> <span class="n">zero_based_indexing</span> <span class="k">else</span> <span class="n">index</span> <span class="o">-</span> <span class="mi">1</span><span class="p">)</span> -</span><span id="parse_json_extract_path-1029"><a href="#parse_json_extract_path-1029"><span class="linenos">1029</span></a> <span class="p">)</span> -</span><span id="parse_json_extract_path-1030"><a href="#parse_json_extract_path-1030"><span class="linenos">1030</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="parse_json_extract_path-1031"><a href="#parse_json_extract_path-1031"><span class="linenos">1031</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">text</span><span class="p">))</span> -</span><span id="parse_json_extract_path-1032"><a href="#parse_json_extract_path-1032"><span class="linenos">1032</span></a> -</span><span id="parse_json_extract_path-1033"><a href="#parse_json_extract_path-1033"><span class="linenos">1033</span></a> <span class="c1"># This is done to avoid failing in the expression validator due to the arg count</span> -</span><span id="parse_json_extract_path-1034"><a href="#parse_json_extract_path-1034"><span class="linenos">1034</span></a> <span class="k">del</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">:]</span> -</span><span id="parse_json_extract_path-1035"><a href="#parse_json_extract_path-1035"><span class="linenos">1035</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">segments</span><span class="p">))</span> -</span><span id="parse_json_extract_path-1036"><a href="#parse_json_extract_path-1036"><span class="linenos">1036</span></a> -</span><span id="parse_json_extract_path-1037"><a href="#parse_json_extract_path-1037"><span class="linenos">1037</span></a> <span class="k">return</span> <span class="n">_parse_json_extract_path</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="parse_json_extract_path-995"><a href="#parse_json_extract_path-995"><span class="linenos"> 995</span></a><span class="k">def</span> <span class="nf">parse_json_extract_path</span><span class="p">(</span> +</span><span id="parse_json_extract_path-996"><a href="#parse_json_extract_path-996"><span class="linenos"> 996</span></a> <span class="n">expr_type</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Type</span><span class="p">[</span><span class="n">F</span><span class="p">],</span> <span class="n">zero_based_indexing</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="parse_json_extract_path-997"><a href="#parse_json_extract_path-997"><span class="linenos"> 997</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">],</span> <span class="n">F</span><span class="p">]:</span> +</span><span id="parse_json_extract_path-998"><a href="#parse_json_extract_path-998"><span class="linenos"> 998</span></a> <span class="k">def</span> <span class="nf">_parse_json_extract_path</span><span class="p">(</span><span class="n">args</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">)</span> <span class="o">-></span> <span class="n">F</span><span class="p">:</span> +</span><span id="parse_json_extract_path-999"><a href="#parse_json_extract_path-999"><span class="linenos"> 999</span></a> <span class="n">segments</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">List</span><span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">]</span> <span class="o">=</span> <span class="p">[</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathRoot</span><span class="p">()]</span> +</span><span id="parse_json_extract_path-1000"><a href="#parse_json_extract_path-1000"><span class="linenos">1000</span></a> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">:]:</span> +</span><span id="parse_json_extract_path-1001"><a href="#parse_json_extract_path-1001"><span class="linenos">1001</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">):</span> +</span><span id="parse_json_extract_path-1002"><a href="#parse_json_extract_path-1002"><span class="linenos">1002</span></a> <span class="c1"># We use the fallback parser because we can't really transpile non-literals safely</span> +</span><span id="parse_json_extract_path-1003"><a href="#parse_json_extract_path-1003"><span class="linenos">1003</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="o">.</span><span class="n">from_arg_list</span><span class="p">(</span><span class="n">args</span><span class="p">)</span> +</span><span id="parse_json_extract_path-1004"><a href="#parse_json_extract_path-1004"><span class="linenos">1004</span></a> +</span><span id="parse_json_extract_path-1005"><a href="#parse_json_extract_path-1005"><span class="linenos">1005</span></a> <span class="n">text</span> <span class="o">=</span> <span class="n">arg</span><span class="o">.</span><span class="n">name</span> +</span><span id="parse_json_extract_path-1006"><a href="#parse_json_extract_path-1006"><span class="linenos">1006</span></a> <span class="k">if</span> <span class="n">is_int</span><span class="p">(</span><span class="n">text</span><span class="p">):</span> +</span><span id="parse_json_extract_path-1007"><a href="#parse_json_extract_path-1007"><span class="linenos">1007</span></a> <span class="n">index</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">text</span><span class="p">)</span> +</span><span id="parse_json_extract_path-1008"><a href="#parse_json_extract_path-1008"><span class="linenos">1008</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span> +</span><span id="parse_json_extract_path-1009"><a href="#parse_json_extract_path-1009"><span class="linenos">1009</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">index</span> <span class="k">if</span> <span class="n">zero_based_indexing</span> <span class="k">else</span> <span class="n">index</span> <span class="o">-</span> <span class="mi">1</span><span class="p">)</span> +</span><span id="parse_json_extract_path-1010"><a href="#parse_json_extract_path-1010"><span class="linenos">1010</span></a> <span class="p">)</span> +</span><span id="parse_json_extract_path-1011"><a href="#parse_json_extract_path-1011"><span class="linenos">1011</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="parse_json_extract_path-1012"><a href="#parse_json_extract_path-1012"><span class="linenos">1012</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">text</span><span class="p">))</span> +</span><span id="parse_json_extract_path-1013"><a href="#parse_json_extract_path-1013"><span class="linenos">1013</span></a> +</span><span id="parse_json_extract_path-1014"><a href="#parse_json_extract_path-1014"><span class="linenos">1014</span></a> <span class="c1"># This is done to avoid failing in the expression validator due to the arg count</span> +</span><span id="parse_json_extract_path-1015"><a href="#parse_json_extract_path-1015"><span class="linenos">1015</span></a> <span class="k">del</span> <span class="n">args</span><span class="p">[</span><span class="mi">2</span><span class="p">:]</span> +</span><span id="parse_json_extract_path-1016"><a href="#parse_json_extract_path-1016"><span class="linenos">1016</span></a> <span class="k">return</span> <span class="n">expr_type</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">seq_get</span><span class="p">(</span><span class="n">args</span><span class="p">,</span> <span class="mi">0</span><span class="p">),</span> <span class="n">expression</span><span class="o">=</span><span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">(</span><span class="n">expressions</span><span class="o">=</span><span class="n">segments</span><span class="p">))</span> +</span><span id="parse_json_extract_path-1017"><a href="#parse_json_extract_path-1017"><span class="linenos">1017</span></a> +</span><span id="parse_json_extract_path-1018"><a href="#parse_json_extract_path-1018"><span class="linenos">1018</span></a> <span class="k">return</span> <span class="n">_parse_json_extract_path</span> </span></pre></div> @@ -5067,28 +5007,28 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#json_extract_segments"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="json_extract_segments-1040"><a href="#json_extract_segments-1040"><span class="linenos">1040</span></a><span class="k">def</span> <span class="nf">json_extract_segments</span><span class="p">(</span> -</span><span id="json_extract_segments-1041"><a href="#json_extract_segments-1041"><span class="linenos">1041</span></a> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">quoted_index</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> -</span><span id="json_extract_segments-1042"><a href="#json_extract_segments-1042"><span class="linenos">1042</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> -</span><span id="json_extract_segments-1043"><a href="#json_extract_segments-1043"><span class="linenos">1043</span></a> <span class="k">def</span> <span class="nf">_json_extract_segments</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="json_extract_segments-1044"><a href="#json_extract_segments-1044"><span class="linenos">1044</span></a> <span class="n">path</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span> -</span><span id="json_extract_segments-1045"><a href="#json_extract_segments-1045"><span class="linenos">1045</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">):</span> -</span><span id="json_extract_segments-1046"><a href="#json_extract_segments-1046"><span class="linenos">1046</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> -</span><span id="json_extract_segments-1047"><a href="#json_extract_segments-1047"><span class="linenos">1047</span></a> -</span><span id="json_extract_segments-1048"><a href="#json_extract_segments-1048"><span class="linenos">1048</span></a> <span class="n">segments</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="json_extract_segments-1049"><a href="#json_extract_segments-1049"><span class="linenos">1049</span></a> <span class="k">for</span> <span class="n">segment</span> <span class="ow">in</span> <span class="n">path</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> -</span><span id="json_extract_segments-1050"><a href="#json_extract_segments-1050"><span class="linenos">1050</span></a> <span class="n">path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">segment</span><span class="p">)</span> -</span><span id="json_extract_segments-1051"><a href="#json_extract_segments-1051"><span class="linenos">1051</span></a> <span class="k">if</span> <span class="n">path</span><span class="p">:</span> -</span><span id="json_extract_segments-1052"><a href="#json_extract_segments-1052"><span class="linenos">1052</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span> -</span><span id="json_extract_segments-1053"><a href="#json_extract_segments-1053"><span class="linenos">1053</span></a> <span class="n">quoted_index</span> <span class="ow">or</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">)</span> -</span><span id="json_extract_segments-1054"><a href="#json_extract_segments-1054"><span class="linenos">1054</span></a> <span class="p">):</span> -</span><span id="json_extract_segments-1055"><a href="#json_extract_segments-1055"><span class="linenos">1055</span></a> <span class="n">path</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_START</span><span class="si">}{</span><span class="n">path</span><span class="si">}{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_END</span><span class="si">}</span><span class="s2">"</span> -</span><span id="json_extract_segments-1056"><a href="#json_extract_segments-1056"><span class="linenos">1056</span></a> -</span><span id="json_extract_segments-1057"><a href="#json_extract_segments-1057"><span class="linenos">1057</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">path</span><span class="p">)</span> -</span><span id="json_extract_segments-1058"><a href="#json_extract_segments-1058"><span class="linenos">1058</span></a> -</span><span id="json_extract_segments-1059"><a href="#json_extract_segments-1059"><span class="linenos">1059</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="o">*</span><span class="n">segments</span><span class="p">)</span> -</span><span id="json_extract_segments-1060"><a href="#json_extract_segments-1060"><span class="linenos">1060</span></a> -</span><span id="json_extract_segments-1061"><a href="#json_extract_segments-1061"><span class="linenos">1061</span></a> <span class="k">return</span> <span class="n">_json_extract_segments</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="json_extract_segments-1021"><a href="#json_extract_segments-1021"><span class="linenos">1021</span></a><span class="k">def</span> <span class="nf">json_extract_segments</span><span class="p">(</span> +</span><span id="json_extract_segments-1022"><a href="#json_extract_segments-1022"><span class="linenos">1022</span></a> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">quoted_index</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">True</span> +</span><span id="json_extract_segments-1023"><a href="#json_extract_segments-1023"><span class="linenos">1023</span></a><span class="p">)</span> <span class="o">-></span> <span class="n">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">Generator</span><span class="p">,</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">],</span> <span class="nb">str</span><span class="p">]:</span> +</span><span id="json_extract_segments-1024"><a href="#json_extract_segments-1024"><span class="linenos">1024</span></a> <span class="k">def</span> <span class="nf">_json_extract_segments</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">JSON_EXTRACT_TYPE</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="json_extract_segments-1025"><a href="#json_extract_segments-1025"><span class="linenos">1025</span></a> <span class="n">path</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span> +</span><span id="json_extract_segments-1026"><a href="#json_extract_segments-1026"><span class="linenos">1026</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">path</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPath</span><span class="p">):</span> +</span><span id="json_extract_segments-1027"><a href="#json_extract_segments-1027"><span class="linenos">1027</span></a> <span class="k">return</span> <span class="n">rename_func</span><span class="p">(</span><span class="n">name</span><span class="p">)(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">)</span> +</span><span id="json_extract_segments-1028"><a href="#json_extract_segments-1028"><span class="linenos">1028</span></a> +</span><span id="json_extract_segments-1029"><a href="#json_extract_segments-1029"><span class="linenos">1029</span></a> <span class="n">segments</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="json_extract_segments-1030"><a href="#json_extract_segments-1030"><span class="linenos">1030</span></a> <span class="k">for</span> <span class="n">segment</span> <span class="ow">in</span> <span class="n">path</span><span class="o">.</span><span class="n">expressions</span><span class="p">:</span> +</span><span id="json_extract_segments-1031"><a href="#json_extract_segments-1031"><span class="linenos">1031</span></a> <span class="n">path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sql</span><span class="p">(</span><span class="n">segment</span><span class="p">)</span> +</span><span id="json_extract_segments-1032"><a href="#json_extract_segments-1032"><span class="linenos">1032</span></a> <span class="k">if</span> <span class="n">path</span><span class="p">:</span> +</span><span id="json_extract_segments-1033"><a href="#json_extract_segments-1033"><span class="linenos">1033</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathPart</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span> +</span><span id="json_extract_segments-1034"><a href="#json_extract_segments-1034"><span class="linenos">1034</span></a> <span class="n">quoted_index</span> <span class="ow">or</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">segment</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathSubscript</span><span class="p">)</span> +</span><span id="json_extract_segments-1035"><a href="#json_extract_segments-1035"><span class="linenos">1035</span></a> <span class="p">):</span> +</span><span id="json_extract_segments-1036"><a href="#json_extract_segments-1036"><span class="linenos">1036</span></a> <span class="n">path</span> <span class="o">=</span> <span class="sa">f</span><span class="s2">"</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_START</span><span class="si">}{</span><span class="n">path</span><span class="si">}{</span><span class="bp">self</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">QUOTE_END</span><span class="si">}</span><span class="s2">"</span> +</span><span id="json_extract_segments-1037"><a href="#json_extract_segments-1037"><span class="linenos">1037</span></a> +</span><span id="json_extract_segments-1038"><a href="#json_extract_segments-1038"><span class="linenos">1038</span></a> <span class="n">segments</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">path</span><span class="p">)</span> +</span><span id="json_extract_segments-1039"><a href="#json_extract_segments-1039"><span class="linenos">1039</span></a> +</span><span id="json_extract_segments-1040"><a href="#json_extract_segments-1040"><span class="linenos">1040</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">func</span><span class="p">(</span><span class="n">name</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="o">*</span><span class="n">segments</span><span class="p">)</span> +</span><span id="json_extract_segments-1041"><a href="#json_extract_segments-1041"><span class="linenos">1041</span></a> +</span><span id="json_extract_segments-1042"><a href="#json_extract_segments-1042"><span class="linenos">1042</span></a> <span class="k">return</span> <span class="n">_json_extract_segments</span> </span></pre></div> @@ -5106,11 +5046,11 @@ columns are removed from the create statement.</p> </div> <a class="headerlink" href="#json_path_key_only_name"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="json_path_key_only_name-1064"><a href="#json_path_key_only_name-1064"><span class="linenos">1064</span></a><span class="k">def</span> <span class="nf">json_path_key_only_name</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> -</span><span id="json_path_key_only_name-1065"><a href="#json_path_key_only_name-1065"><span class="linenos">1065</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathWildcard</span><span class="p">):</span> -</span><span id="json_path_key_only_name-1066"><a href="#json_path_key_only_name-1066"><span class="linenos">1066</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"Unsupported wildcard in JSONPathKey expression"</span><span class="p">)</span> -</span><span id="json_path_key_only_name-1067"><a href="#json_path_key_only_name-1067"><span class="linenos">1067</span></a> -</span><span id="json_path_key_only_name-1068"><a href="#json_path_key_only_name-1068"><span class="linenos">1068</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">name</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="json_path_key_only_name-1045"><a href="#json_path_key_only_name-1045"><span class="linenos">1045</span></a><span class="k">def</span> <span class="nf">json_path_key_only_name</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Generator</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathKey</span><span class="p">)</span> <span class="o">-></span> <span class="nb">str</span><span class="p">:</span> +</span><span id="json_path_key_only_name-1046"><a href="#json_path_key_only_name-1046"><span class="linenos">1046</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">JSONPathWildcard</span><span class="p">):</span> +</span><span id="json_path_key_only_name-1047"><a href="#json_path_key_only_name-1047"><span class="linenos">1047</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">unsupported</span><span class="p">(</span><span class="s2">"Unsupported wildcard in JSONPathKey expression"</span><span class="p">)</span> +</span><span id="json_path_key_only_name-1048"><a href="#json_path_key_only_name-1048"><span class="linenos">1048</span></a> +</span><span id="json_path_key_only_name-1049"><a href="#json_path_key_only_name-1049"><span class="linenos">1049</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">name</span> </span></pre></div> |