Update extra_transition_domain_helper.py

This commit is contained in:
zandercymatics 2023-11-08 11:51:59 -07:00
parent 025f08f3d0
commit 0bc03047b8
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -615,8 +615,8 @@ class LoadExtraTransitionDomain:
# TODO - change name # TODO - change name
@dataclass @dataclass
class PatternMap: class FileDataHolder:
"""Helper class that holds data and metadata about a requested file. """Helper class that holds data about a requested file.
filename: str -> The desired filename to target. If no filename is given, filename: str -> The desired filename to target. If no filename is given,
it is assumed that you are passing in a filename pattern and it will look it is assumed that you are passing in a filename pattern and it will look
@ -806,7 +806,7 @@ class ExtraTransitionDomain:
a list of values. a list of values.
return example: return example:
EnumFilenames.AUTHORITY_ADHOC: PatternMap( EnumFilenames.AUTHORITY_ADHOC: FileDataHolder(
authority_adhoc_filename, authority_adhoc_filename,
self.strip_date_regex, self.strip_date_regex,
AuthorityAdhoc, AuthorityAdhoc,
@ -815,7 +815,7 @@ class ExtraTransitionDomain:
""" """
file_data = {} file_data = {}
for file_type, filename, data_type, id_field in pattern_map_params: for file_type, filename, data_type, id_field in pattern_map_params:
file_data[file_type] = PatternMap( file_data[file_type] = FileDataHolder(
filename, filename,
self.strip_date_regex, self.strip_date_regex,
data_type, data_type,
@ -879,7 +879,7 @@ class ExtraTransitionDomain:
def clear_file_data(self): def clear_file_data(self):
for item in self.file_data.values(): for item in self.file_data.values():
file_type: PatternMap = item file_type: FileDataHolder = item
file_type.data = {} file_type.data = {}
def parse_csv_file( def parse_csv_file(