34 lines
823 B
Python
34 lines
823 B
Python
|
import stripe
|
||
|
|
from app import app
|
||
|
|
|
||
|
|
from util.invoice import renderInvoiceToPdf
|
||
|
|
|
||
|
|
from data import model
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
def sendInvoice(invoice_id):
|
||
|
|
invoice = stripe.Invoice.retrieve(invoice_id)
|
||
|
|
if not invoice['customer']:
|
||
|
|
print 'No customer found'
|
||
|
|
return
|
||
|
|
|
||
|
|
customer_id = invoice['customer']
|
||
|
|
user = model.user.get_user_or_org_by_customer_id(customer_id)
|
||
|
|
if not user:
|
||
|
|
print 'No user found for customer %s' % (customer_id)
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
with app.app_context():
|
||
|
|
file_data = renderInvoiceToPdf(invoice, user)
|
||
|
|
with open('invoice.pdf', 'wb') as f:
|
||
|
|
f.write(file_data)
|
||
|
|
|
||
|
|
print 'Invoice output as invoice.pdf'
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Generate an invoice')
|
||
|
|
parser.add_argument('invoice_id', help='The invoice ID')
|
||
|
|
args = parser.parse_args()
|
||
|
|
sendInvoice(args.invoice_id)
|