Merge pull request #35 from tpokorra/TP-202109-graph_for_wallet

show graph for the whole wallet
This commit was merged in pull request #35.
This commit is contained in:
2021-09-20 06:34:53 +02:00
committed by GitHub
4 changed files with 182 additions and 5 deletions
+13
View File
@@ -150,3 +150,16 @@ class Calc:
total_tax_free += out["value_tax_free"]
return (total_investment, current_value, total_tax_free, rateEUR.rate, rateUSD.rate, last_updated, out)
def GetWalletGraphs(self, userid):
out = {}
out["graphs"] = []
out["graphs"].append({"id": "w1", "label": "1 week", "period": "number_of_days=7"})
out["graphs"].append({"id": "m1", "active": "active", "label": "1 month", "period": "number_of_days=30"})
out["graphs"].append({"id": "m6", "label": "6 months", "period": "number_of_days=180"})
out["graphs"].append({"id": "y1", "label": "1 year", "period": "number_of_days=365"})
out["graphs"].append({"id": "y3", "label": "3 years", "period": "number_of_days=" + str(365*3)})
out["graphs"].append({"id": "y5", "label": "5 years", "period": "number_of_days=" + str(365*5)})
out["graphs"].append({"id": "y10", "label": "10 years", "period": "number_of_days=" + str(365*10)})
return (out["graphs"])
+118
View File
@@ -7,6 +7,7 @@ from matplotlib.figure import Figure
import matplotlib.ticker as plticker
import matplotlib.dates as mdates
from django.http import HttpResponse
from decimal import Decimal
class Graph:
@@ -113,3 +114,120 @@ class Graph:
canvas = FigureCanvasAgg(fig)
canvas.print_png(response)
return response
def wallet_graph_days(self, Userid, Fiat, DayDiff):
x = []
y = []
y2 = []
with connection.cursor() as cursor:
# first get the relevant crypto currencies
sql = "SELECT distinct crypto_currency FROM transaction where owner_id = %s"
cursor.execute(sql, [Userid,])
currency_rows = cursor.fetchall()
cryptos = []
for currency_row in currency_rows:
cryptos.append(currency_row[0])
# loop through the days
start_date = datetime.datetime.today().replace(hour=0, minute=0, second=0, microsecond=0) - datetime.timedelta(days=int(DayDiff))
end_date = datetime.datetime.today().replace(hour=0, minute=0, second=0, microsecond=0)
delta = datetime.timedelta(days=1)
while start_date <= end_date:
date = start_date
start_date += delta
total_fiat_amount = Decimal(0)
total_fiat_value = Decimal(0)
# get the average exchange rates for this day
sql = """SELECT AVG(rate), crypto_currency
FROM exchangerate
WHERE fiat_currency = %s
AND crypto_currency IN %s
AND datetime_valid BETWEEN %s AND %s
GROUP BY crypto_currency"""
cursor.execute(sql, [Fiat, cryptos, date, date + delta])
rate_rows = cursor.fetchall()
# get the current amount on that day
sql = """SELECT crypto_currency, transaction_type, SUM(crypto_amount), SUM(fiat_amount), SUM(crypto_fee), SUM(fiat_fee)
FROM transaction
WHERE owner_id = %s
AND date_valid <= %s
GROUP BY crypto_currency, transaction_type"""
cursor.execute(sql, [Userid, date])
amount_rows = cursor.fetchall()
for rate_row in rate_rows:
rate = rate_row[0]
crypto = rate_row[1]
total_crypto_amount = Decimal(0)
for amount_row in amount_rows:
tr_crypto = amount_row[0]
if tr_crypto == crypto:
tr_type = amount_row[1]
tr_amount = amount_row[2]
tr_fiat_amount = amount_row[3]
tr_crypto_fee = amount_row[4]
tr_fiat_fee = amount_row[5]
if tr_amount:
if tr_type == "S":
total_crypto_amount -= tr_amount
total_fiat_amount += tr_fiat_amount
elif tr_type == "B":
total_crypto_amount += tr_amount
total_fiat_amount -= tr_fiat_amount
if tr_crypto_fee:
total_crypto_amount -= tr_crypto_fee
if tr_fiat_fee:
total_fiat_value -= tr_fiat_fee
total_fiat_amount -= tr_fiat_fee
total_fiat_value += total_crypto_amount * rate
x.append(date)
y.append(total_fiat_value)
y2.append(total_fiat_amount)
if len(x) == 0:
raise Exception("no data for this time range in the database")
fig_size = plt.rcParams["figure.figsize"]
fig_size[0] = 5
fig_size[1] = 4
plt.rcParams["figure.figsize"] = fig_size
fig = Figure()
ax = fig.add_subplot(1, 1, 1)
ax.plot(x, y, color='green', linestyle='dashed', linewidth = 2,
marker='o', markerfacecolor='blue', markersize=2)
ax.plot(x, y2, color='blue', linestyle='dashed', linewidth = 2,
marker='o', markerfacecolor='red', markersize=2)
if int(DayDiff) <= 30:
hours = mdates.DayLocator(interval = 6)
elif int(DayDiff) <= 180:
hours = mdates.DayLocator(interval = 30)
elif int(DayDiff) <= 360:
hours = mdates.DayLocator(interval = int((int(DayDiff)/6)))
else:
hours = mdates.DayLocator(interval = 360)
h_fmt = mdates.DateFormatter('%b %d\n%Y')
ax.xaxis.set_major_locator(hours)
ax.xaxis.set_major_formatter(h_fmt)
#ax.set_xlabel("time")
ax.set_ylabel(Fiat)
Crypto = "Wallet"
ax.set_title(('%s in %s' % (Crypto, Fiat)))
response = HttpResponse(content_type = 'image/png')
canvas = FigureCanvasAgg(fig)
canvas.print_png(response)
return response
+40 -1
View File
@@ -118,8 +118,47 @@
{% endfor %}
<br/><br/><br/>
<input type="checkbox" id="accWALLET" />
<label for="accWALLET">
My Wallet
{% if wallet.rateEUR %}{{wallet.rateEUR|formatcurrency}} EUR{% endif %}
</label>
<div class="content">
<ul class="tab-nav" id="tabnavWallet">
<li>
<a class="button" href="#w1Wallet">1 week</a>
</li>
<li>
<a class="button active" href="#m1Wallet">1 month</a>
</li>
<li>
<a class="button" href="#m6Wallet">6 months</a>
</li>
<li>
<a class="button" href="#y1Wallet">1 year</a>
</li>
<li>
<a class="button" href="#y3Wallet">3 years</a>
</li>
<li>
<a class="button" href="#y5Wallet">5 years</a>
</li>
<li>
<a class="button" href="#y10Wallet">10 years</a>
</li>
</ul>
<div class="tab-content" id="tabcontentWallet">
{% for g in wallet.graphs %}
<div class="tab-pane {{g.active}}" id="{{ g.id }}Wallet">
<h4>{{ g.label }}</h4>
<img src="/graph?crypto=WALLET&fiat=EUR&{{g.period}}" loading="lazy"><br/>
</div>
{% endfor %}
</div>
</div>
<br/><br/><br/>
</div>
{% endblock %}
+7
View File
@@ -25,8 +25,11 @@ def monitor(request):
django_timezone = pytz.timezone(settings.TIME_ZONE)
wallet = {"rateEUR": current_value, "graphs": calc.GetWalletGraphs(request.user.id)}
return render(request,"monitor.html",
{'cryptos':cryptos,
'wallet': wallet,
'total_tax_free': total_tax_free,
'total_investment': total_investment,
'current_value': current_value,
@@ -42,6 +45,10 @@ def graph(request):
Crypto = request.GET['crypto']
Fiat = request.GET['fiat']
if Crypto == "WALLET":
if 'number_of_days' in request.GET:
return Graph().wallet_graph_days(request.user.id, Fiat, request.GET['number_of_days'])
else:
if 'number_of_days' in request.GET:
return Graph().graph_days(Crypto, Fiat, request.GET['number_of_days'])
if 'number_of_hours' in request.GET: