Found more code examples from STM. Used claude to generate a summary (CLAUDE.md)

2026-02-11 06:05:20 -05:00
parent cf0edba776
commit c4b970cbde
33 changed files with 6077 additions and 1 deletions
@@ -0,0 +1,168 @@
+'''
+Attempting to automatically close invoices in the AS400 using data from Jomar.. weee...
+
+Created on: 2017-11-20
+
+@author: Wes
+'''
+import pyodbc, time, logging, sys, os, StringIO, csv, sqlite3
+from sqlalchemy import create_engine
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from FTP.alert import send_alert_html
+
+RECIPIENTS = ['admin@example.com', 'user1@example.com']
+SUBJECT = 'AS400 Invoice Inteface Exceptions'
+MESSAGE = None
+
+AS400_LIB = 'STMDATA'
+logger = None
+LOG_LEVEL = logging.DEBUG
+
+INVOICE = 0
+INVOICE_LINE = 1
+TICKET = 2
+U_NUMBER = 3
+INV_YARDS = 4
+FIN_YARDS =5
+SHIP_DATE = 6
+INV_DATE = 7
+
+#engine = create_engine("mssql+pyodbc://jsisbprod", echo=False)
+for row in sqlite3.connect('D:\\FTP\\ftp.db').execute("SELECT * FROM connection_strings WHERE code = 'wfa'"):
+    engine = create_engine(row[1], echo=False)
+Base = declarative_base(engine)
+metadata = Base.metadata
+
+def initializeLogger(name='log'):
+    global logger
+    global MESSAGE
+    #logger = logging.getLogger(__name__)
+    logger = logging.getLogger(name)
+    if len(logger.handlers) > 0:
+        return logger
+    logger.setLevel(LOG_LEVEL)
+    handler = logging.FileHandler('D:\\' + name + '.log')
+    handler.setLevel(LOG_LEVEL)
+    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
+    
+    #Also send logs > DEBUG to stdout
+    ch = logging.StreamHandler(sys.stdout)
+    ch.setLevel(logging.DEBUG)
+    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
+    ch.setFormatter(formatter)
+    logger.addHandler(ch)
+
+    #Capture log messages in a string for e-mail alert
+    MESSAGE = StringIO.StringIO()
+    sh = logging.StreamHandler(MESSAGE)
+    sh.setLevel(logging.DEBUG)
+    sh.setFormatter(formatter)
+    logger.addHandler(sh)
+
+def loadSession():
+    global engine
+    Session = sessionmaker(bind=engine)
+    session = Session()
+    return session
+
+def build_extract_query(d):
+    q = ("SELECT r1rnr,r1lfn,RIGHT(rtrnr, 5) rtrnr,urbpo,rtilg,rtpglg,lsdat,'***DATE***' as invdte "
+         "FROM txuyre01 "
+         "LEFT JOIN invoice_rolls ON rtbnr=r1bnr AND rtplt=r1plt AND rtinv=r1rnr AND r1lfn=rtafn "
+         "LEFT JOIN TXUYUV00 ON RTALN = UBLN "
+         "LEFT JOIN TXUYLS00 ON RTDLN = LSNR "
+         "WHERE r1rnr IN (SELECT RRNR FROM TXUYRE00 WHERE RDATLS = '***DATE***') "
+         "AND R1ANR NOT LIKE 'FREIGHT' AND R1ART NOT LIKE 'SR'"
+         "ORDER BY r1lfn,r1rnr")
+    return q.replace('***DATE***',d)
+
+def build_insert_statement(r):
+    q = 'INSERT INTO ' + AS400_LIB + '.JSIINV VALUES (INVNUM,INVLIN,TICK#,\'ORDERA\',INVYDS,FINYDS,SHPDTE,INVDTE)'
+    q = q.replace('INVNUM',r[INVOICE])
+    q = q.replace('INVLIN',str(r[INVOICE_LINE]))
+    q = q.replace('TICK#',r[TICKET])
+    q = q.replace('ORDERA',r[U_NUMBER])
+    q = q.replace('INVYDS',str(r[INV_YARDS]))
+    q = q.replace('FINYDS',str(r[FIN_YARDS]))
+    q = q.replace('SHPDTE',str(r[SHIP_DATE]))
+    q = q.replace('INVDTE',str(r[INV_DATE]))
+    return q
+
+def AII_main(d):
+    os.chdir('D:\\')
+    initializeLogger('AS400_Invoice_Interface')
+    logger.info('Connecting to JSISBPROD...')
+    session = loadSession()
+
+    extract = []
+
+    logger.info('Running extract query for: %s', d)
+    logger.debug(build_extract_query(d))
+    for row in session.execute(build_extract_query(d)):
+        extract.append(row)
+
+    logger.debug('Disconnecting from JSISBPROD...')
+    session.close()
+
+    logger.info('Opening connection with AS400...')
+    connection = pyodbc.connect(
+        driver='{iSeries Access ODBC Driver}',
+        system='SUNBURY',
+        uid=os.environ['AS400_UID'],
+        pwd=os.environ['AS400_PWD'],
+        autocommit=True)
+    c1 = connection.cursor()
+
+    if 'STMDATA' in AS400_LIB:
+        try:
+            logger.info('Backing up rows from STMDATA.JSIINV --> WES.JSIINV_BAK')
+            c1.execute('INSERT INTO WES.JSIINV_BAK SELECT * FROM STMDATA.JSIINV')
+        except:
+            logger.warn('***Failed to backup rows currently in STMDATA.JSIINV***')
+    
+    logger.info('Deleting old rows from %s.JSIINV', AS400_LIB)
+    c1.execute('DELETE FROM ' + AS400_LIB + '.JSIINV')
+
+    exceptions = []
+
+    logger.info('Inserting rows into %s.JSIINV', AS400_LIB)
+    for row in extract:
+        # Ticket will be null for credits
+        if row[TICKET] and row[U_NUMBER] and row[TICKET].isdigit():
+            #print build_insert_statement(row)
+            c1.execute(build_insert_statement(row))
+        else:
+            exceptions.append(row)
+
+    logger.debug('Disconnecting from AS400...')
+    c1.close()
+
+    if len(exceptions) > 0:
+        #Open file for writing
+        logger.debug('Dumping exceptions to csv file...')
+        fname = "AII_ERRORS_" + d + ".csv"
+        logger.debug('Creating %s. Using \'excel\' dialect for csv writer.', fname)
+        f = open(fname, 'wb')
+        writer = csv.writer(f, dialect='excel')
+        writer.writerow(['Invoice #', 'Invoice Line', 'Ticket', 'AS400 Order', 'Invoiced Yards', 'Ship Date', 'Invoice Date'])
+
+        for row in exceptions:
+            writer.writerow(row)
+
+        f.close()
+
+        try:
+            logger.debug('Sending e-mail...')
+            send_alert_html(RECIPIENTS, (SUBJECT + " - " + d), attachments=[fname], message=MESSAGE.getvalue())
+        except:
+            logger.error('Unable to send e-mail. Is the internet out?')
+
+        logger.debug('Deleting %s', fname)
+        os.remove(fname)
+
+if __name__ == '__main__':
+    AII_main(sys.argv[1])
+    #print build_extract_query(sys.argv[1])