diff --git a/CenRa_FLUX/flux_editor.py b/CenRa_FLUX/flux_editor.py index e07aa33..d89cb73 100644 --- a/CenRa_FLUX/flux_editor.py +++ b/CenRa_FLUX/flux_editor.py @@ -231,12 +231,20 @@ class Flux_Editor(QDialog, EDITOR_CLASS): cur.execute(custom_list) list_schema = cur.fetchall() - SQLraster = """SELECT concat(r_table_schema,'.',r_table_name) from public.raster_columns; """ - cur.execute(SQLraster) - list_raster = cur.fetchall() - RasterList = [] - for rasterFind in list_raster: - RasterList.append(rasterFind[0]) + SQLcountRaster = """SELECT schemaname,viewname FROM pg_catalog.pg_views + WHERE schemaname LIKE 'public' AND viewname LIKE 'raster_columns';""" + cur.execute(SQLcountRaster) + RasterIF = len(cur.fetchall()) + + if RasterIF == 1: + SQLloadRaster = """SELECT concat(r_table_schema,'.',r_table_name) from public.raster_columns; """ + cur.execute(SQLloadRaster) + list_raster = cur.fetchall() + RasterList = [] + for rasterFind in list_raster: + RasterList.append(rasterFind[0]) + else: + RasterList = [] self.tableWidget.setRowCount(len(list_schema)) self.tableWidget.setColumnCount(4) @@ -374,8 +382,8 @@ class Flux_Editor(QDialog, EDITOR_CLASS): # def displayOnWindows(type, uri, name): # p = [] - SQLraster = """SELECT concat(r_table_schema,'.',r_table_name) from public.raster_columns; """ - cur.execute(SQLraster) + SQLloadRaster = """SELECT concat(r_table_schema,'.',r_table_name) from public.raster_columns; """ + cur.execute(SQLloadRaster) list_raster = cur.fetchall() RasterList = [] for rasterFind in list_raster: