Skip to content

Commit 6614ed6

Browse files
author
Alan Christie
committed
refactor: Minor adjustments
1 parent 7302974 commit 6614ed6

File tree

1 file changed

+20
-19
lines changed

1 file changed

+20
-19
lines changed

tools/coins.py

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -40,25 +40,18 @@ def main(c_args: argparse.Namespace) -> None:
4040
if args.verbose:
4141
pprint(p_rv.msg)
4242

43+
product_id: str = p_rv.msg["product"]["product"]["id"]
44+
product_name: str = p_rv.msg["product"]["product"]["name"]
4345
allowance: Decimal = Decimal(p_rv.msg["product"]["coins"]["allowance"])
4446
allowance_multiplier: Decimal = Decimal(p_rv.msg["product"]["coins"]["allowance_multiplier"])
4547
limit: Decimal = Decimal(p_rv.msg["product"]["coins"]["limit"])
4648
overspend_multiplier: Decimal = Decimal(p_rv.msg["product"]["coins"]["overspend_multiplier"])
4749

4850
remaining_days: int = p_rv.msg["product"]["coins"]["remaining_days"]
4951

50-
invoice: Dict[str, Any] = {
51-
"Allowance": str(allowance),
52-
"Allowance Multiplier": str(allowance_multiplier),
53-
"Limit": str(limit),
54-
"Overspend Multiplier": str(overspend_multiplier),
55-
}
56-
5752
# Get the product's charges...
5853
pc_rv: AsApiRv = AsApi.get_product_charges(token, product_id=args.product)
5954
assert pc_rv.success
60-
invoice["From"] = pc_rv.msg["from"]
61-
invoice["Until"] = pc_rv.msg["until"]
6255
if args.verbose:
6356
pprint(pc_rv.msg)
6457

@@ -84,16 +77,24 @@ def main(c_args: argparse.Namespace) -> None:
8477
total_processing_coins += Decimal(item["coins"])
8578
num_processing_charges += 1
8679

87-
# Accumulate processing coins
88-
total_processing_coins: Decimal = Decimal()
89-
90-
invoice["Billing Day"] = p_rv.msg["product"]["coins"]["billing_day"]
91-
invoice["Remaining Days"] = remaining_days
92-
invoice["Current Burn Rate"] = str(burn_rate)
93-
invoice["Number of Storage Charges"] = num_storage_charges
94-
invoice["Accrued Storage Coins"] = str(total_storage_coins)
95-
invoice["Number of Processing Charges"] = num_processing_charges
96-
invoice["Accrued Processing Coins"] = str(total_processing_coins)
80+
invoice: Dict[str, Any] = {
81+
"Product": product_id,
82+
"Product Name": product_name,
83+
"Claim ID": pc_rv.msg.get("claim_id", "Unclaimed"),
84+
"Claim Name": pc_rv.msg.get("claim_name", "Undefined"),
85+
"Allowance": str(allowance),
86+
"Allowance Multiplier": str(allowance_multiplier),
87+
"Limit": str(limit),
88+
"Overspend Multiplier": str(overspend_multiplier),
89+
"From": pc_rv.msg["from"], "Until": pc_rv.msg["until"],
90+
"Billing Day": p_rv.msg["product"]["coins"]["billing_day"],
91+
"Remaining Days": remaining_days,
92+
"Current Burn Rate": str(burn_rate),
93+
"Number of Storage Charges": num_storage_charges,
94+
"Accrued Storage Coins": str(total_storage_coins),
95+
"Number of Processing Charges": num_processing_charges,
96+
"Accrued Processing Coins": str(total_processing_coins),
97+
}
9798

9899
total_coins: Decimal = total_storage_coins + total_processing_coins
99100

0 commit comments

Comments
 (0)