diff options
Diffstat (limited to 'docs/sqlglot/optimizer/annotate_types.html')
| -rw-r--r-- | docs/sqlglot/optimizer/annotate_types.html | 1434 |
1 files changed, 721 insertions, 713 deletions
diff --git a/docs/sqlglot/optimizer/annotate_types.html b/docs/sqlglot/optimizer/annotate_types.html index 72b4abc..de56fa9 100644 --- a/docs/sqlglot/optimizer/annotate_types.html +++ b/docs/sqlglot/optimizer/annotate_types.html @@ -366,334 +366,338 @@ </span><span id="L-263"><a href="#L-263"><span class="linenos">263</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_timeunit</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> </span><span id="L-264"><a href="#L-264"><span class="linenos">264</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> </span><span id="L-265"><a href="#L-265"><span class="linenos">265</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_div</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> -</span><span id="L-266"><a href="#L-266"><span class="linenos">266</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Filter</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">),</span> -</span><span id="L-267"><a href="#L-267"><span class="linenos">267</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">If</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"true"</span><span class="p">,</span> <span class="s2">"false"</span><span class="p">),</span> -</span><span id="L-268"><a href="#L-268"><span class="linenos">268</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">INTERVAL</span><span class="p">),</span> -</span><span id="L-269"><a href="#L-269"><span class="linenos">269</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Least</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="L-270"><a href="#L-270"><span class="linenos">270</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_literal</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> -</span><span id="L-271"><a href="#L-271"><span class="linenos">271</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> -</span><span id="L-272"><a href="#L-272"><span class="linenos">272</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="L-273"><a href="#L-273"><span class="linenos">273</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="L-274"><a href="#L-274"><span class="linenos">274</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Null</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">NULL</span><span class="p">),</span> -</span><span id="L-275"><a href="#L-275"><span class="linenos">275</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Nullif</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">),</span> -</span><span id="L-276"><a href="#L-276"><span class="linenos">276</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">UNKNOWN</span><span class="p">),</span> -</span><span id="L-277"><a href="#L-277"><span class="linenos">277</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Sum</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">promote</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> -</span><span id="L-278"><a href="#L-278"><span class="linenos">278</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">TryCast</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"to"</span><span class="p">]),</span> -</span><span id="L-279"><a href="#L-279"><span class="linenos">279</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> -</span><span id="L-280"><a href="#L-280"><span class="linenos">280</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Struct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">struct</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> -</span><span id="L-281"><a href="#L-281"><span class="linenos">281</span></a> <span class="p">}</span> -</span><span id="L-282"><a href="#L-282"><span class="linenos">282</span></a> -</span><span id="L-283"><a href="#L-283"><span class="linenos">283</span></a> <span class="n">NESTED_TYPES</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="L-284"><a href="#L-284"><span class="linenos">284</span></a> <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">ARRAY</span><span class="p">,</span> -</span><span id="L-285"><a href="#L-285"><span class="linenos">285</span></a> <span class="p">}</span> -</span><span id="L-286"><a href="#L-286"><span class="linenos">286</span></a> -</span><span id="L-287"><a href="#L-287"><span class="linenos">287</span></a> <span class="c1"># Specifies what types a given type can be coerced into (autofilled)</span> -</span><span id="L-288"><a href="#L-288"><span class="linenos">288</span></a> <span class="n">COERCES_TO</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]</span> <span class="o">=</span> <span class="p">{}</span> -</span><span id="L-289"><a href="#L-289"><span class="linenos">289</span></a> -</span><span id="L-290"><a href="#L-290"><span class="linenos">290</span></a> <span class="c1"># Coercion functions for binary operations.</span> -</span><span id="L-291"><a href="#L-291"><span class="linenos">291</span></a> <span class="c1"># Map of type pairs to a callable that takes both sides of the binary operation and returns the resulting type.</span> -</span><span id="L-292"><a href="#L-292"><span class="linenos">292</span></a> <span class="n">BINARY_COERCIONS</span><span class="p">:</span> <span class="n">BinaryCoercions</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="L-293"><a href="#L-293"><span class="linenos">293</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="L-294"><a href="#L-294"><span class="linenos">294</span></a> <span class="p">{</span> -</span><span id="L-295"><a href="#L-295"><span class="linenos">295</span></a> <span class="p">(</span><span class="n">t</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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date_literal</span><span class="p">(</span> -</span><span id="L-296"><a href="#L-296"><span class="linenos">296</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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-297"><a href="#L-297"><span class="linenos">297</span></a> <span class="p">)</span> -</span><span id="L-298"><a href="#L-298"><span class="linenos">298</span></a> <span class="k">for</span> <span class="n">t</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> -</span><span id="L-299"><a href="#L-299"><span class="linenos">299</span></a> <span class="p">}</span> -</span><span id="L-300"><a href="#L-300"><span class="linenos">300</span></a> <span class="p">),</span> -</span><span id="L-301"><a href="#L-301"><span class="linenos">301</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="L-302"><a href="#L-302"><span class="linenos">302</span></a> <span class="p">{</span> -</span><span id="L-303"><a href="#L-303"><span class="linenos">303</span></a> <span class="c1"># text + numeric will yield the numeric type to match most dialects' semantics</span> -</span><span id="L-304"><a href="#L-304"><span class="linenos">304</span></a> <span class="p">(</span><span class="n">text</span><span class="p">,</span> <span class="n">numeric</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span> -</span><span id="L-305"><a href="#L-305"><span class="linenos">305</span></a> <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="p">,</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> <span class="k">else</span> <span class="n">r</span><span class="o">.</span><span class="n">type</span> -</span><span id="L-306"><a href="#L-306"><span class="linenos">306</span></a> <span class="p">)</span> -</span><span id="L-307"><a href="#L-307"><span class="linenos">307</span></a> <span class="k">for</span> <span class="n">text</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> -</span><span id="L-308"><a href="#L-308"><span class="linenos">308</span></a> <span class="k">for</span> <span class="n">numeric</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> -</span><span id="L-309"><a href="#L-309"><span class="linenos">309</span></a> <span class="p">}</span> -</span><span id="L-310"><a href="#L-310"><span class="linenos">310</span></a> <span class="p">),</span> -</span><span id="L-311"><a href="#L-311"><span class="linenos">311</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="L-312"><a href="#L-312"><span class="linenos">312</span></a> <span class="p">{</span> -</span><span id="L-313"><a href="#L-313"><span class="linenos">313</span></a> <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 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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date</span><span class="p">(</span> -</span><span id="L-314"><a href="#L-314"><span class="linenos">314</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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-315"><a href="#L-315"><span class="linenos">315</span></a> <span class="p">),</span> -</span><span id="L-316"><a href="#L-316"><span class="linenos">316</span></a> <span class="p">}</span> -</span><span id="L-317"><a href="#L-317"><span class="linenos">317</span></a> <span class="p">),</span> -</span><span id="L-318"><a href="#L-318"><span class="linenos">318</span></a> <span class="p">}</span> -</span><span id="L-319"><a href="#L-319"><span class="linenos">319</span></a> -</span><span id="L-320"><a href="#L-320"><span class="linenos">320</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> -</span><span id="L-321"><a href="#L-321"><span class="linenos">321</span></a> <span class="bp">self</span><span class="p">,</span> -</span><span id="L-322"><a href="#L-322"><span class="linenos">322</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> -</span><span id="L-323"><a href="#L-323"><span class="linenos">323</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="L-324"><a href="#L-324"><span class="linenos">324</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="L-325"><a href="#L-325"><span class="linenos">325</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="L-326"><a href="#L-326"><span class="linenos">326</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> -</span><span id="L-327"><a href="#L-327"><span class="linenos">327</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> -</span><span id="L-328"><a href="#L-328"><span class="linenos">328</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> -</span><span id="L-329"><a href="#L-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> -</span><span id="L-330"><a href="#L-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> -</span><span id="L-331"><a href="#L-331"><span class="linenos">331</span></a> -</span><span id="L-332"><a href="#L-332"><span class="linenos">332</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> -</span><span id="L-333"><a href="#L-333"><span class="linenos">333</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> -</span><span id="L-334"><a href="#L-334"><span class="linenos">334</span></a> -</span><span id="L-335"><a href="#L-335"><span class="linenos">335</span></a> <span class="k">def</span> <span class="nf">_set_type</span><span class="p">(</span> -</span><span id="L-336"><a href="#L-336"><span class="linenos">336</span></a> <span class="bp">self</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">target_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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> -</span><span id="L-337"><a href="#L-337"><span class="linenos">337</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> -</span><span id="L-338"><a href="#L-338"><span class="linenos">338</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> <span class="o">=</span> <span class="n">target_type</span> <span class="c1"># type: ignore</span> -</span><span id="L-339"><a href="#L-339"><span class="linenos">339</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> -</span><span id="L-340"><a href="#L-340"><span class="linenos">340</span></a> -</span><span id="L-341"><a href="#L-341"><span class="linenos">341</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-342"><a href="#L-342"><span class="linenos">342</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> -</span><span id="L-343"><a href="#L-343"><span class="linenos">343</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> -</span><span id="L-344"><a href="#L-344"><span class="linenos">344</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> -</span><span id="L-345"><a href="#L-345"><span class="linenos">345</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> -</span><span id="L-346"><a href="#L-346"><span class="linenos">346</span></a> <span class="k">continue</span> -</span><span id="L-347"><a href="#L-347"><span class="linenos">347</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> -</span><span id="L-348"><a href="#L-348"><span class="linenos">348</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="L-349"><a href="#L-349"><span class="linenos">349</span></a> -</span><span id="L-350"><a href="#L-350"><span class="linenos">350</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> -</span><span id="L-351"><a href="#L-351"><span class="linenos">351</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> -</span><span id="L-352"><a href="#L-352"><span class="linenos">352</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> -</span><span id="L-353"><a href="#L-353"><span class="linenos">353</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-354"><a href="#L-354"><span class="linenos">354</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> -</span><span id="L-355"><a href="#L-355"><span class="linenos">355</span></a> -</span><span id="L-356"><a href="#L-356"><span class="linenos">356</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> -</span><span id="L-357"><a href="#L-357"><span class="linenos">357</span></a> <span class="k">continue</span> -</span><span id="L-358"><a href="#L-358"><span class="linenos">358</span></a> -</span><span id="L-359"><a href="#L-359"><span class="linenos">359</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="L-360"><a href="#L-360"><span class="linenos">360</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> -</span><span id="L-361"><a href="#L-361"><span class="linenos">361</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> -</span><span id="L-362"><a href="#L-362"><span class="linenos">362</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> -</span><span id="L-363"><a href="#L-363"><span class="linenos">363</span></a> <span class="n">values</span><span class="p">,</span> -</span><span id="L-364"><a href="#L-364"><span class="linenos">364</span></a> <span class="p">)</span> -</span><span id="L-365"><a href="#L-365"><span class="linenos">365</span></a> <span class="p">}</span> -</span><span id="L-366"><a href="#L-366"><span class="linenos">366</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-367"><a href="#L-367"><span class="linenos">367</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="L-368"><a href="#L-368"><span class="linenos">368</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> -</span><span id="L-369"><a href="#L-369"><span class="linenos">369</span></a> <span class="p">}</span> -</span><span id="L-370"><a href="#L-370"><span class="linenos">370</span></a> -</span><span id="L-371"><a href="#L-371"><span class="linenos">371</span></a> <span class="c1"># First annotate the current scope's column references</span> -</span><span id="L-372"><a href="#L-372"><span class="linenos">372</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> -</span><span id="L-373"><a href="#L-373"><span class="linenos">373</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> -</span><span id="L-374"><a href="#L-374"><span class="linenos">374</span></a> <span class="k">continue</span> -</span><span id="L-375"><a href="#L-375"><span class="linenos">375</span></a> -</span><span id="L-376"><a href="#L-376"><span class="linenos">376</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> -</span><span id="L-377"><a href="#L-377"><span class="linenos">377</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> -</span><span id="L-378"><a href="#L-378"><span class="linenos">378</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> -</span><span id="L-379"><a href="#L-379"><span class="linenos">379</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> -</span><span id="L-380"><a href="#L-380"><span class="linenos">380</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="L-381"><a href="#L-381"><span class="linenos">381</span></a> -</span><span id="L-382"><a href="#L-382"><span class="linenos">382</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> -</span><span id="L-383"><a href="#L-383"><span class="linenos">383</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-384"><a href="#L-384"><span class="linenos">384</span></a> -</span><span id="L-385"><a href="#L-385"><span class="linenos">385</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> -</span><span id="L-386"><a href="#L-386"><span class="linenos">386</span></a> -</span><span id="L-387"><a href="#L-387"><span class="linenos">387</span></a> <span class="k">def</span> <span class="nf">_maybe_annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-388"><a href="#L-388"><span class="linenos">388</span></a> <span class="k">if</span> <span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> -</span><span id="L-389"><a href="#L-389"><span class="linenos">389</span></a> <span class="k">return</span> <span class="n">expression</span> <span class="c1"># We've already inferred the expression's type</span> -</span><span id="L-390"><a href="#L-390"><span class="linenos">390</span></a> -</span><span id="L-391"><a href="#L-391"><span class="linenos">391</span></a> <span class="n">annotator</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="vm">__class__</span><span class="p">)</span> -</span><span id="L-392"><a href="#L-392"><span class="linenos">392</span></a> -</span><span id="L-393"><a href="#L-393"><span class="linenos">393</span></a> <span class="k">return</span> <span class="p">(</span> -</span><span id="L-394"><a href="#L-394"><span class="linenos">394</span></a> <span class="n">annotator</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-395"><a href="#L-395"><span class="linenos">395</span></a> <span class="k">if</span> <span class="n">annotator</span> -</span><span id="L-396"><a href="#L-396"><span class="linenos">396</span></a> <span class="k">else</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> -</span><span id="L-397"><a href="#L-397"><span class="linenos">397</span></a> <span class="p">)</span> -</span><span id="L-398"><a href="#L-398"><span class="linenos">398</span></a> -</span><span id="L-399"><a href="#L-399"><span class="linenos">399</span></a> <span class="k">def</span> <span class="nf">_annotate_args</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-400"><a href="#L-400"><span class="linenos">400</span></a> <span class="k">for</span> <span class="n">_</span><span class="p">,</span> <span class="n">value</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">iter_expressions</span><span class="p">():</span> -</span><span id="L-401"><a href="#L-401"><span class="linenos">401</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> -</span><span id="L-402"><a href="#L-402"><span class="linenos">402</span></a> -</span><span id="L-403"><a href="#L-403"><span class="linenos">403</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-404"><a href="#L-404"><span class="linenos">404</span></a> -</span><span id="L-405"><a href="#L-405"><span class="linenos">405</span></a> <span class="k">def</span> <span class="nf">_maybe_coerce</span><span class="p">(</span> -</span><span id="L-406"><a href="#L-406"><span class="linenos">406</span></a> <span class="bp">self</span><span class="p">,</span> <span class="n">type1</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">type2</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> -</span><span id="L-407"><a href="#L-407"><span class="linenos">407</span></a> <span class="p">)</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">:</span> -</span><span id="L-408"><a href="#L-408"><span class="linenos">408</span></a> <span class="n">type1_value</span> <span class="o">=</span> <span class="n">type1</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type1</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type1</span> -</span><span id="L-409"><a href="#L-409"><span class="linenos">409</span></a> <span class="n">type2_value</span> <span class="o">=</span> <span class="n">type2</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type2</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type2</span> -</span><span id="L-410"><a href="#L-410"><span class="linenos">410</span></a> -</span><span id="L-411"><a href="#L-411"><span class="linenos">411</span></a> <span class="c1"># We propagate the NULL / UNKNOWN types upwards if found</span> -</span><span id="L-412"><a href="#L-412"><span class="linenos">412</span></a> <span class="k">if</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> -</span><span id="L-413"><a href="#L-413"><span class="linenos">413</span></a> <span class="k">return</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">NULL</span> -</span><span id="L-414"><a href="#L-414"><span class="linenos">414</span></a> <span class="k">if</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">UNKNOWN</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> -</span><span id="L-415"><a href="#L-415"><span class="linenos">415</span></a> <span class="k">return</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">UNKNOWN</span> -</span><span id="L-416"><a href="#L-416"><span class="linenos">416</span></a> -</span><span id="L-417"><a href="#L-417"><span class="linenos">417</span></a> <span class="k">if</span> <span class="n">type1_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> -</span><span id="L-418"><a href="#L-418"><span class="linenos">418</span></a> <span class="k">return</span> <span class="n">type1</span> -</span><span id="L-419"><a href="#L-419"><span class="linenos">419</span></a> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> -</span><span id="L-420"><a href="#L-420"><span class="linenos">420</span></a> <span class="k">return</span> <span class="n">type2</span> -</span><span id="L-421"><a href="#L-421"><span class="linenos">421</span></a> -</span><span id="L-422"><a href="#L-422"><span class="linenos">422</span></a> <span class="k">return</span> <span class="n">type2_value</span> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="p">{})</span> <span class="k">else</span> <span class="n">type1_value</span> <span class="c1"># type: ignore</span> -</span><span id="L-423"><a href="#L-423"><span class="linenos">423</span></a> -</span><span id="L-424"><a href="#L-424"><span class="linenos">424</span></a> <span class="c1"># Note: the following "no_type_check" decorators were added because mypy was yelling due</span> -</span><span id="L-425"><a href="#L-425"><span class="linenos">425</span></a> <span class="c1"># to assigning Type values to expression.type (since its getter returns Optional[DataType]).</span> -</span><span id="L-426"><a href="#L-426"><span class="linenos">426</span></a> <span class="c1"># This is a known mypy issue: https://github.com/python/mypy/issues/3004</span> -</span><span id="L-427"><a href="#L-427"><span class="linenos">427</span></a> -</span><span id="L-428"><a href="#L-428"><span class="linenos">428</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-429"><a href="#L-429"><span class="linenos">429</span></a> <span class="k">def</span> <span class="nf">_annotate_binary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">B</span><span class="p">)</span> <span class="o">-></span> <span class="n">B</span><span class="p">:</span> -</span><span id="L-430"><a href="#L-430"><span class="linenos">430</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-431"><a href="#L-431"><span class="linenos">431</span></a> -</span><span id="L-432"><a href="#L-432"><span class="linenos">432</span></a> <span class="n">left</span><span class="p">,</span> <span class="n">right</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">right</span> -</span><span id="L-433"><a href="#L-433"><span class="linenos">433</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">left</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> -</span><span id="L-434"><a href="#L-434"><span class="linenos">434</span></a> -</span><span id="L-435"><a href="#L-435"><span class="linenos">435</span></a> <span class="k">if</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">Connector</span><span class="p">):</span> -</span><span id="L-436"><a href="#L-436"><span class="linenos">436</span></a> <span class="k">if</span> <span class="n">left_type</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">NULL</span> <span class="ow">and</span> <span class="n">right_type</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">NULL</span><span class="p">:</span> -</span><span id="L-437"><a href="#L-437"><span class="linenos">437</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">NULL</span><span class="p">)</span> -</span><span id="L-438"><a href="#L-438"><span class="linenos">438</span></a> <span class="k">elif</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">):</span> -</span><span id="L-439"><a href="#L-439"><span class="linenos">439</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="L-440"><a href="#L-440"><span class="linenos">440</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="L-441"><a href="#L-441"><span class="linenos">441</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">build</span><span class="p">(</span><span class="s2">"NULLABLE"</span><span class="p">,</span> <span class="n">expressions</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">build</span><span class="p">(</span><span class="s2">"BOOLEAN"</span><span class="p">)),</span> -</span><span id="L-442"><a href="#L-442"><span class="linenos">442</span></a> <span class="p">)</span> -</span><span id="L-443"><a href="#L-443"><span class="linenos">443</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-444"><a href="#L-444"><span class="linenos">444</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="L-445"><a href="#L-445"><span class="linenos">445</span></a> <span class="k">elif</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">Predicate</span><span class="p">):</span> -</span><span id="L-446"><a href="#L-446"><span class="linenos">446</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="L-447"><a href="#L-447"><span class="linenos">447</span></a> <span class="k">elif</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">:</span> -</span><span id="L-448"><a href="#L-448"><span class="linenos">448</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">[(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)](</span><span class="n">left</span><span class="p">,</span> <span class="n">right</span><span class="p">))</span> -</span><span id="L-449"><a href="#L-449"><span class="linenos">449</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-450"><a href="#L-450"><span class="linenos">450</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> -</span><span id="L-451"><a href="#L-451"><span class="linenos">451</span></a> -</span><span id="L-452"><a href="#L-452"><span class="linenos">452</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-453"><a href="#L-453"><span class="linenos">453</span></a> -</span><span id="L-454"><a href="#L-454"><span class="linenos">454</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-455"><a href="#L-455"><span class="linenos">455</span></a> <span class="k">def</span> <span class="nf">_annotate_unary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-456"><a href="#L-456"><span class="linenos">456</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-457"><a href="#L-457"><span class="linenos">457</span></a> -</span><span id="L-458"><a href="#L-458"><span class="linenos">458</span></a> <span class="k">if</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">Condition</span><span class="p">)</span> <span class="ow">and</span> <span class="ow">not</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">Paren</span><span class="p">):</span> -</span><span id="L-459"><a href="#L-459"><span class="linenos">459</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="L-460"><a href="#L-460"><span class="linenos">460</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-461"><a href="#L-461"><span class="linenos">461</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</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">type</span><span class="p">)</span> -</span><span id="L-462"><a href="#L-462"><span class="linenos">462</span></a> -</span><span id="L-463"><a href="#L-463"><span class="linenos">463</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-464"><a href="#L-464"><span class="linenos">464</span></a> -</span><span id="L-465"><a href="#L-465"><span class="linenos">465</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-466"><a href="#L-466"><span class="linenos">466</span></a> <span class="k">def</span> <span class="nf">_annotate_literal</span><span class="p">(</span><span class="bp">self</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">Literal</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> -</span><span id="L-467"><a href="#L-467"><span class="linenos">467</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> -</span><span id="L-468"><a href="#L-468"><span class="linenos">468</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">VARCHAR</span><span class="p">)</span> -</span><span id="L-469"><a href="#L-469"><span class="linenos">469</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_int</span><span class="p">:</span> -</span><span id="L-470"><a href="#L-470"><span class="linenos">470</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">INT</span><span class="p">)</span> -</span><span id="L-471"><a href="#L-471"><span class="linenos">471</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-472"><a href="#L-472"><span class="linenos">472</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> -</span><span id="L-473"><a href="#L-473"><span class="linenos">473</span></a> -</span><span id="L-474"><a href="#L-474"><span class="linenos">474</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-475"><a href="#L-475"><span class="linenos">475</span></a> -</span><span id="L-476"><a href="#L-476"><span class="linenos">476</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-477"><a href="#L-477"><span class="linenos">477</span></a> <span class="k">def</span> <span class="nf">_annotate_with_type</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> <span class="n">target_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="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-478"><a href="#L-478"><span class="linenos">478</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">target_type</span><span class="p">)</span> -</span><span id="L-479"><a href="#L-479"><span class="linenos">479</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-480"><a href="#L-480"><span class="linenos">480</span></a> -</span><span id="L-481"><a href="#L-481"><span class="linenos">481</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-482"><a href="#L-482"><span class="linenos">482</span></a> <span class="k">def</span> <span class="nf">_annotate_struct_value</span><span class="p">(</span> -</span><span id="L-483"><a href="#L-483"><span class="linenos">483</span></a> <span class="bp">self</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><span id="L-484"><a href="#L-484"><span class="linenos">484</span></a> <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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">]</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">:</span> -</span><span id="L-485"><a href="#L-485"><span class="linenos">485</span></a> <span class="n">alias</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">"alias"</span><span class="p">)</span> -</span><span id="L-486"><a href="#L-486"><span class="linenos">486</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> -</span><span id="L-487"><a href="#L-487"><span class="linenos">487</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">alias</span><span class="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="L-488"><a href="#L-488"><span class="linenos">488</span></a> -</span><span id="L-489"><a href="#L-489"><span class="linenos">489</span></a> <span class="c1"># Case: key = value or key := value</span> -</span><span id="L-490"><a href="#L-490"><span class="linenos">490</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> -</span><span id="L-491"><a href="#L-491"><span class="linenos">491</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</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="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="L-492"><a href="#L-492"><span class="linenos">492</span></a> -</span><span id="L-493"><a href="#L-493"><span class="linenos">493</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> -</span><span id="L-494"><a href="#L-494"><span class="linenos">494</span></a> -</span><span id="L-495"><a href="#L-495"><span class="linenos">495</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="L-496"><a href="#L-496"><span class="linenos">496</span></a> <span class="k">def</span> <span class="nf">_annotate_by_args</span><span class="p">(</span> -</span><span id="L-497"><a href="#L-497"><span class="linenos">497</span></a> <span class="bp">self</span><span class="p">,</span> -</span><span id="L-498"><a href="#L-498"><span class="linenos">498</span></a> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> -</span><span id="L-499"><a href="#L-499"><span class="linenos">499</span></a> <span class="o">*</span><span class="n">args</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> -</span><span id="L-500"><a href="#L-500"><span class="linenos">500</span></a> <span class="n">promote</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="L-501"><a href="#L-501"><span class="linenos">501</span></a> <span class="n">array</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="L-502"><a href="#L-502"><span class="linenos">502</span></a> <span class="n">struct</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="L-503"><a href="#L-503"><span class="linenos">503</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="L-504"><a href="#L-504"><span class="linenos">504</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-505"><a href="#L-505"><span class="linenos">505</span></a> -</span><span id="L-506"><a href="#L-506"><span class="linenos">506</span></a> <span class="n">expressions</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="o">=</span> <span class="p">[]</span> -</span><span id="L-507"><a href="#L-507"><span class="linenos">507</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><span id="L-508"><a href="#L-508"><span class="linenos">508</span></a> <span class="n">arg_expr</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="n">arg</span><span class="p">)</span> -</span><span id="L-509"><a href="#L-509"><span class="linenos">509</span></a> <span class="n">expressions</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">expr</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">ensure_list</span><span class="p">(</span><span class="n">arg_expr</span><span class="p">)</span> <span class="k">if</span> <span class="n">expr</span><span class="p">)</span> -</span><span id="L-510"><a href="#L-510"><span class="linenos">510</span></a> -</span><span id="L-511"><a href="#L-511"><span class="linenos">511</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="kc">None</span> -</span><span id="L-512"><a href="#L-512"><span class="linenos">512</span></a> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">:</span> -</span><span id="L-513"><a href="#L-513"><span class="linenos">513</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">last_datatype</span> <span class="ow">or</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">,</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="L-514"><a href="#L-514"><span class="linenos">514</span></a> -</span><span id="L-515"><a href="#L-515"><span class="linenos">515</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">last_datatype</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">UNKNOWN</span><span class="p">)</span> -</span><span id="L-516"><a href="#L-516"><span class="linenos">516</span></a> -</span><span id="L-517"><a href="#L-517"><span class="linenos">517</span></a> <span class="k">if</span> <span class="n">promote</span><span class="p">:</span> -</span><span id="L-518"><a href="#L-518"><span class="linenos">518</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span><span class="p">:</span> -</span><span id="L-519"><a href="#L-519"><span class="linenos">519</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> -</span><span id="L-520"><a href="#L-520"><span class="linenos">520</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">FLOAT_TYPES</span><span class="p">:</span> -</span><span id="L-521"><a href="#L-521"><span class="linenos">521</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> -</span><span id="L-522"><a href="#L-522"><span class="linenos">522</span></a> -</span><span id="L-523"><a href="#L-523"><span class="linenos">523</span></a> <span class="k">if</span> <span class="n">array</span><span class="p">:</span> -</span><span id="L-524"><a href="#L-524"><span class="linenos">524</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="L-525"><a href="#L-525"><span class="linenos">525</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="L-526"><a href="#L-526"><span class="linenos">526</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> -</span><span id="L-527"><a href="#L-527"><span class="linenos">527</span></a> <span class="n">this</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">ARRAY</span><span class="p">,</span> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">],</span> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span> -</span><span id="L-528"><a href="#L-528"><span class="linenos">528</span></a> <span class="p">),</span> -</span><span id="L-529"><a href="#L-529"><span class="linenos">529</span></a> <span class="p">)</span> -</span><span id="L-530"><a href="#L-530"><span class="linenos">530</span></a> -</span><span id="L-531"><a href="#L-531"><span class="linenos">531</span></a> <span class="k">if</span> <span class="n">struct</span><span class="p">:</span> -</span><span id="L-532"><a href="#L-532"><span class="linenos">532</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="L-533"><a href="#L-533"><span class="linenos">533</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="L-534"><a href="#L-534"><span class="linenos">534</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> -</span><span id="L-535"><a href="#L-535"><span class="linenos">535</span></a> <span class="n">this</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">STRUCT</span><span class="p">,</span> -</span><span id="L-536"><a href="#L-536"><span class="linenos">536</span></a> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">_annotate_struct_value</span><span class="p">(</span><span class="n">expr</span><span class="p">)</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">],</span> -</span><span id="L-537"><a href="#L-537"><span class="linenos">537</span></a> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> -</span><span id="L-538"><a href="#L-538"><span class="linenos">538</span></a> <span class="p">),</span> -</span><span id="L-539"><a href="#L-539"><span class="linenos">539</span></a> <span class="p">)</span> -</span><span id="L-540"><a href="#L-540"><span class="linenos">540</span></a> -</span><span id="L-541"><a href="#L-541"><span class="linenos">541</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-542"><a href="#L-542"><span class="linenos">542</span></a> -</span><span id="L-543"><a href="#L-543"><span class="linenos">543</span></a> <span class="k">def</span> <span class="nf">_annotate_timeunit</span><span class="p">(</span> -</span><span id="L-544"><a href="#L-544"><span class="linenos">544</span></a> <span class="bp">self</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">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> -</span><span id="L-545"><a href="#L-545"><span class="linenos">545</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> -</span><span id="L-546"><a href="#L-546"><span class="linenos">546</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-547"><a href="#L-547"><span class="linenos">547</span></a> -</span><span id="L-548"><a href="#L-548"><span class="linenos">548</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span><span class="p">:</span> -</span><span id="L-549"><a href="#L-549"><span class="linenos">549</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date_literal</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">unit</span><span class="p">)</span> -</span><span id="L-550"><a href="#L-550"><span class="linenos">550</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEMPORAL_TYPES</span><span class="p">:</span> -</span><span id="L-551"><a href="#L-551"><span class="linenos">551</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date</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">unit</span><span class="p">)</span> -</span><span id="L-552"><a href="#L-552"><span class="linenos">552</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-553"><a href="#L-553"><span class="linenos">553</span></a> <span class="n">datatype</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">UNKNOWN</span> -</span><span id="L-554"><a href="#L-554"><span class="linenos">554</span></a> -</span><span id="L-555"><a href="#L-555"><span class="linenos">555</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">datatype</span><span class="p">)</span> -</span><span id="L-556"><a href="#L-556"><span class="linenos">556</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-557"><a href="#L-557"><span class="linenos">557</span></a> -</span><span id="L-558"><a href="#L-558"><span class="linenos">558</span></a> <span class="k">def</span> <span class="nf">_annotate_bracket</span><span class="p">(</span><span class="bp">self</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">Bracket</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Bracket</span><span class="p">:</span> -</span><span id="L-559"><a href="#L-559"><span class="linenos">559</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-560"><a href="#L-560"><span class="linenos">560</span></a> -</span><span id="L-561"><a href="#L-561"><span class="linenos">561</span></a> <span class="n">bracket_arg</span> <span class="o">=</span> <span class="n">expression</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-562"><a href="#L-562"><span class="linenos">562</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><span id="L-563"><a href="#L-563"><span class="linenos">563</span></a> -</span><span id="L-564"><a href="#L-564"><span class="linenos">564</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">):</span> -</span><span id="L-565"><a href="#L-565"><span class="linenos">565</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="L-566"><a href="#L-566"><span class="linenos">566</span></a> <span class="k">elif</span> <span class="n">this</span><span class="o">.</span><span class="n">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">ARRAY</span><span class="p">):</span> -</span><span id="L-567"><a href="#L-567"><span class="linenos">567</span></a> <span class="n">contained_type</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">expressions</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</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">UNKNOWN</span> -</span><span id="L-568"><a href="#L-568"><span class="linenos">568</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">contained_type</span><span class="p">)</span> +</span><span id="L-266"><a href="#L-266"><span class="linenos">266</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_explode</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> +</span><span id="L-267"><a href="#L-267"><span class="linenos">267</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Filter</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">),</span> +</span><span id="L-268"><a href="#L-268"><span class="linenos">268</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">If</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"true"</span><span class="p">,</span> <span class="s2">"false"</span><span class="p">),</span> +</span><span id="L-269"><a href="#L-269"><span class="linenos">269</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">INTERVAL</span><span class="p">),</span> +</span><span id="L-270"><a href="#L-270"><span class="linenos">270</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Least</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="L-271"><a href="#L-271"><span class="linenos">271</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_literal</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> +</span><span id="L-272"><a href="#L-272"><span class="linenos">272</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> +</span><span id="L-273"><a href="#L-273"><span class="linenos">273</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="L-274"><a href="#L-274"><span class="linenos">274</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="L-275"><a href="#L-275"><span class="linenos">275</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Null</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">NULL</span><span class="p">),</span> +</span><span id="L-276"><a href="#L-276"><span class="linenos">276</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Nullif</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">),</span> +</span><span id="L-277"><a href="#L-277"><span class="linenos">277</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">UNKNOWN</span><span class="p">),</span> +</span><span id="L-278"><a href="#L-278"><span class="linenos">278</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Sum</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">promote</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> +</span><span id="L-279"><a href="#L-279"><span class="linenos">279</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">TryCast</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"to"</span><span class="p">]),</span> +</span><span id="L-280"><a href="#L-280"><span class="linenos">280</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> +</span><span id="L-281"><a href="#L-281"><span class="linenos">281</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Struct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">struct</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> +</span><span id="L-282"><a href="#L-282"><span class="linenos">282</span></a> <span class="p">}</span> +</span><span id="L-283"><a href="#L-283"><span class="linenos">283</span></a> +</span><span id="L-284"><a href="#L-284"><span class="linenos">284</span></a> <span class="n">NESTED_TYPES</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="L-285"><a href="#L-285"><span class="linenos">285</span></a> <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">ARRAY</span><span class="p">,</span> +</span><span id="L-286"><a href="#L-286"><span class="linenos">286</span></a> <span class="p">}</span> +</span><span id="L-287"><a href="#L-287"><span class="linenos">287</span></a> +</span><span id="L-288"><a href="#L-288"><span class="linenos">288</span></a> <span class="c1"># Specifies what types a given type can be coerced into (autofilled)</span> +</span><span id="L-289"><a href="#L-289"><span class="linenos">289</span></a> <span class="n">COERCES_TO</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]</span> <span class="o">=</span> <span class="p">{}</span> +</span><span id="L-290"><a href="#L-290"><span class="linenos">290</span></a> +</span><span id="L-291"><a href="#L-291"><span class="linenos">291</span></a> <span class="c1"># Coercion functions for binary operations.</span> +</span><span id="L-292"><a href="#L-292"><span class="linenos">292</span></a> <span class="c1"># Map of type pairs to a callable that takes both sides of the binary operation and returns the resulting type.</span> +</span><span id="L-293"><a href="#L-293"><span class="linenos">293</span></a> <span class="n">BINARY_COERCIONS</span><span class="p">:</span> <span class="n">BinaryCoercions</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="L-294"><a href="#L-294"><span class="linenos">294</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="L-295"><a href="#L-295"><span class="linenos">295</span></a> <span class="p">{</span> +</span><span id="L-296"><a href="#L-296"><span class="linenos">296</span></a> <span class="p">(</span><span class="n">t</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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date_literal</span><span class="p">(</span> +</span><span id="L-297"><a href="#L-297"><span class="linenos">297</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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-298"><a href="#L-298"><span class="linenos">298</span></a> <span class="p">)</span> +</span><span id="L-299"><a href="#L-299"><span class="linenos">299</span></a> <span class="k">for</span> <span class="n">t</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> +</span><span id="L-300"><a href="#L-300"><span class="linenos">300</span></a> <span class="p">}</span> +</span><span id="L-301"><a href="#L-301"><span class="linenos">301</span></a> <span class="p">),</span> +</span><span id="L-302"><a href="#L-302"><span class="linenos">302</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="L-303"><a href="#L-303"><span class="linenos">303</span></a> <span class="p">{</span> +</span><span id="L-304"><a href="#L-304"><span class="linenos">304</span></a> <span class="c1"># text + numeric will yield the numeric type to match most dialects' semantics</span> +</span><span id="L-305"><a href="#L-305"><span class="linenos">305</span></a> <span class="p">(</span><span class="n">text</span><span class="p">,</span> <span class="n">numeric</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span> +</span><span id="L-306"><a href="#L-306"><span class="linenos">306</span></a> <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="p">,</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> <span class="k">else</span> <span class="n">r</span><span class="o">.</span><span class="n">type</span> +</span><span id="L-307"><a href="#L-307"><span class="linenos">307</span></a> <span class="p">)</span> +</span><span id="L-308"><a href="#L-308"><span class="linenos">308</span></a> <span class="k">for</span> <span class="n">text</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> +</span><span id="L-309"><a href="#L-309"><span class="linenos">309</span></a> <span class="k">for</span> <span class="n">numeric</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> +</span><span id="L-310"><a href="#L-310"><span class="linenos">310</span></a> <span class="p">}</span> +</span><span id="L-311"><a href="#L-311"><span class="linenos">311</span></a> <span class="p">),</span> +</span><span id="L-312"><a href="#L-312"><span class="linenos">312</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="L-313"><a href="#L-313"><span class="linenos">313</span></a> <span class="p">{</span> +</span><span id="L-314"><a href="#L-314"><span class="linenos">314</span></a> <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 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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date</span><span class="p">(</span> +</span><span id="L-315"><a href="#L-315"><span class="linenos">315</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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-316"><a href="#L-316"><span class="linenos">316</span></a> <span class="p">),</span> +</span><span id="L-317"><a href="#L-317"><span class="linenos">317</span></a> <span class="p">}</span> +</span><span id="L-318"><a href="#L-318"><span class="linenos">318</span></a> <span class="p">),</span> +</span><span id="L-319"><a href="#L-319"><span class="linenos">319</span></a> <span class="p">}</span> +</span><span id="L-320"><a href="#L-320"><span class="linenos">320</span></a> +</span><span id="L-321"><a href="#L-321"><span class="linenos">321</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> +</span><span id="L-322"><a href="#L-322"><span class="linenos">322</span></a> <span class="bp">self</span><span class="p">,</span> +</span><span id="L-323"><a href="#L-323"><span class="linenos">323</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> +</span><span id="L-324"><a href="#L-324"><span class="linenos">324</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="L-325"><a href="#L-325"><span class="linenos">325</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="L-326"><a href="#L-326"><span class="linenos">326</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="L-327"><a href="#L-327"><span class="linenos">327</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> +</span><span id="L-328"><a href="#L-328"><span class="linenos">328</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> +</span><span id="L-329"><a href="#L-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> +</span><span id="L-330"><a href="#L-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> +</span><span id="L-331"><a href="#L-331"><span class="linenos">331</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> +</span><span id="L-332"><a href="#L-332"><span class="linenos">332</span></a> +</span><span id="L-333"><a href="#L-333"><span class="linenos">333</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> +</span><span id="L-334"><a href="#L-334"><span class="linenos">334</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> +</span><span id="L-335"><a href="#L-335"><span class="linenos">335</span></a> +</span><span id="L-336"><a href="#L-336"><span class="linenos">336</span></a> <span class="k">def</span> <span class="nf">_set_type</span><span class="p">(</span> +</span><span id="L-337"><a href="#L-337"><span class="linenos">337</span></a> <span class="bp">self</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">target_type</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">DataType</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="p">]</span> +</span><span id="L-338"><a href="#L-338"><span class="linenos">338</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> +</span><span id="L-339"><a href="#L-339"><span class="linenos">339</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> <span class="o">=</span> <span class="n">target_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">UNKNOWN</span> <span class="c1"># type: ignore</span> +</span><span id="L-340"><a href="#L-340"><span class="linenos">340</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> +</span><span id="L-341"><a href="#L-341"><span class="linenos">341</span></a> +</span><span id="L-342"><a href="#L-342"><span class="linenos">342</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-343"><a href="#L-343"><span class="linenos">343</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> +</span><span id="L-344"><a href="#L-344"><span class="linenos">344</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> +</span><span id="L-345"><a href="#L-345"><span class="linenos">345</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> +</span><span id="L-346"><a href="#L-346"><span class="linenos">346</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> +</span><span id="L-347"><a href="#L-347"><span class="linenos">347</span></a> <span class="k">continue</span> +</span><span id="L-348"><a href="#L-348"><span class="linenos">348</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> +</span><span id="L-349"><a href="#L-349"><span class="linenos">349</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="L-350"><a href="#L-350"><span class="linenos">350</span></a> +</span><span id="L-351"><a href="#L-351"><span class="linenos">351</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> +</span><span id="L-352"><a href="#L-352"><span class="linenos">352</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> +</span><span id="L-353"><a href="#L-353"><span class="linenos">353</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> +</span><span id="L-354"><a href="#L-354"><span class="linenos">354</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-355"><a href="#L-355"><span class="linenos">355</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> +</span><span id="L-356"><a href="#L-356"><span class="linenos">356</span></a> +</span><span id="L-357"><a href="#L-357"><span class="linenos">357</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> +</span><span id="L-358"><a href="#L-358"><span class="linenos">358</span></a> <span class="k">continue</span> +</span><span id="L-359"><a href="#L-359"><span class="linenos">359</span></a> +</span><span id="L-360"><a href="#L-360"><span class="linenos">360</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="L-361"><a href="#L-361"><span class="linenos">361</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> +</span><span id="L-362"><a href="#L-362"><span class="linenos">362</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> +</span><span id="L-363"><a href="#L-363"><span class="linenos">363</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> +</span><span id="L-364"><a href="#L-364"><span class="linenos">364</span></a> <span class="n">values</span><span class="p">,</span> +</span><span id="L-365"><a href="#L-365"><span class="linenos">365</span></a> <span class="p">)</span> +</span><span id="L-366"><a href="#L-366"><span class="linenos">366</span></a> <span class="p">}</span> +</span><span id="L-367"><a href="#L-367"><span class="linenos">367</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-368"><a href="#L-368"><span class="linenos">368</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="L-369"><a href="#L-369"><span class="linenos">369</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> +</span><span id="L-370"><a href="#L-370"><span class="linenos">370</span></a> <span class="p">}</span> +</span><span id="L-371"><a href="#L-371"><span class="linenos">371</span></a> +</span><span id="L-372"><a href="#L-372"><span class="linenos">372</span></a> <span class="c1"># First annotate the current scope's column references</span> +</span><span id="L-373"><a href="#L-373"><span class="linenos">373</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> +</span><span id="L-374"><a href="#L-374"><span class="linenos">374</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> +</span><span id="L-375"><a href="#L-375"><span class="linenos">375</span></a> <span class="k">continue</span> +</span><span id="L-376"><a href="#L-376"><span class="linenos">376</span></a> +</span><span id="L-377"><a href="#L-377"><span class="linenos">377</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> +</span><span id="L-378"><a href="#L-378"><span class="linenos">378</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> +</span><span id="L-379"><a href="#L-379"><span class="linenos">379</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> +</span><span id="L-380"><a href="#L-380"><span class="linenos">380</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> +</span><span id="L-381"><a href="#L-381"><span class="linenos">381</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="L-382"><a href="#L-382"><span class="linenos">382</span></a> +</span><span id="L-383"><a href="#L-383"><span class="linenos">383</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> +</span><span id="L-384"><a href="#L-384"><span class="linenos">384</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-385"><a href="#L-385"><span class="linenos">385</span></a> +</span><span id="L-386"><a href="#L-386"><span class="linenos">386</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> +</span><span id="L-387"><a href="#L-387"><span class="linenos">387</span></a> +</span><span id="L-388"><a href="#L-388"><span class="linenos">388</span></a> <span class="k">def</span> <span class="nf">_maybe_annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-389"><a href="#L-389"><span class="linenos">389</span></a> <span class="k">if</span> <span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> +</span><span id="L-390"><a href="#L-390"><span class="linenos">390</span></a> <span class="k">return</span> <span class="n">expression</span> <span class="c1"># We've already inferred the expression's type</span> +</span><span id="L-391"><a href="#L-391"><span class="linenos">391</span></a> +</span><span id="L-392"><a href="#L-392"><span class="linenos">392</span></a> <span class="n">annotator</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="vm">__class__</span><span class="p">)</span> +</span><span id="L-393"><a href="#L-393"><span class="linenos">393</span></a> +</span><span id="L-394"><a href="#L-394"><span class="linenos">394</span></a> <span class="k">return</span> <span class="p">(</span> +</span><span id="L-395"><a href="#L-395"><span class="linenos">395</span></a> <span class="n">annotator</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-396"><a href="#L-396"><span class="linenos">396</span></a> <span class="k">if</span> <span class="n">annotator</span> +</span><span id="L-397"><a href="#L-397"><span class="linenos">397</span></a> <span class="k">else</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> +</span><span id="L-398"><a href="#L-398"><span class="linenos">398</span></a> <span class="p">)</span> +</span><span id="L-399"><a href="#L-399"><span class="linenos">399</span></a> +</span><span id="L-400"><a href="#L-400"><span class="linenos">400</span></a> <span class="k">def</span> <span class="nf">_annotate_args</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-401"><a href="#L-401"><span class="linenos">401</span></a> <span class="k">for</span> <span class="n">_</span><span class="p">,</span> <span class="n">value</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">iter_expressions</span><span class="p">():</span> +</span><span id="L-402"><a href="#L-402"><span class="linenos">402</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> +</span><span id="L-403"><a href="#L-403"><span class="linenos">403</span></a> +</span><span id="L-404"><a href="#L-404"><span class="linenos">404</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-405"><a href="#L-405"><span class="linenos">405</span></a> +</span><span id="L-406"><a href="#L-406"><span class="linenos">406</span></a> <span class="k">def</span> <span class="nf">_maybe_coerce</span><span class="p">(</span> +</span><span id="L-407"><a href="#L-407"><span class="linenos">407</span></a> <span class="bp">self</span><span class="p">,</span> <span class="n">type1</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">type2</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> +</span><span id="L-408"><a href="#L-408"><span class="linenos">408</span></a> <span class="p">)</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">:</span> +</span><span id="L-409"><a href="#L-409"><span class="linenos">409</span></a> <span class="n">type1_value</span> <span class="o">=</span> <span class="n">type1</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type1</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type1</span> +</span><span id="L-410"><a href="#L-410"><span class="linenos">410</span></a> <span class="n">type2_value</span> <span class="o">=</span> <span class="n">type2</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type2</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type2</span> +</span><span id="L-411"><a href="#L-411"><span class="linenos">411</span></a> +</span><span id="L-412"><a href="#L-412"><span class="linenos">412</span></a> <span class="c1"># We propagate the NULL / UNKNOWN types upwards if found</span> +</span><span id="L-413"><a href="#L-413"><span class="linenos">413</span></a> <span class="k">if</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> +</span><span id="L-414"><a href="#L-414"><span class="linenos">414</span></a> <span class="k">return</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">NULL</span> +</span><span id="L-415"><a href="#L-415"><span class="linenos">415</span></a> <span class="k">if</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">UNKNOWN</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> +</span><span id="L-416"><a href="#L-416"><span class="linenos">416</span></a> <span class="k">return</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">UNKNOWN</span> +</span><span id="L-417"><a href="#L-417"><span class="linenos">417</span></a> +</span><span id="L-418"><a href="#L-418"><span class="linenos">418</span></a> <span class="k">if</span> <span class="n">type1_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> +</span><span id="L-419"><a href="#L-419"><span class="linenos">419</span></a> <span class="k">return</span> <span class="n">type1</span> +</span><span id="L-420"><a href="#L-420"><span class="linenos">420</span></a> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> +</span><span id="L-421"><a href="#L-421"><span class="linenos">421</span></a> <span class="k">return</span> <span class="n">type2</span> +</span><span id="L-422"><a href="#L-422"><span class="linenos">422</span></a> +</span><span id="L-423"><a href="#L-423"><span class="linenos">423</span></a> <span class="k">return</span> <span class="n">type2_value</span> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="p">{})</span> <span class="k">else</span> <span class="n">type1_value</span> <span class="c1"># type: ignore</span> +</span><span id="L-424"><a href="#L-424"><span class="linenos">424</span></a> +</span><span id="L-425"><a href="#L-425"><span class="linenos">425</span></a> <span class="c1"># Note: the following "no_type_check" decorators were added because mypy was yelling due</span> +</span><span id="L-426"><a href="#L-426"><span class="linenos">426</span></a> <span class="c1"># to assigning Type values to expression.type (since its getter returns Optional[DataType]).</span> +</span><span id="L-427"><a href="#L-427"><span class="linenos">427</span></a> <span class="c1"># This is a known mypy issue: https://github.com/python/mypy/issues/3004</span> +</span><span id="L-428"><a href="#L-428"><span class="linenos">428</span></a> +</span><span id="L-429"><a href="#L-429"><span class="linenos">429</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-430"><a href="#L-430"><span class="linenos">430</span></a> <span class="k">def</span> <span class="nf">_annotate_binary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">B</span><span class="p">)</span> <span class="o">-></span> <span class="n">B</span><span class="p">:</span> +</span><span id="L-431"><a href="#L-431"><span class="linenos">431</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-432"><a href="#L-432"><span class="linenos">432</span></a> +</span><span id="L-433"><a href="#L-433"><span class="linenos">433</span></a> <span class="n">left</span><span class="p">,</span> <span class="n">right</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">right</span> +</span><span id="L-434"><a href="#L-434"><span class="linenos">434</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">left</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> +</span><span id="L-435"><a href="#L-435"><span class="linenos">435</span></a> +</span><span id="L-436"><a href="#L-436"><span class="linenos">436</span></a> <span class="k">if</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">Connector</span><span class="p">):</span> +</span><span id="L-437"><a href="#L-437"><span class="linenos">437</span></a> <span class="k">if</span> <span class="n">left_type</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">NULL</span> <span class="ow">and</span> <span class="n">right_type</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">NULL</span><span class="p">:</span> +</span><span id="L-438"><a href="#L-438"><span class="linenos">438</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">NULL</span><span class="p">)</span> +</span><span id="L-439"><a href="#L-439"><span class="linenos">439</span></a> <span class="k">elif</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">):</span> +</span><span id="L-440"><a href="#L-440"><span class="linenos">440</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="L-441"><a href="#L-441"><span class="linenos">441</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="L-442"><a href="#L-442"><span class="linenos">442</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">build</span><span class="p">(</span><span class="s2">"NULLABLE"</span><span class="p">,</span> <span class="n">expressions</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">build</span><span class="p">(</span><span class="s2">"BOOLEAN"</span><span class="p">)),</span> +</span><span id="L-443"><a href="#L-443"><span class="linenos">443</span></a> <span class="p">)</span> +</span><span id="L-444"><a href="#L-444"><span class="linenos">444</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-445"><a href="#L-445"><span class="linenos">445</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="L-446"><a href="#L-446"><span class="linenos">446</span></a> <span class="k">elif</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">Predicate</span><span class="p">):</span> +</span><span id="L-447"><a href="#L-447"><span class="linenos">447</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="L-448"><a href="#L-448"><span class="linenos">448</span></a> <span class="k">elif</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">:</span> +</span><span id="L-449"><a href="#L-449"><span class="linenos">449</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">[(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)](</span><span class="n">left</span><span class="p">,</span> <span class="n">right</span><span class="p">))</span> +</span><span id="L-450"><a href="#L-450"><span class="linenos">450</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-451"><a href="#L-451"><span class="linenos">451</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> +</span><span id="L-452"><a href="#L-452"><span class="linenos">452</span></a> +</span><span id="L-453"><a href="#L-453"><span class="linenos">453</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-454"><a href="#L-454"><span class="linenos">454</span></a> +</span><span id="L-455"><a href="#L-455"><span class="linenos">455</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-456"><a href="#L-456"><span class="linenos">456</span></a> <span class="k">def</span> <span class="nf">_annotate_unary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-457"><a href="#L-457"><span class="linenos">457</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-458"><a href="#L-458"><span class="linenos">458</span></a> +</span><span id="L-459"><a href="#L-459"><span class="linenos">459</span></a> <span class="k">if</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">Condition</span><span class="p">)</span> <span class="ow">and</span> <span class="ow">not</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">Paren</span><span class="p">):</span> +</span><span id="L-460"><a href="#L-460"><span class="linenos">460</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="L-461"><a href="#L-461"><span class="linenos">461</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-462"><a href="#L-462"><span class="linenos">462</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</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">type</span><span class="p">)</span> +</span><span id="L-463"><a href="#L-463"><span class="linenos">463</span></a> +</span><span id="L-464"><a href="#L-464"><span class="linenos">464</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-465"><a href="#L-465"><span class="linenos">465</span></a> +</span><span id="L-466"><a href="#L-466"><span class="linenos">466</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-467"><a href="#L-467"><span class="linenos">467</span></a> <span class="k">def</span> <span class="nf">_annotate_literal</span><span class="p">(</span><span class="bp">self</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">Literal</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> +</span><span id="L-468"><a href="#L-468"><span class="linenos">468</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> +</span><span id="L-469"><a href="#L-469"><span class="linenos">469</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">VARCHAR</span><span class="p">)</span> +</span><span id="L-470"><a href="#L-470"><span class="linenos">470</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_int</span><span class="p">:</span> +</span><span id="L-471"><a href="#L-471"><span class="linenos">471</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">INT</span><span class="p">)</span> +</span><span id="L-472"><a href="#L-472"><span class="linenos">472</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-473"><a href="#L-473"><span class="linenos">473</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> +</span><span id="L-474"><a href="#L-474"><span class="linenos">474</span></a> +</span><span id="L-475"><a href="#L-475"><span class="linenos">475</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-476"><a href="#L-476"><span class="linenos">476</span></a> +</span><span id="L-477"><a href="#L-477"><span class="linenos">477</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-478"><a href="#L-478"><span class="linenos">478</span></a> <span class="k">def</span> <span class="nf">_annotate_with_type</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> <span class="n">target_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="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-479"><a href="#L-479"><span class="linenos">479</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">target_type</span><span class="p">)</span> +</span><span id="L-480"><a href="#L-480"><span class="linenos">480</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-481"><a href="#L-481"><span class="linenos">481</span></a> +</span><span id="L-482"><a href="#L-482"><span class="linenos">482</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-483"><a href="#L-483"><span class="linenos">483</span></a> <span class="k">def</span> <span class="nf">_annotate_struct_value</span><span class="p">(</span> +</span><span id="L-484"><a href="#L-484"><span class="linenos">484</span></a> <span class="bp">self</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><span id="L-485"><a href="#L-485"><span class="linenos">485</span></a> <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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">]</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">:</span> +</span><span id="L-486"><a href="#L-486"><span class="linenos">486</span></a> <span class="n">alias</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">"alias"</span><span class="p">)</span> +</span><span id="L-487"><a href="#L-487"><span class="linenos">487</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> +</span><span id="L-488"><a href="#L-488"><span class="linenos">488</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">alias</span><span class="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="L-489"><a href="#L-489"><span class="linenos">489</span></a> +</span><span id="L-490"><a href="#L-490"><span class="linenos">490</span></a> <span class="c1"># Case: key = value or key := value</span> +</span><span id="L-491"><a href="#L-491"><span class="linenos">491</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> +</span><span id="L-492"><a href="#L-492"><span class="linenos">492</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</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="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="L-493"><a href="#L-493"><span class="linenos">493</span></a> +</span><span id="L-494"><a href="#L-494"><span class="linenos">494</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> +</span><span id="L-495"><a href="#L-495"><span class="linenos">495</span></a> +</span><span id="L-496"><a href="#L-496"><span class="linenos">496</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="L-497"><a href="#L-497"><span class="linenos">497</span></a> <span class="k">def</span> <span class="nf">_annotate_by_args</span><span class="p">(</span> +</span><span id="L-498"><a href="#L-498"><span class="linenos">498</span></a> <span class="bp">self</span><span class="p">,</span> +</span><span id="L-499"><a href="#L-499"><span class="linenos">499</span></a> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> +</span><span id="L-500"><a href="#L-500"><span class="linenos">500</span></a> <span class="o">*</span><span class="n">args</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> +</span><span id="L-501"><a href="#L-501"><span class="linenos">501</span></a> <span class="n">promote</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="L-502"><a href="#L-502"><span class="linenos">502</span></a> <span class="n">array</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="L-503"><a href="#L-503"><span class="linenos">503</span></a> <span class="n">struct</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="L-504"><a href="#L-504"><span class="linenos">504</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="L-505"><a href="#L-505"><span class="linenos">505</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-506"><a href="#L-506"><span class="linenos">506</span></a> +</span><span id="L-507"><a href="#L-507"><span class="linenos">507</span></a> <span class="n">expressions</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="o">=</span> <span class="p">[]</span> +</span><span id="L-508"><a href="#L-508"><span class="linenos">508</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><span id="L-509"><a href="#L-509"><span class="linenos">509</span></a> <span class="n">arg_expr</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="n">arg</span><span class="p">)</span> +</span><span id="L-510"><a href="#L-510"><span class="linenos">510</span></a> <span class="n">expressions</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">expr</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">ensure_list</span><span class="p">(</span><span class="n">arg_expr</span><span class="p">)</span> <span class="k">if</span> <span class="n">expr</span><span class="p">)</span> +</span><span id="L-511"><a href="#L-511"><span class="linenos">511</span></a> +</span><span id="L-512"><a href="#L-512"><span class="linenos">512</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="kc">None</span> +</span><span id="L-513"><a href="#L-513"><span class="linenos">513</span></a> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">:</span> +</span><span id="L-514"><a href="#L-514"><span class="linenos">514</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">last_datatype</span> <span class="ow">or</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">,</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="L-515"><a href="#L-515"><span class="linenos">515</span></a> +</span><span id="L-516"><a href="#L-516"><span class="linenos">516</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">last_datatype</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">UNKNOWN</span><span class="p">)</span> +</span><span id="L-517"><a href="#L-517"><span class="linenos">517</span></a> +</span><span id="L-518"><a href="#L-518"><span class="linenos">518</span></a> <span class="k">if</span> <span class="n">promote</span><span class="p">:</span> +</span><span id="L-519"><a href="#L-519"><span class="linenos">519</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span><span class="p">:</span> +</span><span id="L-520"><a href="#L-520"><span class="linenos">520</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> +</span><span id="L-521"><a href="#L-521"><span class="linenos">521</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">FLOAT_TYPES</span><span class="p">:</span> +</span><span id="L-522"><a href="#L-522"><span class="linenos">522</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> +</span><span id="L-523"><a href="#L-523"><span class="linenos">523</span></a> +</span><span id="L-524"><a href="#L-524"><span class="linenos">524</span></a> <span class="k">if</span> <span class="n">array</span><span class="p">:</span> +</span><span id="L-525"><a href="#L-525"><span class="linenos">525</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="L-526"><a href="#L-526"><span class="linenos">526</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="L-527"><a href="#L-527"><span class="linenos">527</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> +</span><span id="L-528"><a href="#L-528"><span class="linenos">528</span></a> <span class="n">this</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">ARRAY</span><span class="p">,</span> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">],</span> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span> +</span><span id="L-529"><a href="#L-529"><span class="linenos">529</span></a> <span class="p">),</span> +</span><span id="L-530"><a href="#L-530"><span class="linenos">530</span></a> <span class="p">)</span> +</span><span id="L-531"><a href="#L-531"><span class="linenos">531</span></a> +</span><span id="L-532"><a href="#L-532"><span class="linenos">532</span></a> <span class="k">if</span> <span class="n">struct</span><span class="p">:</span> +</span><span id="L-533"><a href="#L-533"><span class="linenos">533</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="L-534"><a href="#L-534"><span class="linenos">534</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="L-535"><a href="#L-535"><span class="linenos">535</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> +</span><span id="L-536"><a href="#L-536"><span class="linenos">536</span></a> <span class="n">this</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">STRUCT</span><span class="p">,</span> +</span><span id="L-537"><a href="#L-537"><span class="linenos">537</span></a> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">_annotate_struct_value</span><span class="p">(</span><span class="n">expr</span><span class="p">)</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">],</span> +</span><span id="L-538"><a href="#L-538"><span class="linenos">538</span></a> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> +</span><span id="L-539"><a href="#L-539"><span class="linenos">539</span></a> <span class="p">),</span> +</span><span id="L-540"><a href="#L-540"><span class="linenos">540</span></a> <span class="p">)</span> +</span><span id="L-541"><a href="#L-541"><span class="linenos">541</span></a> +</span><span id="L-542"><a href="#L-542"><span class="linenos">542</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-543"><a href="#L-543"><span class="linenos">543</span></a> +</span><span id="L-544"><a href="#L-544"><span class="linenos">544</span></a> <span class="k">def</span> <span class="nf">_annotate_timeunit</span><span class="p">(</span> +</span><span id="L-545"><a href="#L-545"><span class="linenos">545</span></a> <span class="bp">self</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">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> +</span><span id="L-546"><a href="#L-546"><span class="linenos">546</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> +</span><span id="L-547"><a href="#L-547"><span class="linenos">547</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-548"><a href="#L-548"><span class="linenos">548</span></a> +</span><span id="L-549"><a href="#L-549"><span class="linenos">549</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span><span class="p">:</span> +</span><span id="L-550"><a href="#L-550"><span class="linenos">550</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date_literal</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">unit</span><span class="p">)</span> +</span><span id="L-551"><a href="#L-551"><span class="linenos">551</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEMPORAL_TYPES</span><span class="p">:</span> +</span><span id="L-552"><a href="#L-552"><span class="linenos">552</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date</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">unit</span><span class="p">)</span> +</span><span id="L-553"><a href="#L-553"><span class="linenos">553</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-554"><a href="#L-554"><span class="linenos">554</span></a> <span class="n">datatype</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">UNKNOWN</span> +</span><span id="L-555"><a href="#L-555"><span class="linenos">555</span></a> +</span><span id="L-556"><a href="#L-556"><span class="linenos">556</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">datatype</span><span class="p">)</span> +</span><span id="L-557"><a href="#L-557"><span class="linenos">557</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-558"><a href="#L-558"><span class="linenos">558</span></a> +</span><span id="L-559"><a href="#L-559"><span class="linenos">559</span></a> <span class="k">def</span> <span class="nf">_annotate_bracket</span><span class="p">(</span><span class="bp">self</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">Bracket</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Bracket</span><span class="p">:</span> +</span><span id="L-560"><a href="#L-560"><span class="linenos">560</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-561"><a href="#L-561"><span class="linenos">561</span></a> +</span><span id="L-562"><a href="#L-562"><span class="linenos">562</span></a> <span class="n">bracket_arg</span> <span class="o">=</span> <span class="n">expression</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-563"><a href="#L-563"><span class="linenos">563</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><span id="L-564"><a href="#L-564"><span class="linenos">564</span></a> +</span><span id="L-565"><a href="#L-565"><span class="linenos">565</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">):</span> +</span><span id="L-566"><a href="#L-566"><span class="linenos">566</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="L-567"><a href="#L-567"><span class="linenos">567</span></a> <span class="k">elif</span> <span class="n">this</span><span class="o">.</span><span class="n">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">ARRAY</span><span class="p">):</span> +</span><span id="L-568"><a href="#L-568"><span class="linenos">568</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">type</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-569"><a href="#L-569"><span class="linenos">569</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">))</span> <span class="ow">and</span> <span class="n">bracket_arg</span> <span class="ow">in</span> <span class="n">this</span><span class="o">.</span><span class="n">keys</span><span class="p">:</span> </span><span id="L-570"><a href="#L-570"><span class="linenos">570</span></a> <span class="n">index</span> <span class="o">=</span> <span class="n">this</span><span class="o">.</span><span class="n">keys</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">)</span> </span><span id="L-571"><a href="#L-571"><span class="linenos">571</span></a> <span class="n">value</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">values</span><span class="p">,</span> <span class="n">index</span><span class="p">)</span> -</span><span id="L-572"><a href="#L-572"><span class="linenos">572</span></a> <span class="n">value_type</span> <span class="o">=</span> <span class="n">value</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">value</span> <span class="k">else</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">UNKNOWN</span> -</span><span id="L-573"><a href="#L-573"><span class="linenos">573</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">value_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">UNKNOWN</span><span class="p">)</span> -</span><span id="L-574"><a href="#L-574"><span class="linenos">574</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-575"><a href="#L-575"><span class="linenos">575</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> -</span><span id="L-576"><a href="#L-576"><span class="linenos">576</span></a> -</span><span id="L-577"><a href="#L-577"><span class="linenos">577</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="L-578"><a href="#L-578"><span class="linenos">578</span></a> -</span><span id="L-579"><a href="#L-579"><span class="linenos">579</span></a> <span class="k">def</span> <span class="nf">_annotate_div</span><span class="p">(</span><span class="bp">self</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">Div</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> -</span><span id="L-580"><a href="#L-580"><span class="linenos">580</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="L-581"><a href="#L-581"><span class="linenos">581</span></a> -</span><span id="L-582"><a href="#L-582"><span class="linenos">582</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="o">.</span><span class="n">type</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">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="c1"># type: ignore</span> -</span><span id="L-583"><a href="#L-583"><span class="linenos">583</span></a> -</span><span id="L-584"><a href="#L-584"><span class="linenos">584</span></a> <span class="k">if</span> <span class="p">(</span> -</span><span id="L-585"><a href="#L-585"><span class="linenos">585</span></a> <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">"typed"</span><span class="p">)</span> -</span><span id="L-586"><a href="#L-586"><span class="linenos">586</span></a> <span class="ow">and</span> <span class="n">left_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> -</span><span id="L-587"><a href="#L-587"><span class="linenos">587</span></a> <span class="ow">and</span> <span class="n">right_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> -</span><span id="L-588"><a href="#L-588"><span class="linenos">588</span></a> <span class="p">):</span> -</span><span id="L-589"><a href="#L-589"><span class="linenos">589</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> -</span><span id="L-590"><a href="#L-590"><span class="linenos">590</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="L-591"><a href="#L-591"><span class="linenos">591</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> -</span><span id="L-592"><a href="#L-592"><span class="linenos">592</span></a> -</span><span id="L-593"><a href="#L-593"><span class="linenos">593</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-572"><a href="#L-572"><span class="linenos">572</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">value</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">value</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> +</span><span id="L-573"><a href="#L-573"><span class="linenos">573</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-574"><a href="#L-574"><span class="linenos">574</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> +</span><span id="L-575"><a href="#L-575"><span class="linenos">575</span></a> +</span><span id="L-576"><a href="#L-576"><span class="linenos">576</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-577"><a href="#L-577"><span class="linenos">577</span></a> +</span><span id="L-578"><a href="#L-578"><span class="linenos">578</span></a> <span class="k">def</span> <span class="nf">_annotate_div</span><span class="p">(</span><span class="bp">self</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">Div</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> +</span><span id="L-579"><a href="#L-579"><span class="linenos">579</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-580"><a href="#L-580"><span class="linenos">580</span></a> +</span><span id="L-581"><a href="#L-581"><span class="linenos">581</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="o">.</span><span class="n">type</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">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="c1"># type: ignore</span> +</span><span id="L-582"><a href="#L-582"><span class="linenos">582</span></a> +</span><span id="L-583"><a href="#L-583"><span class="linenos">583</span></a> <span class="k">if</span> <span class="p">(</span> +</span><span id="L-584"><a href="#L-584"><span class="linenos">584</span></a> <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">"typed"</span><span class="p">)</span> +</span><span id="L-585"><a href="#L-585"><span class="linenos">585</span></a> <span class="ow">and</span> <span class="n">left_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> +</span><span id="L-586"><a href="#L-586"><span class="linenos">586</span></a> <span class="ow">and</span> <span class="n">right_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> +</span><span id="L-587"><a href="#L-587"><span class="linenos">587</span></a> <span class="p">):</span> +</span><span id="L-588"><a href="#L-588"><span class="linenos">588</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> +</span><span id="L-589"><a href="#L-589"><span class="linenos">589</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="L-590"><a href="#L-590"><span class="linenos">590</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> +</span><span id="L-591"><a href="#L-591"><span class="linenos">591</span></a> +</span><span id="L-592"><a href="#L-592"><span class="linenos">592</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="L-593"><a href="#L-593"><span class="linenos">593</span></a> +</span><span id="L-594"><a href="#L-594"><span class="linenos">594</span></a> <span class="k">def</span> <span class="nf">_annotate_explode</span><span class="p">(</span><span class="bp">self</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">Explode</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">:</span> +</span><span id="L-595"><a href="#L-595"><span class="linenos">595</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="L-596"><a href="#L-596"><span class="linenos">596</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">seq_get</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">type</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-597"><a href="#L-597"><span class="linenos">597</span></a> <span class="k">return</span> <span class="n">expression</span> </span></pre></div> @@ -960,334 +964,338 @@ </span><span id="TypeAnnotator-264"><a href="#TypeAnnotator-264"><span class="linenos">264</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_timeunit</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> </span><span id="TypeAnnotator-265"><a href="#TypeAnnotator-265"><span class="linenos">265</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Distinct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> </span><span id="TypeAnnotator-266"><a href="#TypeAnnotator-266"><span class="linenos">266</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_div</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> -</span><span id="TypeAnnotator-267"><a href="#TypeAnnotator-267"><span class="linenos">267</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Filter</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">),</span> -</span><span id="TypeAnnotator-268"><a href="#TypeAnnotator-268"><span class="linenos">268</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">If</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"true"</span><span class="p">,</span> <span class="s2">"false"</span><span class="p">),</span> -</span><span id="TypeAnnotator-269"><a href="#TypeAnnotator-269"><span class="linenos">269</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">INTERVAL</span><span class="p">),</span> -</span><span id="TypeAnnotator-270"><a href="#TypeAnnotator-270"><span class="linenos">270</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Least</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="TypeAnnotator-271"><a href="#TypeAnnotator-271"><span class="linenos">271</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_literal</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> -</span><span id="TypeAnnotator-272"><a href="#TypeAnnotator-272"><span class="linenos">272</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> -</span><span id="TypeAnnotator-273"><a href="#TypeAnnotator-273"><span class="linenos">273</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="TypeAnnotator-274"><a href="#TypeAnnotator-274"><span class="linenos">274</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> -</span><span id="TypeAnnotator-275"><a href="#TypeAnnotator-275"><span class="linenos">275</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Null</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">NULL</span><span class="p">),</span> -</span><span id="TypeAnnotator-276"><a href="#TypeAnnotator-276"><span class="linenos">276</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Nullif</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">),</span> -</span><span id="TypeAnnotator-277"><a href="#TypeAnnotator-277"><span class="linenos">277</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">UNKNOWN</span><span class="p">),</span> -</span><span id="TypeAnnotator-278"><a href="#TypeAnnotator-278"><span class="linenos">278</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Sum</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">promote</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> -</span><span id="TypeAnnotator-279"><a href="#TypeAnnotator-279"><span class="linenos">279</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">TryCast</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"to"</span><span class="p">]),</span> -</span><span id="TypeAnnotator-280"><a href="#TypeAnnotator-280"><span class="linenos">280</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> -</span><span id="TypeAnnotator-281"><a href="#TypeAnnotator-281"><span class="linenos">281</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Struct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">struct</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> -</span><span id="TypeAnnotator-282"><a href="#TypeAnnotator-282"><span class="linenos">282</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-283"><a href="#TypeAnnotator-283"><span class="linenos">283</span></a> -</span><span id="TypeAnnotator-284"><a href="#TypeAnnotator-284"><span class="linenos">284</span></a> <span class="n">NESTED_TYPES</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator-285"><a href="#TypeAnnotator-285"><span class="linenos">285</span></a> <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">ARRAY</span><span class="p">,</span> -</span><span id="TypeAnnotator-286"><a href="#TypeAnnotator-286"><span class="linenos">286</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-287"><a href="#TypeAnnotator-287"><span class="linenos">287</span></a> -</span><span id="TypeAnnotator-288"><a href="#TypeAnnotator-288"><span class="linenos">288</span></a> <span class="c1"># Specifies what types a given type can be coerced into (autofilled)</span> -</span><span id="TypeAnnotator-289"><a href="#TypeAnnotator-289"><span class="linenos">289</span></a> <span class="n">COERCES_TO</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]</span> <span class="o">=</span> <span class="p">{}</span> -</span><span id="TypeAnnotator-290"><a href="#TypeAnnotator-290"><span class="linenos">290</span></a> -</span><span id="TypeAnnotator-291"><a href="#TypeAnnotator-291"><span class="linenos">291</span></a> <span class="c1"># Coercion functions for binary operations.</span> -</span><span id="TypeAnnotator-292"><a href="#TypeAnnotator-292"><span class="linenos">292</span></a> <span class="c1"># Map of type pairs to a callable that takes both sides of the binary operation and returns the resulting type.</span> -</span><span id="TypeAnnotator-293"><a href="#TypeAnnotator-293"><span class="linenos">293</span></a> <span class="n">BINARY_COERCIONS</span><span class="p">:</span> <span class="n">BinaryCoercions</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator-294"><a href="#TypeAnnotator-294"><span class="linenos">294</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="TypeAnnotator-295"><a href="#TypeAnnotator-295"><span class="linenos">295</span></a> <span class="p">{</span> -</span><span id="TypeAnnotator-296"><a href="#TypeAnnotator-296"><span class="linenos">296</span></a> <span class="p">(</span><span class="n">t</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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date_literal</span><span class="p">(</span> -</span><span id="TypeAnnotator-297"><a href="#TypeAnnotator-297"><span class="linenos">297</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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="TypeAnnotator-298"><a href="#TypeAnnotator-298"><span class="linenos">298</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-299"><a href="#TypeAnnotator-299"><span class="linenos">299</span></a> <span class="k">for</span> <span class="n">t</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> -</span><span id="TypeAnnotator-300"><a href="#TypeAnnotator-300"><span class="linenos">300</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-301"><a href="#TypeAnnotator-301"><span class="linenos">301</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-302"><a href="#TypeAnnotator-302"><span class="linenos">302</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="TypeAnnotator-303"><a href="#TypeAnnotator-303"><span class="linenos">303</span></a> <span class="p">{</span> -</span><span id="TypeAnnotator-304"><a href="#TypeAnnotator-304"><span class="linenos">304</span></a> <span class="c1"># text + numeric will yield the numeric type to match most dialects' semantics</span> -</span><span id="TypeAnnotator-305"><a href="#TypeAnnotator-305"><span class="linenos">305</span></a> <span class="p">(</span><span class="n">text</span><span class="p">,</span> <span class="n">numeric</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span> -</span><span id="TypeAnnotator-306"><a href="#TypeAnnotator-306"><span class="linenos">306</span></a> <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="p">,</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> <span class="k">else</span> <span class="n">r</span><span class="o">.</span><span class="n">type</span> -</span><span id="TypeAnnotator-307"><a href="#TypeAnnotator-307"><span class="linenos">307</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-308"><a href="#TypeAnnotator-308"><span class="linenos">308</span></a> <span class="k">for</span> <span class="n">text</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> -</span><span id="TypeAnnotator-309"><a href="#TypeAnnotator-309"><span class="linenos">309</span></a> <span class="k">for</span> <span class="n">numeric</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> -</span><span id="TypeAnnotator-310"><a href="#TypeAnnotator-310"><span class="linenos">310</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-311"><a href="#TypeAnnotator-311"><span class="linenos">311</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-312"><a href="#TypeAnnotator-312"><span class="linenos">312</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> -</span><span id="TypeAnnotator-313"><a href="#TypeAnnotator-313"><span class="linenos">313</span></a> <span class="p">{</span> -</span><span id="TypeAnnotator-314"><a href="#TypeAnnotator-314"><span class="linenos">314</span></a> <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 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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date</span><span class="p">(</span> -</span><span id="TypeAnnotator-315"><a href="#TypeAnnotator-315"><span class="linenos">315</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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="TypeAnnotator-316"><a href="#TypeAnnotator-316"><span class="linenos">316</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-317"><a href="#TypeAnnotator-317"><span class="linenos">317</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-318"><a href="#TypeAnnotator-318"><span class="linenos">318</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-319"><a href="#TypeAnnotator-319"><span class="linenos">319</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-320"><a href="#TypeAnnotator-320"><span class="linenos">320</span></a> -</span><span id="TypeAnnotator-321"><a href="#TypeAnnotator-321"><span class="linenos">321</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> -</span><span id="TypeAnnotator-322"><a href="#TypeAnnotator-322"><span class="linenos">322</span></a> <span class="bp">self</span><span class="p">,</span> -</span><span id="TypeAnnotator-323"><a href="#TypeAnnotator-323"><span class="linenos">323</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> -</span><span id="TypeAnnotator-324"><a href="#TypeAnnotator-324"><span class="linenos">324</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator-325"><a href="#TypeAnnotator-325"><span class="linenos">325</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator-326"><a href="#TypeAnnotator-326"><span class="linenos">326</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator-327"><a href="#TypeAnnotator-327"><span class="linenos">327</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> -</span><span id="TypeAnnotator-328"><a href="#TypeAnnotator-328"><span class="linenos">328</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> -</span><span id="TypeAnnotator-329"><a href="#TypeAnnotator-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> -</span><span id="TypeAnnotator-330"><a href="#TypeAnnotator-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> -</span><span id="TypeAnnotator-331"><a href="#TypeAnnotator-331"><span class="linenos">331</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> -</span><span id="TypeAnnotator-332"><a href="#TypeAnnotator-332"><span class="linenos">332</span></a> -</span><span id="TypeAnnotator-333"><a href="#TypeAnnotator-333"><span class="linenos">333</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> -</span><span id="TypeAnnotator-334"><a href="#TypeAnnotator-334"><span class="linenos">334</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> -</span><span id="TypeAnnotator-335"><a href="#TypeAnnotator-335"><span class="linenos">335</span></a> -</span><span id="TypeAnnotator-336"><a href="#TypeAnnotator-336"><span class="linenos">336</span></a> <span class="k">def</span> <span class="nf">_set_type</span><span class="p">(</span> -</span><span id="TypeAnnotator-337"><a href="#TypeAnnotator-337"><span class="linenos">337</span></a> <span class="bp">self</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">target_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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> -</span><span id="TypeAnnotator-338"><a href="#TypeAnnotator-338"><span class="linenos">338</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> -</span><span id="TypeAnnotator-339"><a href="#TypeAnnotator-339"><span class="linenos">339</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> <span class="o">=</span> <span class="n">target_type</span> <span class="c1"># type: ignore</span> -</span><span id="TypeAnnotator-340"><a href="#TypeAnnotator-340"><span class="linenos">340</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> -</span><span id="TypeAnnotator-341"><a href="#TypeAnnotator-341"><span class="linenos">341</span></a> -</span><span id="TypeAnnotator-342"><a href="#TypeAnnotator-342"><span class="linenos">342</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-343"><a href="#TypeAnnotator-343"><span class="linenos">343</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> -</span><span id="TypeAnnotator-344"><a href="#TypeAnnotator-344"><span class="linenos">344</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> -</span><span id="TypeAnnotator-345"><a href="#TypeAnnotator-345"><span class="linenos">345</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> -</span><span id="TypeAnnotator-346"><a href="#TypeAnnotator-346"><span class="linenos">346</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> -</span><span id="TypeAnnotator-347"><a href="#TypeAnnotator-347"><span class="linenos">347</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator-348"><a href="#TypeAnnotator-348"><span class="linenos">348</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> -</span><span id="TypeAnnotator-349"><a href="#TypeAnnotator-349"><span class="linenos">349</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="TypeAnnotator-350"><a href="#TypeAnnotator-350"><span class="linenos">350</span></a> -</span><span id="TypeAnnotator-351"><a href="#TypeAnnotator-351"><span class="linenos">351</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> -</span><span id="TypeAnnotator-352"><a href="#TypeAnnotator-352"><span class="linenos">352</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> -</span><span id="TypeAnnotator-353"><a href="#TypeAnnotator-353"><span class="linenos">353</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> -</span><span id="TypeAnnotator-354"><a href="#TypeAnnotator-354"><span class="linenos">354</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-355"><a href="#TypeAnnotator-355"><span class="linenos">355</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> -</span><span id="TypeAnnotator-356"><a href="#TypeAnnotator-356"><span class="linenos">356</span></a> -</span><span id="TypeAnnotator-357"><a href="#TypeAnnotator-357"><span class="linenos">357</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> -</span><span id="TypeAnnotator-358"><a href="#TypeAnnotator-358"><span class="linenos">358</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator-359"><a href="#TypeAnnotator-359"><span class="linenos">359</span></a> -</span><span id="TypeAnnotator-360"><a href="#TypeAnnotator-360"><span class="linenos">360</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator-361"><a href="#TypeAnnotator-361"><span class="linenos">361</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> -</span><span id="TypeAnnotator-362"><a href="#TypeAnnotator-362"><span class="linenos">362</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> -</span><span id="TypeAnnotator-363"><a href="#TypeAnnotator-363"><span class="linenos">363</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> -</span><span id="TypeAnnotator-364"><a href="#TypeAnnotator-364"><span class="linenos">364</span></a> <span class="n">values</span><span class="p">,</span> -</span><span id="TypeAnnotator-365"><a href="#TypeAnnotator-365"><span class="linenos">365</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-366"><a href="#TypeAnnotator-366"><span class="linenos">366</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-367"><a href="#TypeAnnotator-367"><span class="linenos">367</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-368"><a href="#TypeAnnotator-368"><span class="linenos">368</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator-369"><a href="#TypeAnnotator-369"><span class="linenos">369</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> -</span><span id="TypeAnnotator-370"><a href="#TypeAnnotator-370"><span class="linenos">370</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator-371"><a href="#TypeAnnotator-371"><span class="linenos">371</span></a> -</span><span id="TypeAnnotator-372"><a href="#TypeAnnotator-372"><span class="linenos">372</span></a> <span class="c1"># First annotate the current scope's column references</span> -</span><span id="TypeAnnotator-373"><a href="#TypeAnnotator-373"><span class="linenos">373</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> -</span><span id="TypeAnnotator-374"><a href="#TypeAnnotator-374"><span class="linenos">374</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> -</span><span id="TypeAnnotator-375"><a href="#TypeAnnotator-375"><span class="linenos">375</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator-376"><a href="#TypeAnnotator-376"><span class="linenos">376</span></a> -</span><span id="TypeAnnotator-377"><a href="#TypeAnnotator-377"><span class="linenos">377</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> -</span><span id="TypeAnnotator-378"><a href="#TypeAnnotator-378"><span class="linenos">378</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> -</span><span id="TypeAnnotator-379"><a href="#TypeAnnotator-379"><span class="linenos">379</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> -</span><span id="TypeAnnotator-380"><a href="#TypeAnnotator-380"><span class="linenos">380</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> -</span><span id="TypeAnnotator-381"><a href="#TypeAnnotator-381"><span class="linenos">381</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-382"><a href="#TypeAnnotator-382"><span class="linenos">382</span></a> -</span><span id="TypeAnnotator-383"><a href="#TypeAnnotator-383"><span class="linenos">383</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> -</span><span id="TypeAnnotator-384"><a href="#TypeAnnotator-384"><span class="linenos">384</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-385"><a href="#TypeAnnotator-385"><span class="linenos">385</span></a> -</span><span id="TypeAnnotator-386"><a href="#TypeAnnotator-386"><span class="linenos">386</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> -</span><span id="TypeAnnotator-387"><a href="#TypeAnnotator-387"><span class="linenos">387</span></a> -</span><span id="TypeAnnotator-388"><a href="#TypeAnnotator-388"><span class="linenos">388</span></a> <span class="k">def</span> <span class="nf">_maybe_annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-389"><a href="#TypeAnnotator-389"><span class="linenos">389</span></a> <span class="k">if</span> <span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> -</span><span id="TypeAnnotator-390"><a href="#TypeAnnotator-390"><span class="linenos">390</span></a> <span class="k">return</span> <span class="n">expression</span> <span class="c1"># We've already inferred the expression's type</span> -</span><span id="TypeAnnotator-391"><a href="#TypeAnnotator-391"><span class="linenos">391</span></a> -</span><span id="TypeAnnotator-392"><a href="#TypeAnnotator-392"><span class="linenos">392</span></a> <span class="n">annotator</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="vm">__class__</span><span class="p">)</span> -</span><span id="TypeAnnotator-393"><a href="#TypeAnnotator-393"><span class="linenos">393</span></a> -</span><span id="TypeAnnotator-394"><a href="#TypeAnnotator-394"><span class="linenos">394</span></a> <span class="k">return</span> <span class="p">(</span> -</span><span id="TypeAnnotator-395"><a href="#TypeAnnotator-395"><span class="linenos">395</span></a> <span class="n">annotator</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="TypeAnnotator-396"><a href="#TypeAnnotator-396"><span class="linenos">396</span></a> <span class="k">if</span> <span class="n">annotator</span> -</span><span id="TypeAnnotator-397"><a href="#TypeAnnotator-397"><span class="linenos">397</span></a> <span class="k">else</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> -</span><span id="TypeAnnotator-398"><a href="#TypeAnnotator-398"><span class="linenos">398</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-399"><a href="#TypeAnnotator-399"><span class="linenos">399</span></a> -</span><span id="TypeAnnotator-400"><a href="#TypeAnnotator-400"><span class="linenos">400</span></a> <span class="k">def</span> <span class="nf">_annotate_args</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-401"><a href="#TypeAnnotator-401"><span class="linenos">401</span></a> <span class="k">for</span> <span class="n">_</span><span class="p">,</span> <span class="n">value</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">iter_expressions</span><span class="p">():</span> -</span><span id="TypeAnnotator-402"><a href="#TypeAnnotator-402"><span class="linenos">402</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> -</span><span id="TypeAnnotator-403"><a href="#TypeAnnotator-403"><span class="linenos">403</span></a> -</span><span id="TypeAnnotator-404"><a href="#TypeAnnotator-404"><span class="linenos">404</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-405"><a href="#TypeAnnotator-405"><span class="linenos">405</span></a> -</span><span id="TypeAnnotator-406"><a href="#TypeAnnotator-406"><span class="linenos">406</span></a> <span class="k">def</span> <span class="nf">_maybe_coerce</span><span class="p">(</span> -</span><span id="TypeAnnotator-407"><a href="#TypeAnnotator-407"><span class="linenos">407</span></a> <span class="bp">self</span><span class="p">,</span> <span class="n">type1</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">type2</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> -</span><span id="TypeAnnotator-408"><a href="#TypeAnnotator-408"><span class="linenos">408</span></a> <span class="p">)</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">:</span> -</span><span id="TypeAnnotator-409"><a href="#TypeAnnotator-409"><span class="linenos">409</span></a> <span class="n">type1_value</span> <span class="o">=</span> <span class="n">type1</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type1</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type1</span> -</span><span id="TypeAnnotator-410"><a href="#TypeAnnotator-410"><span class="linenos">410</span></a> <span class="n">type2_value</span> <span class="o">=</span> <span class="n">type2</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type2</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type2</span> -</span><span id="TypeAnnotator-411"><a href="#TypeAnnotator-411"><span class="linenos">411</span></a> -</span><span id="TypeAnnotator-412"><a href="#TypeAnnotator-412"><span class="linenos">412</span></a> <span class="c1"># We propagate the NULL / UNKNOWN types upwards if found</span> -</span><span id="TypeAnnotator-413"><a href="#TypeAnnotator-413"><span class="linenos">413</span></a> <span class="k">if</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> -</span><span id="TypeAnnotator-414"><a href="#TypeAnnotator-414"><span class="linenos">414</span></a> <span class="k">return</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">NULL</span> -</span><span id="TypeAnnotator-415"><a href="#TypeAnnotator-415"><span class="linenos">415</span></a> <span class="k">if</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">UNKNOWN</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> -</span><span id="TypeAnnotator-416"><a href="#TypeAnnotator-416"><span class="linenos">416</span></a> <span class="k">return</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">UNKNOWN</span> -</span><span id="TypeAnnotator-417"><a href="#TypeAnnotator-417"><span class="linenos">417</span></a> -</span><span id="TypeAnnotator-418"><a href="#TypeAnnotator-418"><span class="linenos">418</span></a> <span class="k">if</span> <span class="n">type1_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-419"><a href="#TypeAnnotator-419"><span class="linenos">419</span></a> <span class="k">return</span> <span class="n">type1</span> -</span><span id="TypeAnnotator-420"><a href="#TypeAnnotator-420"><span class="linenos">420</span></a> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-421"><a href="#TypeAnnotator-421"><span class="linenos">421</span></a> <span class="k">return</span> <span class="n">type2</span> -</span><span id="TypeAnnotator-422"><a href="#TypeAnnotator-422"><span class="linenos">422</span></a> -</span><span id="TypeAnnotator-423"><a href="#TypeAnnotator-423"><span class="linenos">423</span></a> <span class="k">return</span> <span class="n">type2_value</span> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="p">{})</span> <span class="k">else</span> <span class="n">type1_value</span> <span class="c1"># type: ignore</span> -</span><span id="TypeAnnotator-424"><a href="#TypeAnnotator-424"><span class="linenos">424</span></a> -</span><span id="TypeAnnotator-425"><a href="#TypeAnnotator-425"><span class="linenos">425</span></a> <span class="c1"># Note: the following "no_type_check" decorators were added because mypy was yelling due</span> -</span><span id="TypeAnnotator-426"><a href="#TypeAnnotator-426"><span class="linenos">426</span></a> <span class="c1"># to assigning Type values to expression.type (since its getter returns Optional[DataType]).</span> -</span><span id="TypeAnnotator-427"><a href="#TypeAnnotator-427"><span class="linenos">427</span></a> <span class="c1"># This is a known mypy issue: https://github.com/python/mypy/issues/3004</span> -</span><span id="TypeAnnotator-428"><a href="#TypeAnnotator-428"><span class="linenos">428</span></a> -</span><span id="TypeAnnotator-429"><a href="#TypeAnnotator-429"><span class="linenos">429</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-430"><a href="#TypeAnnotator-430"><span class="linenos">430</span></a> <span class="k">def</span> <span class="nf">_annotate_binary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">B</span><span class="p">)</span> <span class="o">-></span> <span class="n">B</span><span class="p">:</span> -</span><span id="TypeAnnotator-431"><a href="#TypeAnnotator-431"><span class="linenos">431</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-432"><a href="#TypeAnnotator-432"><span class="linenos">432</span></a> -</span><span id="TypeAnnotator-433"><a href="#TypeAnnotator-433"><span class="linenos">433</span></a> <span class="n">left</span><span class="p">,</span> <span class="n">right</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">right</span> -</span><span id="TypeAnnotator-434"><a href="#TypeAnnotator-434"><span class="linenos">434</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">left</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> -</span><span id="TypeAnnotator-435"><a href="#TypeAnnotator-435"><span class="linenos">435</span></a> -</span><span id="TypeAnnotator-436"><a href="#TypeAnnotator-436"><span class="linenos">436</span></a> <span class="k">if</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">Connector</span><span class="p">):</span> -</span><span id="TypeAnnotator-437"><a href="#TypeAnnotator-437"><span class="linenos">437</span></a> <span class="k">if</span> <span class="n">left_type</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">NULL</span> <span class="ow">and</span> <span class="n">right_type</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">NULL</span><span class="p">:</span> -</span><span id="TypeAnnotator-438"><a href="#TypeAnnotator-438"><span class="linenos">438</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">NULL</span><span class="p">)</span> -</span><span id="TypeAnnotator-439"><a href="#TypeAnnotator-439"><span class="linenos">439</span></a> <span class="k">elif</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">):</span> -</span><span id="TypeAnnotator-440"><a href="#TypeAnnotator-440"><span class="linenos">440</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="TypeAnnotator-441"><a href="#TypeAnnotator-441"><span class="linenos">441</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="TypeAnnotator-442"><a href="#TypeAnnotator-442"><span class="linenos">442</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">build</span><span class="p">(</span><span class="s2">"NULLABLE"</span><span class="p">,</span> <span class="n">expressions</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">build</span><span class="p">(</span><span class="s2">"BOOLEAN"</span><span class="p">)),</span> -</span><span id="TypeAnnotator-443"><a href="#TypeAnnotator-443"><span class="linenos">443</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-444"><a href="#TypeAnnotator-444"><span class="linenos">444</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-445"><a href="#TypeAnnotator-445"><span class="linenos">445</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="TypeAnnotator-446"><a href="#TypeAnnotator-446"><span class="linenos">446</span></a> <span class="k">elif</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">Predicate</span><span class="p">):</span> -</span><span id="TypeAnnotator-447"><a href="#TypeAnnotator-447"><span class="linenos">447</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="TypeAnnotator-448"><a href="#TypeAnnotator-448"><span class="linenos">448</span></a> <span class="k">elif</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">:</span> -</span><span id="TypeAnnotator-449"><a href="#TypeAnnotator-449"><span class="linenos">449</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">[(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)](</span><span class="n">left</span><span class="p">,</span> <span class="n">right</span><span class="p">))</span> -</span><span id="TypeAnnotator-450"><a href="#TypeAnnotator-450"><span class="linenos">450</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-451"><a href="#TypeAnnotator-451"><span class="linenos">451</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> -</span><span id="TypeAnnotator-452"><a href="#TypeAnnotator-452"><span class="linenos">452</span></a> -</span><span id="TypeAnnotator-453"><a href="#TypeAnnotator-453"><span class="linenos">453</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-454"><a href="#TypeAnnotator-454"><span class="linenos">454</span></a> -</span><span id="TypeAnnotator-455"><a href="#TypeAnnotator-455"><span class="linenos">455</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-456"><a href="#TypeAnnotator-456"><span class="linenos">456</span></a> <span class="k">def</span> <span class="nf">_annotate_unary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-457"><a href="#TypeAnnotator-457"><span class="linenos">457</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-458"><a href="#TypeAnnotator-458"><span class="linenos">458</span></a> -</span><span id="TypeAnnotator-459"><a href="#TypeAnnotator-459"><span class="linenos">459</span></a> <span class="k">if</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">Condition</span><span class="p">)</span> <span class="ow">and</span> <span class="ow">not</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">Paren</span><span class="p">):</span> -</span><span id="TypeAnnotator-460"><a href="#TypeAnnotator-460"><span class="linenos">460</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> -</span><span id="TypeAnnotator-461"><a href="#TypeAnnotator-461"><span class="linenos">461</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-462"><a href="#TypeAnnotator-462"><span class="linenos">462</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</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">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-463"><a href="#TypeAnnotator-463"><span class="linenos">463</span></a> -</span><span id="TypeAnnotator-464"><a href="#TypeAnnotator-464"><span class="linenos">464</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-465"><a href="#TypeAnnotator-465"><span class="linenos">465</span></a> -</span><span id="TypeAnnotator-466"><a href="#TypeAnnotator-466"><span class="linenos">466</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-467"><a href="#TypeAnnotator-467"><span class="linenos">467</span></a> <span class="k">def</span> <span class="nf">_annotate_literal</span><span class="p">(</span><span class="bp">self</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">Literal</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> -</span><span id="TypeAnnotator-468"><a href="#TypeAnnotator-468"><span class="linenos">468</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> -</span><span id="TypeAnnotator-469"><a href="#TypeAnnotator-469"><span class="linenos">469</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">VARCHAR</span><span class="p">)</span> -</span><span id="TypeAnnotator-470"><a href="#TypeAnnotator-470"><span class="linenos">470</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_int</span><span class="p">:</span> -</span><span id="TypeAnnotator-471"><a href="#TypeAnnotator-471"><span class="linenos">471</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">INT</span><span class="p">)</span> -</span><span id="TypeAnnotator-472"><a href="#TypeAnnotator-472"><span class="linenos">472</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-473"><a href="#TypeAnnotator-473"><span class="linenos">473</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> -</span><span id="TypeAnnotator-474"><a href="#TypeAnnotator-474"><span class="linenos">474</span></a> -</span><span id="TypeAnnotator-475"><a href="#TypeAnnotator-475"><span class="linenos">475</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-476"><a href="#TypeAnnotator-476"><span class="linenos">476</span></a> -</span><span id="TypeAnnotator-477"><a href="#TypeAnnotator-477"><span class="linenos">477</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-478"><a href="#TypeAnnotator-478"><span class="linenos">478</span></a> <span class="k">def</span> <span class="nf">_annotate_with_type</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> <span class="n">target_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="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-479"><a href="#TypeAnnotator-479"><span class="linenos">479</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">target_type</span><span class="p">)</span> -</span><span id="TypeAnnotator-480"><a href="#TypeAnnotator-480"><span class="linenos">480</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-481"><a href="#TypeAnnotator-481"><span class="linenos">481</span></a> -</span><span id="TypeAnnotator-482"><a href="#TypeAnnotator-482"><span class="linenos">482</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-483"><a href="#TypeAnnotator-483"><span class="linenos">483</span></a> <span class="k">def</span> <span class="nf">_annotate_struct_value</span><span class="p">(</span> -</span><span id="TypeAnnotator-484"><a href="#TypeAnnotator-484"><span class="linenos">484</span></a> <span class="bp">self</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><span id="TypeAnnotator-485"><a href="#TypeAnnotator-485"><span class="linenos">485</span></a> <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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">]</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">:</span> -</span><span id="TypeAnnotator-486"><a href="#TypeAnnotator-486"><span class="linenos">486</span></a> <span class="n">alias</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">"alias"</span><span class="p">)</span> -</span><span id="TypeAnnotator-487"><a href="#TypeAnnotator-487"><span class="linenos">487</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> -</span><span id="TypeAnnotator-488"><a href="#TypeAnnotator-488"><span class="linenos">488</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">alias</span><span class="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-489"><a href="#TypeAnnotator-489"><span class="linenos">489</span></a> -</span><span id="TypeAnnotator-490"><a href="#TypeAnnotator-490"><span class="linenos">490</span></a> <span class="c1"># Case: key = value or key := value</span> -</span><span id="TypeAnnotator-491"><a href="#TypeAnnotator-491"><span class="linenos">491</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> -</span><span id="TypeAnnotator-492"><a href="#TypeAnnotator-492"><span class="linenos">492</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</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="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-493"><a href="#TypeAnnotator-493"><span class="linenos">493</span></a> -</span><span id="TypeAnnotator-494"><a href="#TypeAnnotator-494"><span class="linenos">494</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> -</span><span id="TypeAnnotator-495"><a href="#TypeAnnotator-495"><span class="linenos">495</span></a> -</span><span id="TypeAnnotator-496"><a href="#TypeAnnotator-496"><span class="linenos">496</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> -</span><span id="TypeAnnotator-497"><a href="#TypeAnnotator-497"><span class="linenos">497</span></a> <span class="k">def</span> <span class="nf">_annotate_by_args</span><span class="p">(</span> -</span><span id="TypeAnnotator-498"><a href="#TypeAnnotator-498"><span class="linenos">498</span></a> <span class="bp">self</span><span class="p">,</span> -</span><span id="TypeAnnotator-499"><a href="#TypeAnnotator-499"><span class="linenos">499</span></a> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> -</span><span id="TypeAnnotator-500"><a href="#TypeAnnotator-500"><span class="linenos">500</span></a> <span class="o">*</span><span class="n">args</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> -</span><span id="TypeAnnotator-501"><a href="#TypeAnnotator-501"><span class="linenos">501</span></a> <span class="n">promote</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="TypeAnnotator-502"><a href="#TypeAnnotator-502"><span class="linenos">502</span></a> <span class="n">array</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="TypeAnnotator-503"><a href="#TypeAnnotator-503"><span class="linenos">503</span></a> <span class="n">struct</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> -</span><span id="TypeAnnotator-504"><a href="#TypeAnnotator-504"><span class="linenos">504</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator-505"><a href="#TypeAnnotator-505"><span class="linenos">505</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-506"><a href="#TypeAnnotator-506"><span class="linenos">506</span></a> -</span><span id="TypeAnnotator-507"><a href="#TypeAnnotator-507"><span class="linenos">507</span></a> <span class="n">expressions</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="o">=</span> <span class="p">[]</span> -</span><span id="TypeAnnotator-508"><a href="#TypeAnnotator-508"><span class="linenos">508</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><span id="TypeAnnotator-509"><a href="#TypeAnnotator-509"><span class="linenos">509</span></a> <span class="n">arg_expr</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="n">arg</span><span class="p">)</span> -</span><span id="TypeAnnotator-510"><a href="#TypeAnnotator-510"><span class="linenos">510</span></a> <span class="n">expressions</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">expr</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">ensure_list</span><span class="p">(</span><span class="n">arg_expr</span><span class="p">)</span> <span class="k">if</span> <span class="n">expr</span><span class="p">)</span> -</span><span id="TypeAnnotator-511"><a href="#TypeAnnotator-511"><span class="linenos">511</span></a> -</span><span id="TypeAnnotator-512"><a href="#TypeAnnotator-512"><span class="linenos">512</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="kc">None</span> -</span><span id="TypeAnnotator-513"><a href="#TypeAnnotator-513"><span class="linenos">513</span></a> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">:</span> -</span><span id="TypeAnnotator-514"><a href="#TypeAnnotator-514"><span class="linenos">514</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">last_datatype</span> <span class="ow">or</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">,</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-515"><a href="#TypeAnnotator-515"><span class="linenos">515</span></a> -</span><span id="TypeAnnotator-516"><a href="#TypeAnnotator-516"><span class="linenos">516</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">last_datatype</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">UNKNOWN</span><span class="p">)</span> -</span><span id="TypeAnnotator-517"><a href="#TypeAnnotator-517"><span class="linenos">517</span></a> -</span><span id="TypeAnnotator-518"><a href="#TypeAnnotator-518"><span class="linenos">518</span></a> <span class="k">if</span> <span class="n">promote</span><span class="p">:</span> -</span><span id="TypeAnnotator-519"><a href="#TypeAnnotator-519"><span class="linenos">519</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-520"><a href="#TypeAnnotator-520"><span class="linenos">520</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> -</span><span id="TypeAnnotator-521"><a href="#TypeAnnotator-521"><span class="linenos">521</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">FLOAT_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-522"><a href="#TypeAnnotator-522"><span class="linenos">522</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> -</span><span id="TypeAnnotator-523"><a href="#TypeAnnotator-523"><span class="linenos">523</span></a> -</span><span id="TypeAnnotator-524"><a href="#TypeAnnotator-524"><span class="linenos">524</span></a> <span class="k">if</span> <span class="n">array</span><span class="p">:</span> -</span><span id="TypeAnnotator-525"><a href="#TypeAnnotator-525"><span class="linenos">525</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="TypeAnnotator-526"><a href="#TypeAnnotator-526"><span class="linenos">526</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="TypeAnnotator-527"><a href="#TypeAnnotator-527"><span class="linenos">527</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> -</span><span id="TypeAnnotator-528"><a href="#TypeAnnotator-528"><span class="linenos">528</span></a> <span class="n">this</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">ARRAY</span><span class="p">,</span> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">],</span> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span> -</span><span id="TypeAnnotator-529"><a href="#TypeAnnotator-529"><span class="linenos">529</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-530"><a href="#TypeAnnotator-530"><span class="linenos">530</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-531"><a href="#TypeAnnotator-531"><span class="linenos">531</span></a> -</span><span id="TypeAnnotator-532"><a href="#TypeAnnotator-532"><span class="linenos">532</span></a> <span class="k">if</span> <span class="n">struct</span><span class="p">:</span> -</span><span id="TypeAnnotator-533"><a href="#TypeAnnotator-533"><span class="linenos">533</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> -</span><span id="TypeAnnotator-534"><a href="#TypeAnnotator-534"><span class="linenos">534</span></a> <span class="n">expression</span><span class="p">,</span> -</span><span id="TypeAnnotator-535"><a href="#TypeAnnotator-535"><span class="linenos">535</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> -</span><span id="TypeAnnotator-536"><a href="#TypeAnnotator-536"><span class="linenos">536</span></a> <span class="n">this</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">STRUCT</span><span class="p">,</span> -</span><span id="TypeAnnotator-537"><a href="#TypeAnnotator-537"><span class="linenos">537</span></a> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">_annotate_struct_value</span><span class="p">(</span><span class="n">expr</span><span class="p">)</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">],</span> -</span><span id="TypeAnnotator-538"><a href="#TypeAnnotator-538"><span class="linenos">538</span></a> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> -</span><span id="TypeAnnotator-539"><a href="#TypeAnnotator-539"><span class="linenos">539</span></a> <span class="p">),</span> -</span><span id="TypeAnnotator-540"><a href="#TypeAnnotator-540"><span class="linenos">540</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator-541"><a href="#TypeAnnotator-541"><span class="linenos">541</span></a> -</span><span id="TypeAnnotator-542"><a href="#TypeAnnotator-542"><span class="linenos">542</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-543"><a href="#TypeAnnotator-543"><span class="linenos">543</span></a> -</span><span id="TypeAnnotator-544"><a href="#TypeAnnotator-544"><span class="linenos">544</span></a> <span class="k">def</span> <span class="nf">_annotate_timeunit</span><span class="p">(</span> -</span><span id="TypeAnnotator-545"><a href="#TypeAnnotator-545"><span class="linenos">545</span></a> <span class="bp">self</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">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> -</span><span id="TypeAnnotator-546"><a href="#TypeAnnotator-546"><span class="linenos">546</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> -</span><span id="TypeAnnotator-547"><a href="#TypeAnnotator-547"><span class="linenos">547</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-548"><a href="#TypeAnnotator-548"><span class="linenos">548</span></a> -</span><span id="TypeAnnotator-549"><a href="#TypeAnnotator-549"><span class="linenos">549</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-550"><a href="#TypeAnnotator-550"><span class="linenos">550</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date_literal</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">unit</span><span class="p">)</span> -</span><span id="TypeAnnotator-551"><a href="#TypeAnnotator-551"><span class="linenos">551</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEMPORAL_TYPES</span><span class="p">:</span> -</span><span id="TypeAnnotator-552"><a href="#TypeAnnotator-552"><span class="linenos">552</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date</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">unit</span><span class="p">)</span> -</span><span id="TypeAnnotator-553"><a href="#TypeAnnotator-553"><span class="linenos">553</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-554"><a href="#TypeAnnotator-554"><span class="linenos">554</span></a> <span class="n">datatype</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">UNKNOWN</span> -</span><span id="TypeAnnotator-555"><a href="#TypeAnnotator-555"><span class="linenos">555</span></a> -</span><span id="TypeAnnotator-556"><a href="#TypeAnnotator-556"><span class="linenos">556</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">datatype</span><span class="p">)</span> -</span><span id="TypeAnnotator-557"><a href="#TypeAnnotator-557"><span class="linenos">557</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-558"><a href="#TypeAnnotator-558"><span class="linenos">558</span></a> -</span><span id="TypeAnnotator-559"><a href="#TypeAnnotator-559"><span class="linenos">559</span></a> <span class="k">def</span> <span class="nf">_annotate_bracket</span><span class="p">(</span><span class="bp">self</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">Bracket</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Bracket</span><span class="p">:</span> -</span><span id="TypeAnnotator-560"><a href="#TypeAnnotator-560"><span class="linenos">560</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-561"><a href="#TypeAnnotator-561"><span class="linenos">561</span></a> -</span><span id="TypeAnnotator-562"><a href="#TypeAnnotator-562"><span class="linenos">562</span></a> <span class="n">bracket_arg</span> <span class="o">=</span> <span class="n">expression</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="TypeAnnotator-563"><a href="#TypeAnnotator-563"><span class="linenos">563</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><span id="TypeAnnotator-564"><a href="#TypeAnnotator-564"><span class="linenos">564</span></a> -</span><span id="TypeAnnotator-565"><a href="#TypeAnnotator-565"><span class="linenos">565</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">):</span> -</span><span id="TypeAnnotator-566"><a href="#TypeAnnotator-566"><span class="linenos">566</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator-567"><a href="#TypeAnnotator-567"><span class="linenos">567</span></a> <span class="k">elif</span> <span class="n">this</span><span class="o">.</span><span class="n">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">ARRAY</span><span class="p">):</span> -</span><span id="TypeAnnotator-568"><a href="#TypeAnnotator-568"><span class="linenos">568</span></a> <span class="n">contained_type</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">expressions</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</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">UNKNOWN</span> -</span><span id="TypeAnnotator-569"><a href="#TypeAnnotator-569"><span class="linenos">569</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">contained_type</span><span class="p">)</span> +</span><span id="TypeAnnotator-267"><a href="#TypeAnnotator-267"><span class="linenos">267</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_explode</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> +</span><span id="TypeAnnotator-268"><a href="#TypeAnnotator-268"><span class="linenos">268</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Filter</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">),</span> +</span><span id="TypeAnnotator-269"><a href="#TypeAnnotator-269"><span class="linenos">269</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">If</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"true"</span><span class="p">,</span> <span class="s2">"false"</span><span class="p">),</span> +</span><span id="TypeAnnotator-270"><a href="#TypeAnnotator-270"><span class="linenos">270</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Interval</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">INTERVAL</span><span class="p">),</span> +</span><span id="TypeAnnotator-271"><a href="#TypeAnnotator-271"><span class="linenos">271</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Least</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="TypeAnnotator-272"><a href="#TypeAnnotator-272"><span class="linenos">272</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_literal</span><span class="p">(</span><span class="n">e</span><span class="p">),</span> +</span><span id="TypeAnnotator-273"><a href="#TypeAnnotator-273"><span class="linenos">273</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> +</span><span id="TypeAnnotator-274"><a href="#TypeAnnotator-274"><span class="linenos">274</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Max</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="TypeAnnotator-275"><a href="#TypeAnnotator-275"><span class="linenos">275</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Min</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">),</span> +</span><span id="TypeAnnotator-276"><a href="#TypeAnnotator-276"><span class="linenos">276</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Null</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">NULL</span><span class="p">),</span> +</span><span id="TypeAnnotator-277"><a href="#TypeAnnotator-277"><span class="linenos">277</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Nullif</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expression"</span><span class="p">),</span> +</span><span id="TypeAnnotator-278"><a href="#TypeAnnotator-278"><span class="linenos">278</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">UNKNOWN</span><span class="p">),</span> +</span><span id="TypeAnnotator-279"><a href="#TypeAnnotator-279"><span class="linenos">279</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Sum</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"this"</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">promote</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> +</span><span id="TypeAnnotator-280"><a href="#TypeAnnotator-280"><span class="linenos">280</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">TryCast</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="s2">"to"</span><span class="p">]),</span> +</span><span id="TypeAnnotator-281"><a href="#TypeAnnotator-281"><span class="linenos">281</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</span><span class="p">(</span><span class="n">e</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">MAP</span><span class="p">),</span> +</span><span id="TypeAnnotator-282"><a href="#TypeAnnotator-282"><span class="linenos">282</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">Struct</span><span class="p">:</span> <span class="k">lambda</span> <span class="bp">self</span><span class="p">,</span> <span class="n">e</span><span class="p">:</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_by_args</span><span class="p">(</span><span class="n">e</span><span class="p">,</span> <span class="s2">"expressions"</span><span class="p">,</span> <span class="n">struct</span><span class="o">=</span><span class="kc">True</span><span class="p">),</span> +</span><span id="TypeAnnotator-283"><a href="#TypeAnnotator-283"><span class="linenos">283</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-284"><a href="#TypeAnnotator-284"><span class="linenos">284</span></a> +</span><span id="TypeAnnotator-285"><a href="#TypeAnnotator-285"><span class="linenos">285</span></a> <span class="n">NESTED_TYPES</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator-286"><a href="#TypeAnnotator-286"><span class="linenos">286</span></a> <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">ARRAY</span><span class="p">,</span> +</span><span id="TypeAnnotator-287"><a href="#TypeAnnotator-287"><span class="linenos">287</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-288"><a href="#TypeAnnotator-288"><span class="linenos">288</span></a> +</span><span id="TypeAnnotator-289"><a href="#TypeAnnotator-289"><span class="linenos">289</span></a> <span class="c1"># Specifies what types a given type can be coerced into (autofilled)</span> +</span><span id="TypeAnnotator-290"><a href="#TypeAnnotator-290"><span class="linenos">290</span></a> <span class="n">COERCES_TO</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]</span> <span class="o">=</span> <span class="p">{}</span> +</span><span id="TypeAnnotator-291"><a href="#TypeAnnotator-291"><span class="linenos">291</span></a> +</span><span id="TypeAnnotator-292"><a href="#TypeAnnotator-292"><span class="linenos">292</span></a> <span class="c1"># Coercion functions for binary operations.</span> +</span><span id="TypeAnnotator-293"><a href="#TypeAnnotator-293"><span class="linenos">293</span></a> <span class="c1"># Map of type pairs to a callable that takes both sides of the binary operation and returns the resulting type.</span> +</span><span id="TypeAnnotator-294"><a href="#TypeAnnotator-294"><span class="linenos">294</span></a> <span class="n">BINARY_COERCIONS</span><span class="p">:</span> <span class="n">BinaryCoercions</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator-295"><a href="#TypeAnnotator-295"><span class="linenos">295</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="TypeAnnotator-296"><a href="#TypeAnnotator-296"><span class="linenos">296</span></a> <span class="p">{</span> +</span><span id="TypeAnnotator-297"><a href="#TypeAnnotator-297"><span class="linenos">297</span></a> <span class="p">(</span><span class="n">t</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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date_literal</span><span class="p">(</span> +</span><span id="TypeAnnotator-298"><a href="#TypeAnnotator-298"><span class="linenos">298</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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="TypeAnnotator-299"><a href="#TypeAnnotator-299"><span class="linenos">299</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-300"><a href="#TypeAnnotator-300"><span class="linenos">300</span></a> <span class="k">for</span> <span class="n">t</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> +</span><span id="TypeAnnotator-301"><a href="#TypeAnnotator-301"><span class="linenos">301</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-302"><a href="#TypeAnnotator-302"><span class="linenos">302</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-303"><a href="#TypeAnnotator-303"><span class="linenos">303</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="TypeAnnotator-304"><a href="#TypeAnnotator-304"><span class="linenos">304</span></a> <span class="p">{</span> +</span><span id="TypeAnnotator-305"><a href="#TypeAnnotator-305"><span class="linenos">305</span></a> <span class="c1"># text + numeric will yield the numeric type to match most dialects' semantics</span> +</span><span id="TypeAnnotator-306"><a href="#TypeAnnotator-306"><span class="linenos">306</span></a> <span class="p">(</span><span class="n">text</span><span class="p">,</span> <span class="n">numeric</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span> +</span><span id="TypeAnnotator-307"><a href="#TypeAnnotator-307"><span class="linenos">307</span></a> <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="p">,</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">l</span><span class="o">.</span><span class="n">type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> <span class="k">else</span> <span class="n">r</span><span class="o">.</span><span class="n">type</span> +</span><span id="TypeAnnotator-308"><a href="#TypeAnnotator-308"><span class="linenos">308</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-309"><a href="#TypeAnnotator-309"><span class="linenos">309</span></a> <span class="k">for</span> <span class="n">text</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span> +</span><span id="TypeAnnotator-310"><a href="#TypeAnnotator-310"><span class="linenos">310</span></a> <span class="k">for</span> <span class="n">numeric</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">NUMERIC_TYPES</span> +</span><span id="TypeAnnotator-311"><a href="#TypeAnnotator-311"><span class="linenos">311</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-312"><a href="#TypeAnnotator-312"><span class="linenos">312</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-313"><a href="#TypeAnnotator-313"><span class="linenos">313</span></a> <span class="o">**</span><span class="n">swap_all</span><span class="p">(</span> +</span><span id="TypeAnnotator-314"><a href="#TypeAnnotator-314"><span class="linenos">314</span></a> <span class="p">{</span> +</span><span id="TypeAnnotator-315"><a href="#TypeAnnotator-315"><span class="linenos">315</span></a> <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 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">INTERVAL</span><span class="p">):</span> <span class="k">lambda</span> <span class="n">l</span><span class="p">,</span> <span class="n">r</span><span class="p">:</span> <span class="n">_coerce_date</span><span class="p">(</span> +</span><span id="TypeAnnotator-316"><a href="#TypeAnnotator-316"><span class="linenos">316</span></a> <span class="n">l</span><span class="p">,</span> <span class="n">r</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="TypeAnnotator-317"><a href="#TypeAnnotator-317"><span class="linenos">317</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-318"><a href="#TypeAnnotator-318"><span class="linenos">318</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-319"><a href="#TypeAnnotator-319"><span class="linenos">319</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-320"><a href="#TypeAnnotator-320"><span class="linenos">320</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-321"><a href="#TypeAnnotator-321"><span class="linenos">321</span></a> +</span><span id="TypeAnnotator-322"><a href="#TypeAnnotator-322"><span class="linenos">322</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> +</span><span id="TypeAnnotator-323"><a href="#TypeAnnotator-323"><span class="linenos">323</span></a> <span class="bp">self</span><span class="p">,</span> +</span><span id="TypeAnnotator-324"><a href="#TypeAnnotator-324"><span class="linenos">324</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> +</span><span id="TypeAnnotator-325"><a href="#TypeAnnotator-325"><span class="linenos">325</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator-326"><a href="#TypeAnnotator-326"><span class="linenos">326</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator-327"><a href="#TypeAnnotator-327"><span class="linenos">327</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator-328"><a href="#TypeAnnotator-328"><span class="linenos">328</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> +</span><span id="TypeAnnotator-329"><a href="#TypeAnnotator-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> +</span><span id="TypeAnnotator-330"><a href="#TypeAnnotator-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> +</span><span id="TypeAnnotator-331"><a href="#TypeAnnotator-331"><span class="linenos">331</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> +</span><span id="TypeAnnotator-332"><a href="#TypeAnnotator-332"><span class="linenos">332</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> +</span><span id="TypeAnnotator-333"><a href="#TypeAnnotator-333"><span class="linenos">333</span></a> +</span><span id="TypeAnnotator-334"><a href="#TypeAnnotator-334"><span class="linenos">334</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> +</span><span id="TypeAnnotator-335"><a href="#TypeAnnotator-335"><span class="linenos">335</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> +</span><span id="TypeAnnotator-336"><a href="#TypeAnnotator-336"><span class="linenos">336</span></a> +</span><span id="TypeAnnotator-337"><a href="#TypeAnnotator-337"><span class="linenos">337</span></a> <span class="k">def</span> <span class="nf">_set_type</span><span class="p">(</span> +</span><span id="TypeAnnotator-338"><a href="#TypeAnnotator-338"><span class="linenos">338</span></a> <span class="bp">self</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">target_type</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">DataType</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="p">]</span> +</span><span id="TypeAnnotator-339"><a href="#TypeAnnotator-339"><span class="linenos">339</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> +</span><span id="TypeAnnotator-340"><a href="#TypeAnnotator-340"><span class="linenos">340</span></a> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> <span class="o">=</span> <span class="n">target_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">UNKNOWN</span> <span class="c1"># type: ignore</span> +</span><span id="TypeAnnotator-341"><a href="#TypeAnnotator-341"><span class="linenos">341</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="o">.</span><span class="n">add</span><span class="p">(</span><span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">))</span> +</span><span id="TypeAnnotator-342"><a href="#TypeAnnotator-342"><span class="linenos">342</span></a> +</span><span id="TypeAnnotator-343"><a href="#TypeAnnotator-343"><span class="linenos">343</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-344"><a href="#TypeAnnotator-344"><span class="linenos">344</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> +</span><span id="TypeAnnotator-345"><a href="#TypeAnnotator-345"><span class="linenos">345</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> +</span><span id="TypeAnnotator-346"><a href="#TypeAnnotator-346"><span class="linenos">346</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> +</span><span id="TypeAnnotator-347"><a href="#TypeAnnotator-347"><span class="linenos">347</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> +</span><span id="TypeAnnotator-348"><a href="#TypeAnnotator-348"><span class="linenos">348</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator-349"><a href="#TypeAnnotator-349"><span class="linenos">349</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> +</span><span id="TypeAnnotator-350"><a href="#TypeAnnotator-350"><span class="linenos">350</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="TypeAnnotator-351"><a href="#TypeAnnotator-351"><span class="linenos">351</span></a> +</span><span id="TypeAnnotator-352"><a href="#TypeAnnotator-352"><span class="linenos">352</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> +</span><span id="TypeAnnotator-353"><a href="#TypeAnnotator-353"><span class="linenos">353</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> +</span><span id="TypeAnnotator-354"><a href="#TypeAnnotator-354"><span class="linenos">354</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> +</span><span id="TypeAnnotator-355"><a href="#TypeAnnotator-355"><span class="linenos">355</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-356"><a href="#TypeAnnotator-356"><span class="linenos">356</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> +</span><span id="TypeAnnotator-357"><a href="#TypeAnnotator-357"><span class="linenos">357</span></a> +</span><span id="TypeAnnotator-358"><a href="#TypeAnnotator-358"><span class="linenos">358</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> +</span><span id="TypeAnnotator-359"><a href="#TypeAnnotator-359"><span class="linenos">359</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator-360"><a href="#TypeAnnotator-360"><span class="linenos">360</span></a> +</span><span id="TypeAnnotator-361"><a href="#TypeAnnotator-361"><span class="linenos">361</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator-362"><a href="#TypeAnnotator-362"><span class="linenos">362</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> +</span><span id="TypeAnnotator-363"><a href="#TypeAnnotator-363"><span class="linenos">363</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> +</span><span id="TypeAnnotator-364"><a href="#TypeAnnotator-364"><span class="linenos">364</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> +</span><span id="TypeAnnotator-365"><a href="#TypeAnnotator-365"><span class="linenos">365</span></a> <span class="n">values</span><span class="p">,</span> +</span><span id="TypeAnnotator-366"><a href="#TypeAnnotator-366"><span class="linenos">366</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-367"><a href="#TypeAnnotator-367"><span class="linenos">367</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-368"><a href="#TypeAnnotator-368"><span class="linenos">368</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-369"><a href="#TypeAnnotator-369"><span class="linenos">369</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator-370"><a href="#TypeAnnotator-370"><span class="linenos">370</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> +</span><span id="TypeAnnotator-371"><a href="#TypeAnnotator-371"><span class="linenos">371</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator-372"><a href="#TypeAnnotator-372"><span class="linenos">372</span></a> +</span><span id="TypeAnnotator-373"><a href="#TypeAnnotator-373"><span class="linenos">373</span></a> <span class="c1"># First annotate the current scope's column references</span> +</span><span id="TypeAnnotator-374"><a href="#TypeAnnotator-374"><span class="linenos">374</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> +</span><span id="TypeAnnotator-375"><a href="#TypeAnnotator-375"><span class="linenos">375</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> +</span><span id="TypeAnnotator-376"><a href="#TypeAnnotator-376"><span class="linenos">376</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator-377"><a href="#TypeAnnotator-377"><span class="linenos">377</span></a> +</span><span id="TypeAnnotator-378"><a href="#TypeAnnotator-378"><span class="linenos">378</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> +</span><span id="TypeAnnotator-379"><a href="#TypeAnnotator-379"><span class="linenos">379</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> +</span><span id="TypeAnnotator-380"><a href="#TypeAnnotator-380"><span class="linenos">380</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> +</span><span id="TypeAnnotator-381"><a href="#TypeAnnotator-381"><span class="linenos">381</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> +</span><span id="TypeAnnotator-382"><a href="#TypeAnnotator-382"><span class="linenos">382</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-383"><a href="#TypeAnnotator-383"><span class="linenos">383</span></a> +</span><span id="TypeAnnotator-384"><a href="#TypeAnnotator-384"><span class="linenos">384</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> +</span><span id="TypeAnnotator-385"><a href="#TypeAnnotator-385"><span class="linenos">385</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-386"><a href="#TypeAnnotator-386"><span class="linenos">386</span></a> +</span><span id="TypeAnnotator-387"><a href="#TypeAnnotator-387"><span class="linenos">387</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> +</span><span id="TypeAnnotator-388"><a href="#TypeAnnotator-388"><span class="linenos">388</span></a> +</span><span id="TypeAnnotator-389"><a href="#TypeAnnotator-389"><span class="linenos">389</span></a> <span class="k">def</span> <span class="nf">_maybe_annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-390"><a href="#TypeAnnotator-390"><span class="linenos">390</span></a> <span class="k">if</span> <span class="nb">id</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> +</span><span id="TypeAnnotator-391"><a href="#TypeAnnotator-391"><span class="linenos">391</span></a> <span class="k">return</span> <span class="n">expression</span> <span class="c1"># We've already inferred the expression's type</span> +</span><span id="TypeAnnotator-392"><a href="#TypeAnnotator-392"><span class="linenos">392</span></a> +</span><span id="TypeAnnotator-393"><a href="#TypeAnnotator-393"><span class="linenos">393</span></a> <span class="n">annotator</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">expression</span><span class="o">.</span><span class="vm">__class__</span><span class="p">)</span> +</span><span id="TypeAnnotator-394"><a href="#TypeAnnotator-394"><span class="linenos">394</span></a> +</span><span id="TypeAnnotator-395"><a href="#TypeAnnotator-395"><span class="linenos">395</span></a> <span class="k">return</span> <span class="p">(</span> +</span><span id="TypeAnnotator-396"><a href="#TypeAnnotator-396"><span class="linenos">396</span></a> <span class="n">annotator</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="TypeAnnotator-397"><a href="#TypeAnnotator-397"><span class="linenos">397</span></a> <span class="k">if</span> <span class="n">annotator</span> +</span><span id="TypeAnnotator-398"><a href="#TypeAnnotator-398"><span class="linenos">398</span></a> <span class="k">else</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_with_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> +</span><span id="TypeAnnotator-399"><a href="#TypeAnnotator-399"><span class="linenos">399</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-400"><a href="#TypeAnnotator-400"><span class="linenos">400</span></a> +</span><span id="TypeAnnotator-401"><a href="#TypeAnnotator-401"><span class="linenos">401</span></a> <span class="k">def</span> <span class="nf">_annotate_args</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-402"><a href="#TypeAnnotator-402"><span class="linenos">402</span></a> <span class="k">for</span> <span class="n">_</span><span class="p">,</span> <span class="n">value</span> <span class="ow">in</span> <span class="n">expression</span><span class="o">.</span><span class="n">iter_expressions</span><span class="p">():</span> +</span><span id="TypeAnnotator-403"><a href="#TypeAnnotator-403"><span class="linenos">403</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">value</span><span class="p">)</span> +</span><span id="TypeAnnotator-404"><a href="#TypeAnnotator-404"><span class="linenos">404</span></a> +</span><span id="TypeAnnotator-405"><a href="#TypeAnnotator-405"><span class="linenos">405</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-406"><a href="#TypeAnnotator-406"><span class="linenos">406</span></a> +</span><span id="TypeAnnotator-407"><a href="#TypeAnnotator-407"><span class="linenos">407</span></a> <span class="k">def</span> <span class="nf">_maybe_coerce</span><span class="p">(</span> +</span><span id="TypeAnnotator-408"><a href="#TypeAnnotator-408"><span class="linenos">408</span></a> <span class="bp">self</span><span class="p">,</span> <span class="n">type1</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">type2</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span> +</span><span id="TypeAnnotator-409"><a href="#TypeAnnotator-409"><span class="linenos">409</span></a> <span class="p">)</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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">:</span> +</span><span id="TypeAnnotator-410"><a href="#TypeAnnotator-410"><span class="linenos">410</span></a> <span class="n">type1_value</span> <span class="o">=</span> <span class="n">type1</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type1</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type1</span> +</span><span id="TypeAnnotator-411"><a href="#TypeAnnotator-411"><span class="linenos">411</span></a> <span class="n">type2_value</span> <span class="o">=</span> <span class="n">type2</span><span class="o">.</span><span class="n">this</span> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">type2</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">)</span> <span class="k">else</span> <span class="n">type2</span> +</span><span id="TypeAnnotator-412"><a href="#TypeAnnotator-412"><span class="linenos">412</span></a> +</span><span id="TypeAnnotator-413"><a href="#TypeAnnotator-413"><span class="linenos">413</span></a> <span class="c1"># We propagate the NULL / UNKNOWN types upwards if found</span> +</span><span id="TypeAnnotator-414"><a href="#TypeAnnotator-414"><span class="linenos">414</span></a> <span class="k">if</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> +</span><span id="TypeAnnotator-415"><a href="#TypeAnnotator-415"><span class="linenos">415</span></a> <span class="k">return</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">NULL</span> +</span><span id="TypeAnnotator-416"><a href="#TypeAnnotator-416"><span class="linenos">416</span></a> <span class="k">if</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">UNKNOWN</span> <span class="ow">in</span> <span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="n">type2_value</span><span class="p">):</span> +</span><span id="TypeAnnotator-417"><a href="#TypeAnnotator-417"><span class="linenos">417</span></a> <span class="k">return</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">UNKNOWN</span> +</span><span id="TypeAnnotator-418"><a href="#TypeAnnotator-418"><span class="linenos">418</span></a> +</span><span id="TypeAnnotator-419"><a href="#TypeAnnotator-419"><span class="linenos">419</span></a> <span class="k">if</span> <span class="n">type1_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-420"><a href="#TypeAnnotator-420"><span class="linenos">420</span></a> <span class="k">return</span> <span class="n">type1</span> +</span><span id="TypeAnnotator-421"><a href="#TypeAnnotator-421"><span class="linenos">421</span></a> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">NESTED_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-422"><a href="#TypeAnnotator-422"><span class="linenos">422</span></a> <span class="k">return</span> <span class="n">type2</span> +</span><span id="TypeAnnotator-423"><a href="#TypeAnnotator-423"><span class="linenos">423</span></a> +</span><span id="TypeAnnotator-424"><a href="#TypeAnnotator-424"><span class="linenos">424</span></a> <span class="k">return</span> <span class="n">type2_value</span> <span class="k">if</span> <span class="n">type2_value</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">type1_value</span><span class="p">,</span> <span class="p">{})</span> <span class="k">else</span> <span class="n">type1_value</span> <span class="c1"># type: ignore</span> +</span><span id="TypeAnnotator-425"><a href="#TypeAnnotator-425"><span class="linenos">425</span></a> +</span><span id="TypeAnnotator-426"><a href="#TypeAnnotator-426"><span class="linenos">426</span></a> <span class="c1"># Note: the following "no_type_check" decorators were added because mypy was yelling due</span> +</span><span id="TypeAnnotator-427"><a href="#TypeAnnotator-427"><span class="linenos">427</span></a> <span class="c1"># to assigning Type values to expression.type (since its getter returns Optional[DataType]).</span> +</span><span id="TypeAnnotator-428"><a href="#TypeAnnotator-428"><span class="linenos">428</span></a> <span class="c1"># This is a known mypy issue: https://github.com/python/mypy/issues/3004</span> +</span><span id="TypeAnnotator-429"><a href="#TypeAnnotator-429"><span class="linenos">429</span></a> +</span><span id="TypeAnnotator-430"><a href="#TypeAnnotator-430"><span class="linenos">430</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-431"><a href="#TypeAnnotator-431"><span class="linenos">431</span></a> <span class="k">def</span> <span class="nf">_annotate_binary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">B</span><span class="p">)</span> <span class="o">-></span> <span class="n">B</span><span class="p">:</span> +</span><span id="TypeAnnotator-432"><a href="#TypeAnnotator-432"><span class="linenos">432</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-433"><a href="#TypeAnnotator-433"><span class="linenos">433</span></a> +</span><span id="TypeAnnotator-434"><a href="#TypeAnnotator-434"><span class="linenos">434</span></a> <span class="n">left</span><span class="p">,</span> <span class="n">right</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="p">,</span> <span class="n">expression</span><span class="o">.</span><span class="n">right</span> +</span><span id="TypeAnnotator-435"><a href="#TypeAnnotator-435"><span class="linenos">435</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">left</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span><span class="p">,</span> <span class="n">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> +</span><span id="TypeAnnotator-436"><a href="#TypeAnnotator-436"><span class="linenos">436</span></a> +</span><span id="TypeAnnotator-437"><a href="#TypeAnnotator-437"><span class="linenos">437</span></a> <span class="k">if</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">Connector</span><span class="p">):</span> +</span><span id="TypeAnnotator-438"><a href="#TypeAnnotator-438"><span class="linenos">438</span></a> <span class="k">if</span> <span class="n">left_type</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">NULL</span> <span class="ow">and</span> <span class="n">right_type</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">NULL</span><span class="p">:</span> +</span><span id="TypeAnnotator-439"><a href="#TypeAnnotator-439"><span class="linenos">439</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">NULL</span><span class="p">)</span> +</span><span id="TypeAnnotator-440"><a href="#TypeAnnotator-440"><span class="linenos">440</span></a> <span class="k">elif</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">NULL</span> <span class="ow">in</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">):</span> +</span><span id="TypeAnnotator-441"><a href="#TypeAnnotator-441"><span class="linenos">441</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="TypeAnnotator-442"><a href="#TypeAnnotator-442"><span class="linenos">442</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="TypeAnnotator-443"><a href="#TypeAnnotator-443"><span class="linenos">443</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">build</span><span class="p">(</span><span class="s2">"NULLABLE"</span><span class="p">,</span> <span class="n">expressions</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">build</span><span class="p">(</span><span class="s2">"BOOLEAN"</span><span class="p">)),</span> +</span><span id="TypeAnnotator-444"><a href="#TypeAnnotator-444"><span class="linenos">444</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-445"><a href="#TypeAnnotator-445"><span class="linenos">445</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-446"><a href="#TypeAnnotator-446"><span class="linenos">446</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="TypeAnnotator-447"><a href="#TypeAnnotator-447"><span class="linenos">447</span></a> <span class="k">elif</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">Predicate</span><span class="p">):</span> +</span><span id="TypeAnnotator-448"><a href="#TypeAnnotator-448"><span class="linenos">448</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="TypeAnnotator-449"><a href="#TypeAnnotator-449"><span class="linenos">449</span></a> <span class="k">elif</span> <span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">:</span> +</span><span id="TypeAnnotator-450"><a href="#TypeAnnotator-450"><span class="linenos">450</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span><span class="p">[(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">)](</span><span class="n">left</span><span class="p">,</span> <span class="n">right</span><span class="p">))</span> +</span><span id="TypeAnnotator-451"><a href="#TypeAnnotator-451"><span class="linenos">451</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-452"><a href="#TypeAnnotator-452"><span class="linenos">452</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> +</span><span id="TypeAnnotator-453"><a href="#TypeAnnotator-453"><span class="linenos">453</span></a> +</span><span id="TypeAnnotator-454"><a href="#TypeAnnotator-454"><span class="linenos">454</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-455"><a href="#TypeAnnotator-455"><span class="linenos">455</span></a> +</span><span id="TypeAnnotator-456"><a href="#TypeAnnotator-456"><span class="linenos">456</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-457"><a href="#TypeAnnotator-457"><span class="linenos">457</span></a> <span class="k">def</span> <span class="nf">_annotate_unary</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-458"><a href="#TypeAnnotator-458"><span class="linenos">458</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-459"><a href="#TypeAnnotator-459"><span class="linenos">459</span></a> +</span><span id="TypeAnnotator-460"><a href="#TypeAnnotator-460"><span class="linenos">460</span></a> <span class="k">if</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">Condition</span><span class="p">)</span> <span class="ow">and</span> <span class="ow">not</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">Paren</span><span class="p">):</span> +</span><span id="TypeAnnotator-461"><a href="#TypeAnnotator-461"><span class="linenos">461</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BOOLEAN</span><span class="p">)</span> +</span><span id="TypeAnnotator-462"><a href="#TypeAnnotator-462"><span class="linenos">462</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-463"><a href="#TypeAnnotator-463"><span class="linenos">463</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</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">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-464"><a href="#TypeAnnotator-464"><span class="linenos">464</span></a> +</span><span id="TypeAnnotator-465"><a href="#TypeAnnotator-465"><span class="linenos">465</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-466"><a href="#TypeAnnotator-466"><span class="linenos">466</span></a> +</span><span id="TypeAnnotator-467"><a href="#TypeAnnotator-467"><span class="linenos">467</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-468"><a href="#TypeAnnotator-468"><span class="linenos">468</span></a> <span class="k">def</span> <span class="nf">_annotate_literal</span><span class="p">(</span><span class="bp">self</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">Literal</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Literal</span><span class="p">:</span> +</span><span id="TypeAnnotator-469"><a href="#TypeAnnotator-469"><span class="linenos">469</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_string</span><span class="p">:</span> +</span><span id="TypeAnnotator-470"><a href="#TypeAnnotator-470"><span class="linenos">470</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">VARCHAR</span><span class="p">)</span> +</span><span id="TypeAnnotator-471"><a href="#TypeAnnotator-471"><span class="linenos">471</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">is_int</span><span class="p">:</span> +</span><span id="TypeAnnotator-472"><a href="#TypeAnnotator-472"><span class="linenos">472</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">INT</span><span class="p">)</span> +</span><span id="TypeAnnotator-473"><a href="#TypeAnnotator-473"><span class="linenos">473</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-474"><a href="#TypeAnnotator-474"><span class="linenos">474</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> +</span><span id="TypeAnnotator-475"><a href="#TypeAnnotator-475"><span class="linenos">475</span></a> +</span><span id="TypeAnnotator-476"><a href="#TypeAnnotator-476"><span class="linenos">476</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-477"><a href="#TypeAnnotator-477"><span class="linenos">477</span></a> +</span><span id="TypeAnnotator-478"><a href="#TypeAnnotator-478"><span class="linenos">478</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-479"><a href="#TypeAnnotator-479"><span class="linenos">479</span></a> <span class="k">def</span> <span class="nf">_annotate_with_type</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> <span class="n">target_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="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-480"><a href="#TypeAnnotator-480"><span class="linenos">480</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">target_type</span><span class="p">)</span> +</span><span id="TypeAnnotator-481"><a href="#TypeAnnotator-481"><span class="linenos">481</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-482"><a href="#TypeAnnotator-482"><span class="linenos">482</span></a> +</span><span id="TypeAnnotator-483"><a href="#TypeAnnotator-483"><span class="linenos">483</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-484"><a href="#TypeAnnotator-484"><span class="linenos">484</span></a> <span class="k">def</span> <span class="nf">_annotate_struct_value</span><span class="p">(</span> +</span><span id="TypeAnnotator-485"><a href="#TypeAnnotator-485"><span class="linenos">485</span></a> <span class="bp">self</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><span id="TypeAnnotator-486"><a href="#TypeAnnotator-486"><span class="linenos">486</span></a> <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">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">]</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">:</span> +</span><span id="TypeAnnotator-487"><a href="#TypeAnnotator-487"><span class="linenos">487</span></a> <span class="n">alias</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">"alias"</span><span class="p">)</span> +</span><span id="TypeAnnotator-488"><a href="#TypeAnnotator-488"><span class="linenos">488</span></a> <span class="k">if</span> <span class="n">alias</span><span class="p">:</span> +</span><span id="TypeAnnotator-489"><a href="#TypeAnnotator-489"><span class="linenos">489</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</span><span class="p">(</span><span class="n">this</span><span class="o">=</span><span class="n">alias</span><span class="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-490"><a href="#TypeAnnotator-490"><span class="linenos">490</span></a> +</span><span id="TypeAnnotator-491"><a href="#TypeAnnotator-491"><span class="linenos">491</span></a> <span class="c1"># Case: key = value or key := value</span> +</span><span id="TypeAnnotator-492"><a href="#TypeAnnotator-492"><span class="linenos">492</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="p">:</span> +</span><span id="TypeAnnotator-493"><a href="#TypeAnnotator-493"><span class="linenos">493</span></a> <span class="k">return</span> <span class="n">exp</span><span class="o">.</span><span class="n">ColumnDef</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="o">.</span><span class="n">copy</span><span class="p">(),</span> <span class="n">kind</span><span class="o">=</span><span class="n">expression</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-494"><a href="#TypeAnnotator-494"><span class="linenos">494</span></a> +</span><span id="TypeAnnotator-495"><a href="#TypeAnnotator-495"><span class="linenos">495</span></a> <span class="k">return</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span> +</span><span id="TypeAnnotator-496"><a href="#TypeAnnotator-496"><span class="linenos">496</span></a> +</span><span id="TypeAnnotator-497"><a href="#TypeAnnotator-497"><span class="linenos">497</span></a> <span class="nd">@t</span><span class="o">.</span><span class="n">no_type_check</span> +</span><span id="TypeAnnotator-498"><a href="#TypeAnnotator-498"><span class="linenos">498</span></a> <span class="k">def</span> <span class="nf">_annotate_by_args</span><span class="p">(</span> +</span><span id="TypeAnnotator-499"><a href="#TypeAnnotator-499"><span class="linenos">499</span></a> <span class="bp">self</span><span class="p">,</span> +</span><span id="TypeAnnotator-500"><a href="#TypeAnnotator-500"><span class="linenos">500</span></a> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">,</span> +</span><span id="TypeAnnotator-501"><a href="#TypeAnnotator-501"><span class="linenos">501</span></a> <span class="o">*</span><span class="n">args</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> +</span><span id="TypeAnnotator-502"><a href="#TypeAnnotator-502"><span class="linenos">502</span></a> <span class="n">promote</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="TypeAnnotator-503"><a href="#TypeAnnotator-503"><span class="linenos">503</span></a> <span class="n">array</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="TypeAnnotator-504"><a href="#TypeAnnotator-504"><span class="linenos">504</span></a> <span class="n">struct</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span> +</span><span id="TypeAnnotator-505"><a href="#TypeAnnotator-505"><span class="linenos">505</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator-506"><a href="#TypeAnnotator-506"><span class="linenos">506</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-507"><a href="#TypeAnnotator-507"><span class="linenos">507</span></a> +</span><span id="TypeAnnotator-508"><a href="#TypeAnnotator-508"><span class="linenos">508</span></a> <span class="n">expressions</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="o">=</span> <span class="p">[]</span> +</span><span id="TypeAnnotator-509"><a href="#TypeAnnotator-509"><span class="linenos">509</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><span id="TypeAnnotator-510"><a href="#TypeAnnotator-510"><span class="linenos">510</span></a> <span class="n">arg_expr</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="n">arg</span><span class="p">)</span> +</span><span id="TypeAnnotator-511"><a href="#TypeAnnotator-511"><span class="linenos">511</span></a> <span class="n">expressions</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">expr</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">ensure_list</span><span class="p">(</span><span class="n">arg_expr</span><span class="p">)</span> <span class="k">if</span> <span class="n">expr</span><span class="p">)</span> +</span><span id="TypeAnnotator-512"><a href="#TypeAnnotator-512"><span class="linenos">512</span></a> +</span><span id="TypeAnnotator-513"><a href="#TypeAnnotator-513"><span class="linenos">513</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="kc">None</span> +</span><span id="TypeAnnotator-514"><a href="#TypeAnnotator-514"><span class="linenos">514</span></a> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">:</span> +</span><span id="TypeAnnotator-515"><a href="#TypeAnnotator-515"><span class="linenos">515</span></a> <span class="n">last_datatype</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">last_datatype</span> <span class="ow">or</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">,</span> <span class="n">expr</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-516"><a href="#TypeAnnotator-516"><span class="linenos">516</span></a> +</span><span id="TypeAnnotator-517"><a href="#TypeAnnotator-517"><span class="linenos">517</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">last_datatype</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">UNKNOWN</span><span class="p">)</span> +</span><span id="TypeAnnotator-518"><a href="#TypeAnnotator-518"><span class="linenos">518</span></a> +</span><span id="TypeAnnotator-519"><a href="#TypeAnnotator-519"><span class="linenos">519</span></a> <span class="k">if</span> <span class="n">promote</span><span class="p">:</span> +</span><span id="TypeAnnotator-520"><a href="#TypeAnnotator-520"><span class="linenos">520</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-521"><a href="#TypeAnnotator-521"><span class="linenos">521</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> +</span><span id="TypeAnnotator-522"><a href="#TypeAnnotator-522"><span class="linenos">522</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">FLOAT_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-523"><a href="#TypeAnnotator-523"><span class="linenos">523</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">DOUBLE</span><span class="p">)</span> +</span><span id="TypeAnnotator-524"><a href="#TypeAnnotator-524"><span class="linenos">524</span></a> +</span><span id="TypeAnnotator-525"><a href="#TypeAnnotator-525"><span class="linenos">525</span></a> <span class="k">if</span> <span class="n">array</span><span class="p">:</span> +</span><span id="TypeAnnotator-526"><a href="#TypeAnnotator-526"><span class="linenos">526</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="TypeAnnotator-527"><a href="#TypeAnnotator-527"><span class="linenos">527</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="TypeAnnotator-528"><a href="#TypeAnnotator-528"><span class="linenos">528</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> +</span><span id="TypeAnnotator-529"><a href="#TypeAnnotator-529"><span class="linenos">529</span></a> <span class="n">this</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">ARRAY</span><span class="p">,</span> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="n">expression</span><span class="o">.</span><span class="n">type</span><span class="p">],</span> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span> +</span><span id="TypeAnnotator-530"><a href="#TypeAnnotator-530"><span class="linenos">530</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-531"><a href="#TypeAnnotator-531"><span class="linenos">531</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-532"><a href="#TypeAnnotator-532"><span class="linenos">532</span></a> +</span><span id="TypeAnnotator-533"><a href="#TypeAnnotator-533"><span class="linenos">533</span></a> <span class="k">if</span> <span class="n">struct</span><span class="p">:</span> +</span><span id="TypeAnnotator-534"><a href="#TypeAnnotator-534"><span class="linenos">534</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span> +</span><span id="TypeAnnotator-535"><a href="#TypeAnnotator-535"><span class="linenos">535</span></a> <span class="n">expression</span><span class="p">,</span> +</span><span id="TypeAnnotator-536"><a href="#TypeAnnotator-536"><span class="linenos">536</span></a> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="p">(</span> +</span><span id="TypeAnnotator-537"><a href="#TypeAnnotator-537"><span class="linenos">537</span></a> <span class="n">this</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">STRUCT</span><span class="p">,</span> +</span><span id="TypeAnnotator-538"><a href="#TypeAnnotator-538"><span class="linenos">538</span></a> <span class="n">expressions</span><span class="o">=</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">_annotate_struct_value</span><span class="p">(</span><span class="n">expr</span><span class="p">)</span> <span class="k">for</span> <span class="n">expr</span> <span class="ow">in</span> <span class="n">expressions</span><span class="p">],</span> +</span><span id="TypeAnnotator-539"><a href="#TypeAnnotator-539"><span class="linenos">539</span></a> <span class="n">nested</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> +</span><span id="TypeAnnotator-540"><a href="#TypeAnnotator-540"><span class="linenos">540</span></a> <span class="p">),</span> +</span><span id="TypeAnnotator-541"><a href="#TypeAnnotator-541"><span class="linenos">541</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator-542"><a href="#TypeAnnotator-542"><span class="linenos">542</span></a> +</span><span id="TypeAnnotator-543"><a href="#TypeAnnotator-543"><span class="linenos">543</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-544"><a href="#TypeAnnotator-544"><span class="linenos">544</span></a> +</span><span id="TypeAnnotator-545"><a href="#TypeAnnotator-545"><span class="linenos">545</span></a> <span class="k">def</span> <span class="nf">_annotate_timeunit</span><span class="p">(</span> +</span><span id="TypeAnnotator-546"><a href="#TypeAnnotator-546"><span class="linenos">546</span></a> <span class="bp">self</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">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span> +</span><span id="TypeAnnotator-547"><a href="#TypeAnnotator-547"><span class="linenos">547</span></a> <span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">TimeUnit</span> <span class="o">|</span> <span class="n">exp</span><span class="o">.</span><span class="n">DateTrunc</span><span class="p">:</span> +</span><span id="TypeAnnotator-548"><a href="#TypeAnnotator-548"><span class="linenos">548</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-549"><a href="#TypeAnnotator-549"><span class="linenos">549</span></a> +</span><span id="TypeAnnotator-550"><a href="#TypeAnnotator-550"><span class="linenos">550</span></a> <span class="k">if</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEXT_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-551"><a href="#TypeAnnotator-551"><span class="linenos">551</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date_literal</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">unit</span><span class="p">)</span> +</span><span id="TypeAnnotator-552"><a href="#TypeAnnotator-552"><span class="linenos">552</span></a> <span class="k">elif</span> <span class="n">expression</span><span class="o">.</span><span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">TEMPORAL_TYPES</span><span class="p">:</span> +</span><span id="TypeAnnotator-553"><a href="#TypeAnnotator-553"><span class="linenos">553</span></a> <span class="n">datatype</span> <span class="o">=</span> <span class="n">_coerce_date</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">unit</span><span class="p">)</span> +</span><span id="TypeAnnotator-554"><a href="#TypeAnnotator-554"><span class="linenos">554</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-555"><a href="#TypeAnnotator-555"><span class="linenos">555</span></a> <span class="n">datatype</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">UNKNOWN</span> +</span><span id="TypeAnnotator-556"><a href="#TypeAnnotator-556"><span class="linenos">556</span></a> +</span><span id="TypeAnnotator-557"><a href="#TypeAnnotator-557"><span class="linenos">557</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">datatype</span><span class="p">)</span> +</span><span id="TypeAnnotator-558"><a href="#TypeAnnotator-558"><span class="linenos">558</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-559"><a href="#TypeAnnotator-559"><span class="linenos">559</span></a> +</span><span id="TypeAnnotator-560"><a href="#TypeAnnotator-560"><span class="linenos">560</span></a> <span class="k">def</span> <span class="nf">_annotate_bracket</span><span class="p">(</span><span class="bp">self</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">Bracket</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Bracket</span><span class="p">:</span> +</span><span id="TypeAnnotator-561"><a href="#TypeAnnotator-561"><span class="linenos">561</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-562"><a href="#TypeAnnotator-562"><span class="linenos">562</span></a> +</span><span id="TypeAnnotator-563"><a href="#TypeAnnotator-563"><span class="linenos">563</span></a> <span class="n">bracket_arg</span> <span class="o">=</span> <span class="n">expression</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="TypeAnnotator-564"><a href="#TypeAnnotator-564"><span class="linenos">564</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><span id="TypeAnnotator-565"><a href="#TypeAnnotator-565"><span class="linenos">565</span></a> +</span><span id="TypeAnnotator-566"><a href="#TypeAnnotator-566"><span class="linenos">566</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Slice</span><span class="p">):</span> +</span><span id="TypeAnnotator-567"><a href="#TypeAnnotator-567"><span class="linenos">567</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">this</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator-568"><a href="#TypeAnnotator-568"><span class="linenos">568</span></a> <span class="k">elif</span> <span class="n">this</span><span class="o">.</span><span class="n">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">ARRAY</span><span class="p">):</span> +</span><span id="TypeAnnotator-569"><a href="#TypeAnnotator-569"><span class="linenos">569</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">type</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="TypeAnnotator-570"><a href="#TypeAnnotator-570"><span class="linenos">570</span></a> <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">this</span><span class="p">,</span> <span class="p">(</span><span class="n">exp</span><span class="o">.</span><span class="n">Map</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">VarMap</span><span class="p">))</span> <span class="ow">and</span> <span class="n">bracket_arg</span> <span class="ow">in</span> <span class="n">this</span><span class="o">.</span><span class="n">keys</span><span class="p">:</span> </span><span id="TypeAnnotator-571"><a href="#TypeAnnotator-571"><span class="linenos">571</span></a> <span class="n">index</span> <span class="o">=</span> <span class="n">this</span><span class="o">.</span><span class="n">keys</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="n">bracket_arg</span><span class="p">)</span> </span><span id="TypeAnnotator-572"><a href="#TypeAnnotator-572"><span class="linenos">572</span></a> <span class="n">value</span> <span class="o">=</span> <span class="n">seq_get</span><span class="p">(</span><span class="n">this</span><span class="o">.</span><span class="n">values</span><span class="p">,</span> <span class="n">index</span><span class="p">)</span> -</span><span id="TypeAnnotator-573"><a href="#TypeAnnotator-573"><span class="linenos">573</span></a> <span class="n">value_type</span> <span class="o">=</span> <span class="n">value</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">value</span> <span class="k">else</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">UNKNOWN</span> -</span><span id="TypeAnnotator-574"><a href="#TypeAnnotator-574"><span class="linenos">574</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">value_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">UNKNOWN</span><span class="p">)</span> -</span><span id="TypeAnnotator-575"><a href="#TypeAnnotator-575"><span class="linenos">575</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-576"><a href="#TypeAnnotator-576"><span class="linenos">576</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> -</span><span id="TypeAnnotator-577"><a href="#TypeAnnotator-577"><span class="linenos">577</span></a> -</span><span id="TypeAnnotator-578"><a href="#TypeAnnotator-578"><span class="linenos">578</span></a> <span class="k">return</span> <span class="n">expression</span> -</span><span id="TypeAnnotator-579"><a href="#TypeAnnotator-579"><span class="linenos">579</span></a> -</span><span id="TypeAnnotator-580"><a href="#TypeAnnotator-580"><span class="linenos">580</span></a> <span class="k">def</span> <span class="nf">_annotate_div</span><span class="p">(</span><span class="bp">self</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">Div</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> -</span><span id="TypeAnnotator-581"><a href="#TypeAnnotator-581"><span class="linenos">581</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator-582"><a href="#TypeAnnotator-582"><span class="linenos">582</span></a> -</span><span id="TypeAnnotator-583"><a href="#TypeAnnotator-583"><span class="linenos">583</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="o">.</span><span class="n">type</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">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="c1"># type: ignore</span> -</span><span id="TypeAnnotator-584"><a href="#TypeAnnotator-584"><span class="linenos">584</span></a> -</span><span id="TypeAnnotator-585"><a href="#TypeAnnotator-585"><span class="linenos">585</span></a> <span class="k">if</span> <span class="p">(</span> -</span><span id="TypeAnnotator-586"><a href="#TypeAnnotator-586"><span class="linenos">586</span></a> <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">"typed"</span><span class="p">)</span> -</span><span id="TypeAnnotator-587"><a href="#TypeAnnotator-587"><span class="linenos">587</span></a> <span class="ow">and</span> <span class="n">left_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> -</span><span id="TypeAnnotator-588"><a href="#TypeAnnotator-588"><span class="linenos">588</span></a> <span class="ow">and</span> <span class="n">right_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> -</span><span id="TypeAnnotator-589"><a href="#TypeAnnotator-589"><span class="linenos">589</span></a> <span class="p">):</span> -</span><span id="TypeAnnotator-590"><a href="#TypeAnnotator-590"><span class="linenos">590</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> -</span><span id="TypeAnnotator-591"><a href="#TypeAnnotator-591"><span class="linenos">591</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator-592"><a href="#TypeAnnotator-592"><span class="linenos">592</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> -</span><span id="TypeAnnotator-593"><a href="#TypeAnnotator-593"><span class="linenos">593</span></a> -</span><span id="TypeAnnotator-594"><a href="#TypeAnnotator-594"><span class="linenos">594</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-573"><a href="#TypeAnnotator-573"><span class="linenos">573</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">value</span><span class="o">.</span><span class="n">type</span> <span class="k">if</span> <span class="n">value</span> <span class="k">else</span> <span class="kc">None</span><span class="p">)</span> +</span><span id="TypeAnnotator-574"><a href="#TypeAnnotator-574"><span class="linenos">574</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-575"><a href="#TypeAnnotator-575"><span class="linenos">575</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">UNKNOWN</span><span class="p">)</span> +</span><span id="TypeAnnotator-576"><a href="#TypeAnnotator-576"><span class="linenos">576</span></a> +</span><span id="TypeAnnotator-577"><a href="#TypeAnnotator-577"><span class="linenos">577</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-578"><a href="#TypeAnnotator-578"><span class="linenos">578</span></a> +</span><span id="TypeAnnotator-579"><a href="#TypeAnnotator-579"><span class="linenos">579</span></a> <span class="k">def</span> <span class="nf">_annotate_div</span><span class="p">(</span><span class="bp">self</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">Div</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Div</span><span class="p">:</span> +</span><span id="TypeAnnotator-580"><a href="#TypeAnnotator-580"><span class="linenos">580</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-581"><a href="#TypeAnnotator-581"><span class="linenos">581</span></a> +</span><span id="TypeAnnotator-582"><a href="#TypeAnnotator-582"><span class="linenos">582</span></a> <span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span> <span class="o">=</span> <span class="n">expression</span><span class="o">.</span><span class="n">left</span><span class="o">.</span><span class="n">type</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">right</span><span class="o">.</span><span class="n">type</span><span class="o">.</span><span class="n">this</span> <span class="c1"># type: ignore</span> +</span><span id="TypeAnnotator-583"><a href="#TypeAnnotator-583"><span class="linenos">583</span></a> +</span><span id="TypeAnnotator-584"><a href="#TypeAnnotator-584"><span class="linenos">584</span></a> <span class="k">if</span> <span class="p">(</span> +</span><span id="TypeAnnotator-585"><a href="#TypeAnnotator-585"><span class="linenos">585</span></a> <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">"typed"</span><span class="p">)</span> +</span><span id="TypeAnnotator-586"><a href="#TypeAnnotator-586"><span class="linenos">586</span></a> <span class="ow">and</span> <span class="n">left_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> +</span><span id="TypeAnnotator-587"><a href="#TypeAnnotator-587"><span class="linenos">587</span></a> <span class="ow">and</span> <span class="n">right_type</span> <span class="ow">in</span> <span class="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">INTEGER_TYPES</span> +</span><span id="TypeAnnotator-588"><a href="#TypeAnnotator-588"><span class="linenos">588</span></a> <span class="p">):</span> +</span><span id="TypeAnnotator-589"><a href="#TypeAnnotator-589"><span class="linenos">589</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</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">DataType</span><span class="o">.</span><span class="n">Type</span><span class="o">.</span><span class="n">BIGINT</span><span class="p">)</span> +</span><span id="TypeAnnotator-590"><a href="#TypeAnnotator-590"><span class="linenos">590</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator-591"><a href="#TypeAnnotator-591"><span class="linenos">591</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_coerce</span><span class="p">(</span><span class="n">left_type</span><span class="p">,</span> <span class="n">right_type</span><span class="p">))</span> +</span><span id="TypeAnnotator-592"><a href="#TypeAnnotator-592"><span class="linenos">592</span></a> +</span><span id="TypeAnnotator-593"><a href="#TypeAnnotator-593"><span class="linenos">593</span></a> <span class="k">return</span> <span class="n">expression</span> +</span><span id="TypeAnnotator-594"><a href="#TypeAnnotator-594"><span class="linenos">594</span></a> +</span><span id="TypeAnnotator-595"><a href="#TypeAnnotator-595"><span class="linenos">595</span></a> <span class="k">def</span> <span class="nf">_annotate_explode</span><span class="p">(</span><span class="bp">self</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">Explode</span><span class="p">)</span> <span class="o">-></span> <span class="n">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">:</span> +</span><span id="TypeAnnotator-596"><a href="#TypeAnnotator-596"><span class="linenos">596</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_annotate_args</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator-597"><a href="#TypeAnnotator-597"><span class="linenos">597</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">expression</span><span class="p">,</span> <span class="n">seq_get</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">type</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="TypeAnnotator-598"><a href="#TypeAnnotator-598"><span class="linenos">598</span></a> <span class="k">return</span> <span class="n">expression</span> </span></pre></div> @@ -1303,20 +1311,20 @@ </div> <a class="headerlink" href="#TypeAnnotator.__init__"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="TypeAnnotator.__init__-321"><a href="#TypeAnnotator.__init__-321"><span class="linenos">321</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> -</span><span id="TypeAnnotator.__init__-322"><a href="#TypeAnnotator.__init__-322"><span class="linenos">322</span></a> <span class="bp">self</span><span class="p">,</span> -</span><span id="TypeAnnotator.__init__-323"><a href="#TypeAnnotator.__init__-323"><span class="linenos">323</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> -</span><span id="TypeAnnotator.__init__-324"><a href="#TypeAnnotator.__init__-324"><span class="linenos">324</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator.__init__-325"><a href="#TypeAnnotator.__init__-325"><span class="linenos">325</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator.__init__-326"><a href="#TypeAnnotator.__init__-326"><span class="linenos">326</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> -</span><span id="TypeAnnotator.__init__-327"><a href="#TypeAnnotator.__init__-327"><span class="linenos">327</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> -</span><span id="TypeAnnotator.__init__-328"><a href="#TypeAnnotator.__init__-328"><span class="linenos">328</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> -</span><span id="TypeAnnotator.__init__-329"><a href="#TypeAnnotator.__init__-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> -</span><span id="TypeAnnotator.__init__-330"><a href="#TypeAnnotator.__init__-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> -</span><span id="TypeAnnotator.__init__-331"><a href="#TypeAnnotator.__init__-331"><span class="linenos">331</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> -</span><span id="TypeAnnotator.__init__-332"><a href="#TypeAnnotator.__init__-332"><span class="linenos">332</span></a> -</span><span id="TypeAnnotator.__init__-333"><a href="#TypeAnnotator.__init__-333"><span class="linenos">333</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> -</span><span id="TypeAnnotator.__init__-334"><a href="#TypeAnnotator.__init__-334"><span class="linenos">334</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="TypeAnnotator.__init__-322"><a href="#TypeAnnotator.__init__-322"><span class="linenos">322</span></a> <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span> +</span><span id="TypeAnnotator.__init__-323"><a href="#TypeAnnotator.__init__-323"><span class="linenos">323</span></a> <span class="bp">self</span><span class="p">,</span> +</span><span id="TypeAnnotator.__init__-324"><a href="#TypeAnnotator.__init__-324"><span class="linenos">324</span></a> <span class="n">schema</span><span class="p">:</span> <span class="n">Schema</span><span class="p">,</span> +</span><span id="TypeAnnotator.__init__-325"><a href="#TypeAnnotator.__init__-325"><span class="linenos">325</span></a> <span class="n">annotators</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="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">t</span><span class="o">.</span><span class="n">Callable</span><span class="p">[[</span><span class="n">TypeAnnotator</span><span class="p">,</span> <span class="n">E</span><span class="p">],</span> <span class="n">E</span><span class="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator.__init__-326"><a href="#TypeAnnotator.__init__-326"><span class="linenos">326</span></a> <span class="n">coerces_to</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="n">exp</span><span class="o">.</span><span class="n">DataType</span><span class="o">.</span><span class="n">Type</span><span class="p">,</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</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="p">]]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator.__init__-327"><a href="#TypeAnnotator.__init__-327"><span class="linenos">327</span></a> <span class="n">binary_coercions</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">BinaryCoercions</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> +</span><span id="TypeAnnotator.__init__-328"><a href="#TypeAnnotator.__init__-328"><span class="linenos">328</span></a> <span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> +</span><span id="TypeAnnotator.__init__-329"><a href="#TypeAnnotator.__init__-329"><span class="linenos">329</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span> <span class="o">=</span> <span class="n">schema</span> +</span><span id="TypeAnnotator.__init__-330"><a href="#TypeAnnotator.__init__-330"><span class="linenos">330</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">annotators</span> <span class="o">=</span> <span class="n">annotators</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">ANNOTATORS</span> +</span><span id="TypeAnnotator.__init__-331"><a href="#TypeAnnotator.__init__-331"><span class="linenos">331</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">coerces_to</span> <span class="o">=</span> <span class="n">coerces_to</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">COERCES_TO</span> +</span><span id="TypeAnnotator.__init__-332"><a href="#TypeAnnotator.__init__-332"><span class="linenos">332</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">binary_coercions</span> <span class="o">=</span> <span class="n">binary_coercions</span> <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">BINARY_COERCIONS</span> +</span><span id="TypeAnnotator.__init__-333"><a href="#TypeAnnotator.__init__-333"><span class="linenos">333</span></a> +</span><span id="TypeAnnotator.__init__-334"><a href="#TypeAnnotator.__init__-334"><span class="linenos">334</span></a> <span class="c1"># Caches the ids of annotated sub-Expressions, to ensure we only visit them once</span> +</span><span id="TypeAnnotator.__init__-335"><a href="#TypeAnnotator.__init__-335"><span class="linenos">335</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_visited</span><span class="p">:</span> <span class="n">t</span><span class="o">.</span><span class="n">Set</span><span class="p">[</span><span class="nb">int</span><span class="p">]</span> <span class="o">=</span> <span class="nb">set</span><span class="p">()</span> </span></pre></div> @@ -1327,7 +1335,7 @@ <div class="attr variable"> <span class="name">TYPE_TO_EXPRESSIONS</span><span class="annotation">: Dict[<a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>, Set[Type[<a href="../expressions.html#Expression">sqlglot.expressions.Expression</a>]]]</span> = <input id="TypeAnnotator.TYPE_TO_EXPRESSIONS-view-value" class="view-value-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> - <label class="view-value-button pdoc-button" for="TypeAnnotator.TYPE_TO_EXPRESSIONS-view-value"></label><span class="default_value">{<Type.BIGINT: 'BIGINT'>: {<class '<a href="../expressions.html#Length">sqlglot.expressions.Length</a>'>, <class '<a href="../expressions.html#ArraySize">sqlglot.expressions.ArraySize</a>'>, <class '<a href="../expressions.html#ApproxDistinct">sqlglot.expressions.ApproxDistinct</a>'>, <class '<a href="../expressions.html#Count">sqlglot.expressions.Count</a>'>}, <Type.BOOLEAN: 'BOOLEAN'>: {<class '<a href="../expressions.html#In">sqlglot.expressions.In</a>'>, <class '<a href="../expressions.html#Boolean">sqlglot.expressions.Boolean</a>'>, <class '<a href="../expressions.html#Between">sqlglot.expressions.Between</a>'>, <class '<a href="../expressions.html#RegexpLike">sqlglot.expressions.RegexpLike</a>'>}, <Type.DATE: 'DATE'>: {<class '<a href="../expressions.html#DiToDate">sqlglot.expressions.DiToDate</a>'>, <class '<a href="../expressions.html#DateFromParts">sqlglot.expressions.DateFromParts</a>'>, <class '<a href="../expressions.html#CurrentDate">sqlglot.expressions.CurrentDate</a>'>, <class '<a href="../expressions.html#Date">sqlglot.expressions.Date</a>'>, <class '<a href="../expressions.html#StrToDate">sqlglot.expressions.StrToDate</a>'>, <class '<a href="../expressions.html#TimeStrToDate">sqlglot.expressions.TimeStrToDate</a>'>, <class '<a href="../expressions.html#TsOrDsToDate">sqlglot.expressions.TsOrDsToDate</a>'>, <class '<a href="../expressions.html#DateStrToDate">sqlglot.expressions.DateStrToDate</a>'>}, <Type.DATETIME: 'DATETIME'>: {<class '<a href="../expressions.html#DatetimeAdd">sqlglot.expressions.DatetimeAdd</a>'>, <class '<a href="../expressions.html#DatetimeSub">sqlglot.expressions.DatetimeSub</a>'>, <class '<a href="../expressions.html#CurrentDatetime">sqlglot.expressions.CurrentDatetime</a>'>}, <Type.DOUBLE: 'DOUBLE'>: {<class '<a href="../expressions.html#Log2">sqlglot.expressions.Log2</a>'>, <class '<a href="../expressions.html#Log">sqlglot.expressions.Log</a>'>, <class '<a href="../expressions.html#StddevSamp">sqlglot.expressions.StddevSamp</a>'>, <class '<a href="../expressions.html#Ln">sqlglot.expressions.Ln</a>'>, <class '<a href="../expressions.html#StddevPop">sqlglot.expressions.StddevPop</a>'>, <class '<a href="../expressions.html#Avg">sqlglot.expressions.Avg</a>'>, <class '<a href="../expressions.html#Div">sqlglot.expressions.Div</a>'>, <class '<a href="../expressions.html#ApproxQuantile">sqlglot.expressions.ApproxQuantile</a>'>, <class '<a href="../expressions.html#Stddev">sqlglot.expressions.Stddev</a>'>, <class '<a href="../expressions.html#Round">sqlglot.expressions.Round</a>'>, <class '<a href="../expressions.html#Pow">sqlglot.expressions.Pow</a>'>, <class '<a href="../expressions.html#Quantile">sqlglot.expressions.Quantile</a>'>, <class '<a href="../expressions.html#Sqrt">sqlglot.expressions.Sqrt</a>'>, <class '<a href="../expressions.html#SafeDivide">sqlglot.expressions.SafeDivide</a>'>, <class '<a href="../expressions.html#VariancePop">sqlglot.expressions.VariancePop</a>'>, <class '<a href="../expressions.html#Exp">sqlglot.expressions.Exp</a>'>, <class '<a href="../expressions.html#Variance">sqlglot.expressions.Variance</a>'>, <class '<a href="../expressions.html#Log10">sqlglot.expressions.Log10</a>'>}, <Type.INT: 'INT'>: {<class '<a href="../expressions.html#Floor">sqlglot.expressions.Floor</a>'>, <class '<a href="../expressions.html#Extract">sqlglot.expressions.Extract</a>'>, <class '<a href="../expressions.html#TimeDiff">sqlglot.expressions.TimeDiff</a>'>, <class '<a href="../expressions.html#Levenshtein">sqlglot.expressions.Levenshtein</a>'>, <class '<a href="../expressions.html#DateDiff">sqlglot.expressions.DateDiff</a>'>, <class '<a href="../expressions.html#TsOrDiToDi">sqlglot.expressions.TsOrDiToDi</a>'>, <class '<a href="../expressions.html#DateToDi">sqlglot.expressions.DateToDi</a>'>, <class '<a href="../expressions.html#StrPosition">sqlglot.expressions.StrPosition</a>'>, <class '<a href="../expressions.html#TimestampDiff">sqlglot.expressions.TimestampDiff</a>'>, <class '<a href="../expressions.html#Ceil">sqlglot.expressions.Ceil</a>'>, <class '<a href="../expressions.html#DatetimeDiff">sqlglot.expressions.DatetimeDiff</a>'>}, <Type.JSON: 'JSON'>: {<class '<a href="../expressions.html#ParseJSON">sqlglot.expressions.ParseJSON</a>'>}, <Type.TIMESTAMP: 'TIMESTAMP'>: {<class '<a href="../expressions.html#TimestampAdd">sqlglot.expressions.TimestampAdd</a>'>, <class '<a href="../expressions.html#CurrentTime">sqlglot.expressions.CurrentTime</a>'>, <class '<a href="../expressions.html#Timestamp">sqlglot.expressions.Timestamp</a>'>, <class '<a href="../expressions.html#UnixToTime">sqlglot.expressions.UnixToTime</a>'>, <class '<a href="../expressions.html#StrToTime">sqlglot.expressions.StrToTime</a>'>, <class '<a href="../expressions.html#TimeSub">sqlglot.expressions.TimeSub</a>'>, <class '<a href="../expressions.html#TimeStrToTime">sqlglot.expressions.TimeStrToTime</a>'>, <class '<a href="../expressions.html#TimeAdd">sqlglot.expressions.TimeAdd</a>'>, <class '<a href="../expressions.html#TimestampSub">sqlglot.expressions.TimestampSub</a>'>, <class '<a href="../expressions.html#CurrentTimestamp">sqlglot.expressions.CurrentTimestamp</a>'>}, <Type.TINYINT: 'TINYINT'>: {<class '<a href="../expressions.html#Month">sqlglot.expressions.Month</a>'>, <class '<a href="../expressions.html#Day">sqlglot.expressions.Day</a>'>, <class '<a href="../expressions.html#Week">sqlglot.expressions.Week</a>'>, <class '<a href="../expressions.html#Year">sqlglot.expressions.Year</a>'>}, <Type.VARCHAR: 'VARCHAR'>: {<class '<a href="../expressions.html#Substring">sqlglot.expressions.Substring</a>'>, <class '<a href="../expressions.html#TimeToStr">sqlglot.expressions.TimeToStr</a>'>, <class '<a href="../expressions.html#UnixToStr">sqlglot.expressions.UnixToStr</a>'>, <class '<a href="../expressions.html#UnixToTimeStr">sqlglot.expressions.UnixToTimeStr</a>'>, <class '<a href="../expressions.html#Initcap">sqlglot.expressions.Initcap</a>'>, <class '<a href="../expressions.html#Trim">sqlglot.expressions.Trim</a>'>, <class '<a href="../expressions.html#ConcatWs">sqlglot.expressions.ConcatWs</a>'>, <class '<a href="../expressions.html#Lower">sqlglot.expressions.Lower</a>'>, <class '<a href="../expressions.html#Concat">sqlglot.expressions.Concat</a>'>, <class '<a href="../expressions.html#ArrayConcat">sqlglot.expressions.ArrayConcat</a>'>, <class '<a href="../expressions.html#GroupConcat">sqlglot.expressions.GroupConcat</a>'>, <class '<a href="../expressions.html#DateToDateStr">sqlglot.expressions.DateToDateStr</a>'>, <class '<a href="../expressions.html#TsOrDsToDateStr">sqlglot.expressions.TsOrDsToDateStr</a>'>, <class '<a href="../expressions.html#TimeToTimeStr">sqlglot.expressions.TimeToTimeStr</a>'>, <class '<a href="../expressions.html#Upper">sqlglot.expressions.Upper</a>'>}}</span> + <label class="view-value-button pdoc-button" for="TypeAnnotator.TYPE_TO_EXPRESSIONS-view-value"></label><span class="default_value">{<Type.BIGINT: 'BIGINT'>: {<class '<a href="../expressions.html#Length">sqlglot.expressions.Length</a>'>, <class '<a href="../expressions.html#ApproxDistinct">sqlglot.expressions.ApproxDistinct</a>'>, <class '<a href="../expressions.html#Count">sqlglot.expressions.Count</a>'>, <class '<a href="../expressions.html#ArraySize">sqlglot.expressions.ArraySize</a>'>}, <Type.BOOLEAN: 'BOOLEAN'>: {<class '<a href="../expressions.html#In">sqlglot.expressions.In</a>'>, <class '<a href="../expressions.html#RegexpLike">sqlglot.expressions.RegexpLike</a>'>, <class '<a href="../expressions.html#Between">sqlglot.expressions.Between</a>'>, <class '<a href="../expressions.html#Boolean">sqlglot.expressions.Boolean</a>'>}, <Type.DATE: 'DATE'>: {<class '<a href="../expressions.html#TsOrDsToDate">sqlglot.expressions.TsOrDsToDate</a>'>, <class '<a href="../expressions.html#DiToDate">sqlglot.expressions.DiToDate</a>'>, <class '<a href="../expressions.html#DateFromParts">sqlglot.expressions.DateFromParts</a>'>, <class '<a href="../expressions.html#Date">sqlglot.expressions.Date</a>'>, <class '<a href="../expressions.html#TimeStrToDate">sqlglot.expressions.TimeStrToDate</a>'>, <class '<a href="../expressions.html#StrToDate">sqlglot.expressions.StrToDate</a>'>, <class '<a href="../expressions.html#CurrentDate">sqlglot.expressions.CurrentDate</a>'>, <class '<a href="../expressions.html#DateStrToDate">sqlglot.expressions.DateStrToDate</a>'>}, <Type.DATETIME: 'DATETIME'>: {<class '<a href="../expressions.html#DatetimeAdd">sqlglot.expressions.DatetimeAdd</a>'>, <class '<a href="../expressions.html#DatetimeSub">sqlglot.expressions.DatetimeSub</a>'>, <class '<a href="../expressions.html#CurrentDatetime">sqlglot.expressions.CurrentDatetime</a>'>}, <Type.DOUBLE: 'DOUBLE'>: {<class '<a href="../expressions.html#Log2">sqlglot.expressions.Log2</a>'>, <class '<a href="../expressions.html#Avg">sqlglot.expressions.Avg</a>'>, <class '<a href="../expressions.html#Log">sqlglot.expressions.Log</a>'>, <class '<a href="../expressions.html#Pow">sqlglot.expressions.Pow</a>'>, <class '<a href="../expressions.html#Round">sqlglot.expressions.Round</a>'>, <class '<a href="../expressions.html#StddevSamp">sqlglot.expressions.StddevSamp</a>'>, <class '<a href="../expressions.html#Ln">sqlglot.expressions.Ln</a>'>, <class '<a href="../expressions.html#StddevPop">sqlglot.expressions.StddevPop</a>'>, <class '<a href="../expressions.html#Stddev">sqlglot.expressions.Stddev</a>'>, <class '<a href="../expressions.html#Sqrt">sqlglot.expressions.Sqrt</a>'>, <class '<a href="../expressions.html#ApproxQuantile">sqlglot.expressions.ApproxQuantile</a>'>, <class '<a href="../expressions.html#SafeDivide">sqlglot.expressions.SafeDivide</a>'>, <class '<a href="../expressions.html#VariancePop">sqlglot.expressions.VariancePop</a>'>, <class '<a href="../expressions.html#Quantile">sqlglot.expressions.Quantile</a>'>, <class '<a href="../expressions.html#Variance">sqlglot.expressions.Variance</a>'>, <class '<a href="../expressions.html#Log10">sqlglot.expressions.Log10</a>'>, <class '<a href="../expressions.html#Exp">sqlglot.expressions.Exp</a>'>, <class '<a href="../expressions.html#Div">sqlglot.expressions.Div</a>'>}, <Type.INT: 'INT'>: {<class '<a href="../expressions.html#Floor">sqlglot.expressions.Floor</a>'>, <class '<a href="../expressions.html#Levenshtein">sqlglot.expressions.Levenshtein</a>'>, <class '<a href="../expressions.html#Extract">sqlglot.expressions.Extract</a>'>, <class '<a href="../expressions.html#DatetimeDiff">sqlglot.expressions.DatetimeDiff</a>'>, <class '<a href="../expressions.html#TimeDiff">sqlglot.expressions.TimeDiff</a>'>, <class '<a href="../expressions.html#TsOrDiToDi">sqlglot.expressions.TsOrDiToDi</a>'>, <class '<a href="../expressions.html#DateToDi">sqlglot.expressions.DateToDi</a>'>, <class '<a href="../expressions.html#Ceil">sqlglot.expressions.Ceil</a>'>, <class '<a href="../expressions.html#StrPosition">sqlglot.expressions.StrPosition</a>'>, <class '<a href="../expressions.html#TimestampDiff">sqlglot.expressions.TimestampDiff</a>'>, <class '<a href="../expressions.html#DateDiff">sqlglot.expressions.DateDiff</a>'>}, <Type.JSON: 'JSON'>: {<class '<a href="../expressions.html#ParseJSON">sqlglot.expressions.ParseJSON</a>'>}, <Type.TIMESTAMP: 'TIMESTAMP'>: {<class '<a href="../expressions.html#TimestampSub">sqlglot.expressions.TimestampSub</a>'>, <class '<a href="../expressions.html#TimestampAdd">sqlglot.expressions.TimestampAdd</a>'>, <class '<a href="../expressions.html#UnixToTime">sqlglot.expressions.UnixToTime</a>'>, <class '<a href="../expressions.html#Timestamp">sqlglot.expressions.Timestamp</a>'>, <class '<a href="../expressions.html#CurrentTimestamp">sqlglot.expressions.CurrentTimestamp</a>'>, <class '<a href="../expressions.html#TimeStrToTime">sqlglot.expressions.TimeStrToTime</a>'>, <class '<a href="../expressions.html#TimeSub">sqlglot.expressions.TimeSub</a>'>, <class '<a href="../expressions.html#CurrentTime">sqlglot.expressions.CurrentTime</a>'>, <class '<a href="../expressions.html#StrToTime">sqlglot.expressions.StrToTime</a>'>, <class '<a href="../expressions.html#TimeAdd">sqlglot.expressions.TimeAdd</a>'>}, <Type.TINYINT: 'TINYINT'>: {<class '<a href="../expressions.html#Month">sqlglot.expressions.Month</a>'>, <class '<a href="../expressions.html#Week">sqlglot.expressions.Week</a>'>, <class '<a href="../expressions.html#Day">sqlglot.expressions.Day</a>'>, <class '<a href="../expressions.html#Year">sqlglot.expressions.Year</a>'>}, <Type.VARCHAR: 'VARCHAR'>: {<class '<a href="../expressions.html#TimeToStr">sqlglot.expressions.TimeToStr</a>'>, <class '<a href="../expressions.html#UnixToStr">sqlglot.expressions.UnixToStr</a>'>, <class '<a href="../expressions.html#ConcatWs">sqlglot.expressions.ConcatWs</a>'>, <class '<a href="../expressions.html#UnixToTimeStr">sqlglot.expressions.UnixToTimeStr</a>'>, <class '<a href="../expressions.html#Substring">sqlglot.expressions.Substring</a>'>, <class '<a href="../expressions.html#Trim">sqlglot.expressions.Trim</a>'>, <class '<a href="../expressions.html#Concat">sqlglot.expressions.Concat</a>'>, <class '<a href="../expressions.html#ArrayConcat">sqlglot.expressions.ArrayConcat</a>'>, <class '<a href="../expressions.html#GroupConcat">sqlglot.expressions.GroupConcat</a>'>, <class '<a href="../expressions.html#Initcap">sqlglot.expressions.Initcap</a>'>, <class '<a href="../expressions.html#Lower">sqlglot.expressions.Lower</a>'>, <class '<a href="../expressions.html#TsOrDsToDateStr">sqlglot.expressions.TsOrDsToDateStr</a>'>, <class '<a href="../expressions.html#DateToDateStr">sqlglot.expressions.DateToDateStr</a>'>, <class '<a href="../expressions.html#TimeToTimeStr">sqlglot.expressions.TimeToTimeStr</a>'>, <class '<a href="../expressions.html#Upper">sqlglot.expressions.Upper</a>'>}}</span> </div> @@ -1340,7 +1348,7 @@ <div class="attr variable"> <span class="name">ANNOTATORS</span><span class="annotation">: Dict</span> = <input id="TypeAnnotator.ANNOTATORS-view-value" class="view-value-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> - <label class="view-value-button pdoc-button" for="TypeAnnotator.ANNOTATORS-view-value"></label><span class="default_value">{<class '<a href="../expressions.html#Alias">sqlglot.expressions.Alias</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseNot">sqlglot.expressions.BitwiseNot</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Neg">sqlglot.expressions.Neg</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Not">sqlglot.expressions.Not</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Paren">sqlglot.expressions.Paren</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#PivotAlias">sqlglot.expressions.PivotAlias</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Unary">sqlglot.expressions.Unary</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Add">sqlglot.expressions.Add</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#And">sqlglot.expressions.And</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayContained">sqlglot.expressions.ArrayContained</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayContains">sqlglot.expressions.ArrayContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayOverlaps">sqlglot.expressions.ArrayOverlaps</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Binary">sqlglot.expressions.Binary</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseAnd">sqlglot.expressions.BitwiseAnd</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseLeftShift">sqlglot.expressions.BitwiseLeftShift</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseOr">sqlglot.expressions.BitwiseOr</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseRightShift">sqlglot.expressions.BitwiseRightShift</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseXor">sqlglot.expressions.BitwiseXor</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Collate">sqlglot.expressions.Collate</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Connector">sqlglot.expressions.Connector</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#DPipe">sqlglot.expressions.DPipe</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Distance">sqlglot.expressions.Distance</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Div">sqlglot.expressions.Div</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Dot">sqlglot.expressions.Dot</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#EQ">sqlglot.expressions.EQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Escape">sqlglot.expressions.Escape</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#GT">sqlglot.expressions.GT</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#GTE">sqlglot.expressions.GTE</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Glob">sqlglot.expressions.Glob</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ILike">sqlglot.expressions.ILike</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ILikeAny">sqlglot.expressions.ILikeAny</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#IntDiv">sqlglot.expressions.IntDiv</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Is">sqlglot.expressions.Is</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONArrayContains">sqlglot.expressions.JSONArrayContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBContains">sqlglot.expressions.JSONBContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBExtract">sqlglot.expressions.JSONBExtract</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBExtractScalar">sqlglot.expressions.JSONBExtractScalar</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONExtract">sqlglot.expressions.JSONExtract</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONExtractScalar">sqlglot.expressions.JSONExtractScalar</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Kwarg">sqlglot.expressions.Kwarg</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LT">sqlglot.expressions.LT</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LTE">sqlglot.expressions.LTE</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Like">sqlglot.expressions.Like</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LikeAny">sqlglot.expressions.LikeAny</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Mod">sqlglot.expressions.Mod</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Mul">sqlglot.expressions.Mul</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NEQ">sqlglot.expressions.NEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NullSafeEQ">sqlglot.expressions.NullSafeEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NullSafeNEQ">sqlglot.expressions.NullSafeNEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Operator">sqlglot.expressions.Operator</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Or">sqlglot.expressions.Or</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Overlaps">sqlglot.expressions.Overlaps</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Pow">sqlglot.expressions.Pow</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#PropertyEQ">sqlglot.expressions.PropertyEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#RegexpILike">sqlglot.expressions.RegexpILike</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#RegexpLike">sqlglot.expressions.RegexpLike</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#SimilarTo">sqlglot.expressions.SimilarTo</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Slice">sqlglot.expressions.Slice</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Sub">sqlglot.expressions.Sub</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Xor">sqlglot.expressions.Xor</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Length">sqlglot.expressions.Length</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ArraySize">sqlglot.expressions.ArraySize</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ApproxDistinct">sqlglot.expressions.ApproxDistinct</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Count">sqlglot.expressions.Count</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#In">sqlglot.expressions.In</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Boolean">sqlglot.expressions.Boolean</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Between">sqlglot.expressions.Between</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DiToDate">sqlglot.expressions.DiToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateFromParts">sqlglot.expressions.DateFromParts</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentDate">sqlglot.expressions.CurrentDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Date">sqlglot.expressions.Date</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrToDate">sqlglot.expressions.StrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeStrToDate">sqlglot.expressions.TimeStrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDsToDate">sqlglot.expressions.TsOrDsToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateStrToDate">sqlglot.expressions.DateStrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeAdd">sqlglot.expressions.DatetimeAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeSub">sqlglot.expressions.DatetimeSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentDatetime">sqlglot.expressions.CurrentDatetime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log2">sqlglot.expressions.Log2</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log">sqlglot.expressions.Log</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StddevSamp">sqlglot.expressions.StddevSamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Ln">sqlglot.expressions.Ln</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StddevPop">sqlglot.expressions.StddevPop</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Avg">sqlglot.expressions.Avg</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ApproxQuantile">sqlglot.expressions.ApproxQuantile</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Stddev">sqlglot.expressions.Stddev</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Round">sqlglot.expressions.Round</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Quantile">sqlglot.expressions.Quantile</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Sqrt">sqlglot.expressions.Sqrt</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#SafeDivide">sqlglot.expressions.SafeDivide</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#VariancePop">sqlglot.expressions.VariancePop</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Exp">sqlglot.expressions.Exp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Variance">sqlglot.expressions.Variance</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log10">sqlglot.expressions.Log10</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Floor">sqlglot.expressions.Floor</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Extract">sqlglot.expressions.Extract</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeDiff">sqlglot.expressions.TimeDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Levenshtein">sqlglot.expressions.Levenshtein</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateDiff">sqlglot.expressions.DateDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDiToDi">sqlglot.expressions.TsOrDiToDi</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateToDi">sqlglot.expressions.DateToDi</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrPosition">sqlglot.expressions.StrPosition</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampDiff">sqlglot.expressions.TimestampDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Ceil">sqlglot.expressions.Ceil</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeDiff">sqlglot.expressions.DatetimeDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ParseJSON">sqlglot.expressions.ParseJSON</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampAdd">sqlglot.expressions.TimestampAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentTime">sqlglot.expressions.CurrentTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Timestamp">sqlglot.expressions.Timestamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToTime">sqlglot.expressions.UnixToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrToTime">sqlglot.expressions.StrToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeSub">sqlglot.expressions.TimeSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeStrToTime">sqlglot.expressions.TimeStrToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeAdd">sqlglot.expressions.TimeAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampSub">sqlglot.expressions.TimestampSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentTimestamp">sqlglot.expressions.CurrentTimestamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Month">sqlglot.expressions.Month</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Day">sqlglot.expressions.Day</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Week">sqlglot.expressions.Week</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Year">sqlglot.expressions.Year</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Substring">sqlglot.expressions.Substring</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeToStr">sqlglot.expressions.TimeToStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToStr">sqlglot.expressions.UnixToStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToTimeStr">sqlglot.expressions.UnixToTimeStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Initcap">sqlglot.expressions.Initcap</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Trim">sqlglot.expressions.Trim</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ConcatWs">sqlglot.expressions.ConcatWs</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Lower">sqlglot.expressions.Lower</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Concat">sqlglot.expressions.Concat</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ArrayConcat">sqlglot.expressions.ArrayConcat</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#GroupConcat">sqlglot.expressions.GroupConcat</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateToDateStr">sqlglot.expressions.DateToDateStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDsToDateStr">sqlglot.expressions.TsOrDsToDateStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeToTimeStr">sqlglot.expressions.TimeToTimeStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Upper">sqlglot.expressions.Upper</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Abs">sqlglot.expressions.Abs</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Anonymous">sqlglot.expressions.Anonymous</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Array">sqlglot.expressions.Array</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#ArrayAgg">sqlglot.expressions.ArrayAgg</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Bracket">sqlglot.expressions.Bracket</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Cast">sqlglot.expressions.Cast</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Case">sqlglot.expressions.Case</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Coalesce">sqlglot.expressions.Coalesce</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DataType">sqlglot.expressions.DataType</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateAdd">sqlglot.expressions.DateAdd</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateSub">sqlglot.expressions.DateSub</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateTrunc">sqlglot.expressions.DateTrunc</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Distinct">sqlglot.expressions.Distinct</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Filter">sqlglot.expressions.Filter</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#If">sqlglot.expressions.If</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Interval">sqlglot.expressions.Interval</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Least">sqlglot.expressions.Least</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Literal">sqlglot.expressions.Literal</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Map">sqlglot.expressions.Map</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Max">sqlglot.expressions.Max</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Min">sqlglot.expressions.Min</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Null">sqlglot.expressions.Null</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Nullif">sqlglot.expressions.Nullif</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Sum">sqlglot.expressions.Sum</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#TryCast">sqlglot.expressions.TryCast</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#VarMap">sqlglot.expressions.VarMap</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Struct">sqlglot.expressions.Struct</a>'>: <function TypeAnnotator.<lambda>>}</span> + <label class="view-value-button pdoc-button" for="TypeAnnotator.ANNOTATORS-view-value"></label><span class="default_value">{<class '<a href="../expressions.html#Alias">sqlglot.expressions.Alias</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseNot">sqlglot.expressions.BitwiseNot</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Neg">sqlglot.expressions.Neg</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Not">sqlglot.expressions.Not</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Paren">sqlglot.expressions.Paren</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#PivotAlias">sqlglot.expressions.PivotAlias</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Unary">sqlglot.expressions.Unary</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Add">sqlglot.expressions.Add</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#And">sqlglot.expressions.And</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayContained">sqlglot.expressions.ArrayContained</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayContains">sqlglot.expressions.ArrayContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ArrayOverlaps">sqlglot.expressions.ArrayOverlaps</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Binary">sqlglot.expressions.Binary</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseAnd">sqlglot.expressions.BitwiseAnd</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseLeftShift">sqlglot.expressions.BitwiseLeftShift</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseOr">sqlglot.expressions.BitwiseOr</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseRightShift">sqlglot.expressions.BitwiseRightShift</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#BitwiseXor">sqlglot.expressions.BitwiseXor</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Collate">sqlglot.expressions.Collate</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Connector">sqlglot.expressions.Connector</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#DPipe">sqlglot.expressions.DPipe</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Distance">sqlglot.expressions.Distance</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Div">sqlglot.expressions.Div</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Dot">sqlglot.expressions.Dot</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#EQ">sqlglot.expressions.EQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Escape">sqlglot.expressions.Escape</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#GT">sqlglot.expressions.GT</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#GTE">sqlglot.expressions.GTE</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Glob">sqlglot.expressions.Glob</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ILike">sqlglot.expressions.ILike</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#ILikeAny">sqlglot.expressions.ILikeAny</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#IntDiv">sqlglot.expressions.IntDiv</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Is">sqlglot.expressions.Is</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONArrayContains">sqlglot.expressions.JSONArrayContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBContains">sqlglot.expressions.JSONBContains</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBExtract">sqlglot.expressions.JSONBExtract</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONBExtractScalar">sqlglot.expressions.JSONBExtractScalar</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONExtract">sqlglot.expressions.JSONExtract</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#JSONExtractScalar">sqlglot.expressions.JSONExtractScalar</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Kwarg">sqlglot.expressions.Kwarg</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LT">sqlglot.expressions.LT</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LTE">sqlglot.expressions.LTE</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Like">sqlglot.expressions.Like</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#LikeAny">sqlglot.expressions.LikeAny</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Mod">sqlglot.expressions.Mod</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Mul">sqlglot.expressions.Mul</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NEQ">sqlglot.expressions.NEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NullSafeEQ">sqlglot.expressions.NullSafeEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#NullSafeNEQ">sqlglot.expressions.NullSafeNEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Operator">sqlglot.expressions.Operator</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Or">sqlglot.expressions.Or</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Overlaps">sqlglot.expressions.Overlaps</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Pow">sqlglot.expressions.Pow</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#PropertyEQ">sqlglot.expressions.PropertyEQ</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#RegexpILike">sqlglot.expressions.RegexpILike</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#RegexpLike">sqlglot.expressions.RegexpLike</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#SimilarTo">sqlglot.expressions.SimilarTo</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Slice">sqlglot.expressions.Slice</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Sub">sqlglot.expressions.Sub</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Xor">sqlglot.expressions.Xor</a>'>: <function TypeAnnotator.<dictcomp>.<lambda>>, <class '<a href="../expressions.html#Length">sqlglot.expressions.Length</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ApproxDistinct">sqlglot.expressions.ApproxDistinct</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Count">sqlglot.expressions.Count</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ArraySize">sqlglot.expressions.ArraySize</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#In">sqlglot.expressions.In</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Between">sqlglot.expressions.Between</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Boolean">sqlglot.expressions.Boolean</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDsToDate">sqlglot.expressions.TsOrDsToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DiToDate">sqlglot.expressions.DiToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateFromParts">sqlglot.expressions.DateFromParts</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Date">sqlglot.expressions.Date</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeStrToDate">sqlglot.expressions.TimeStrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrToDate">sqlglot.expressions.StrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentDate">sqlglot.expressions.CurrentDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateStrToDate">sqlglot.expressions.DateStrToDate</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeAdd">sqlglot.expressions.DatetimeAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeSub">sqlglot.expressions.DatetimeSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentDatetime">sqlglot.expressions.CurrentDatetime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log2">sqlglot.expressions.Log2</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Avg">sqlglot.expressions.Avg</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log">sqlglot.expressions.Log</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Round">sqlglot.expressions.Round</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StddevSamp">sqlglot.expressions.StddevSamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Ln">sqlglot.expressions.Ln</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StddevPop">sqlglot.expressions.StddevPop</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Stddev">sqlglot.expressions.Stddev</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Sqrt">sqlglot.expressions.Sqrt</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ApproxQuantile">sqlglot.expressions.ApproxQuantile</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#SafeDivide">sqlglot.expressions.SafeDivide</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#VariancePop">sqlglot.expressions.VariancePop</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Quantile">sqlglot.expressions.Quantile</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Variance">sqlglot.expressions.Variance</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Log10">sqlglot.expressions.Log10</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Exp">sqlglot.expressions.Exp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Floor">sqlglot.expressions.Floor</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Levenshtein">sqlglot.expressions.Levenshtein</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Extract">sqlglot.expressions.Extract</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DatetimeDiff">sqlglot.expressions.DatetimeDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeDiff">sqlglot.expressions.TimeDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDiToDi">sqlglot.expressions.TsOrDiToDi</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateToDi">sqlglot.expressions.DateToDi</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Ceil">sqlglot.expressions.Ceil</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrPosition">sqlglot.expressions.StrPosition</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampDiff">sqlglot.expressions.TimestampDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateDiff">sqlglot.expressions.DateDiff</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ParseJSON">sqlglot.expressions.ParseJSON</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampSub">sqlglot.expressions.TimestampSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimestampAdd">sqlglot.expressions.TimestampAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToTime">sqlglot.expressions.UnixToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Timestamp">sqlglot.expressions.Timestamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentTimestamp">sqlglot.expressions.CurrentTimestamp</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeStrToTime">sqlglot.expressions.TimeStrToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeSub">sqlglot.expressions.TimeSub</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#CurrentTime">sqlglot.expressions.CurrentTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#StrToTime">sqlglot.expressions.StrToTime</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeAdd">sqlglot.expressions.TimeAdd</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Month">sqlglot.expressions.Month</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Week">sqlglot.expressions.Week</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Day">sqlglot.expressions.Day</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Year">sqlglot.expressions.Year</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeToStr">sqlglot.expressions.TimeToStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToStr">sqlglot.expressions.UnixToStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ConcatWs">sqlglot.expressions.ConcatWs</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#UnixToTimeStr">sqlglot.expressions.UnixToTimeStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Substring">sqlglot.expressions.Substring</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Trim">sqlglot.expressions.Trim</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Concat">sqlglot.expressions.Concat</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#ArrayConcat">sqlglot.expressions.ArrayConcat</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#GroupConcat">sqlglot.expressions.GroupConcat</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Initcap">sqlglot.expressions.Initcap</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Lower">sqlglot.expressions.Lower</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TsOrDsToDateStr">sqlglot.expressions.TsOrDsToDateStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#DateToDateStr">sqlglot.expressions.DateToDateStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#TimeToTimeStr">sqlglot.expressions.TimeToTimeStr</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Upper">sqlglot.expressions.Upper</a>'>: <function _annotate_with_type_lambda.<locals>.<lambda>>, <class '<a href="../expressions.html#Abs">sqlglot.expressions.Abs</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Anonymous">sqlglot.expressions.Anonymous</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Array">sqlglot.expressions.Array</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#ArrayAgg">sqlglot.expressions.ArrayAgg</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Bracket">sqlglot.expressions.Bracket</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Cast">sqlglot.expressions.Cast</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Case">sqlglot.expressions.Case</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Coalesce">sqlglot.expressions.Coalesce</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DataType">sqlglot.expressions.DataType</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateAdd">sqlglot.expressions.DateAdd</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateSub">sqlglot.expressions.DateSub</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#DateTrunc">sqlglot.expressions.DateTrunc</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Distinct">sqlglot.expressions.Distinct</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Explode">sqlglot.expressions.Explode</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Filter">sqlglot.expressions.Filter</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#If">sqlglot.expressions.If</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Interval">sqlglot.expressions.Interval</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Least">sqlglot.expressions.Least</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Literal">sqlglot.expressions.Literal</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Map">sqlglot.expressions.Map</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Max">sqlglot.expressions.Max</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Min">sqlglot.expressions.Min</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Null">sqlglot.expressions.Null</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Nullif">sqlglot.expressions.Nullif</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Sum">sqlglot.expressions.Sum</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#TryCast">sqlglot.expressions.TryCast</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#VarMap">sqlglot.expressions.VarMap</a>'>: <function TypeAnnotator.<lambda>>, <class '<a href="../expressions.html#Struct">sqlglot.expressions.Struct</a>'>: <function TypeAnnotator.<lambda>>}</span> </div> @@ -1365,7 +1373,7 @@ <div class="attr variable"> <span class="name">COERCES_TO</span><span class="annotation">: Dict[<a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>, Set[<a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>]]</span> = <input id="TypeAnnotator.COERCES_TO-view-value" class="view-value-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> - <label class="view-value-button pdoc-button" for="TypeAnnotator.COERCES_TO-view-value"></label><span class="default_value">{<Type.TEXT: 'TEXT'>: set(), <Type.NVARCHAR: 'NVARCHAR'>: {<Type.TEXT: 'TEXT'>}, <Type.VARCHAR: 'VARCHAR'>: {<Type.TEXT: 'TEXT'>, <Type.NVARCHAR: 'NVARCHAR'>}, <Type.NCHAR: 'NCHAR'>: {<Type.VARCHAR: 'VARCHAR'>, <Type.TEXT: 'TEXT'>, <Type.NVARCHAR: 'NVARCHAR'>}, <Type.CHAR: 'CHAR'>: {<Type.VARCHAR: 'VARCHAR'>, <Type.TEXT: 'TEXT'>, <Type.NVARCHAR: 'NVARCHAR'>, <Type.NCHAR: 'NCHAR'>}, <Type.DOUBLE: 'DOUBLE'>: set(), <Type.FLOAT: 'FLOAT'>: {<Type.DOUBLE: 'DOUBLE'>}, <Type.DECIMAL: 'DECIMAL'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>}, <Type.BIGINT: 'BIGINT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.DECIMAL: 'DECIMAL'>}, <Type.INT: 'INT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.DECIMAL: 'DECIMAL'>, <Type.BIGINT: 'BIGINT'>}, <Type.SMALLINT: 'SMALLINT'>: {<Type.INT: 'INT'>, <Type.DECIMAL: 'DECIMAL'>, <Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.BIGINT: 'BIGINT'>}, <Type.TINYINT: 'TINYINT'>: {<Type.INT: 'INT'>, <Type.DECIMAL: 'DECIMAL'>, <Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.SMALLINT: 'SMALLINT'>, <Type.BIGINT: 'BIGINT'>}, <Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>: set(), <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>}, <Type.TIMESTAMP: 'TIMESTAMP'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>}, <Type.DATETIME: 'DATETIME'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMP: 'TIMESTAMP'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>}, <Type.DATE: 'DATE'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMP: 'TIMESTAMP'>, <Type.DATETIME: 'DATETIME'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>}}</span> + <label class="view-value-button pdoc-button" for="TypeAnnotator.COERCES_TO-view-value"></label><span class="default_value">{<Type.TEXT: 'TEXT'>: set(), <Type.NVARCHAR: 'NVARCHAR'>: {<Type.TEXT: 'TEXT'>}, <Type.VARCHAR: 'VARCHAR'>: {<Type.TEXT: 'TEXT'>, <Type.NVARCHAR: 'NVARCHAR'>}, <Type.NCHAR: 'NCHAR'>: {<Type.TEXT: 'TEXT'>, <Type.VARCHAR: 'VARCHAR'>, <Type.NVARCHAR: 'NVARCHAR'>}, <Type.CHAR: 'CHAR'>: {<Type.TEXT: 'TEXT'>, <Type.VARCHAR: 'VARCHAR'>, <Type.NCHAR: 'NCHAR'>, <Type.NVARCHAR: 'NVARCHAR'>}, <Type.DOUBLE: 'DOUBLE'>: set(), <Type.FLOAT: 'FLOAT'>: {<Type.DOUBLE: 'DOUBLE'>}, <Type.DECIMAL: 'DECIMAL'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>}, <Type.BIGINT: 'BIGINT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.DECIMAL: 'DECIMAL'>}, <Type.INT: 'INT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.FLOAT: 'FLOAT'>, <Type.BIGINT: 'BIGINT'>, <Type.DECIMAL: 'DECIMAL'>}, <Type.SMALLINT: 'SMALLINT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.INT: 'INT'>, <Type.FLOAT: 'FLOAT'>, <Type.BIGINT: 'BIGINT'>, <Type.DECIMAL: 'DECIMAL'>}, <Type.TINYINT: 'TINYINT'>: {<Type.DOUBLE: 'DOUBLE'>, <Type.INT: 'INT'>, <Type.FLOAT: 'FLOAT'>, <Type.SMALLINT: 'SMALLINT'>, <Type.BIGINT: 'BIGINT'>, <Type.DECIMAL: 'DECIMAL'>}, <Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>: set(), <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>}, <Type.TIMESTAMP: 'TIMESTAMP'>: {<Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>}, <Type.DATETIME: 'DATETIME'>: {<Type.TIMESTAMP: 'TIMESTAMP'>, <Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>}, <Type.DATE: 'DATE'>: {<Type.TIMESTAMP: 'TIMESTAMP'>, <Type.TIMESTAMPLTZ: 'TIMESTAMPLTZ'>, <Type.TIMESTAMPTZ: 'TIMESTAMPTZ'>, <Type.DATETIME: 'DATETIME'>}}</span> </div> @@ -1378,7 +1386,7 @@ <div class="attr variable"> <span class="name">BINARY_COERCIONS</span><span class="annotation">: Dict[Tuple[<a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>, <a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>], Callable[[<a href="../expressions.html#Expression">sqlglot.expressions.Expression</a>, <a href="../expressions.html#Expression">sqlglot.expressions.Expression</a>], <a href="../expressions.html#DataType.Type">sqlglot.expressions.DataType.Type</a>]]</span> = <input id="TypeAnnotator.BINARY_COERCIONS-view-value" class="view-value-toggle-state" type="checkbox" aria-hidden="true" tabindex="-1"> - <label class="view-value-button pdoc-button" for="TypeAnnotator.BINARY_COERCIONS-view-value"></label><span class="default_value">{(<Type.CHAR: 'CHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DATE: 'DATE'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.DATE: 'DATE'>): <function TypeAnnotator.<lambda>>}</span> + <label class="view-value-button pdoc-button" for="TypeAnnotator.BINARY_COERCIONS-view-value"></label><span class="default_value">{(<Type.TEXT: 'TEXT'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TEXT: 'TEXT'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NCHAR: 'NCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.CHAR: 'CHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.VARCHAR: 'VARCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT256: 'INT256'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT: 'INT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.FLOAT: 'FLOAT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.TINYINT: 'TINYINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.BIT: 'BIT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.INT128: 'INT128'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.DOUBLE: 'DOUBLE'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.SMALLINT: 'SMALLINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.NVARCHAR: 'NVARCHAR'>, <Type.BIGINT: 'BIGINT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.TEXT: 'TEXT'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.NCHAR: 'NCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.CHAR: 'CHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.VARCHAR: 'VARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT256: 'INT256'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT: 'INT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.FLOAT: 'FLOAT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.TINYINT: 'TINYINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIT: 'BIT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.INT128: 'INT128'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DOUBLE: 'DOUBLE'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.SMALLINT: 'SMALLINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.BIGINT: 'BIGINT'>, <Type.NVARCHAR: 'NVARCHAR'>): <function TypeAnnotator.<dictcomp>.<lambda>>, (<Type.DATE: 'DATE'>, <Type.INTERVAL: 'INTERVAL'>): <function TypeAnnotator.<lambda>>, (<Type.INTERVAL: 'INTERVAL'>, <Type.DATE: 'DATE'>): <function TypeAnnotator.<lambda>>}</span> </div> @@ -1442,51 +1450,51 @@ </div> <a class="headerlink" href="#TypeAnnotator.annotate"></a> - <div class="pdoc-code codehilite"><pre><span></span><span id="TypeAnnotator.annotate-342"><a href="#TypeAnnotator.annotate-342"><span class="linenos">342</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-343"><a href="#TypeAnnotator.annotate-343"><span class="linenos">343</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-344"><a href="#TypeAnnotator.annotate-344"><span class="linenos">344</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> -</span><span id="TypeAnnotator.annotate-345"><a href="#TypeAnnotator.annotate-345"><span class="linenos">345</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> -</span><span id="TypeAnnotator.annotate-346"><a href="#TypeAnnotator.annotate-346"><span class="linenos">346</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-347"><a href="#TypeAnnotator.annotate-347"><span class="linenos">347</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator.annotate-348"><a href="#TypeAnnotator.annotate-348"><span class="linenos">348</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-349"><a href="#TypeAnnotator.annotate-349"><span class="linenos">349</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> -</span><span id="TypeAnnotator.annotate-350"><a href="#TypeAnnotator.annotate-350"><span class="linenos">350</span></a> -</span><span id="TypeAnnotator.annotate-351"><a href="#TypeAnnotator.annotate-351"><span class="linenos">351</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-352"><a href="#TypeAnnotator.annotate-352"><span class="linenos">352</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-353"><a href="#TypeAnnotator.annotate-353"><span class="linenos">353</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> -</span><span id="TypeAnnotator.annotate-354"><a href="#TypeAnnotator.annotate-354"><span class="linenos">354</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-355"><a href="#TypeAnnotator.annotate-355"><span class="linenos">355</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> -</span><span id="TypeAnnotator.annotate-356"><a href="#TypeAnnotator.annotate-356"><span class="linenos">356</span></a> -</span><span id="TypeAnnotator.annotate-357"><a href="#TypeAnnotator.annotate-357"><span class="linenos">357</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-358"><a href="#TypeAnnotator.annotate-358"><span class="linenos">358</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator.annotate-359"><a href="#TypeAnnotator.annotate-359"><span class="linenos">359</span></a> -</span><span id="TypeAnnotator.annotate-360"><a href="#TypeAnnotator.annotate-360"><span class="linenos">360</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator.annotate-361"><a href="#TypeAnnotator.annotate-361"><span class="linenos">361</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> -</span><span id="TypeAnnotator.annotate-362"><a href="#TypeAnnotator.annotate-362"><span class="linenos">362</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> -</span><span id="TypeAnnotator.annotate-363"><a href="#TypeAnnotator.annotate-363"><span class="linenos">363</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> -</span><span id="TypeAnnotator.annotate-364"><a href="#TypeAnnotator.annotate-364"><span class="linenos">364</span></a> <span class="n">values</span><span class="p">,</span> -</span><span id="TypeAnnotator.annotate-365"><a href="#TypeAnnotator.annotate-365"><span class="linenos">365</span></a> <span class="p">)</span> -</span><span id="TypeAnnotator.annotate-366"><a href="#TypeAnnotator.annotate-366"><span class="linenos">366</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator.annotate-367"><a href="#TypeAnnotator.annotate-367"><span class="linenos">367</span></a> <span class="k">else</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-368"><a href="#TypeAnnotator.annotate-368"><span class="linenos">368</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> -</span><span id="TypeAnnotator.annotate-369"><a href="#TypeAnnotator.annotate-369"><span class="linenos">369</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> -</span><span id="TypeAnnotator.annotate-370"><a href="#TypeAnnotator.annotate-370"><span class="linenos">370</span></a> <span class="p">}</span> -</span><span id="TypeAnnotator.annotate-371"><a href="#TypeAnnotator.annotate-371"><span class="linenos">371</span></a> -</span><span id="TypeAnnotator.annotate-372"><a href="#TypeAnnotator.annotate-372"><span class="linenos">372</span></a> <span class="c1"># First annotate the current scope's column references</span> -</span><span id="TypeAnnotator.annotate-373"><a href="#TypeAnnotator.annotate-373"><span class="linenos">373</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-374"><a href="#TypeAnnotator.annotate-374"><span class="linenos">374</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> -</span><span id="TypeAnnotator.annotate-375"><a href="#TypeAnnotator.annotate-375"><span class="linenos">375</span></a> <span class="k">continue</span> -</span><span id="TypeAnnotator.annotate-376"><a href="#TypeAnnotator.annotate-376"><span class="linenos">376</span></a> -</span><span id="TypeAnnotator.annotate-377"><a href="#TypeAnnotator.annotate-377"><span class="linenos">377</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> -</span><span id="TypeAnnotator.annotate-378"><a href="#TypeAnnotator.annotate-378"><span class="linenos">378</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> -</span><span id="TypeAnnotator.annotate-379"><a href="#TypeAnnotator.annotate-379"><span class="linenos">379</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> -</span><span id="TypeAnnotator.annotate-380"><a href="#TypeAnnotator.annotate-380"><span class="linenos">380</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> -</span><span id="TypeAnnotator.annotate-381"><a href="#TypeAnnotator.annotate-381"><span class="linenos">381</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> -</span><span id="TypeAnnotator.annotate-382"><a href="#TypeAnnotator.annotate-382"><span class="linenos">382</span></a> -</span><span id="TypeAnnotator.annotate-383"><a href="#TypeAnnotator.annotate-383"><span class="linenos">383</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> -</span><span id="TypeAnnotator.annotate-384"><a href="#TypeAnnotator.annotate-384"><span class="linenos">384</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> -</span><span id="TypeAnnotator.annotate-385"><a href="#TypeAnnotator.annotate-385"><span class="linenos">385</span></a> -</span><span id="TypeAnnotator.annotate-386"><a href="#TypeAnnotator.annotate-386"><span class="linenos">386</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> + <div class="pdoc-code codehilite"><pre><span></span><span id="TypeAnnotator.annotate-343"><a href="#TypeAnnotator.annotate-343"><span class="linenos">343</span></a> <span class="k">def</span> <span class="nf">annotate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">expression</span><span class="p">:</span> <span class="n">E</span><span class="p">)</span> <span class="o">-></span> <span class="n">E</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-344"><a href="#TypeAnnotator.annotate-344"><span class="linenos">344</span></a> <span class="k">for</span> <span class="n">scope</span> <span class="ow">in</span> <span class="n">traverse_scope</span><span class="p">(</span><span class="n">expression</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-345"><a href="#TypeAnnotator.annotate-345"><span class="linenos">345</span></a> <span class="n">selects</span> <span class="o">=</span> <span class="p">{}</span> +</span><span id="TypeAnnotator.annotate-346"><a href="#TypeAnnotator.annotate-346"><span class="linenos">346</span></a> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">source</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">items</span><span class="p">():</span> +</span><span id="TypeAnnotator.annotate-347"><a href="#TypeAnnotator.annotate-347"><span class="linenos">347</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">Scope</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-348"><a href="#TypeAnnotator.annotate-348"><span class="linenos">348</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator.annotate-349"><a href="#TypeAnnotator.annotate-349"><span class="linenos">349</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">UDTF</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-350"><a href="#TypeAnnotator.annotate-350"><span class="linenos">350</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[]</span> +</span><span id="TypeAnnotator.annotate-351"><a href="#TypeAnnotator.annotate-351"><span class="linenos">351</span></a> +</span><span id="TypeAnnotator.annotate-352"><a href="#TypeAnnotator.annotate-352"><span class="linenos">352</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Lateral</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-353"><a href="#TypeAnnotator.annotate-353"><span class="linenos">353</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</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">exp</span><span class="o">.</span><span class="n">Explode</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-354"><a href="#TypeAnnotator.annotate-354"><span class="linenos">354</span></a> <span class="n">values</span> <span class="o">=</span> <span class="p">[</span><span class="n">source</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">this</span><span class="p">]</span> +</span><span id="TypeAnnotator.annotate-355"><a href="#TypeAnnotator.annotate-355"><span class="linenos">355</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-356"><a href="#TypeAnnotator.annotate-356"><span class="linenos">356</span></a> <span class="n">values</span> <span class="o">=</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">expressions</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">expressions</span> +</span><span id="TypeAnnotator.annotate-357"><a href="#TypeAnnotator.annotate-357"><span class="linenos">357</span></a> +</span><span id="TypeAnnotator.annotate-358"><a href="#TypeAnnotator.annotate-358"><span class="linenos">358</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">values</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-359"><a href="#TypeAnnotator.annotate-359"><span class="linenos">359</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator.annotate-360"><a href="#TypeAnnotator.annotate-360"><span class="linenos">360</span></a> +</span><span id="TypeAnnotator.annotate-361"><a href="#TypeAnnotator.annotate-361"><span class="linenos">361</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator.annotate-362"><a href="#TypeAnnotator.annotate-362"><span class="linenos">362</span></a> <span class="n">alias</span><span class="p">:</span> <span class="n">column</span> +</span><span id="TypeAnnotator.annotate-363"><a href="#TypeAnnotator.annotate-363"><span class="linenos">363</span></a> <span class="k">for</span> <span class="n">alias</span><span class="p">,</span> <span class="n">column</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span> +</span><span id="TypeAnnotator.annotate-364"><a href="#TypeAnnotator.annotate-364"><span class="linenos">364</span></a> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">alias_column_names</span><span class="p">,</span> +</span><span id="TypeAnnotator.annotate-365"><a href="#TypeAnnotator.annotate-365"><span class="linenos">365</span></a> <span class="n">values</span><span class="p">,</span> +</span><span id="TypeAnnotator.annotate-366"><a href="#TypeAnnotator.annotate-366"><span class="linenos">366</span></a> <span class="p">)</span> +</span><span id="TypeAnnotator.annotate-367"><a href="#TypeAnnotator.annotate-367"><span class="linenos">367</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator.annotate-368"><a href="#TypeAnnotator.annotate-368"><span class="linenos">368</span></a> <span class="k">else</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-369"><a href="#TypeAnnotator.annotate-369"><span class="linenos">369</span></a> <span class="n">selects</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{</span> +</span><span id="TypeAnnotator.annotate-370"><a href="#TypeAnnotator.annotate-370"><span class="linenos">370</span></a> <span class="n">select</span><span class="o">.</span><span class="n">alias_or_name</span><span class="p">:</span> <span class="n">select</span> <span class="k">for</span> <span class="n">select</span> <span class="ow">in</span> <span class="n">source</span><span class="o">.</span><span class="n">expression</span><span class="o">.</span><span class="n">selects</span> +</span><span id="TypeAnnotator.annotate-371"><a href="#TypeAnnotator.annotate-371"><span class="linenos">371</span></a> <span class="p">}</span> +</span><span id="TypeAnnotator.annotate-372"><a href="#TypeAnnotator.annotate-372"><span class="linenos">372</span></a> +</span><span id="TypeAnnotator.annotate-373"><a href="#TypeAnnotator.annotate-373"><span class="linenos">373</span></a> <span class="c1"># First annotate the current scope's column references</span> +</span><span id="TypeAnnotator.annotate-374"><a href="#TypeAnnotator.annotate-374"><span class="linenos">374</span></a> <span class="k">for</span> <span class="n">col</span> <span class="ow">in</span> <span class="n">scope</span><span class="o">.</span><span class="n">columns</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-375"><a href="#TypeAnnotator.annotate-375"><span class="linenos">375</span></a> <span class="k">if</span> <span class="ow">not</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">:</span> +</span><span id="TypeAnnotator.annotate-376"><a href="#TypeAnnotator.annotate-376"><span class="linenos">376</span></a> <span class="k">continue</span> +</span><span id="TypeAnnotator.annotate-377"><a href="#TypeAnnotator.annotate-377"><span class="linenos">377</span></a> +</span><span id="TypeAnnotator.annotate-378"><a href="#TypeAnnotator.annotate-378"><span class="linenos">378</span></a> <span class="n">source</span> <span class="o">=</span> <span class="n">scope</span><span class="o">.</span><span class="n">sources</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">)</span> +</span><span id="TypeAnnotator.annotate-379"><a href="#TypeAnnotator.annotate-379"><span class="linenos">379</span></a> <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">exp</span><span class="o">.</span><span class="n">Table</span><span class="p">):</span> +</span><span id="TypeAnnotator.annotate-380"><a href="#TypeAnnotator.annotate-380"><span class="linenos">380</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">schema</span><span class="o">.</span><span class="n">get_column_type</span><span class="p">(</span><span class="n">source</span><span class="p">,</span> <span class="n">col</span><span class="p">))</span> +</span><span id="TypeAnnotator.annotate-381"><a href="#TypeAnnotator.annotate-381"><span class="linenos">381</span></a> <span class="k">elif</span> <span class="n">source</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">table</span> <span class="ow">in</span> <span class="n">selects</span> <span class="ow">and</span> <span class="n">col</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">]:</span> +</span><span id="TypeAnnotator.annotate-382"><a href="#TypeAnnotator.annotate-382"><span class="linenos">382</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_set_type</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="n">selects</span><span class="p">[</span><span class="n">col</span><span class="o">.</span><span class="n">table</span><span class="p">][</span><span class="n">col</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">type</span><span class="p">)</span> +</span><span id="TypeAnnotator.annotate-383"><a href="#TypeAnnotator.annotate-383"><span class="linenos">383</span></a> +</span><span id="TypeAnnotator.annotate-384"><a href="#TypeAnnotator.annotate-384"><span class="linenos">384</span></a> <span class="c1"># Then (possibly) annotate the remaining expressions in the scope</span> +</span><span id="TypeAnnotator.annotate-385"><a href="#TypeAnnotator.annotate-385"><span class="linenos">385</span></a> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">scope</span><span class="o">.</span><span class="n">expression</span><span class="p">)</span> +</span><span id="TypeAnnotator.annotate-386"><a href="#TypeAnnotator.annotate-386"><span class="linenos">386</span></a> +</span><span id="TypeAnnotator.annotate-387"><a href="#TypeAnnotator.annotate-387"><span class="linenos">387</span></a> <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_maybe_annotate</span><span class="p">(</span><span class="n">expression</span><span class="p">)</span> <span class="c1"># This takes care of non-traversable expressions</span> </span></pre></div> |