Skip to content

Commit 741da20

Browse files
committed
Fix database functions for SQLAlchemy 1
1 parent ba9afaf commit 741da20

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

isimip_publisher/utils/database.py

+11-4
Original file line numberDiff line numberDiff line change
@@ -740,8 +740,7 @@ def update_views(session):
740740

741741

742742
def update_words_view(session):
743-
engine = session.get_bind()
744-
if 'words' in inspect(engine).get_materialized_view_names():
743+
if 'words' in get_materialized_view_names(session):
745744
session.connection().execute(text('''
746745
REFRESH MATERIALIZED VIEW words
747746
'''))
@@ -757,8 +756,7 @@ def update_words_view(session):
757756

758757

759758
def update_identifiers_view(session):
760-
engine = session.get_bind()
761-
if 'identifiers' in inspect(engine).get_materialized_view_names():
759+
if 'identifiers' in get_materialized_view_names(session):
762760
session.connection().execute(text('''
763761
REFRESH MATERIALIZED VIEW identifiers
764762
'''))
@@ -777,3 +775,12 @@ def update_identifiers_view(session):
777775
CREATE INDEX ON identifiers(identifier)
778776
'''))
779777
logger.debug('create identifiers view')
778+
779+
780+
def get_materialized_view_names(session):
781+
engine = session.get_bind()
782+
try:
783+
return inspect(engine).get_materialized_view_names()
784+
except AttributeError:
785+
# for SQLAlchemy < 2
786+
return inspect(engine).get_view_names()

0 commit comments

Comments
 (0)