1. support mypy check, tox -e mypy or tox -e pep8 2. fix error of mypy check Change-Id: I41b0013d271f3c7d3a28e1ea6dd0b083893d8983
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
# Copyright 2021 99cloud
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import annotations
|
|
|
|
from oslo_policy import _parser
|
|
|
|
from .base import Enforcer, UserContext
|
|
from .manager import get_service_rules
|
|
|
|
ENFORCER = Enforcer()
|
|
|
|
|
|
def setup() -> None:
|
|
service_rules = get_service_rules()
|
|
all_api_rules = []
|
|
for service, rules in service_rules.items():
|
|
api_rules = []
|
|
for rule in rules:
|
|
# Update rule name with prefix service.
|
|
rule.name = f"{service}:{rule.name}"
|
|
# Update check
|
|
rule.check_str = rule.check_str.replace("rule:", f"rule:{service}:")
|
|
rule.check = _parser.parse_rule(rule.check_str)
|
|
# Update basic check
|
|
rule.basic_check_str = rule.basic_check_str.replace("rule:", f"rule:{service}:")
|
|
rule.basic_check = _parser.parse_rule(rule.basic_check_str)
|
|
api_rules.append(rule)
|
|
all_api_rules.extend(api_rules)
|
|
|
|
ENFORCER.register_rules(all_api_rules)
|
|
|
|
|
|
__all__ = (
|
|
"ENFORCER",
|
|
"UserContext",
|
|
"setup",
|
|
)
|