ormlambda 3.34.1__py3-none-any.whl → 3.34.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
ormlambda/__init__.py CHANGED
@@ -24,3 +24,36 @@ from .model.base_model import (
24
24
  ) # COMMENT: to avoid relative import we need to import BaseModel after import Table,Column, ForeignKey, IRepositoryBase and Disassembler
25
25
 
26
26
  from .engine import create_engine, URL, make_url # noqa: F401
27
+
28
+ from .sql.sqltypes import ( # noqa: F401
29
+ JSON as JSON,
30
+ UUID as UUID,
31
+ NullType as NullType,
32
+ INTEGER as INTEGER,
33
+ INT as INT,
34
+ SMALLINTEGER as SMALLINTEGER,
35
+ BIGINTEGER as BIGINTEGER,
36
+ NUMERIC as NUMERIC,
37
+ FLOAT as FLOAT,
38
+ REAL as REAL,
39
+ DOUBLE as DOUBLE,
40
+ DECIMAL as DECIMAL,
41
+ STRING as STRING,
42
+ TEXT as TEXT,
43
+ UNICODE as UNICODE,
44
+ UNICODETEXT as UNICODETEXT,
45
+ CHAR as CHAR,
46
+ NCHAR as NCHAR,
47
+ BLOB as BLOB,
48
+ VARCHAR as VARCHAR,
49
+ NVARCHAR as NVARCHAR,
50
+ DATE as DATE,
51
+ TIME as TIME,
52
+ DATETIME as DATETIME,
53
+ TIMESTAMP as TIMESTAMP,
54
+ BOOLEAN as BOOLEAN,
55
+ LARGEBINARY as LARGEBINARY,
56
+ VARBINARY as VARBINARY,
57
+ ENUM as ENUM,
58
+ POINT as POINT,
59
+ )
@@ -1,8 +1,84 @@
1
1
  from . import base
2
2
  from . import mysqlconnector
3
3
 
4
- # default dialect
5
- base.dialect = dialect = mysqlconnector.dialect
4
+ from .base import BIGINT
5
+ from .base import BINARY
6
+ from .base import BIT
7
+ from .base import BLOB
8
+ from .base import BOOLEAN
9
+ from .base import CHAR
10
+ from .base import DATE
11
+ from .base import DATETIME
12
+ from .base import DECIMAL
13
+ from .base import DOUBLE
14
+ from .base import FLOAT
15
+ from .base import INTEGER
16
+ from .base import LONGBLOB
17
+ from .base import LONGTEXT
18
+ from .base import MEDIUMBLOB
19
+ from .base import MEDIUMINT
20
+ from .base import MEDIUMTEXT
21
+ from .base import NCHAR
22
+ from .base import NUMERIC
23
+ from .base import NVARCHAR
24
+ from .base import REAL
25
+ from .base import SMALLINT
26
+ from .base import TEXT
27
+ from .base import TIME
28
+ from .base import TIMESTAMP
29
+ from .base import TINYBLOB
30
+ from .base import TINYINT
31
+ from .base import TINYTEXT
32
+ from .base import VARBINARY
33
+ from .base import VARCHAR
34
+ from .base import YEAR
35
+
36
+
6
37
 
38
+ __all__ = (
39
+ "BIGINT",
40
+ "BINARY",
41
+ "BIT",
42
+ "BLOB",
43
+ "BOOLEAN",
44
+ "CHAR",
45
+ "DATE",
46
+ "DATETIME",
47
+ "DECIMAL",
48
+ "DOUBLE",
49
+ "ENUM",
50
+ "FLOAT",
51
+ "INET4",
52
+ "INET6",
53
+ "INTEGER",
54
+ "INTEGER",
55
+ "JSON",
56
+ "LONGBLOB",
57
+ "LONGTEXT",
58
+ "MEDIUMBLOB",
59
+ "MEDIUMINT",
60
+ "MEDIUMTEXT",
61
+ "NCHAR",
62
+ "NVARCHAR",
63
+ "NUMERIC",
64
+ "SET",
65
+ "SMALLINT",
66
+ "REAL",
67
+ "TEXT",
68
+ "TIME",
69
+ "TIMESTAMP",
70
+ "TINYBLOB",
71
+ "TINYINT",
72
+ "TINYTEXT",
73
+ "VARBINARY",
74
+ "VARCHAR",
75
+ "YEAR",
76
+ "dialect",
77
+ "insert",
78
+ "Insert",
79
+ "match",
80
+ )
7
81
 
8
- from .types import * # noqa: F403
82
+
83
+ # default dialect
84
+ base.dialect = dialect = mysqlconnector.dialect
@@ -10,7 +10,10 @@ if TYPE_CHECKING:
10
10
 
11
11
  from .types import (
12
12
  _NumericType,
13
+ _NumericCommonType,
13
14
  _StringType,
15
+ VARCHAR,
16
+ CHAR,
14
17
  NUMERIC,
15
18
  DECIMAL,
16
19
  DOUBLE,
@@ -36,7 +39,16 @@ from .types import (
36
39
  MEDIUMBLOB,
37
40
  LONGBLOB,
38
41
  )
39
- from ormlambda.sql.sqltypes import BLOB
42
+ from ormlambda.sql.sqltypes import (
43
+ LARGEBINARY,
44
+ BLOB,
45
+ BOOLEAN,
46
+ DATE,
47
+ UUID,
48
+ VARBINARY,
49
+ BINARY,
50
+ )
51
+
40
52
 
41
53
  from ormlambda.databases.my_sql import MySQLRepository, MySQLCaster
42
54
 
@@ -73,7 +85,7 @@ class MySQLCompiler(compiler.SQLCompiler):
73
85
  """Overridden from base SQLCompiler value"""
74
86
 
75
87
  def visit_select(self, select: Select, **kw):
76
- return f"{select.CLAUSE} {select.COLUMNS} FROM {select.FROM.query(self.dialect,**kw)}"
88
+ return f"{select.CLAUSE} {select.COLUMNS} FROM {select.FROM.query(self.dialect, **kw)}"
77
89
 
78
90
  def visit_group_by(self, groupby: GroupBy, **kw):
79
91
  column = groupby._create_query(self.dialect, **kw)
@@ -110,7 +122,7 @@ class MySQLCompiler(compiler.SQLCompiler):
110
122
  context = ClauseInfoContext(table_context=order._context._table_context, clause_context=None) if order._context else None
111
123
  for index, clause in enumerate(order._convert_into_clauseInfo(columns, context, dialect=self.dialect)):
112
124
  clause.alias_clause = None
113
- string_columns.append(f"{clause.query(self.dialect,**kw)} {str(order._order_type[index])}")
125
+ string_columns.append(f"{clause.query(self.dialect, **kw)} {str(order._order_type[index])}")
114
126
 
115
127
  return f"{order.FUNCTION_NAME()} {', '.join(string_columns)}"
116
128
 
@@ -122,7 +134,7 @@ class MySQLCompiler(compiler.SQLCompiler):
122
134
 
123
135
  class MySQLDDLCompiler(compiler.DDLCompiler):
124
136
  def get_column_specification(self, column: Column, **kwargs):
125
- colspec = column.column_name + " " + self.dialect.type_compiler_instance.process(column.dtype)
137
+ colspec = column.column_name + " " + self.dialect.type_compiler_instance.process(column.dbtype)
126
138
  default = self.get_column_default_string(column)
127
139
  if default is not None:
128
140
  colspec += " DEFAULT " + default
@@ -164,7 +176,7 @@ class MySQLTypeCompiler(compiler.GenericTypeCompiler):
164
176
  return getattr(type_, name, defaults.get(name))
165
177
 
166
178
  if attr("charset"):
167
- charset = f"CHARACTER SET {attr("charset")}"
179
+ charset = f"CHARACTER SET {attr('charset')}"
168
180
  elif attr("ascii"):
169
181
  charset = "ASCII"
170
182
  elif attr("unicode"):
@@ -184,183 +196,238 @@ class MySQLTypeCompiler(compiler.GenericTypeCompiler):
184
196
  return " ".join([c for c in ("NATIONAL", spec, collation) if c is not None])
185
197
  return " ".join([c for c in (spec, charset, collation) if c is not None])
186
198
 
187
- def visit_INTEGER(self, type_: INTEGER, **kw):
188
- if self.mysql_type(type_) and type_.display_width is not None:
199
+ def _mysql_type(self, type_, **kw):
200
+ return isinstance(type, _StringType | _NumericCommonType)
201
+
202
+ def visit_NUMERIC(self, type_: NUMERIC, **kw):
203
+ if type_.precision is None:
204
+ return self._extend_numeric(type_, "NUMERIC")
205
+ elif type_.scale is None:
189
206
  return self._extend_numeric(
190
207
  type_,
191
- f"INTEGER(%({type_.display_width})s)",
208
+ f"NUMERIC({type_.precision})",
192
209
  )
193
210
  else:
194
- return self._extend_numeric(type_, "INTEGER")
195
-
196
- def visit_VARCHAR(self, type_, **kw):
197
- if type_.length is None:
198
- raise ValueError("VARCHAR requires a length on dialect %s" % self.dialect.name)
199
- return self._extend_string(type_, {}, "VARCHAR(%d)" % type_.length)
200
-
201
- def visit_CHAR(self, type_, **kw):
202
- if type_.length is not None:
203
- return self._extend_string(type_, {}, "CHAR(%(length)s)" % {"length": type_.length})
204
- else:
205
- return self._extend_string(type_, {}, "CHAR")
206
-
207
- def visit_NUMERIC(self, type_: NUMERIC, **kw):
208
- return "NUMERIC"
211
+ return self._extend_numeric(
212
+ type_,
213
+ f"NUMERIC({type_.precision}, {type_.scale})",
214
+ )
209
215
 
210
216
  def visit_DECIMAL(self, type_: DECIMAL, **kw):
211
- return "DECIMAL"
217
+ if type_.precision is None:
218
+ return self._extend_numeric(type_, "DECIMAL")
219
+ elif type_.scale is None:
220
+ return self._extend_numeric(
221
+ type_,
222
+ f"DECIMAL({type_.precision})",
223
+ )
224
+ else:
225
+ return self._extend_numeric(
226
+ type_,
227
+ f"DECIMAL({type_.precision}, {type_.scale})",
228
+ )
212
229
 
213
230
  def visit_DOUBLE(self, type_: DOUBLE, **kw):
214
- return "DOUBLE"
231
+ if type_.precision is not None and type_.scale is not None:
232
+ return self._extend_numeric(
233
+ type_,
234
+ f"DOUBLE({type_.precision}, {type_.scale})",
235
+ )
236
+ else:
237
+ return self._extend_numeric(type_, "DOUBLE")
215
238
 
216
239
  def visit_REAL(self, type_: REAL, **kw):
217
- return "REAL"
240
+ if type_.precision is not None and type_.scale is not None:
241
+ return self._extend_numeric(
242
+ type_,
243
+ f"REAL({type_.precision}, {type_.scale})",
244
+ )
245
+ else:
246
+ return self._extend_numeric(type_, "REAL")
218
247
 
219
248
  def visit_FLOAT(self, type_: FLOAT, **kw):
220
- return "FLOAT"
249
+ if self._mysql_type(type_) and type_.scale is not None and type_.precision is not None:
250
+ return self._extend_numeric(type_, f"FLOAT({type_.precision}, {type_.scale})")
251
+ elif type_.precision is not None:
252
+ return self._extend_numeric(type_, f"FLOAT({type_.precision})")
253
+ else:
254
+ return self._extend_numeric(type_, "FLOAT")
255
+
256
+ def visit_INTEGER(self, type_: INTEGER, **kw):
257
+ if self._mysql_type(type_) and type_.display_width is not None:
258
+ return self._extend_numeric(
259
+ type_,
260
+ f"INTEGER({type_.display_width})",
261
+ )
262
+ else:
263
+ return self._extend_numeric(type_, "INTEGER")
221
264
 
222
265
  def visit_BIGINT(self, type_: BIGINT, **kw):
223
- return "BIGINT"
266
+ if self._mysql_type(type_) and type_.display_width is not None:
267
+ return self._extend_numeric(
268
+ type_,
269
+ f"BIGINT({type_.display_width})",
270
+ )
271
+ else:
272
+ return self._extend_numeric(type_, "BIGINT")
224
273
 
225
274
  def visit_MEDIUMINT(self, type_: MEDIUMINT, **kw):
226
- return "MEDIUMINT"
275
+ if self._mysql_type(type_) and type_.display_width is not None:
276
+ return self._extend_numeric(
277
+ type_,
278
+ f"MEDIUMINT({type_.display_width})",
279
+ )
280
+ else:
281
+ return self._extend_numeric(type_, "MEDIUMINT")
227
282
 
228
283
  def visit_TINYINT(self, type_: TINYINT, **kw):
229
- return "TINYINT"
284
+ if self._mysql_type(type_) and type_.display_width is not None:
285
+ return self._extend_numeric(type_, f"TINYINT({type_.display_width})")
286
+ else:
287
+ return self._extend_numeric(type_, "TINYINT")
230
288
 
231
289
  def visit_SMALLINT(self, type_: SMALLINT, **kw):
232
- return "SMALLINT"
290
+ if self._mysql_type(type_) and type_.display_width is not None:
291
+ return self._extend_numeric(
292
+ type_,
293
+ f"SMALLINT({type_.display_width})",
294
+ )
295
+ else:
296
+ return self._extend_numeric(type_, "SMALLINT")
233
297
 
234
298
  def visit_BIT(self, type_: BIT, **kw):
235
- return "BIT"
299
+ if type_.length is not None:
300
+ return f"BIT({type_.length})"
301
+ else:
302
+ return "BIT"
303
+
304
+ def visit_DATETIME(self, type_: DATETIME, **kw):
305
+ if getattr(type_, "fsp", None):
306
+ return f"DATETIME({type_.fsp})"
307
+ else:
308
+ return "DATETIME"
309
+
310
+ def visit_DATE(self, type_: DATE, **kw):
311
+ return "DATE"
236
312
 
237
313
  def visit_TIME(self, type_: TIME, **kw):
238
- return "TIME"
314
+ if getattr(type_, "fsp", None):
315
+ return f"TIME({type_.fsp})"
316
+ else:
317
+ return "TIME"
239
318
 
240
319
  def visit_TIMESTAMP(self, type_: TIMESTAMP, **kw):
241
- return "TIMESTAMP"
242
-
243
- def visit_DATETIME(self, type_: DATETIME, **kw):
244
- return "DATETIME"
320
+ if getattr(type_, "fsp", None):
321
+ return f"TIMESTAMP({type_.fsp})"
322
+ else:
323
+ return "TIMESTAMP"
245
324
 
246
325
  def visit_YEAR(self, type_: YEAR, **kw):
247
- return "YEAR"
326
+ if type_.display_width is None:
327
+ return "YEAR"
328
+ else:
329
+ return f"YEAR({type_.display_width})"
248
330
 
249
331
  def visit_TEXT(self, type_: TEXT, **kw):
250
332
  if type_.length is not None:
251
333
  return self._extend_string(type_, {}, f"TEXT({type_.length})")
252
- return self._extend_string(type_, {}, "TEXT")
334
+ else:
335
+ return self._extend_string(type_, {}, "TEXT")
253
336
 
254
337
  def visit_TINYTEXT(self, type_: TINYTEXT, **kw):
255
- return "TINYTEXT"
338
+ return self._extend_string(type_, {}, "TINYTEXT")
256
339
 
257
340
  def visit_MEDIUMTEXT(self, type_: MEDIUMTEXT, **kw):
258
- return "MEDIUMTEXT"
341
+ return self._extend_string(type_, {}, "MEDIUMTEXT")
259
342
 
260
343
  def visit_LONGTEXT(self, type_: LONGTEXT, **kw):
261
- return "LONGTEXT"
262
-
263
- def visit_NVARCHAR(self, type_: NVARCHAR, **kw):
264
- return "NVARCHAR"
265
-
266
- def visit_NCHAR(self, type_: NCHAR, **kw):
267
- return "NCHAR"
268
-
269
- def visit_TINYBLOB(self, type_: TINYBLOB, **kw):
270
- return "TINYBLOB"
271
-
272
- def visit_BLOB(self, type_: BLOB, **kw) -> str:
273
- blob = "BLOB"
274
- blob += f"({type_.length})" if type_.length is not None else ""
275
- return blob
276
-
277
- def visit_MEDIUMBLOB(self, type_: MEDIUMBLOB, **kw):
278
- return "MEDIUMBLOB"
279
-
280
- def visit_LONGBLOB(self, type_: LONGBLOB, **kw):
281
- return "LONGBLOB"
282
-
283
- # region visit lowercase
284
-
285
- def visit_integer(self, type_: INTEGER, **kw):
286
- return self.visit_INTEGER(type_, **kw)
287
-
288
- def visit_varchar(self, type_, **kw):
289
- return self.visit_VARCHAR(type_, **kw)
290
-
291
- def visit_char(self, type_, **kw):
292
- return self.visit_CHAR(type_, **kw)
293
-
294
- def visit_numeric(self, type_: NUMERIC, **kw):
295
- return self.visit_NUMERIC(type_, **kw)
296
-
297
- def visit_decimal(self, type_: DECIMAL, **kw):
298
- return self.visit_DECIMAL(type_, **kw)
299
-
300
- def visit_double(self, type_: DOUBLE, **kw):
301
- return self.visit_DOUBLE(type_, **kw)
344
+ return self._extend_string(type_, {}, "LONGTEXT")
302
345
 
303
- def visit_real(self, type_: REAL, **kw):
304
- return self.visit_REAL(type_, **kw)
305
-
306
- def visit_float(self, type_: FLOAT, **kw):
307
- return self.visit_FLOAT(type_, **kw)
308
-
309
- def visit_bigint(self, type_: BIGINT, **kw):
310
- return self.visit_BIGINT(type_, **kw)
311
-
312
- def visit_mediumint(self, type_: MEDIUMINT, **kw):
313
- return self.visit_MEDIUMINT(type_, **kw)
314
-
315
- def visit_tinyint(self, type_: TINYINT, **kw):
316
- return self.visit_TINYINT(type_, **kw)
346
+ def visit_VARCHAR(self, type_: VARCHAR, **kw):
347
+ if type_.length is not None:
348
+ return self._extend_string(type_, {}, f"VARCHAR({type_.length})")
349
+ else:
350
+ raise ValueError(f"VARCHAR requires a length on dialect {self.dialect.name}")
317
351
 
318
- def visit_smallint(self, type_: SMALLINT, **kw):
319
- return self.visit_SMALLINT(type_, **kw)
352
+ def visit_CHAR(self, type_: CHAR, **kw):
353
+ if type_.length is not None:
354
+ return self._extend_string(type_, {}, f"CHAR({type_.length})")
355
+ else:
356
+ return self._extend_string(type_, {}, "CHAR")
320
357
 
321
- def visit_bit(self, type_: BIT, **kw):
322
- return self.visit_BIT(type_, **kw)
358
+ def visit_NVARCHAR(self, type_: NVARCHAR, **kw):
359
+ # We'll actually generate the equiv. "NATIONAL VARCHAR" instead
360
+ # of "NVARCHAR".
361
+ if type_.length is not None:
362
+ return self._extend_string(
363
+ type_,
364
+ {"national": True},
365
+ f"VARCHAR({type_.length})",
366
+ )
367
+ else:
368
+ raise ValueError(f"NVARCHAR requires a length on dialect {self.dialect.name}")
323
369
 
324
- def visit_time(self, type_: TIME, **kw):
325
- return self.visit_TIME(type_, **kw)
370
+ def visit_NCHAR(self, type_: NCHAR, **kw):
371
+ # We'll actually generate the equiv.
372
+ # "NATIONAL CHAR" instead of "NCHAR".
373
+ if type_.length is not None:
374
+ return self._extend_string(
375
+ type_,
376
+ {"national": True},
377
+ f"CHAR({type_.length})",
378
+ )
379
+ else:
380
+ return self._extend_string(type_, {"national": True}, "CHAR")
326
381
 
327
- def visit_timestamp(self, type_: TIMESTAMP, **kw):
328
- return self.visit_TIMESTAMP(type_, **kw)
382
+ def visit_UUID(self, type_: UUID, **kw):
383
+ return "UUID"
329
384
 
330
- def visit_datetime(self, type_: DATETIME, **kw):
331
- return self.visit_DATETIME(type_, **kw)
385
+ def visit_VARBINARY(self, type_: VARBINARY, **kw):
386
+ return f"VARBINARY({type_.length})"
332
387
 
333
- def visit_year(self, type_: YEAR, **kw):
334
- return self.visit_YEAR(type_, **kw)
388
+ def visit_JSON(self, type_, **kw):
389
+ return "JSON"
335
390
 
336
- def visit_text(self, type_: TEXT, **kw):
337
- return self.visit_TEXT(type_, **kw)
391
+ def visit_large_binary(self, type_: LARGEBINARY, **kw):
392
+ return self.visit_BLOB(type_)
338
393
 
339
- def visit_tinytext(self, type_: TINYTEXT, **kw):
340
- return self.visit_TINYTEXT(type_, **kw)
394
+ def visit_enum(self, type_, **kw):
395
+ if not type_.native_enum:
396
+ return super().visit_enum(type_)
397
+ else:
398
+ return self.visit_ENUM(type_, type_.enums)
341
399
 
342
- def visit_mediumtext(self, type_: MEDIUMTEXT, **kw):
343
- return self.visit_MEDIUMTEXT(type_, **kw)
400
+ def visit_BLOB(self, type_: BLOB, **kw):
401
+ if type_.length is not None:
402
+ return f"BLOB({type_.length})"
403
+ else:
404
+ return "BLOB"
344
405
 
345
- def visit_longtext(self, type_: LONGTEXT, **kw):
346
- return self.visit_LONGTEXT(type_, **kw)
406
+ def visit_TINYBLOB(self, type_: TINYBLOB, **kw):
407
+ return "TINYBLOB"
347
408
 
348
- def visit_nvarchar(self, type_: NVARCHAR, **kw):
349
- return self.visit_NVARCHAR(type_, **kw)
409
+ def visit_MEDIUMBLOB(self, type_: MEDIUMBLOB, **kw):
410
+ return "MEDIUMBLOB"
350
411
 
351
- def visit_nchar(self, type_: NCHAR, **kw):
352
- return self.visit_NCHAR(type_, **kw)
412
+ def visit_LONGBLOB(self, type_: LONGBLOB, **kw):
413
+ return "LONGBLOB"
353
414
 
354
- def visit_tinyblob(self, type_: TINYBLOB, **kw):
355
- return self.visit_TINYBLOB(type_, **kw)
415
+ def _visit_enumerated_values(self, name, type_, enumerated_values):
416
+ quoted_enums = []
417
+ for e in enumerated_values:
418
+ if self.dialect.identifier_preparer._double_percents:
419
+ e = e.replace("%", "%%")
420
+ quoted_enums.append(f"'{e.replace("'", "''")}'")
421
+ return self._extend_string(type_, {}, f"{name}({','.join(quoted_enums)})")
356
422
 
357
- def visit_mediumblob(self, type_: MEDIUMBLOB, **kw):
358
- return self.visit_MEDIUMBLOB(type_, **kw)
423
+ def visit_ENUM(self, type_, **kw):
424
+ return self._visit_enumerated_values("ENUM", type_, type_.enums)
359
425
 
360
- def visit_longblob(self, type_: LONGBLOB, **kw):
361
- return self.visit_LONGBLOB(type_, **kw)
426
+ def visit_SET(self, type_, **kw):
427
+ return self._visit_enumerated_values("SET", type_, type_.values)
362
428
 
363
- # endregion
429
+ def visit_BOOLEAN(self, type_: BOOLEAN, **kw):
430
+ return "BOOL"
364
431
 
365
432
 
366
433
  class MySQLDialect(default.DefaultDialect):
@@ -32,7 +32,7 @@ class _NumericType(_NumericCommonType, sqltypes.Numeric): ...
32
32
  class _FloatType(_NumericCommonType, sqltypes.Float):
33
33
  def __init__(self, precision=None, scale=None, asdecimal=True, **kw):
34
34
  if isinstance(self, (REAL, DOUBLE)) and ((precision is None and scale is not None) or (precision is not None and scale is None)):
35
- raise AttributeError("You must specify both precision and scale or omit " "both altogether.")
35
+ raise AttributeError("You must specify both precision and scale or omit both altogether.")
36
36
  super().__init__(precision=precision, asdecimal=asdecimal, **kw)
37
37
  self.scale = scale
38
38
 
@@ -42,6 +42,9 @@ class _IntegerType(_NumericCommonType, sqltypes.Integer):
42
42
  self.display_width = display_width
43
43
  super().__init__(**kw)
44
44
 
45
+ def __repr__(self):
46
+ return f"{type(self).__name__}({self.display_width})"
47
+
45
48
 
46
49
  class _StringType(sqltypes.String):
47
50
  """Base for MySQL string types."""
@@ -9,6 +9,7 @@ from ormlambda.sql.types import (
9
9
  TableType,
10
10
  ColumnType,
11
11
  AliasType,
12
+ TypeEngine,
12
13
  )
13
14
  from .interface import IClauseInfo
14
15
  from ormlambda.sql import ForeignKey
@@ -166,6 +167,18 @@ class ClauseInfo[T: Table](IClauseInfo[T]):
166
167
  return self._column
167
168
  return type(self._column)
168
169
 
170
+ @property
171
+ def dbtype(self)->tp.Optional[TypeEngine]:
172
+ if self._dtype is not None:
173
+ return self._dtype
174
+
175
+ if isinstance(self._column, Column):
176
+ return self._column.dbtype
177
+
178
+ if isinstance(self._column, type):
179
+ return self._column
180
+ return type(self._column)
181
+
169
182
  def query(self, dialect: Dialect, **kwargs) -> str:
170
183
  return self._create_query(dialect, **kwargs)
171
184
 
@@ -1,14 +1,14 @@
1
1
  from __future__ import annotations
2
2
  import abc
3
- from typing import Annotated, Any, Iterable, Type, Optional, TYPE_CHECKING, get_type_hints, overload, get_origin, get_args
3
+ from typing import Annotated, Any, ClassVar, Iterable, Type, Optional, TYPE_CHECKING, get_type_hints, overload, get_origin, get_args
4
4
  from ormlambda.sql.types import TableType, ComparerType, ColumnType
5
5
  from ormlambda import ConditionType
6
+ from ormlambda.sql.type_api import TypeEngine
6
7
 
7
8
  if TYPE_CHECKING:
8
9
  import re
9
10
  from ormlambda import Table
10
11
  from ormlambda.sql.comparer import Comparer, Regex, Like
11
- from ormlambda.sql.type_api import TypeEngine
12
12
 
13
13
 
14
14
  from ormlambda.types import (
@@ -24,10 +24,13 @@ from ormlambda.types import (
24
24
 
25
25
 
26
26
  class Column[TProp]:
27
- PRIVATE_CHAR: str = "_"
27
+ PRIVATE_CHAR: ClassVar[str] = "_"
28
+
29
+ _dbtype: Optional[TypeEngine]
28
30
 
29
31
  __slots__ = (
30
- "dtype",
32
+ "_dtype",
33
+ "_dbtype",
31
34
  "column_name",
32
35
  "table",
33
36
  "is_primary_key",
@@ -111,7 +114,8 @@ class Column[TProp]:
111
114
 
112
115
  def __set__(self, obj, value):
113
116
  if self._check and value is not None:
114
- if not isinstance(value, self.dtype):
117
+ type_ = self.dtype if not isinstance(self.dtype, TypeEngine) else self.dtype.python_type
118
+ if not isinstance(value, type_):
115
119
  raise ValueError(f"The '{self.column_name}' Column from '{self.table.__table_name__}' table expected '{str(self.dtype)}' type. You passed '{type(value).__name__}' type")
116
120
  setattr(obj, self.__private_name, value)
117
121
 
@@ -132,8 +136,6 @@ class Column[TProp]:
132
136
  def _fill_from_annotations[T: Table](self, obj: Type[T], name: str) -> None:
133
137
  """Read the metada when using Annotated typing class, and set the attributes accordingly"""
134
138
 
135
- from ormlambda.sql.type_api import TypeEngine
136
-
137
139
  annotations = get_type_hints(obj, include_extras=True)
138
140
  if name in annotations:
139
141
  annotation = annotations[name]
@@ -163,6 +165,23 @@ class Column[TProp]:
163
165
  self.is_not_null = True
164
166
  return None
165
167
 
168
+ @property
169
+ def dtype(self) -> Type[TProp]:
170
+ return self._dtype
171
+
172
+ @dtype.setter
173
+ def dtype(self, value: Type[TProp] | TypeEngine) -> None:
174
+ if isinstance(value, TypeEngine):
175
+ self._dtype = value.python_type
176
+ self._dbtype = value
177
+ else:
178
+ self._dtype = value
179
+ self._dbtype = None
180
+
181
+ @property
182
+ def dbtype(self) -> TypeEngine[TProp]:
183
+ return self._dbtype if self._dbtype else self.dtype
184
+
166
185
  @abc.abstractmethod
167
186
  def __comparer_creator(self, other: ColumnType, compare: ComparerType) -> Comparer:
168
187
  from ormlambda.sql.comparer import Comparer
ormlambda/sql/sqltypes.py CHANGED
@@ -10,6 +10,7 @@ import ormlambda.util as util
10
10
  from uuid import UUID as _python_UUID
11
11
  from shapely import Point as _python_Point
12
12
 
13
+
13
14
  class _NoArg(enum.Enum):
14
15
  NO_ARG = 0
15
16
 
@@ -391,7 +392,7 @@ class Enum(String, TypeEngine[EnumType]):
391
392
  enum_args = get_args(pt)
392
393
  bad_args = [arg for arg in enum_args if not isinstance(arg, str)]
393
394
  if bad_args:
394
- raise ValueError(f"Can't create string-based Enum datatype from non-string " f"values: {', '.join(repr(x) for x in bad_args)}. Please " f"provide an explicit Enum datatype for this Python type")
395
+ raise ValueError(f"Can't create string-based Enum datatype from non-string values: {', '.join(repr(x) for x in bad_args)}. Please provide an explicit Enum datatype for this Python type")
395
396
  native_enum = False
396
397
  return enum_args, native_enum
397
398
 
@@ -407,7 +408,7 @@ class Enum(String, TypeEngine[EnumType]):
407
408
  elif util.is_pep695(python_type):
408
409
  value = python_type.__value__
409
410
  if not util.is_literal(value):
410
- raise ValueError(f"Can't associate TypeAliasType '{python_type}' to an " "Enum since it's not a direct alias of a Literal. Only " "aliases in this form `type my_alias = Literal['a', " "'b']` are supported when generating Enums.")
411
+ raise ValueError(f"Can't associate TypeAliasType '{python_type}' to an Enum since it's not a direct alias of a Literal. Only aliases in this form `type my_alias = Literal['a', 'b']` are supported when generating Enums.")
411
412
  enum_args, native_enum = process_literal(value)
412
413
 
413
414
  elif isinstance(python_type, type) and issubclass(python_type, enum.Enum):
@@ -439,13 +440,15 @@ class Enum(String, TypeEngine[EnumType]):
439
440
 
440
441
  self._valid_lookup.update([(value, self._valid_lookup[self._object_lookup[value]]) for value in values])
441
442
 
443
+
442
444
  class Point(TypeEngine[_python_Point]):
443
- __visit_name__ = 'point'
445
+ __visit_name__ = "point"
444
446
 
445
447
  @property
446
- def python_type(self)->Type[_python_Point]:
448
+ def python_type(self) -> Type[_python_Point]:
447
449
  return _python_Point
448
-
450
+
451
+
449
452
  class JSON(TypeEngine[Any]):
450
453
  """JSON data type."""
451
454
 
@@ -558,6 +561,12 @@ class BLOB(LargeBinary):
558
561
  __visit_name__ = "BLOB"
559
562
 
560
563
 
564
+ class BINARY(_Binary):
565
+ """The SQL BINARY type."""
566
+
567
+ __visit_name__ = "BINARY"
568
+
569
+
561
570
  class VARCHAR(String):
562
571
  __visit_name__ = "VARCHAR"
563
572
 
@@ -597,6 +606,7 @@ class VARBINARY(Varbinary):
597
606
  class ENUM(Enum):
598
607
  __visit_name__ = "ENUM"
599
608
 
609
+
600
610
  class POINT(Point):
601
611
  __visit_name__ = "POINT"
602
612
 
@@ -623,13 +633,15 @@ _type_dicc: dict[Any, TypeEngine[Any]] = {
623
633
  float: Float(),
624
634
  NoneType: NULLTYPE,
625
635
  dt.datetime: Datetime(timezone=False),
636
+ dt.date: DATE(),
626
637
  bytes: _BINARY,
627
638
  bytearray: _BINARY,
628
639
  bool: Boolean(),
629
640
  enum.Enum: _ENUM,
630
641
  Literal: _ENUM,
631
642
  _python_UUID: UUID(),
632
- _python_Point: POINT()
643
+ _python_Point: POINT(),
644
+ decimal.Decimal: DECIMAL(),
633
645
  }
634
646
  # enderegion
635
647
 
ormlambda/sql/type_api.py CHANGED
@@ -33,3 +33,6 @@ class TypeEngine[T: Any](Element, abc.ABC):
33
33
  if _coerced_type is NULLTYPE:
34
34
  return self
35
35
  return _coerced_type
36
+
37
+ def __repr__(self):
38
+ return f"{TypeEngine.__name__}: {super().__repr__()}"
ormlambda/sql/visitors.py CHANGED
@@ -72,3 +72,6 @@ class Element:
72
72
  # return visitor.visit_unsupported_compilation(self, err, **kw) # type: ignore # noqa: E501
73
73
 
74
74
  cls._compiler_dispatch = _compiler_dispatch
75
+
76
+ def __repr__(self):
77
+ return f"{Element.__name__}: {self.__visit_name__}"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ormlambda
3
- Version: 3.34.1
3
+ Version: 3.34.6
4
4
  Summary: ORM designed to interact with the database (currently with MySQL) using lambda functions and nested functions
5
5
  Author: p-hzamora
6
6
  Author-email: p.hzamora@icloud.com
@@ -1,4 +1,4 @@
1
- ormlambda/__init__.py,sha256=0s-uDYwzCsRAdN1z_dfzi6rb-9EiR3wcbSgQNJHxlyE,772
1
+ ormlambda/__init__.py,sha256=uH9p7xjLHjQ8Dii3h6HWzWpB6pyKKpBvzAQgJ5klp0M,1505
2
2
  ormlambda/caster/__init__.py,sha256=lXzW6TJwgtyFTaBhKBETyGIxEje005S3SPg7PXnYEVQ,137
3
3
  ormlambda/caster/base_caster.py,sha256=c3vCGoKDyJ39kfUiS7sNKhKdjBRYSK1Ie88lwDIXqgE,1774
4
4
  ormlambda/caster/caster.py,sha256=2iilDARnNGGh-NVJKXr7nhxjoaQOLJGk3RnGoSzHpOY,2453
@@ -60,10 +60,10 @@ ormlambda/dialects/default/__init__.py,sha256=N1B0LKEDu7r2-mTF9mBA4ReyhYeDuJDnbQ
60
60
  ormlambda/dialects/default/base.py,sha256=rmK-nV5luN2lphSgisM8TBy3s4JXVgvZa79ATHeVOuU,1088
61
61
  ormlambda/dialects/interface/__init__.py,sha256=IbB5o6ae7XNVVfNt4uqZRZcVK9HHG6eLLK_E0J99kwY,43
62
62
  ormlambda/dialects/interface/dialect.py,sha256=z47GXCi-uJ6Oq7Zm64OHM0q61T3xWCss-uGdcuuXng0,2810
63
- ormlambda/dialects/mysql/__init__.py,sha256=y4gh32agzxxpjuC6x3oKEuCJYBwMYYuvJ-xxBS1rbv4,152
64
- ormlambda/dialects/mysql/base.py,sha256=ocDJprw5PFhZ9vsrrUJdBoqjwRwa3V163wK1p7j-gpc,11688
63
+ ormlambda/dialects/mysql/__init__.py,sha256=1yAEM2MDcp56eSv1MrbtuDVQ5d_EzQ0S7ztSXKVS-eo,1511
64
+ ormlambda/dialects/mysql/base.py,sha256=drHxKpFVVJPWy82rWSD2rFKvQ-GN8imoDmp3_ou4hcw,14535
65
65
  ormlambda/dialects/mysql/mysqlconnector.py,sha256=DOjK7U7LOhjuVQJULzAV8xaRGX0OlBU8APUeRiTcbDY,1378
66
- ormlambda/dialects/mysql/types.py,sha256=lOhdi442saxQJlQ_jgChBETE0fFMGTQLWN0a76R4fkc,23453
66
+ ormlambda/dialects/mysql/types.py,sha256=ZQ75KyUwmNAt48F48CeIiMHgQjO0JeDXrksrrss2ILg,23537
67
67
  ormlambda/dialects/sqlite/__init__.py,sha256=2EMmxD7ORtGoD18GJ_9aC_tPutAq9ormMZJmaJ5nkYA,103
68
68
  ormlambda/dialects/sqlite/base.py,sha256=24LSB461yQDEnXa-TsQt_srJmBCAR6M6419pa40CL_4,1503
69
69
  ormlambda/dialects/sqlite/pysqlite.py,sha256=dZY0NV2IvSTk3DNEDyncstmIABKj3M2xfmhf2dc2zQs,680
@@ -85,7 +85,7 @@ ormlambda/repository/response.py,sha256=TBmWUGGoZChjXVhZXU3F84eUUA1rf7VPeNi92Ae8
85
85
  ormlambda/sql/__init__.py,sha256=0CWQzfxhTRWXozoRsg460o_ZwjW9w4uyL5jQUcD4eHg,121
86
86
  ormlambda/sql/clause_info/__init__.py,sha256=wOr8M0ZW7fJ-OGHok9eExh7fVP6kiwqGckEsKdE0U7A,255
87
87
  ormlambda/sql/clause_info/aggregate_function_base.py,sha256=FCTrXhDvO_XDVBXZK5AiUHonDENHT_MFxW6kn2_--6E,3541
88
- ormlambda/sql/clause_info/clause_info.py,sha256=ypiBjvuij4iKdMAkTPy7jlCa7hZIEi11agsaVCoHtKM,12674
88
+ ormlambda/sql/clause_info/clause_info.py,sha256=OdqAwJgrB66Ypwh_PLFHnkZQk4gHPXXrsn-sEXpmHxo,13014
89
89
  ormlambda/sql/clause_info/clause_info_context.py,sha256=Y32p3x4mqcdNHbAowrKaN_ldnGn7zvaP1UdAkWWryhM,2852
90
90
  ormlambda/sql/clause_info/interface/IAggregate.py,sha256=P-QPaTMAMHVR5M9-ClmL8wsj0uNGG5xpxjuAwWnzKxA,216
91
91
  ormlambda/sql/clause_info/interface/IClauseInfo.py,sha256=Gnr6vArgimCt0oJqW-_q2_5jMi8EDBW1wZ8rEF2uzTk,921
@@ -114,7 +114,7 @@ ormlambda/sql/clauses/update.py,sha256=s4Vl_CmZzUYvdP2ThrTDz2ps_nU3_uwROQNICD06q
114
114
  ormlambda/sql/clauses/upsert.py,sha256=1MlHSzIVXSnCnXx_0HqX_-JT1FAxaIKHImrv-OKjnWU,2250
115
115
  ormlambda/sql/clauses/where.py,sha256=Db1IpXnbqAwzZpaThFDSxkrCNLsop__aJ43cDtcZ2Sk,2129
116
116
  ormlambda/sql/column/__init__.py,sha256=MwBJznOoeNE7tK9oppVtC3784XPkKo0yq9VLlh2lWUY,41
117
- ormlambda/sql/column/column.py,sha256=onneRMZJJd85sFkJZhgfcLTCChC-vHPcS7oukt-eSUY,7380
117
+ ormlambda/sql/column/column.py,sha256=kDmLn5yxpYqPy98FyohHiHtHKDgF3LiERnfQYAnbNmE,7966
118
118
  ormlambda/sql/comparer.py,sha256=_KQ6gYe0eUg5aOL1i70mV8a1Qa1jI5UQr-0VfAMgVzk,5482
119
119
  ormlambda/sql/compiler.py,sha256=dSmF8RYQECPweKZI_TB9stDkLXTJy9FYvAQMNSCY8AY,12967
120
120
  ormlambda/sql/ddl.py,sha256=WhVoG_zlTT9DVZlnokHB8B1SC4ZXQSrXTQJ6vlnlpEY,1644
@@ -126,14 +126,14 @@ ormlambda/sql/functions/max.py,sha256=wXNhFmOlMyD1sKcaBO9asQeXcfY0OvMQPkrlARVXgd
126
126
  ormlambda/sql/functions/min.py,sha256=MTRGXgwzbLgKk9PJpn6JGK1TU91hZe_VsTszngLOh7Q,1570
127
127
  ormlambda/sql/functions/sum.py,sha256=_TN8SitDDBVD6VlsHZH3RpmiE5hkFrGGgC3tuEctKjk,1614
128
128
  ormlambda/sql/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
129
- ormlambda/sql/sqltypes.py,sha256=-YyzA-gPTnARoK29bMS6R3GCNvD4LTKvGH1Ev6YfUEY,14207
129
+ ormlambda/sql/sqltypes.py,sha256=1Ir_Wxz1VjnNOh2zzIf7P7w4tbY0ljlubZMoAEiUkYU,14333
130
130
  ormlambda/sql/table/__init__.py,sha256=LLZcMLjwFxgBCPhdm5UQYYZDIbtYWCKPO9CbcXsy5zI,50
131
131
  ormlambda/sql/table/fields.py,sha256=ovNR3bJ473aKW-2NhbKr0iJlVpgW06jurHLob2IyZU8,2116
132
132
  ormlambda/sql/table/table.py,sha256=U8ByMDEEXIOwfPykDcjzoyNV1WEieN40Zk18EXveX60,6594
133
133
  ormlambda/sql/table/table_constructor.py,sha256=c3Z-1El0onSClYBmAatoUBYUOT70tITVqtsDJMxZ9QU,1092
134
- ormlambda/sql/type_api.py,sha256=FdEUBpdPbHIag--OU3mHY8vxMqyyzTMYygDBCUvGlcU,980
134
+ ormlambda/sql/type_api.py,sha256=vepqMw6hR4zVhhNIYbtX491iVsxAfwHiSiD3uYQUGTE,1067
135
135
  ormlambda/sql/types.py,sha256=lQxC5gbhDPRckGSRJZ4rYSZr-XUvIMMH8WfkN1wtM1g,844
136
- ormlambda/sql/visitors.py,sha256=FJdnaiF86xK699Kq1qfHFSPtmeVsDX_PaUNHAoP8Tck,1967
136
+ ormlambda/sql/visitors.py,sha256=AHXxCEOME8-b_-YhFEzBv-Jd15vSoX5F-jEGrPPNRLE,2052
137
137
  ormlambda/statements/__init__.py,sha256=kmd5K0dXiLQ2xTYlhd2yXeG2l955RtpDkrAIkZlfad8,148
138
138
  ormlambda/statements/base_statement.py,sha256=Dw89C2-eSwtuNc8UqzQMkoVuykf0gIWyA81YoXu6CQQ,4985
139
139
  ormlambda/statements/interfaces/IStatements.py,sha256=b2eQBeEoRivehhopmm-gca6qRSZ3ld06YMEHQEVP1d8,12824
@@ -150,8 +150,8 @@ ormlambda/util/module_tree/dfs_traversal.py,sha256=lSF03G63XtJFLp03ueAmsHMBvhUkj
150
150
  ormlambda/util/module_tree/dynamic_module.py,sha256=vJOqvm5-WKksudXBK1IcTn4WsuLxt1qXNISq-_21sy4,8705
151
151
  ormlambda/util/plugin_loader.py,sha256=p6WLn-MF1bQd2i2GHy98WQjNKmbQdqIUyTFIcBIHC5M,1034
152
152
  ormlambda/util/typing.py,sha256=Z7irz53ui0hoFnJTe06NX4JPl60_thgdjJIxh5gjGGk,169
153
- ormlambda-3.34.1.dist-info/AUTHORS,sha256=uWpOHaCPTOLbVkk5x9McoLwbgzSeCg7yILeDRyMGWGM,606
154
- ormlambda-3.34.1.dist-info/LICENSE,sha256=xBprFw8GJLdHMOoUqDk0427EvjIcbEREvXXVFULuuXU,1080
155
- ormlambda-3.34.1.dist-info/METADATA,sha256=q1FalyMLHthaHwhaCbL2lzNdIq6pkyZdvoVQSeb-Tas,13326
156
- ormlambda-3.34.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
157
- ormlambda-3.34.1.dist-info/RECORD,,
153
+ ormlambda-3.34.6.dist-info/AUTHORS,sha256=uWpOHaCPTOLbVkk5x9McoLwbgzSeCg7yILeDRyMGWGM,606
154
+ ormlambda-3.34.6.dist-info/LICENSE,sha256=xBprFw8GJLdHMOoUqDk0427EvjIcbEREvXXVFULuuXU,1080
155
+ ormlambda-3.34.6.dist-info/METADATA,sha256=dLX5_MrsxoSEMj42Ummw8Cnv7E8829nPkqxLQCH6V4o,13326
156
+ ormlambda-3.34.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
157
+ ormlambda-3.34.6.dist-info/RECORD,,