280 lines
10 KiB
Python
280 lines
10 KiB
Python
'''
|
|
Created on Jul 15, 2015
|
|
|
|
@author: Wes
|
|
'''
|
|
from alert import send_alert_html
|
|
from glob import glob
|
|
|
|
import os, sys, logging, time, StringIO
|
|
|
|
import pyodbc
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.sql.expression import update
|
|
|
|
RECIPIENTS = ['admin@example.com']
|
|
SUBJECT = 'Invoices Transmitted to Momentum.'
|
|
MESSAGE = None
|
|
|
|
TEST_MODE = False
|
|
LOG_LEVEL = logging.DEBUG
|
|
ACC_NUMBERS = ['3398', '4318', '5230', '6600', '4063']
|
|
|
|
engine = create_engine("mssql+pyodbc://@jsisbprod", echo=False)
|
|
Base = declarative_base(engine)
|
|
metadata = Base.metadata
|
|
logger = None
|
|
|
|
########################################################################
|
|
# Jomar Invoice (EDI 810) staging tables #
|
|
########################################################################
|
|
class CSPINV00(Base):
|
|
__tablename__ = 'CSPINV00'
|
|
__table_args__ = {'autoload':True}
|
|
class CSPINV01(Base):
|
|
__tablename__ = 'CSPINV01'
|
|
__table_args__ = {'autoload':True}
|
|
class TransactionTable(Base):
|
|
__tablename__ = 'ftp_trans_numbers'
|
|
__table_args__ = {'autoload':True}
|
|
|
|
########################################################################
|
|
# Jomar Item description tables. Used to look up Pattern name, color, #
|
|
# style, etc. for a given item number #
|
|
########################################################################
|
|
class TXMYAT03(Base):
|
|
__tablename__ = 'TXMYAT03'
|
|
__table_args__ = {'autoload':True}
|
|
########################################################################
|
|
# Jomar Customer master file. Used to look up account names #
|
|
########################################################################
|
|
class CustomerMaster(Base):
|
|
__tablename__ = 'TXUYKD00'
|
|
__table_args__ = {'autoload':True}
|
|
|
|
def loadSession():
|
|
global engine
|
|
Session = sessionmaker(bind=engine)
|
|
session = Session()
|
|
return session
|
|
|
|
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:\\FTP\\logs\\' + name + '.log')
|
|
handler.setLevel(LOG_LEVEL)
|
|
formatter = logging.Formatter('%(asctime)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.INFO)
|
|
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)
|
|
|
|
'''
|
|
Gotta get transaction number from AS400 until ALL transmissions
|
|
are handled on the new system... whoopee
|
|
'''
|
|
def getTransNumber():
|
|
connection = pyodbc.connect(
|
|
driver='{iSeries Access ODBC Driver}',
|
|
system='SUNBURY',
|
|
uid=os.environ['AS400_UID'],
|
|
pwd=os.environ['AS400_PWD'])
|
|
c1 = connection.cursor()
|
|
ret = -1
|
|
|
|
c1.execute('select * from stmdata.MGTX#FL')
|
|
for row in c1:
|
|
ret = int(row[1]) + 1
|
|
|
|
c1.close()
|
|
|
|
return ret
|
|
|
|
'''
|
|
Update the Momentum transaction number in the AS400 database
|
|
'''
|
|
def updateTransNumber(txnum):
|
|
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()
|
|
c1.execute('UPDATE STMDATA.MGTX#FL SET MGSEQ# = ' + str(txnum))
|
|
c1.close()
|
|
|
|
'''
|
|
Grabs any unprocessed Maharam invoices from the 810 EDI staging tables in Jomar.
|
|
Formats data into text document to send to Maharam via FTP.
|
|
Marks invoices as processed.
|
|
'''
|
|
def createMomentumInvoiceDocument():
|
|
global MESSAGE
|
|
initializeLogger('MOM_810_GENERATE')
|
|
xmission = [] #List of strings, the individual lines of the transmission file that will be sent
|
|
invoices_to_flag_processed = []
|
|
|
|
#First line of transmission is T or P. 'Test' or 'Production'
|
|
if(TEST_MODE):
|
|
logger.info('**RUNNING IN TEST MODE**')
|
|
xmission.append('T~\n')
|
|
else:
|
|
logger.info('**RUNNING IN PRODUCTION MODE**')
|
|
xmission.append('P~\n')
|
|
|
|
logger.debug('Connecting to database...')
|
|
session = loadSession()
|
|
|
|
trans_no = -1
|
|
# t = session.query(mf_transaction_number).filter(mf_transaction_number.doc == '810')
|
|
# if t.count() == 1:
|
|
# for row in t:
|
|
# trans_no = row.txnum
|
|
# else:
|
|
# logger.fatal('Failed to retrieve transaction number from database... Something ain\'t right. %s', t.count())
|
|
# sys.exit('Failed to retrieve transaction number from database... Something ain\'t right.')
|
|
trans_no = getTransNumber()
|
|
|
|
#Second line (Control Line) of transmission is essentially a boilerplate
|
|
logger.info('Building Control Line')
|
|
ctrlLine = 'TP~12~5702863800~12~9498338886~'
|
|
ctrlLine += time.strftime('%Y%m%d')
|
|
ctrlLine += '~' + str(trans_no)
|
|
xmission.append(ctrlLine + "\n")
|
|
logger.debug(ctrlLine)
|
|
|
|
logger.info('Querying EDI table for unprocessed invoices...')
|
|
logger.debug('SELECT * FROM CSPINV00 WHERE IVCUSTB IN (\'3398\', \'4318\', \'5230\', \'6600\') ORDER BY IVIDAT;')
|
|
q = session.query(CSPINV00,CustomerMaster).order_by(CSPINV00.IVIDAT).filter(CSPINV00.IVCUSTB.in_(ACC_NUMBERS)).filter(CustomerMaster.KKDN == CSPINV00.IVCUSTB)
|
|
for row in q:
|
|
if(row[0].IVPROCSS == 'Y'):
|
|
continue
|
|
invoices_to_flag_processed.append(row[0].IVINV)
|
|
businessGroup = row[0].IVBNR
|
|
plant = row[0].IVPLT
|
|
invoiceType = row[0].IVITYP
|
|
logger.info('Building header line for invoice: ' + row[0].IVINV)
|
|
|
|
#Translate Sunbury bill-to account number to Momentum vendor number
|
|
momVendor = ''
|
|
if(row[0].IVCUSTB == '3398'):
|
|
momVendor = '6975'
|
|
elif(row[0].IVCUSTB == '4318'):
|
|
momVendor = '0036'
|
|
elif(row[0].IVCUSTB == '5230'):
|
|
momVendor = '7738'
|
|
elif(row[0].IVCUSTB == '6600'):
|
|
momVendor = '2082'
|
|
elif(row[0].IVCUSTB == '4063'):
|
|
momVendor = '2082'
|
|
logger.debug('Sunbury Account: %s --> Momentum Vendor: %s', row[0].IVCUSTB, momVendor)
|
|
|
|
momPO = row[0].IVCPO[-5:]
|
|
|
|
hLine = 'H~'
|
|
hLine += (row[0].IVINV + '~')
|
|
hLine += (str(row[0].IVIDAT) + '~')
|
|
hLine += (momVendor + '~')
|
|
hLine += (momPO + '~')
|
|
hLine += (row[0].IVCUSTB + '~')
|
|
#hLine += (row[0].IVTERMS + '~')
|
|
hLine += ('02~') #Changed to constant '02' per Kim Stolmeier @ Momentum
|
|
hLine += (row[1].KNA2 + '~')
|
|
hLine += (row[0].IVSADD1 + '~')
|
|
hLine += (row[0].IVSCTY + '~')
|
|
hLine += (row[0].IVSST + '~')
|
|
hLine += (row[0].IVSZIP + '~')
|
|
hLine += (row[1].KNA2 + '~')
|
|
hLine += (row[0].IVBADD1 + '~')
|
|
hLine += (row[0].IVBCTY + '~')
|
|
hLine += (row[0].IVBST + '~')
|
|
hLine += (row[0].IVBZIP + '~')
|
|
hLine += (str(row[0].IVFRT) + '~')
|
|
xmission.append(hLine + "\n")
|
|
logger.debug(hLine)
|
|
|
|
logger.info('Querying EDI table for invoice detail lines for invoice: ' + row[0].IVINV)
|
|
logger.debug('SELECT * FROM CSPINV01 LEFT JOIN TXMYAT03 ON TXMYAT03.XANR = CSPINV01.I1ITEM WHERE CSPINV01.I1BNR = \'' + businessGroup + '\' AND CSPINV01.I1PLT = \'' + plant + '\' AND CSPINV01.I1ITYP = \'' + invoiceType + '\' AND CSPINV01.I1INV = \'' + row[0].IVINV + '\';')
|
|
q2 = session.query(CSPINV01).filter(CSPINV01.I1BNR == businessGroup, CSPINV01.I1PLT == plant, CSPINV01.I1ITYP == invoiceType, CSPINV01.I1INV == row[0].IVINV).join(TXMYAT03, TXMYAT03.XANR == CSPINV01.I1ITEM).add_entity(TXMYAT03)
|
|
for row2 in q2:
|
|
if(row2[0].I1ITEM == 'INSURANCE'):
|
|
'''
|
|
@todo: Does Momentum even pay for insurance?
|
|
'''
|
|
logger.warn('Skipping insurance line!')
|
|
continue
|
|
|
|
logger.info('Building detail line for line ' + str(row2[0].I1ILINE))
|
|
dLine = 'D~'
|
|
dLine += (row2[1].XFLX33 + '~')
|
|
dLine += (row2[1].XFLX34 + '~')
|
|
dLine += (row2[1].XFLX35 + '~')
|
|
dLine += ("{0:.3f}".format(row2[0].I1QTY) + '~')
|
|
dLine += ("{0:.2f}".format(row2[0].I1PRI) + '~')
|
|
dLine += (momPO + '~')
|
|
xmission.append(dLine + "\n")
|
|
logger.debug(dLine)
|
|
|
|
if len(xmission) > 2:
|
|
xmission.append('S~' + str(len(xmission)) + '~')
|
|
|
|
#Write transmission data out to file here
|
|
logger.info('Creating D:\\FTP\\SEND_MG\\SBR_810_' + str(trans_no) + '.txt')
|
|
f = open('SEND_MG\\SBR_810_' + str(trans_no) + '.txt', 'w')
|
|
f.writelines(xmission)
|
|
f.close()
|
|
|
|
try:
|
|
logger.info('Sending e-mail alert...')
|
|
send_alert_html(RECIPIENTS, SUBJECT, attachments=glob('D:\\FTP\\SEND_MG\\SBR_810_*.txt'), message=MESSAGE.getvalue())
|
|
except:
|
|
logger.error('Unable to send e-mail. Is the internet out?')
|
|
|
|
#Update EDI tables and transaction number #mf_transaction_number
|
|
if(len(invoices_to_flag_processed) > 0 and not TEST_MODE):
|
|
#Set process flag on these invoices to 'Y'
|
|
ex = update(CSPINV00.__table__).where(CSPINV00.IVINV.in_(invoices_to_flag_processed)).values(IVPROCSS=u"Y")
|
|
session.execute(ex)
|
|
ex = None
|
|
|
|
#Might as well keep the table in the new database up to date
|
|
ex2 = update(TransactionTable.__table__).where(TransactionTable.client.in_(['MOMTEX'])).values(xaction=(trans_no))
|
|
session.execute(ex2)
|
|
ex2 = None
|
|
|
|
logger.debug('Comitting changes to database...')
|
|
session.commit()
|
|
|
|
updateTransNumber(trans_no) #Update transaction number in AS400
|
|
|
|
logger.debug('Closing database connection...')
|
|
session.close()
|
|
else:
|
|
logger.warn('Nothing to send!')
|
|
session.close()
|
|
|
|
MESSAGE.truncate(0)
|
|
|
|
if __name__ == '__main__':
|
|
createMomentumInvoiceDocument()
|