module.py

#!/usr/bin/python3
# coding=utf-8

#   Copyright 2021 getcarrier.io
#
#   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.

""" Module """

from pylon.core.tools import log  # pylint: disable=E0611,E0401
from pylon.core.tools import module  # pylint: disable=E0611,E0401

from .init_db import init_db
from tools import theme, shared


class Module(module.ModuleModel):
    """ Task module """

    def __init__(self, context, descriptor):
        self.context = context
        self.descriptor = descriptor

    def init(self):
        """ Init module """
        log.info('Initializing module')
        init_db()

        self.descriptor.init_api()
        self.descriptor.init_rpcs()
        self.descriptor.init_blueprint()
        self.descriptor.init_slots()

        theme.register_section(
            "security",
            "Security",
            kind="holder",
            location="left",
            permissions={
                "permissions": ["security"],
                "recommended_roles": {
                    "default": {"admin": True, "editor": True, "viewer": True},
                }
            }
        )
        theme.register_subsection(
            "security", "app",
            "App",
            title="Security Application",
            kind="slot",
            prefix="security_app_",
            weight=5,
            permissions={
                "permissions": ["security.app"],
                "recommended_roles": {
                    "default": {"admin": True, "editor": True, "viewer": True},
                }
            }
        )

        theme.register_page(
            "security", "app",
            "results",
            title="Test Results",
            kind="slot",
            prefix="security_app_results_",
            permissions={
                "permissions": ["security.app.reports"],
                "recommended_roles": {
                    "default": {"admin": True, "editor": True, "viewer": True},
                }
            }
        )

        shared.job_type_rpcs.add('security_dast')

    def deinit(self):  # pylint: disable=R0201
        """ De-init module """
        log.info('De-initializing module')