1
0
mirror of https://github.com/quay/quay.git synced 2026-01-26 06:21:37 +03:00
Files
quay/tools/monthlyrevenue.py
Kenny Lee Sin Cheong 5f63b3a7bb chore: drop deprecated tables and remove unused code (PROJQUAY-522) (#2089)
* chore: drop deprecated tables and remove unused code

* isort imports

* migration: check for table existence before drop
2023-08-25 12:17:24 -04:00

51 lines
1.4 KiB
Python

from collections import defaultdict
from typing import DefaultDict, Tuple
from app import billing
offset = 0
def empty_tuple():
return (0, 0)
plan_revenue: DefaultDict[Tuple[int, int], Tuple[int, int]] = defaultdict(empty_tuple)
batch = billing.Customer.all(count=100, offset=offset)
while batch.data:
for cust in batch.data:
if cust.subscription:
sub = cust.subscription
total_customer_revenue = sub.plan.amount * sub.quantity
if cust.discount and cust.discount.coupon:
coupon = cust.discount.coupon
if coupon.percent_off:
total_customer_revenue *= 1 - coupon.percent_off / 100.0
if coupon.amount_off:
total_customer_revenue -= coupon.amount_off
subscribers, revenue = plan_revenue[sub.plan.id]
plan_revenue[sub.plan.id] = (subscribers + 1, revenue + total_customer_revenue)
offset += len(batch.data)
batch = billing.Customer.all(count=100, offset=offset)
def format_money(total_cents):
dollars = total_cents // 100
cents = total_cents % 100
return dollars, cents
total_monthly_revenue = 0
for plan_id, (subs, rev) in list(plan_revenue.items()):
total_monthly_revenue += rev
d, c = format_money(rev)
print("%s: $%d.%02d(%s)" % (plan_id, d, c, subs))
d, c = format_money(total_monthly_revenue)
print("Monthly revenue: $%d.%02d" % (d, c))