mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
35 lines
870 B
Python
35 lines
870 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)
|