2025-04-03 10:39:47 +05:30

47 lines
1.6 KiB
Python

# Copyright 2025 Collate
# Licensed under the Collate Community License, Version 1.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# https://github.com/open-metadata/OpenMetadata/blob/main/ingestion/LICENSE
# 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 airflow.plugins_manager import AirflowPlugin
from airflow.www.app import cached_app
from flask import Blueprint
from openmetadata_managed_apis.api.app import get_blueprint
from openmetadata_managed_apis.api.config import PLUGIN_NAME
from openmetadata_managed_apis.views.rest_api import RestApiView
# Creating View to be used by Plugin
rest_api_view = {"category": "Admin", "name": "REST API Plugin", "view": RestApiView()}
# Creating template Blueprint
template_blueprint = Blueprint(
"template_blueprint",
__name__,
template_folder="views/templates",
)
with cached_app().app_context():
# Import REST API blueprint
api_blueprint = get_blueprint()
class RestApiPlugin(AirflowPlugin):
"""
Creating the RestApiPlugin which extends the AirflowPlugin to import it into Airflow
"""
name = PLUGIN_NAME
operators = []
appbuilder_views = [rest_api_view]
flask_blueprints = [template_blueprint, api_blueprint]
hooks = []
executors = []
admin_views = [rest_api_view]
menu_links = []