Coverage for libs/sdc_etl_libs/sdc_file_helpers/TechnicalStandards/EDI/SDCEDIConverterFactory.py : 73%
Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1from sdc_etl_libs.sdc_dataframe.SDCDataframeEnums import SDCDFTypes
2from sdc_etl_libs.sdc_file_helpers.TechnicalStandards.EDI.SDCEDIConverter import \
3 SDCEDIConverter
4from sdc_etl_libs.sdc_file_helpers.TechnicalStandards.EDI.SDCEDIPandasConverter import \
5 SDCEDIPandasConverter
6from sdc_etl_libs.sdc_file_helpers.TechnicalStandards.EDI.SDCEDISparkConverter import \
7 SDCEDISparkConverter
10class SDCEDIConverterFactory:
12 @staticmethod
13 def get_edi_converter(dataframe_type_, edi_file_obj_, file_name_):
14 if dataframe_type_ == SDCDFTypes.PANDAS.value:
15 return SDCEDIPandasConverter(edi_file_obj_, file_name_)
16 elif dataframe_type_ == SDCDFTypes.SPARK.value:
17 return SDCEDISparkConverter(edi_file_obj_, file_name_)
18 else:
19 return SDCEDIConverter(edi_file_obj_, file_name_)