Added bat file to source correct venv and run script. Reorganized db connection stuff and added check to create db/tables if they don't exist. Uncommented email section and set mark_sent method to only be called if not testing.

This commit is contained in:
Dan Dembinski
2021-04-08 10:58:01 -04:00
parent 82ad72007c
commit 909b12ff6c
3 changed files with 15 additions and 12 deletions

2
CFFInvoice.bat Normal file
View File

@@ -0,0 +1,2 @@
@echo off
cmd /k "cd /d C:\Program Files (x86)\AKSESS\Storefront\CFFInvoice\Scripts & activate & cd /d C:\Program Files (x86)\AKSESS\Storefront\CFFInvoice & python main.py"

View File

@@ -3,9 +3,6 @@ from sqlalchemy import create_engine, Column, Integer, String, REAL, Boolean
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
engine = create_engine('sqlite:///cffInvoice.db')
engine.connect()
Base = declarative_base()
@@ -33,7 +30,7 @@ class Invoice(Base):
ShippingState = Column(String)
ShippingPostalCode = Column(String)
InvoiceID = Column(Integer, primary_key=True, autoincrement=True)
InvoiceSent = Column(Boolean)
InvoiceSent = Column(Boolean, default=0)
class ItemDetail(Base):
@@ -53,7 +50,11 @@ class SentInvoices(Base):
InvoiceID = Column(Integer)
SentDate = Column(String)
engine = create_engine('sqlite:///cffInvoice.db')
engine.connect()
Invoice.__table__.create(bind=engine, checkfirst=True)
ItemDetail.__table__.create(bind=engine, checkfirst=True)
SentInvoices.__table__.create(bind=engine, checkfirst=True)
Session = sessionmaker(bind=engine)
session = Session()

View File

@@ -191,13 +191,13 @@ def send_email(test, record):
msg.add_header('Content-Type', 'text/html')
msg.attach(bodyHTML)
# try:
# with smtplib.SMTP(smtp_server, port, timeout=120) as server:
# # server.sendmail(msg['From'], msg['To'] + ',' + msg['Cc'] + ',' + msg['Bcc'], msg.as_string())
# server.send_message(msg)
# except Exception as e:
# print(e)
if test is True:
try:
with smtplib.SMTP(smtp_server, port, timeout=120) as server:
# server.sendmail(msg['From'], msg['To'] + ',' + msg['Cc'] + ',' + msg['Bcc'], msg.as_string())
server.send_message(msg)
except Exception as e:
print(e)
if test is False:
InvoiceDatabase.mark_sent(InvoiceID, InvoiceDate)
sent_orders.append(OrderNumber)
print(sent_orders)