Skip to content

Commit 9a9c1ed

Browse files
GiorgioPorgioGiorgioPorgio
committed
increase typing to 5 args in select
1 parent e7744e5 commit 9a9c1ed

File tree

2 files changed

+339
-1
lines changed

2 files changed

+339
-1
lines changed

‎scripts/generate_select.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
destiny_path = Path(__file__).parent.parent / "sqlmodel/sql/_expression_select_gen.py"
1414

1515

16-
number_of_types = 4
16+
number_of_types = 5
1717

1818

1919
class Arg(BaseModel):

‎sqlmodel/sql/_expression_select_gen.py‎

Lines changed: 338 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,24 @@
107107
_T3 = TypeVar("_T3")
108108

109109

110+
_TScalar_4 = TypeVar(
111+
"_TScalar_4",
112+
Column, # type: ignore
113+
Sequence, # type: ignore
114+
Mapping, # type: ignore
115+
UUID,
116+
datetime,
117+
float,
118+
int,
119+
bool,
120+
bytes,
121+
str,
122+
None,
123+
)
124+
125+
_T4 = TypeVar("_T4")
126+
127+
110128
# Generated TypeVars end
111129

112130

@@ -358,6 +376,326 @@ def select( # type: ignore
358376
) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _TScalar_3]]: ...
359377

360378

379+
@overload
380+
def select( # type: ignore
381+
__ent0: _TCCA[_T0],
382+
__ent1: _TCCA[_T1],
383+
__ent2: _TCCA[_T2],
384+
__ent3: _TCCA[_T3],
385+
__ent4: _TCCA[_T4],
386+
) -> Select[Tuple[_T0, _T1, _T2, _T3, _T4]]: ...
387+
388+
389+
@overload
390+
def select( # type: ignore
391+
__ent0: _TCCA[_T0],
392+
__ent1: _TCCA[_T1],
393+
__ent2: _TCCA[_T2],
394+
__ent3: _TCCA[_T3],
395+
entity_4: _TScalar_4,
396+
) -> Select[Tuple[_T0, _T1, _T2, _T3, _TScalar_4]]: ...
397+
398+
399+
@overload
400+
def select( # type: ignore
401+
__ent0: _TCCA[_T0],
402+
__ent1: _TCCA[_T1],
403+
__ent2: _TCCA[_T2],
404+
entity_3: _TScalar_3,
405+
__ent4: _TCCA[_T4],
406+
) -> Select[Tuple[_T0, _T1, _T2, _TScalar_3, _T4]]: ...
407+
408+
409+
@overload
410+
def select( # type: ignore
411+
__ent0: _TCCA[_T0],
412+
__ent1: _TCCA[_T1],
413+
__ent2: _TCCA[_T2],
414+
entity_3: _TScalar_3,
415+
entity_4: _TScalar_4,
416+
) -> Select[Tuple[_T0, _T1, _T2, _TScalar_3, _TScalar_4]]: ...
417+
418+
419+
@overload
420+
def select( # type: ignore
421+
__ent0: _TCCA[_T0],
422+
__ent1: _TCCA[_T1],
423+
entity_2: _TScalar_2,
424+
__ent3: _TCCA[_T3],
425+
__ent4: _TCCA[_T4],
426+
) -> Select[Tuple[_T0, _T1, _TScalar_2, _T3, _T4]]: ...
427+
428+
429+
@overload
430+
def select( # type: ignore
431+
__ent0: _TCCA[_T0],
432+
__ent1: _TCCA[_T1],
433+
entity_2: _TScalar_2,
434+
__ent3: _TCCA[_T3],
435+
entity_4: _TScalar_4,
436+
) -> Select[Tuple[_T0, _T1, _TScalar_2, _T3, _TScalar_4]]: ...
437+
438+
439+
@overload
440+
def select( # type: ignore
441+
__ent0: _TCCA[_T0],
442+
__ent1: _TCCA[_T1],
443+
entity_2: _TScalar_2,
444+
entity_3: _TScalar_3,
445+
__ent4: _TCCA[_T4],
446+
) -> Select[Tuple[_T0, _T1, _TScalar_2, _TScalar_3, _T4]]: ...
447+
448+
449+
@overload
450+
def select( # type: ignore
451+
__ent0: _TCCA[_T0],
452+
__ent1: _TCCA[_T1],
453+
entity_2: _TScalar_2,
454+
entity_3: _TScalar_3,
455+
entity_4: _TScalar_4,
456+
) -> Select[Tuple[_T0, _T1, _TScalar_2, _TScalar_3, _TScalar_4]]: ...
457+
458+
459+
@overload
460+
def select( # type: ignore
461+
__ent0: _TCCA[_T0],
462+
entity_1: _TScalar_1,
463+
__ent2: _TCCA[_T2],
464+
__ent3: _TCCA[_T3],
465+
__ent4: _TCCA[_T4],
466+
) -> Select[Tuple[_T0, _TScalar_1, _T2, _T3, _T4]]: ...
467+
468+
469+
@overload
470+
def select( # type: ignore
471+
__ent0: _TCCA[_T0],
472+
entity_1: _TScalar_1,
473+
__ent2: _TCCA[_T2],
474+
__ent3: _TCCA[_T3],
475+
entity_4: _TScalar_4,
476+
) -> Select[Tuple[_T0, _TScalar_1, _T2, _T3, _TScalar_4]]: ...
477+
478+
479+
@overload
480+
def select( # type: ignore
481+
__ent0: _TCCA[_T0],
482+
entity_1: _TScalar_1,
483+
__ent2: _TCCA[_T2],
484+
entity_3: _TScalar_3,
485+
__ent4: _TCCA[_T4],
486+
) -> Select[Tuple[_T0, _TScalar_1, _T2, _TScalar_3, _T4]]: ...
487+
488+
489+
@overload
490+
def select( # type: ignore
491+
__ent0: _TCCA[_T0],
492+
entity_1: _TScalar_1,
493+
__ent2: _TCCA[_T2],
494+
entity_3: _TScalar_3,
495+
entity_4: _TScalar_4,
496+
) -> Select[Tuple[_T0, _TScalar_1, _T2, _TScalar_3, _TScalar_4]]: ...
497+
498+
499+
@overload
500+
def select( # type: ignore
501+
__ent0: _TCCA[_T0],
502+
entity_1: _TScalar_1,
503+
entity_2: _TScalar_2,
504+
__ent3: _TCCA[_T3],
505+
__ent4: _TCCA[_T4],
506+
) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _T3, _T4]]: ...
507+
508+
509+
@overload
510+
def select( # type: ignore
511+
__ent0: _TCCA[_T0],
512+
entity_1: _TScalar_1,
513+
entity_2: _TScalar_2,
514+
__ent3: _TCCA[_T3],
515+
entity_4: _TScalar_4,
516+
) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _T3, _TScalar_4]]: ...
517+
518+
519+
@overload
520+
def select( # type: ignore
521+
__ent0: _TCCA[_T0],
522+
entity_1: _TScalar_1,
523+
entity_2: _TScalar_2,
524+
entity_3: _TScalar_3,
525+
__ent4: _TCCA[_T4],
526+
) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _TScalar_3, _T4]]: ...
527+
528+
529+
@overload
530+
def select( # type: ignore
531+
__ent0: _TCCA[_T0],
532+
entity_1: _TScalar_1,
533+
entity_2: _TScalar_2,
534+
entity_3: _TScalar_3,
535+
entity_4: _TScalar_4,
536+
) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _TScalar_3, _TScalar_4]]: ...
537+
538+
539+
@overload
540+
def select( # type: ignore
541+
entity_0: _TScalar_0,
542+
__ent1: _TCCA[_T1],
543+
__ent2: _TCCA[_T2],
544+
__ent3: _TCCA[_T3],
545+
__ent4: _TCCA[_T4],
546+
) -> Select[Tuple[_TScalar_0, _T1, _T2, _T3, _T4]]: ...
547+
548+
549+
@overload
550+
def select( # type: ignore
551+
entity_0: _TScalar_0,
552+
__ent1: _TCCA[_T1],
553+
__ent2: _TCCA[_T2],
554+
__ent3: _TCCA[_T3],
555+
entity_4: _TScalar_4,
556+
) -> Select[Tuple[_TScalar_0, _T1, _T2, _T3, _TScalar_4]]: ...
557+
558+
559+
@overload
560+
def select( # type: ignore
561+
entity_0: _TScalar_0,
562+
__ent1: _TCCA[_T1],
563+
__ent2: _TCCA[_T2],
564+
entity_3: _TScalar_3,
565+
__ent4: _TCCA[_T4],
566+
) -> Select[Tuple[_TScalar_0, _T1, _T2, _TScalar_3, _T4]]: ...
567+
568+
569+
@overload
570+
def select( # type: ignore
571+
entity_0: _TScalar_0,
572+
__ent1: _TCCA[_T1],
573+
__ent2: _TCCA[_T2],
574+
entity_3: _TScalar_3,
575+
entity_4: _TScalar_4,
576+
) -> Select[Tuple[_TScalar_0, _T1, _T2, _TScalar_3, _TScalar_4]]: ...
577+
578+
579+
@overload
580+
def select( # type: ignore
581+
entity_0: _TScalar_0,
582+
__ent1: _TCCA[_T1],
583+
entity_2: _TScalar_2,
584+
__ent3: _TCCA[_T3],
585+
__ent4: _TCCA[_T4],
586+
) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _T3, _T4]]: ...
587+
588+
589+
@overload
590+
def select( # type: ignore
591+
entity_0: _TScalar_0,
592+
__ent1: _TCCA[_T1],
593+
entity_2: _TScalar_2,
594+
__ent3: _TCCA[_T3],
595+
entity_4: _TScalar_4,
596+
) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _T3, _TScalar_4]]: ...
597+
598+
599+
@overload
600+
def select( # type: ignore
601+
entity_0: _TScalar_0,
602+
__ent1: _TCCA[_T1],
603+
entity_2: _TScalar_2,
604+
entity_3: _TScalar_3,
605+
__ent4: _TCCA[_T4],
606+
) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _TScalar_3, _T4]]: ...
607+
608+
609+
@overload
610+
def select( # type: ignore
611+
entity_0: _TScalar_0,
612+
__ent1: _TCCA[_T1],
613+
entity_2: _TScalar_2,
614+
entity_3: _TScalar_3,
615+
entity_4: _TScalar_4,
616+
) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _TScalar_3, _TScalar_4]]: ...
617+
618+
619+
@overload
620+
def select( # type: ignore
621+
entity_0: _TScalar_0,
622+
entity_1: _TScalar_1,
623+
__ent2: _TCCA[_T2],
624+
__ent3: _TCCA[_T3],
625+
__ent4: _TCCA[_T4],
626+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _T3, _T4]]: ...
627+
628+
629+
@overload
630+
def select( # type: ignore
631+
entity_0: _TScalar_0,
632+
entity_1: _TScalar_1,
633+
__ent2: _TCCA[_T2],
634+
__ent3: _TCCA[_T3],
635+
entity_4: _TScalar_4,
636+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _T3, _TScalar_4]]: ...
637+
638+
639+
@overload
640+
def select( # type: ignore
641+
entity_0: _TScalar_0,
642+
entity_1: _TScalar_1,
643+
__ent2: _TCCA[_T2],
644+
entity_3: _TScalar_3,
645+
__ent4: _TCCA[_T4],
646+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _TScalar_3, _T4]]: ...
647+
648+
649+
@overload
650+
def select( # type: ignore
651+
entity_0: _TScalar_0,
652+
entity_1: _TScalar_1,
653+
__ent2: _TCCA[_T2],
654+
entity_3: _TScalar_3,
655+
entity_4: _TScalar_4,
656+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _TScalar_3, _TScalar_4]]: ...
657+
658+
659+
@overload
660+
def select( # type: ignore
661+
entity_0: _TScalar_0,
662+
entity_1: _TScalar_1,
663+
entity_2: _TScalar_2,
664+
__ent3: _TCCA[_T3],
665+
__ent4: _TCCA[_T4],
666+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _T3, _T4]]: ...
667+
668+
669+
@overload
670+
def select( # type: ignore
671+
entity_0: _TScalar_0,
672+
entity_1: _TScalar_1,
673+
entity_2: _TScalar_2,
674+
__ent3: _TCCA[_T3],
675+
entity_4: _TScalar_4,
676+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _T3, _TScalar_4]]: ...
677+
678+
679+
@overload
680+
def select( # type: ignore
681+
entity_0: _TScalar_0,
682+
entity_1: _TScalar_1,
683+
entity_2: _TScalar_2,
684+
entity_3: _TScalar_3,
685+
__ent4: _TCCA[_T4],
686+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _TScalar_3, _T4]]: ...
687+
688+
689+
@overload
690+
def select( # type: ignore
691+
entity_0: _TScalar_0,
692+
entity_1: _TScalar_1,
693+
entity_2: _TScalar_2,
694+
entity_3: _TScalar_3,
695+
entity_4: _TScalar_4,
696+
) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _TScalar_3, _TScalar_4]]: ...
697+
698+
361699
# Generated overloads end
362700

363701

0 commit comments

Comments
 (0)