Example #1
0
def get_schema(config, dag_name, schema="instance"):
    """
    Gets a schema by name from airflow server. We use the variable api.
    We transform the jsonschema into a marshmallow class

    """
    af_client = Airflow.from_config(config)
    if not af_client.is_alive():
        raise AirflowError(error="Airflow is not accessible")

    schema_json = af_client.get_one_schema(dag_name, schema)
    manager = SchemaManager(schema_json)
    return manager.jsonschema_to_flask()
Example #2
0
from cornflow_client import SchemaManager, get_pulp_jsonschema
import os

fileDir = os.path.dirname(__file__)

manager = SchemaManager(get_pulp_jsonschema())

DataSchema = manager.jsonschema_to_flask()