mirror of
https://gitlab.com/flectra-community/reporting-engine.git
synced 2024-11-15 02:32:05 +00:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
# Copyright 2015-2018 Onestein (<http://www.onestein.eu>)
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
|
|
|
|
import logging
|
|
|
|
from flectra import SUPERUSER_ID
|
|
from flectra import api, modules
|
|
|
|
from flectra.tools import existing_tables, topological_sort
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _bi_view(_name):
|
|
return _name[0:6] == 'x_bve.'
|
|
|
|
|
|
def post_load():
|
|
|
|
def check_tables_exist(self, cr):
|
|
"""
|
|
Verify that all tables are present and try to initialize
|
|
those that are missing.
|
|
"""
|
|
# This monkey patch is meant to avoid that the _logger writes
|
|
# warning and error messages, while running an update all,
|
|
# in case the model is a bi-view-generated model.
|
|
|
|
env = api.Environment(cr, SUPERUSER_ID, {})
|
|
table2model = {
|
|
model._table: name for name, model in env.items()
|
|
if not model._abstract and not _bi_view(name) # here is the patch
|
|
}
|
|
missing_tables = set(table2model).difference(
|
|
existing_tables(cr, table2model))
|
|
|
|
if missing_tables:
|
|
missing = {table2model[table] for table in missing_tables}
|
|
_logger.warning("Models have no table: %s.", ", ".join(missing))
|
|
# recreate missing tables following model dependencies
|
|
deps = {name: model._depends for name, model in env.items()}
|
|
for name in topological_sort(deps):
|
|
if name in missing:
|
|
_logger.info("Recreate table of model %s.", name)
|
|
env[name].init()
|
|
# check again, and log errors if tables are still missing
|
|
missing_tables = set(table2model).difference(
|
|
existing_tables(cr, table2model))
|
|
for table in missing_tables:
|
|
_logger.error("Model %s has no table.", table2model[table])
|
|
|
|
modules.registry.Registry.check_tables_exist = check_tables_exist
|
|
|
|
|
|
def uninstall_hook(cr, registry):
|
|
# delete dirty data that could cause problems
|
|
# while re-installing the module
|
|
cr.execute("""
|
|
delete from ir_model where model like 'x_bve.%'
|
|
""")
|
|
cr.execute("""
|
|
delete from bve_view where model_name like 'x_bve.%'
|
|
""")
|
|
cr.execute("""
|
|
SELECT 'DROP VIEW ' || table_name
|
|
FROM information_schema.views
|
|
WHERE table_schema NOT IN ('pg_catalog', 'information_schema')
|
|
AND table_name like 'x_bve_%'
|
|
""")
|
|
results = list(cr.fetchall())
|
|
for result in results:
|
|
cr.execute(result[0])
|