Coverage for sqlmodel/sql/_expression_select_gen.py: 100%

21 statements  

« prev     ^ index     » next       coverage.py v7.6.1, created at 2024-08-12 19:13 +0000

1# WARNING: do not modify this code, it is generated by _expression_select_gen.py.jinja2 

2 

3from datetime import datetime 1fabcde

4from typing import ( 1fabcde

5 Any, 

6 Mapping, 

7 Sequence, 

8 Tuple, 

9 Type, 

10 TypeVar, 

11 Union, 

12 overload, 

13) 

14from uuid import UUID 1fabcde

15 

16from sqlalchemy import ( 1fabcde

17 Column, 

18) 

19from sqlalchemy.sql.elements import ( 1fabcde

20 SQLCoreOperations, 

21) 

22from sqlalchemy.sql.roles import TypedColumnsClauseRole 1fabcde

23 

24from ._expression_select_cls import Select, SelectOfScalar 1fabcde

25 

26_T = TypeVar("_T") 1fabcde

27 

28 

29_TCCA = Union[ 1fabcde

30 TypedColumnsClauseRole[_T], 

31 SQLCoreOperations[_T], 

32 Type[_T], 

33] 

34 

35# Generated TypeVars start 

36 

37 

38_TScalar_0 = TypeVar( 1fabcde

39 "_TScalar_0", 

40 Column, # type: ignore 

41 Sequence, # type: ignore 

42 Mapping, # type: ignore 

43 UUID, 

44 datetime, 

45 float, 

46 int, 

47 bool, 

48 bytes, 

49 str, 

50 None, 

51) 

52 

53_T0 = TypeVar("_T0") 1fabcde

54 

55 

56_TScalar_1 = TypeVar( 1fabcde

57 "_TScalar_1", 

58 Column, # type: ignore 

59 Sequence, # type: ignore 

60 Mapping, # type: ignore 

61 UUID, 

62 datetime, 

63 float, 

64 int, 

65 bool, 

66 bytes, 

67 str, 

68 None, 

69) 

70 

71_T1 = TypeVar("_T1") 1fabcde

72 

73 

74_TScalar_2 = TypeVar( 1fabcde

75 "_TScalar_2", 

76 Column, # type: ignore 

77 Sequence, # type: ignore 

78 Mapping, # type: ignore 

79 UUID, 

80 datetime, 

81 float, 

82 int, 

83 bool, 

84 bytes, 

85 str, 

86 None, 

87) 

88 

89_T2 = TypeVar("_T2") 1fabcde

90 

91 

92_TScalar_3 = TypeVar( 1fabcde

93 "_TScalar_3", 

94 Column, # type: ignore 

95 Sequence, # type: ignore 

96 Mapping, # type: ignore 

97 UUID, 

98 datetime, 

99 float, 

100 int, 

101 bool, 

102 bytes, 

103 str, 

104 None, 

105) 

106 

107_T3 = TypeVar("_T3") 1fabcde

108 

109 

110# Generated TypeVars end 

111 

112 

113@overload 1fabcde

114def select(__ent0: _TCCA[_T0]) -> SelectOfScalar[_T0]: ... 1fabcde

115 

116 

117@overload 1fabcde

118def select(__ent0: _TScalar_0) -> SelectOfScalar[_TScalar_0]: # type: ignore 1fabcde

119 ... 

120 

121 

122# Generated overloads start 

123 

124 

125@overload 1fabcde

126def select( # type: ignore 1abcde

127 __ent0: _TCCA[_T0], 1fabcde

128 __ent1: _TCCA[_T1], 1fabcde

129) -> Select[Tuple[_T0, _T1]]: ... 1fabcde

130 

131 

132@overload 1fabcde

133def select( # type: ignore 1abcde

134 __ent0: _TCCA[_T0], 1fabcde

135 entity_1: _TScalar_1, 1fabcde

136) -> Select[Tuple[_T0, _TScalar_1]]: ... 1fabcde

137 

138 

139@overload 1fabcde

140def select( # type: ignore 1abcde

141 entity_0: _TScalar_0, 1fabcde

142 __ent1: _TCCA[_T1], 1fabcde

143) -> Select[Tuple[_TScalar_0, _T1]]: ... 1fabcde

144 

145 

146@overload 1fabcde

147def select( # type: ignore 1abcde

148 entity_0: _TScalar_0, 1fabcde

149 entity_1: _TScalar_1, 1fabcde

150) -> Select[Tuple[_TScalar_0, _TScalar_1]]: ... 1fabcde

151 

152 

153@overload 1fabcde

