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)