1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
"""
All the unique changes for the databases package
with the custom Numeric as the deprecated pypostgresql
for backwards compatibility and to make sure the
package can go to SQLAlchemy 2.0+.
"""
import typing
from sqlalchemy import types, util
from sqlalchemy.dialects.postgresql.base import PGDialect, PGExecutionContext
from sqlalchemy.engine import processors
from sqlalchemy.types import Float, Numeric
class PGExecutionContext_psycopg(PGExecutionContext):
...
class PGNumeric(Numeric):
def bind_processor(
self, dialect: typing.Any
) -> typing.Union[str, None]: # pragma: no cover
return processors.to_str
def result_processor(
self, dialect: typing.Any, coltype: typing.Any
) -> typing.Union[float, None]: # pragma: no cover
if self.asdecimal:
return None
else:
return processors.to_float
class PGDialect_psycopg(PGDialect):
colspecs = util.update_copy(
PGDialect.colspecs,
{
types.Numeric: PGNumeric,
types.Float: Float,
},
)
execution_ctx_cls = PGExecutionContext_psycopg
dialect = PGDialect_psycopg
|
