69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
Update X3 with information from middleware
|
||
|
|
To reprocess a file, move it to the correct folder on the FTP
|
||
|
|
"""
|
||
|
|
|
||
|
|
import shutil
|
||
|
|
import yamamotoyama
|
||
|
|
import pathlib
|
||
|
|
|
||
|
|
THIS_DIRECTORY = pathlib.Path(__file__).parent
|
||
|
|
RECEIVED_867_DIRECTORY = THIS_DIRECTORY / "processed_867s"
|
||
|
|
ARCHIVE = RECEIVED_867_DIRECTORY / "archive"
|
||
|
|
|
||
|
|
|
||
|
|
INSERT_867s = """\
|
||
|
|
execute [PROD].[insert_shandex_867_data]
|
||
|
|
:ctrlnum,
|
||
|
|
:credat;
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""
|
||
|
|
Do it!
|
||
|
|
"""
|
||
|
|
process_received_867s()
|
||
|
|
remove_completed_867s()
|
||
|
|
|
||
|
|
|
||
|
|
def remove_completed_867s():
|
||
|
|
"""
|
||
|
|
Clean up all of the files received from the FTP
|
||
|
|
"""
|
||
|
|
for file in RECEIVED_867_DIRECTORY.iterdir():
|
||
|
|
if file.name.endswith('.edi'):
|
||
|
|
shutil.move(file,ARCHIVE / file.name)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def import_received_867s(data):
|
||
|
|
with yamamotoyama.get_connection() as data_base:
|
||
|
|
with data_base.transaction():
|
||
|
|
data_base.bulk_query(INSERT_867s, data)
|
||
|
|
|
||
|
|
|
||
|
|
def process_received_867s():
|
||
|
|
#pull out the control number on the GS line
|
||
|
|
for file in RECEIVED_867_DIRECTORY.iterdir():
|
||
|
|
control_number = ''
|
||
|
|
transaction_date = ''
|
||
|
|
if file.name.endswith('.edi'):
|
||
|
|
with file.open(encoding="utf-8", newline="") as edi_file:
|
||
|
|
for record in edi_file.read().split("~"):
|
||
|
|
fields = record.split("*")
|
||
|
|
if fields[0] == 'GS':
|
||
|
|
control_number = fields[6]
|
||
|
|
if fields[0] == 'BPT':
|
||
|
|
transaction_date = fields[3]
|
||
|
|
data = {
|
||
|
|
"ctrlnum":control_number,
|
||
|
|
"credat":transaction_date
|
||
|
|
}
|
||
|
|
import_received_867s(data)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|