EbookFoundation
/
regluit
mirror of https://github.com/EbookFoundation/regluit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Latest integration of campaign data in our own db plus the spreadsheet you can download from Amazon payments 

to produce CVS file
pull/1/head
Raymond Yee 2012-07-03 07:46:42 -07:00
parent bfe02fc39d
commit 5e3cf5d7aa
1 changed files with 35 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
core/management/commands/campaign_data.py
View File

@ -6,25 +6,32 @@ from regluit.payment.models import Transaction
from django.db.models import Q, F, Count, Sum, Max from django.db.models import Q, F, Count, Sum, Max
from django.contrib.auth.models import User from django.contrib.auth.models import User
from regluit.payment.manager import PaymentManager from regluit.payment.manager import PaymentManager
from decimal import Decimal as D
from regluit.experimental.gutenberg import unicode_csv from regluit.experimental.gutenberg import unicode_csv
def amazon_payments(fname=r"/Users/raymondyee/Downloads/All-Activity-Jan-25-2012-Jun-25-2012.csv"): def amazon_payments(fname=r"/Users/raymondyee/Downloads/All-Activity-With-Balance-Jan-01-2012-Jul-02-2012.csv"):
r0 = unicode_csv.UnicodeReader(f=open(fname), encoding="iso-8859-1") r0 = unicode_csv.UnicodeReader(f=open(fname), encoding="iso-8859-1")
# grab the header # grab the header
header = r0.next() header = r0.next()
print "header", header
r = unicode_csv.UnicodeDictReader(f=open(fname), fieldnames=header, encoding="iso-8859-1") r = unicode_csv.UnicodeDictReader(f=open(fname), fieldnames=header, encoding="iso-8859-1")
return dict([(k['Transaction ID'],k) for k in r]) return dict([(k['Transaction ID'],k) for k in r])
def transactions_with_payment_info(transactions, payments=None): def transactions_with_payment_info(transactions, payments=None):
"""an iterator for transactions / if a dict representing the actual payment data is provided, correlate the two""" """an iterator for transactions / if a dict representing the actual payment data is provided, correlate the two"""
for t in transactions: for t in transactions:
data = {"username":t.user.username, data = {"id": t.id,
"user_id": t.user.id,
"username":t.user.username,
"email":t.user.email, "email":t.user.email,
"campaign_id": t.campaign.id if t.campaign is not None else None, "campaign_id": t.campaign.id if t.campaign is not None else "NULL",
"amount":t.amount, "amount":t.amount,
"status":t.status, "status":t.status,
"local_status":t.local_status, "local_status":t.local_status,
"premium_id": t.premium.id if t.premium is not None else "NULL",
"premium_amount": t.premium.amount if t.premium is not None else "NULL",
"premium_description": t.premium.description if t.premium is not None else "NULL",
"preapproval_key":t.preapproval_key, "preapproval_key":t.preapproval_key,
"approved":t.approved, "approved":t.approved,
"error":t.error} "error":t.error}
@ -34,13 +41,23 @@ def transactions_with_payment_info(transactions, payments=None):
data.update({'payment_transaction_id':payment['Transaction ID'], data.update({'payment_transaction_id':payment['Transaction ID'],
'payment_name': payment['Name'], 'payment_name': payment['Name'],
'payment_status': payment['Status'], 'payment_status': payment['Status'],
'payment_amount': payment['Amount'], 'payment_amount': D(payment['Amount'].replace("$","")),
'payment_fees': payment['Fees'] 'payment_fees': D(payment['Fees'].replace("$",""))
}) })
else: else:
data.update({'payment_transaction_id':None}) data.update({'payment_transaction_id':"NULL",
'payment_name': "NULL",
'payment_status': "NULL",
'payment_amount': "NULL",
'payment_fees': "NULL"
})
else: else:
data.update({'payment_transaction_id':None}) data.update({'payment_transaction_id':"NULL",
'payment_name': "NULL",
'payment_status': "NULL",
'payment_amount': "NULL",
'payment_fees': "NULL"
})
yield data yield data
@ -122,16 +139,25 @@ class Command(BaseCommand):
stats_for_campaign(c3) stats_for_campaign(c3)
out_fname = "/Users/raymondyee/Downloads/unglue_it_trans.csv" out_fname = "/Users/raymondyee/Downloads/unglue_it_trans.csv"
out_headers = ["username", out_headers = ["id",
"user_id",
"username",
"email", "email",
"campaign_id", "campaign_id",
"amount", "amount",
"status", "status",
"local_status", "local_status",
"premium_id",
"premium_amount",
"premium_description",
"preapproval_key", "preapproval_key",
"approved", "approved",
"error", "error",
"payment_transaction_id" "payment_transaction_id",
"payment_name",
"payment_fees",
"payment_status",
"payment_amount"
] ]
f = open(out_fname, "wb") f = open(out_fname, "wb")