37 lines
862 B
Python
37 lines
862 B
Python
|
"""
|
||
|
|
Provides missing Postgresql types.
|
||
|
|
|
||
|
|
Use GeoAlchemy2 to support Postgis types.
|
||
|
|
Implement mockups for Point and LTree type, but generated models won't work out
|
||
|
|
of the box: a better way would be to use a package implementing proper classes
|
||
|
|
which could be imported in the target project.
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
from sqlalchemy.dialects.postgresql.base import ischema_names, PGTypeCompiler
|
||
|
|
from sqlalchemy import types as sqltypes
|
||
|
|
|
||
|
|
try:
|
||
|
|
import geoalchemy2
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
class LTREE(sqltypes.TypeEngine):
|
||
|
|
"""Postgresql LTREE type mockup."""
|
||
|
|
|
||
|
|
__visit_name__ = 'LTREE'
|
||
|
|
|
||
|
|
|
||
|
|
class POINT(sqltypes.TypeEngine):
|
||
|
|
"""Postgresql POINT type mockup."""
|
||
|
|
|
||
|
|
__visit_name__ = 'POINT'
|
||
|
|
|
||
|
|
|
||
|
|
ischema_names['ltree'] = LTREE
|
||
|
|
ischema_names['point'] = POINT
|
||
|
|
|
||
|
|
PGTypeCompiler.visit_LTREE = lambda self, type_: 'LTREE'
|
||
|
|
PGTypeCompiler.visit_POINT = lambda self, type_: 'POINT'
|