msb217 commented on a change in pull request #7596: [AIRFLOW-6870] [WIP] remove imports from models/__init__.py URL: https://github.com/apache/airflow/pull/7596#discussion_r387134095
########## File path: airflow/utils/db.py ########## @@ -540,6 +537,11 @@ def resetdb(): """ Clear out the database """ + from airflow.models.all_models import ( # noqa: F401 # pylint: disable=unused-import + DAG, XCOM_RETURN_KEY, BaseOperator, BaseOperatorLink, BaseJob, Connection, DagBag, + DagModel, DagPickle, DagRun, DagTag, Log, Pool, SerializedDagModel, SkipMixin, SlaMiss, + TaskFail, TaskInstance, TaskReschedule, Variable, XCom + ) Review comment: @potiuk I would have placed `from airflow.models.all_models import *` as a local import as you specified, however, `import *` is only allowed at the module level in python. Therefore, I had to specify all model names individually. I could wrap this into some kind of `load_models` helper function so it's a bit cleaner ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services