Extract task definition logic

Summary: This simplifies a future diff that will extend the logic.

Test Plan: unit

Reviewers: dgibson, alangenfeld, johann

Reviewed By: dgibson

Differential Revision: https://dagster.phacility.com/D8638