From 45b56dc3cbf0fefb51ac4ea8f5814d035e012bbd Mon Sep 17 00:00:00 2001
From: Nathan Woodburn <github@nathan.woodburn.au>
Date: Wed, 8 Nov 2023 12:55:10 +1100
Subject: [PATCH] feat: Add stats page

---
 gift.py | 12 +++++++++++-
 main.py | 10 +++++-----
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/gift.py b/gift.py
index 1bd81c7..769afda 100644
--- a/gift.py
+++ b/gift.py
@@ -116,4 +116,14 @@ def gift(name,email,referer, ip):
         return "Error sending gift:<br>" + send_name.text
 
 
-    return True
\ No newline at end of file
+    return True
+
+def balance():
+    headers = {"Accept": "application/json", "Content-Type": "application/json"}
+    user_info = requests.get(nb_endpoint + "/api/user",headers=headers, cookies=cookies)
+    if user_info.status_code != 200:
+        return "Error getting user info:<br>" + user_info.text
+    user_info = user_info.json()
+    hns_balance = user_info['hns_balance']
+    hns_balance = int(hns_balance)/1000000
+    return hns_balance
diff --git a/main.py b/main.py
index dbd827c..e562c6b 100644
--- a/main.py
+++ b/main.py
@@ -76,11 +76,11 @@ def stats():
 
     # Loop through gifts
     referals = {}
-    for gift in gifts:
-        if gift['referer'] not in referals:
-            referals[gift['referer']] = 1
+    for gift_item in gifts:
+        if gift_item['referer'] not in referals:
+            referals[gift_item['referer']] = 1
         else:
-            referals[gift['referer']] += 1
+            referals[gift_item['referer']] += 1
         
     statsHTML = 'Total gifts: ' + str(len(gifts)) + '<br><br>'
     statsHTML += 'Referals:<br>'
@@ -88,7 +88,7 @@ def stats():
         statsHTML += referal + ': ' + str(referals[referal]) + '<br>'
 
 
-
+    statsHTML += '<br>Remaining HNS: ' + str(gift.balance()) + '<br>'
 
     return render_template('stats.html',address=address,stats=statsHTML)