def generate_csv_header_config(self,
override_filename=None):
""" Adds the csv header to the config.
Added by calling programs within the extend_config method.
"""
if override_filename:
filenames = [override_filename]
else:
filenames = self.config['infiles']
header = csvhelper.Header()
if filenames[0] != '-':
header.load_from_files(filenames, self.config['dialect'])
self.update_config('header', header)
def test_load_and_gets(self):
dialect = csvhelper.Dialect(delimiter='|', quoting=csv.QUOTE_ALL, has_header=True)
fqfn = ttools.make_team_file(self.temp_dir, dialect, 10)
header = csvhelper.Header()
header.load_from_file(fqfn, dialect)
assert len(header.raw_field_names)
assert len(header.field_names)
assert header.get_field_position('role') == 2
assert header.get_field_name(3) == 'name'
assert header.get_field_position_from_any('3') == 3
assert header.get_field_position_from_any('name') == 3
def setup_spec(self,
specs_strings,
spec_type='incl_rec',
header: List[str] = None,
item_count=100):
if header:
header_obj = csvhelper.Header()
header_obj.load_from_list(field_names=header)
else:
header_obj = None
self.spec = mod.Specifications(spec_type=spec_type,
specs_strings=specs_strings,
header=header_obj,
infile_item_count=item_count)
def runner(self,
spec_strings,
spec_type='incl_rec',
item_count=80,
header=None):
""" Creates the Spec Processor object with default settings for the
specification and record length.
"""
if header:
header_obj = csvhelper.Header()
header_obj.load_from_list(field_names=header)
else:
header_obj = None
self.specs = mod.Specifications(spec_type,
specs_strings=spec_strings,
header=header_obj,
infile_item_count=item_count)