Merge branch 'main' of https://github.com/harmonydata/harmony

harmonydata · Jul 8, 2023 · 9a54caa · 9a54caa
2 parents 98675bd + 2ff37f1
commit 9a54caa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/harmony/parsing/wrapper_all_parsers.py b/src/harmony/parsing/wrapper_all_parsers.py
@@ -8,7 +8,7 @@
 
 
 def _get_instruments_from_file(file):
-    if file.file_type == FileType.pdf:
+    if file.file_type == FileType.pdf or file.file_type == FileType.docx:
         instruments_from_this_file = convert_pdf_to_instruments(file)
     elif file.file_type == FileType.txt:
         instruments_from_this_file = convert_text_to_instruments(file)

diff --git a/src/harmony/schemas/enums/file_types.py b/src/harmony/schemas/enums/file_types.py
@@ -4,4 +4,5 @@
 class FileType(str, Enum):
     pdf: str = 'pdf'
     xlsx: str = 'xlsx'
-    txt: str = 'txt'
+    txt: str = 'txt'
+    docx: str = 'docx'
diff --git a/src/harmony/util/file_helper.py b/src/harmony/util/file_helper.py
@@ -12,10 +12,12 @@ def load_instruments_from_local_file(file_name: str) -> List[Instrument]:
         file_type = "pdf"
     elif file_name.lower().endswith("xlsx"):
         file_type = "xlsx"
+    elif file_name.lower().endswith("docx"):
+        file_type = "docx"
     else:
         file_type = "txt"
 
-    if file_type == "pdf" or file_type == "xlsx":
+    if file_type == "pdf" or file_type == "xlsx" or file_type == "docx":
         with open(
                 file_name,
                 "rb") as f: