diff --git a/muk_dbfilter/patch/http.py b/muk_dbfilter/patch/http.py index 243f68a..fdd89c5 100644 --- a/muk_dbfilter/patch/http.py +++ b/muk_dbfilter/patch/http.py @@ -1,35 +1,36 @@ -################################################################################### -# -# Copyright (C) 2018 MuK IT GmbH -# -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU Affero General Public License as -# published by the Free Software Foundation, either version 3 of the -# License, or (at your option) any later version. -# -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU Affero General Public License for more details. -# -# You should have received a copy of the GNU Affero General Public License -# along with this program. If not, see . -# -################################################################################### - -import random -import logging - -from odoo import http, tools -from odoo.http import request - -from odoo.addons.muk_utils.tools.patch import monkey_patch - -_logger = logging.getLogger(__name__) - -@monkey_patch(http) -def db_filter(dbs, httprequest=None): - httprequest = httprequest or request.httprequest - dbs = db_filter.super(dbs, httprequest=httprequest) - filter = httprequest.environ.get('HTTP_X_ODOO_DBFILTER') +################################################################################### +# +# Copyright (C) 2018 MuK IT GmbH +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . +# +################################################################################### + +import random +import re +import logging + +from odoo import http, tools +from odoo.http import request + +from odoo.addons.muk_utils.tools.patch import monkey_patch + +_logger = logging.getLogger(__name__) + +@monkey_patch(http) +def db_filter(dbs, httprequest=None): + httprequest = httprequest or request.httprequest + dbs = db_filter.super(dbs, httprequest=httprequest) + filter = httprequest.environ.get('HTTP_X_ODOO_DBFILTER') return [db for db in dbs if re.match(filter, db)] if filter else dbs \ No newline at end of file