154def select( # type: ignore 1abcde

155 __ent0: _TCCA[_T0], 1fabcde

156 __ent1: _TCCA[_T1], 1fabcde

157 __ent2: _TCCA[_T2], 1fabcde

158) -> Select[Tuple[_T0, _T1, _T2]]: ... 1fabcde

159 

160 

161@overload 1fabcde

162def select( # type: ignore 1abcde

163 __ent0: _TCCA[_T0], 1fabcde

164 __ent1: _TCCA[_T1], 1fabcde

165 entity_2: _TScalar_2, 1fabcde

166) -> Select[Tuple[_T0, _T1, _TScalar_2]]: ... 1fabcde

167 

168 

169@overload 1fabcde

170def select( # type: ignore 1abcde

171 __ent0: _TCCA[_T0], 1fabcde

172 entity_1: _TScalar_1, 1fabcde

173 __ent2: _TCCA[_T2], 1fabcde

174) -> Select[Tuple[_T0, _TScalar_1, _T2]]: ... 1fabcde

175 

176 

177@overload 1fabcde

178def select( # type: ignore 1abcde

179 __ent0: _TCCA[_T0], 1fabcde

180 entity_1: _TScalar_1, 1fabcde

181 entity_2: _TScalar_2, 1fabcde

182) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2]]: ... 1fabcde

183 

184 

185@overload 1fabcde

186def select( # type: ignore 1abcde

187 entity_0: _TScalar_0, 1fabcde

188 __ent1: _TCCA[_T1], 1fabcde

189 __ent2: _TCCA[_T2], 1fabcde

190) -> Select[Tuple[_TScalar_0, _T1, _T2]]: ... 1fabcde

191 

192 

193@overload 1fabcde

194def select( # type: ignore 1abcde

195 entity_0: _TScalar_0, 1fabcde

196 __ent1: _TCCA[_T1], 1fabcde

197 entity_2: _TScalar_2, 1fabcde

198) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2]]: ... 1fabcde

199 

200 

201@overload 1fabcde

202def select( # type: ignore 1abcde

203 entity_0: _TScalar_0, 1fabcde

204 entity_1: _TScalar_1, 1fabcde

205 __ent2: _TCCA[_T2], 1fabcde

206) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2]]: ... 1fabcde

207 

208 

209@overload 1fabcde

210def select( # type: ignore 1abcde

211 entity_0: _TScalar_0, 1fabcde

212 entity_1: _TScalar_1, 1fabcde

213 entity_2: _TScalar_2, 1fabcde

214) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2]]: ... 1fabcde

215 

216 

217@overload 1fabcde

218def select( # type: ignore 1abcde

219 __ent0: _TCCA[_T0], 1fabcde

220 __ent1: _TCCA[_T1], 1fabcde

221 __ent2: _TCCA[_T2], 1fabcde

222 __ent3: _TCCA[_T3], 1fabcde

223) -> Select[Tuple[_T0, _T1, _T2, _T3]]: ... 1fabcde

224 

225 

226@overload 1fabcde

227def select( # type: ignore 1abcde

228 __ent0: _TCCA[_T0], 1fabcde

229 __ent1: _TCCA[_T1], 1fabcde

230 __ent2: _TCCA[_T2], 1fabcde

231 entity_3: _TScalar_3, 1fabcde

232) -> Select[Tuple[_T0, _T1, _T2, _TScalar_3]]: ... 1fabcde

233 

234 

235@overload 1fabcde

236def select( # type: ignore 1abcde

237 __ent0: _TCCA[_T0], 1fabcde

238 __ent1: _TCCA[_T1], 1fabcde

239 entity_2: _TScalar_2, 1fabcde

240 __ent3: _TCCA[_T3], 1fabcde

241) -> Select[Tuple[_T0, _T1, _TScalar_2, _T3]]: ... 1fabcde

242 

243 

244@overload 1fabcde

245def select( # type: ignore 1abcde

246 __ent0: _TCCA[_T0], 1fabcde

247 __ent1: _TCCA[_T1], 1fabcde

248 entity_2: _TScalar_2, 1fabcde

249 entity_3: _TScalar_3, 1fabcde

250) -> Select[Tuple[_T0, _T1, _TScalar_2, _TScalar_3]]: ... 1fabcde

251 

252 

253@overload 1fabcde

254def select( # type: ignore 1abcde

255 __ent0: _TCCA[_T0], 1fabcde

256 entity_1: _TScalar_1, 1fabcde

257 __ent2: _TCCA[_T2], 1fabcde

258 __ent3: _TCCA[_T3], 1fabcde

259) -> Select[Tuple[_T0, _TScalar_1, _T2, _T3]]: ... 1fabcde

260 

261 

262@overload 1fabcde

