Skip to content

Commit f6d36b8

Browse files
committed
Allow selectable default on infer_file_format
1 parent ea36ef4 commit f6d36b8

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

mathics/eval/fileformats/compression.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def eval_ImportZIP(
6868

6969
for element in elements:
7070
member = element.value
71-
file_format = infer_file_format(member).upper()
71+
file_format = infer_file_format(member, "Text").upper()
7272
if file_format not in IMPORTERS.keys():
7373
evaluation.message("Import", "fmtnosup", file_format)
7474
return SymbolFailed

mathics/eval/import_export/importexport.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -696,10 +696,10 @@ def eval_Import_source_only(
696696
return result
697697

698698

699-
def infer_file_format(filename: str) -> Optional[str]:
699+
def infer_file_format(filename: str, default_extension: str = None) -> Optional[str]:
700700
"""
701701
Infer what kind of format filename is in. None is returned if we can't infer
702702
a format.
703703
"""
704704
file_extension = eval_FileExtension(filename).lower()
705-
return FILE_EXTENSION_MAP.get(file_extension, "Text")
705+
return FILE_EXTENSION_MAP.get(file_extension, default_extension)

0 commit comments

Comments
 (0)