__author__ = 'diegopinheiro'
__email__ = 'diegompin@gmail.com'
__github__ = 'https://github.com/diegompin'

'''
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
!!!!! THIS SECTION IS NECESSARY FOR AIRFLOW TO RECOGNIZE THE DAG !!!!! 
from airflow import DAG
!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! 
'''

from mhs.dags.dag_acquisition.aquisition_factory_dag import AcquisitionFactoryDag
from mhs.scripts.acquisition_script.usdsmapper.zipcode_zcta_usdsmapper_acquisition_script import ScriptUSDSMapper


class UDCMapperZipZctaDagFactory(AcquisitionFactoryDag):

    def __init__(self):
        super().__init__()

    def get_params(self):
        return {
            'config_class': "UDSMapper",
            'config_name': "ZIP_ZCTA"
        }

    def get_script(self,  **kwargs):
        return ScriptUSDSMapper()


factory = UDCMapperZipZctaDagFactory()
dag = factory.build_dag()
