backend.py 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. #
  2. # Licensed to the Apache Software Foundation (ASF) under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. The ASF licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing,
  13. # software distributed under the License is distributed on an
  14. # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15. # KIND, either express or implied. See the License for the
  16. # specific language governing permissions and limitations
  17. # under the License.
  18. """Sends lineage metadata to a backend."""
  19. from __future__ import annotations
  20. from typing import TYPE_CHECKING
  21. if TYPE_CHECKING:
  22. from airflow.models.baseoperator import BaseOperator
  23. class LineageBackend:
  24. """Sends lineage metadata to a backend."""
  25. def send_lineage(
  26. self,
  27. operator: BaseOperator,
  28. inlets: list | None = None,
  29. outlets: list | None = None,
  30. context: dict | None = None,
  31. ):
  32. """
  33. Send lineage metadata to a backend.
  34. :param operator: the operator executing a transformation on the inlets and outlets
  35. :param inlets: the inlets to this operator
  36. :param outlets: the outlets from this operator
  37. :param context: the current context of the task instance
  38. """
  39. raise NotImplementedError()