From b3c655bc479fa5e559b0c75b603bad86aef9696d Mon Sep 17 00:00:00 2001
From: forest <forest.n.johnson@gmail.com>
Date: Mon, 26 Jul 2021 15:29:34 -0500
Subject: [PATCH] account-balance cli command

---
 capsulflask/cli.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/capsulflask/cli.py b/capsulflask/cli.py
index c3f29cc..4f211da 100644
--- a/capsulflask/cli.py
+++ b/capsulflask/cli.py
@@ -62,6 +62,18 @@ def sql_script(f, c):
 
     model.connection.commit()
 
+@bp.cli.command('account-balance')
+@click.option('-u', help='users email address')
+@with_appcontext
+def account_balance(u):
+  vms = get_model().list_vms_for_account(u)
+  payments = get_model().list_payments_for_account(u)
+  click.echo(".")
+  click.echo(".")
+  click.echo(get_account_balance(vms, payments, datetime.utcnow()))
+  click.echo(".")
+
+
 
 @bp.cli.command('cron-task')
 @with_appcontext