diff --git a/edi_944_to_table.py b/edi_944_to_table.py index a4ca315..cb12baa 100644 --- a/edi_944_to_table.py +++ b/edi_944_to_table.py @@ -33,6 +33,10 @@ SHANDEX_944_FILENAME_RE = re.compile( r"\A 944_STASH-YAMAMOTOYAMA_ \S+ [.]edi \Z", re.X | re.M | re.S ) +SHANDEX_RETURN_944_FILENAME_RE = re.compile( + r"\A 944_RET_YAMAMOTOYAMA \S+ [.]edi \Z", re.X | re.M | re.S +) + INSERT_RECEIPT = """ execute staging.dbo.shandex_insert_receipt :sdhnum, @@ -78,6 +82,9 @@ def main(): determine_edi_action(edi_filename) # file moved to 997 processing folder to be sent later shutil.move(edi_filename, EDI_997_DIRECTORY / edi_filename.name) + elif SHANDEX_RETURN_944_FILENAME_RE(edi_filename.name): + process_return(edi_filename) + shutil.move(edi_filename, EDI_997_DIRECTORY / edi_filename.name) def determine_edi_action(edi_filename: pathlib.Path):