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)