add header preprocessing to function

Author: briangow

wfdb/io/_header.py

@@ -237,6 +237,25 @@ def get_write_subset(self, spec_type):
         return write_fields
 
 
+    def preprocess_header(self, field):
+        """
+        Get and process the fields as needed before writing
+        """
+        string_field = str(getattr(self, field))
+
+        # Certain fields need extra processing
+        if field == "fs" and isinstance(self.fs, float):
+            if round(self.fs, 8) == float(int(self.fs)):
+                string_field = str(int(self.fs))
+        elif field == "base_time" and "." in string_field:
+            string_field = string_field.rstrip("0")
+        elif field == "base_date":
+            string_field = "/".join(
+                (string_field[8:], string_field[5:7], string_field[:4])
+            )
+
+        return string_field
+
 class HeaderMixin(BaseHeaderMixin):
     """
     Mixin class with single-segment header methods. Inherited by Record class.
@@ -534,19 +553,7 @@ def wr_header_file(self, rec_write_fields, sig_write_fields, write_dir):
         for field in RECORD_SPECS.index:
             # If the field is being used, add it with its delimiter
             if field in rec_write_fields:
-                string_field = str(getattr(self, field))
-
-                # Certain fields need extra processing
-                if field == "fs" and isinstance(self.fs, float):
-                    if round(self.fs, 8) == float(int(self.fs)):
-                        string_field = str(int(self.fs))
-                elif field == "base_time" and "." in string_field:
-                    string_field = string_field.rstrip("0")
-                elif field == "base_date":
-                    string_field = "/".join(
-                        (string_field[8:], string_field[5:7], string_field[:4])
-                    )
-
+                string_field = self.preprocess_header(field)
                 record_line += (
                     RECORD_SPECS.loc[field, "delimiter"] + string_field
                 )
@@ -756,25 +763,8 @@ def wr_header_file(self, write_fields, write_dir):
         for field in RECORD_SPECS.index:
             # If the field is being used, add it with its delimiter
             if field in write_fields:
-                string_field = str(getattr(self, field))
-
-                # Certain fields need extra processing
-                if field == "fs" and isinstance(self.fs, float):
-                    if round(self.fs, 8) == float(int(self.fs)):
-                        string_field = str(int(self.fs))
-                elif field == "base_time" and "." in string_field:
-                    string_field = string_field.rstrip("0")
-                elif field == "base_date":
-                    string_field = "/".join(
-                        (string_field[8:], string_field[5:7], string_field[:4])
-                    )
-
-                record_line += (
-                    RECORD_SPECS.loc[field, "delimiter"] + string_field
-                )
-                # The 'base_counter' field needs to be closed with ')'
-                if field == "base_counter":
-                    record_line += ")"
+                string_field = self.preprocess_header(field)
+                record_line += (RECORD_SPECS.loc[field, "delimiter"] + string_field)
 
         header_lines = [record_line]