263def select( # type: ignore 1abcde

264 __ent0: _TCCA[_T0], 1fabcde

265 entity_1: _TScalar_1, 1fabcde

266 __ent2: _TCCA[_T2], 1fabcde

267 entity_3: _TScalar_3, 1fabcde

268) -> Select[Tuple[_T0, _TScalar_1, _T2, _TScalar_3]]: ... 1fabcde

269 

270 

271@overload 1fabcde

272def select( # type: ignore 1abcde

273 __ent0: _TCCA[_T0], 1fabcde

274 entity_1: _TScalar_1, 1fabcde

275 entity_2: _TScalar_2, 1fabcde

276 __ent3: _TCCA[_T3], 1fabcde

277) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _T3]]: ... 1fabcde

278 

279 

280@overload 1fabcde

281def select( # type: ignore 1abcde

282 __ent0: _TCCA[_T0], 1fabcde

283 entity_1: _TScalar_1, 1fabcde

284 entity_2: _TScalar_2, 1fabcde

285 entity_3: _TScalar_3, 1fabcde

286) -> Select[Tuple[_T0, _TScalar_1, _TScalar_2, _TScalar_3]]: ... 1fabcde

287 

288 

289@overload 1fabcde

290def select( # type: ignore 1abcde

291 entity_0: _TScalar_0, 1fabcde

292 __ent1: _TCCA[_T1], 1fabcde

293 __ent2: _TCCA[_T2], 1fabcde

294 __ent3: _TCCA[_T3], 1fabcde

295) -> Select[Tuple[_TScalar_0, _T1, _T2, _T3]]: ... 1fabcde

296 

297 

298@overload 1fabcde

299def select( # type: ignore 1abcde

300 entity_0: _TScalar_0, 1fabcde

301 __ent1: _TCCA[_T1], 1fabcde

302 __ent2: _TCCA[_T2], 1fabcde

303 entity_3: _TScalar_3, 1fabcde

304) -> Select[Tuple[_TScalar_0, _T1, _T2, _TScalar_3]]: ... 1fabcde

305 

306 

307@overload 1fabcde

308def select( # type: ignore 1abcde

309 entity_0: _TScalar_0, 1fabcde

310 __ent1: _TCCA[_T1], 1fabcde

311 entity_2: _TScalar_2, 1fabcde

312 __ent3: _TCCA[_T3], 1fabcde

313) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _T3]]: ... 1fabcde

314 

315 

316@overload 1fabcde

317def select( # type: ignore 1abcde

318 entity_0: _TScalar_0, 1fabcde

319 __ent1: _TCCA[_T1], 1fabcde

320 entity_2: _TScalar_2, 1fabcde

321 entity_3: _TScalar_3, 1fabcde

322) -> Select[Tuple[_TScalar_0, _T1, _TScalar_2, _TScalar_3]]: ... 1fabcde

323 

324 

325@overload 1fabcde

326def select( # type: ignore 1abcde

327 entity_0: _TScalar_0, 1fabcde

328 entity_1: _TScalar_1, 1fabcde

329 __ent2: _TCCA[_T2], 1fabcde

330 __ent3: _TCCA[_T3], 1fabcde

331) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _T3]]: ... 1fabcde

332 

333 

334@overload 1fabcde

335def select( # type: ignore 1abcde

336 entity_0: _TScalar_0, 1fabcde

337 entity_1: _TScalar_1, 1fabcde

338 __ent2: _TCCA[_T2], 1fabcde

339 entity_3: _TScalar_3, 1fabcde

340) -> Select[Tuple[_TScalar_0, _TScalar_1, _T2, _TScalar_3]]: ... 1fabcde

341 

342 

343@overload 1fabcde

344def select( # type: ignore 1abcde

345 entity_0: _TScalar_0, 1fabcde

346 entity_1: _TScalar_1, 1fabcde

347 entity_2: _TScalar_2, 1fabcde

348 __ent3: _TCCA[_T3], 1fabcde

349) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _T3]]: ... 1fabcde

350 

351 

352@overload 1fabcde

353def select( # type: ignore 1abcde

354 entity_0: _TScalar_0, 1fabcde

355 entity_1: _TScalar_1, 1fabcde

356 entity_2: _TScalar_2, 1fabcde

357 entity_3: _TScalar_3, 1fabcde

358) -> Select[Tuple[_TScalar_0, _TScalar_1, _TScalar_2, _TScalar_3]]: ... 1fabcde

359 

360 

361# Generated overloads end 

362 

363 

364def select(*entities: Any) -> Union[Select, SelectOfScalar]: # type: ignore 1fabcde

365 if len(entities) == 1: 1fabcde

366 return SelectOfScalar(*entities) 1fabcde

367 return Select(*entities) 1fabcde