done
This commit is contained in:
parent
4803aa8518
commit
79d863b112
11
Dockerfile
Normal file
11
Dockerfile
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
FROM python:3
|
||||||
|
|
||||||
|
WORKDIR /usr/src/app
|
||||||
|
|
||||||
|
COPY requirements.txt ./
|
||||||
|
RUN pip install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
|
COPY templates templates
|
||||||
|
COPY main.py main.py
|
||||||
|
|
||||||
|
CMD [ "python", "main.py" ]
|
||||||
102
cache.csv
Normal file
102
cache.csv
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
Date,1 Mo,2 Mo,3 Mo,4 Mo,6 Mo,1 Yr,2 Yr,3 Yr,5 Yr,7 Yr,10 Yr,20 Yr,30 Yr
|
||||||
|
05/24/2024,5.56,5.53,5.46,5.51,5.44,5.21,4.93,4.71,4.53,4.49,4.46,4.65,4.57
|
||||||
|
05/23/2024,5.51,5.48,5.46,5.51,5.44,5.2,4.91,4.71,4.52,4.5,4.47,4.67,4.58
|
||||||
|
05/22/2024,5.49,5.48,5.45,5.5,5.43,5.16,4.86,4.64,4.47,4.44,4.43,4.63,4.55
|
||||||
|
05/21/2024,5.5,5.47,5.45,5.51,5.42,5.14,4.82,4.61,4.43,4.42,4.41,4.65,4.55
|
||||||
|
05/20/2024,5.5,5.47,5.45,5.51,5.43,5.15,4.82,4.62,4.46,4.44,4.44,4.68,4.58
|
||||||
|
05/17/2024,5.5,5.47,5.46,5.5,5.41,5.14,4.83,4.6,4.44,4.43,4.42,4.66,4.56
|
||||||
|
05/16/2024,5.51,5.47,5.45,5.53,5.41,5.13,4.78,4.58,4.4,4.39,4.38,4.62,4.52
|
||||||
|
05/15/2024,5.49,5.46,5.45,5.5,5.4,5.1,4.73,4.51,4.35,4.35,4.36,4.61,4.52
|
||||||
|
05/14/2024,5.5,5.47,5.44,5.51,5.43,5.16,4.81,4.62,4.46,4.45,4.45,4.69,4.59
|
||||||
|
05/13/2024,5.5,5.47,5.45,5.51,5.43,5.16,4.85,4.66,4.5,4.49,4.48,4.72,4.63
|
||||||
|
05/10/2024,5.51,5.47,5.47,5.51,5.43,5.17,4.87,4.65,4.52,4.51,4.5,4.74,4.64
|
||||||
|
05/09/2024,5.51,5.48,5.46,5.5,5.41,5.12,4.8,4.6,4.47,4.46,4.45,4.7,4.6
|
||||||
|
05/08/2024,5.51,5.47,5.45,5.5,5.41,5.13,4.84,4.63,4.5,4.49,4.48,4.73,4.64
|
||||||
|
05/07/2024,5.51,5.48,5.45,5.51,5.41,5.13,4.82,4.6,4.48,4.47,4.47,4.7,4.61
|
||||||
|
05/06/2024,5.51,5.48,5.45,5.5,5.42,5.12,4.82,4.64,4.48,4.48,4.49,4.73,4.64
|
||||||
|
05/03/2024,5.51,5.48,5.45,5.5,5.41,5.12,4.81,4.63,4.48,4.49,4.5,4.75,4.66
|
||||||
|
05/02/2024,5.51,5.47,5.46,5.5,5.42,5.16,4.87,4.71,4.57,4.57,4.58,4.82,4.72
|
||||||
|
05/01/2024,5.47,5.5,5.46,5.51,5.43,5.21,4.96,4.79,4.64,4.64,4.63,4.85,4.74
|
||||||
|
04/30/2024,5.48,5.51,5.46,5.45,5.44,5.25,5.04,4.87,4.72,4.71,4.69,4.9,4.79
|
||||||
|
04/29/2024,5.48,5.51,5.45,5.45,5.43,5.2,4.97,4.8,4.65,4.64,4.63,4.86,4.75
|
||||||
|
04/26/2024,5.48,5.51,5.46,5.45,5.4,5.21,4.96,4.84,4.68,4.68,4.67,4.89,4.78
|
||||||
|
04/25/2024,5.48,5.51,5.47,5.46,5.41,5.21,4.96,4.85,4.7,4.71,4.7,4.93,4.82
|
||||||
|
04/24/2024,5.49,5.5,5.46,5.44,5.4,5.17,4.89,4.78,4.64,4.66,4.65,4.88,4.78
|
||||||
|
04/23/2024,5.49,5.5,5.45,5.44,5.39,5.14,4.86,4.76,4.63,4.62,4.61,4.84,4.73
|
||||||
|
04/22/2024,5.49,5.5,5.42,5.44,5.39,5.16,4.97,4.81,4.66,4.65,4.62,4.84,4.72
|
||||||
|
04/19/2024,5.49,5.51,5.45,5.44,5.39,5.17,4.97,4.81,4.66,4.65,4.62,4.83,4.72
|
||||||
|
04/18/2024,5.49,5.52,5.46,5.44,5.39,5.18,4.98,4.83,4.68,4.67,4.64,4.85,4.74
|
||||||
|
04/17/2024,5.49,5.49,5.45,5.44,5.38,5.16,4.93,4.77,4.62,4.61,4.59,4.81,4.71
|
||||||
|
04/16/2024,5.49,5.51,5.45,5.44,5.39,5.18,4.97,4.83,4.69,4.69,4.67,4.88,4.77
|
||||||
|
04/15/2024,5.49,5.51,5.45,5.44,5.38,5.16,4.93,4.78,4.65,4.65,4.63,4.85,4.74
|
||||||
|
04/12/2024,5.48,5.5,5.45,5.42,5.36,5.13,4.88,4.7,4.54,4.53,4.5,4.73,4.61
|
||||||
|
04/11/2024,5.48,5.51,5.45,5.44,5.38,5.17,4.93,4.77,4.61,4.6,4.56,4.77,4.65
|
||||||
|
04/10/2024,5.49,5.5,5.45,5.44,5.4,5.19,4.97,4.77,4.61,4.59,4.55,4.76,4.64
|
||||||
|
04/09/2024,5.48,5.49,5.43,5.41,5.34,5.03,4.74,4.52,4.37,4.38,4.36,4.6,4.5
|
||||||
|
04/08/2024,5.48,5.49,5.43,5.41,5.35,5.07,4.78,4.6,4.43,4.43,4.42,4.65,4.55
|
||||||
|
04/05/2024,5.47,5.5,5.43,5.41,5.34,5.05,4.73,4.54,4.38,4.39,4.39,4.65,4.54
|
||||||
|
04/04/2024,5.47,5.49,5.41,5.4,5.32,5.0,4.65,4.46,4.3,4.31,4.31,4.57,4.47
|
||||||
|
04/03/2024,5.47,5.44,5.42,5.4,5.33,5.03,4.68,4.48,4.34,4.36,4.36,4.61,4.51
|
||||||
|
04/02/2024,5.49,5.45,5.42,5.4,5.34,5.05,4.7,4.51,4.35,4.37,4.36,4.61,4.51
|
||||||
|
04/01/2024,5.49,5.47,5.44,5.41,5.36,5.06,4.72,4.51,4.34,4.33,4.33,4.58,4.47
|
||||||
|
03/28/2024,5.49,5.48,5.46,5.42,5.38,5.03,4.59,4.4,4.21,4.2,4.2,4.45,4.34
|
||||||
|
03/27/2024,5.5,5.47,5.45,5.41,5.36,4.99,4.54,4.36,4.18,4.18,4.2,4.45,4.36
|
||||||
|
03/26/2024,5.5,5.47,5.46,5.41,5.36,5.0,4.56,4.38,4.22,4.23,4.24,4.49,4.4
|
||||||
|
03/25/2024,5.51,5.48,5.46,5.41,5.36,5.0,4.54,4.39,4.23,4.25,4.25,4.51,4.42
|
||||||
|
03/22/2024,5.51,5.47,5.46,5.4,5.34,4.98,4.59,4.36,4.2,4.22,4.22,4.47,4.39
|
||||||
|
03/21/2024,5.51,5.48,5.48,5.4,5.36,5.01,4.62,4.42,4.26,4.28,4.27,4.53,4.44
|
||||||
|
03/20/2024,5.5,5.47,5.47,5.41,5.36,5.01,4.59,4.41,4.25,4.28,4.27,4.53,4.45
|
||||||
|
03/19/2024,5.52,5.48,5.48,5.41,5.39,5.06,4.68,4.47,4.31,4.31,4.3,4.54,4.44
|
||||||
|
03/18/2024,5.52,5.48,5.48,5.41,5.39,5.06,4.73,4.52,4.36,4.35,4.34,4.57,4.46
|
||||||
|
03/15/2024,5.52,5.48,5.48,5.41,5.38,5.05,4.72,4.51,4.33,4.33,4.31,4.55,4.43
|
||||||
|
03/14/2024,5.52,5.48,5.48,5.42,5.38,5.04,4.68,4.46,4.29,4.3,4.29,4.54,4.44
|
||||||
|
03/13/2024,5.52,5.47,5.48,5.41,5.37,5.01,4.61,4.37,4.19,4.2,4.19,4.45,4.35
|
||||||
|
03/12/2024,5.52,5.48,5.48,5.41,5.37,5.0,4.58,4.33,4.15,4.16,4.16,4.42,4.31
|
||||||
|
03/11/2024,5.5,5.47,5.48,5.41,5.35,4.95,4.51,4.26,4.08,4.09,4.1,4.36,4.26
|
||||||
|
03/08/2024,5.51,5.48,5.46,5.4,5.34,4.92,4.48,4.25,4.06,4.08,4.09,4.36,4.26
|
||||||
|
03/07/2024,5.51,5.48,5.47,5.4,5.34,4.93,4.5,4.28,4.07,4.09,4.09,4.35,4.25
|
||||||
|
03/06/2024,5.5,5.47,5.47,5.4,5.35,4.95,4.55,4.32,4.12,4.12,4.11,4.36,4.24
|
||||||
|
03/05/2024,5.5,5.47,5.47,5.4,5.35,4.94,4.54,4.32,4.13,4.15,4.13,4.39,4.27
|
||||||
|
03/04/2024,5.51,5.49,5.48,5.42,5.37,4.98,4.61,4.39,4.21,4.23,4.22,4.48,4.36
|
||||||
|
03/01/2024,5.54,5.49,5.42,5.41,5.27,4.94,4.54,4.32,4.17,4.2,4.19,4.46,4.33
|
||||||
|
02/29/2024,5.53,5.5,5.45,5.43,5.3,5.01,4.64,4.43,4.26,4.28,4.25,4.51,4.38
|
||||||
|
02/28/2024,5.5,5.51,5.45,5.43,5.31,5.0,4.64,4.44,4.26,4.28,4.27,4.53,4.4
|
||||||
|
02/27/2024,5.5,5.52,5.45,5.47,5.33,5.03,4.7,4.5,4.32,4.34,4.31,4.57,4.44
|
||||||
|
02/26/2024,5.5,5.52,5.47,5.47,5.34,5.03,4.69,4.48,4.29,4.32,4.28,4.53,4.4
|
||||||
|
02/23/2024,5.49,5.51,5.46,5.46,5.32,5.0,4.67,4.45,4.28,4.28,4.26,4.51,4.37
|
||||||
|
02/22/2024,5.49,5.51,5.45,5.45,5.32,5.02,4.69,4.49,4.33,4.35,4.33,4.58,4.47
|
||||||
|
02/21/2024,5.5,5.5,5.44,5.45,5.32,4.98,4.64,4.43,4.3,4.33,4.32,4.59,4.49
|
||||||
|
02/20/2024,5.49,5.49,5.44,5.45,5.32,4.97,4.59,4.38,4.25,4.28,4.27,4.56,4.44
|
||||||
|
02/16/2024,5.48,5.51,5.44,5.45,5.31,4.98,4.64,4.43,4.29,4.31,4.3,4.58,4.45
|
||||||
|
02/15/2024,5.49,5.51,5.43,5.45,5.3,4.93,4.56,4.36,4.22,4.25,4.24,4.54,4.42
|
||||||
|
02/14/2024,5.48,5.51,5.43,5.45,5.31,4.94,4.56,4.38,4.25,4.27,4.27,4.57,4.45
|
||||||
|
02/13/2024,5.48,5.52,5.45,5.46,5.32,4.99,4.64,4.44,4.31,4.33,4.31,4.59,4.46
|
||||||
|
02/12/2024,5.49,5.51,5.43,5.43,5.27,4.87,4.46,4.25,4.13,4.16,4.17,4.48,4.37
|
||||||
|
02/09/2024,5.49,5.51,5.44,5.43,5.26,4.86,4.48,4.25,4.14,4.17,4.17,4.48,4.37
|
||||||
|
02/08/2024,5.49,5.51,5.44,5.42,5.24,4.83,4.46,4.22,4.12,4.15,4.15,4.47,4.36
|
||||||
|
02/07/2024,5.47,5.49,5.43,5.4,5.23,4.83,4.41,4.16,4.06,4.09,4.09,4.41,4.31
|
||||||
|
02/06/2024,5.48,5.5,5.44,5.41,5.23,4.82,4.39,4.14,4.03,4.07,4.09,4.39,4.29
|
||||||
|
02/05/2024,5.49,5.5,5.42,5.42,5.25,4.87,4.46,4.27,4.13,4.16,4.17,4.46,4.35
|
||||||
|
02/02/2024,5.49,5.51,5.43,5.42,5.22,4.81,4.36,4.14,3.99,4.02,4.03,4.33,4.22
|
||||||
|
02/01/2024,5.49,5.51,5.42,5.38,5.15,4.68,4.2,3.96,3.8,3.83,3.87,4.21,4.1
|
||||||
|
01/31/2024,5.53,5.46,5.42,5.4,5.18,4.73,4.27,4.05,3.91,3.95,3.99,4.34,4.22
|
||||||
|
01/30/2024,5.53,5.47,5.42,5.38,5.19,4.8,4.36,4.14,4.0,4.03,4.06,4.4,4.28
|
||||||
|
01/29/2024,5.53,5.46,5.42,5.37,5.19,4.76,4.29,4.1,3.97,4.02,4.08,4.42,4.31
|
||||||
|
01/26/2024,5.54,5.45,5.44,5.39,5.19,4.78,4.34,4.15,4.04,4.1,4.15,4.49,4.38
|
||||||
|
01/25/2024,5.54,5.48,5.44,5.39,5.19,4.76,4.28,4.12,4.01,4.07,4.14,4.49,4.38
|
||||||
|
01/24/2024,5.52,5.44,5.44,5.4,5.22,4.83,4.34,4.19,4.06,4.14,4.18,4.52,4.41
|
||||||
|
01/23/2024,5.53,5.46,5.45,5.38,5.21,4.81,4.31,4.16,4.06,4.11,4.14,4.48,4.38
|
||||||
|
01/22/2024,5.53,5.47,5.46,5.39,5.22,4.83,4.37,4.14,4.03,4.07,4.11,4.44,4.32
|
||||||
|
01/19/2024,5.54,5.47,5.45,5.39,5.21,4.84,4.39,4.18,4.08,4.12,4.15,4.47,4.36
|
||||||
|
01/18/2024,5.53,5.48,5.45,5.39,5.2,4.8,4.34,4.13,4.04,4.1,4.14,4.48,4.37
|
||||||
|
01/17/2024,5.54,5.47,5.47,5.4,5.2,4.8,4.34,4.12,4.02,4.07,4.1,4.42,4.31
|
||||||
|
01/16/2024,5.54,5.47,5.45,5.37,5.18,4.7,4.22,4.02,3.95,4.01,4.07,4.43,4.3
|
||||||
|
01/12/2024,5.55,5.47,5.45,5.37,5.16,4.65,4.14,3.92,3.84,3.91,3.96,4.32,4.2
|
||||||
|
01/11/2024,5.54,5.47,5.46,5.38,5.22,4.75,4.26,4.02,3.9,3.95,3.98,4.32,4.18
|
||||||
|
01/10/2024,5.53,5.46,5.46,5.39,5.23,4.82,4.37,4.1,3.99,4.01,4.04,4.35,4.2
|
||||||
|
01/09/2024,5.53,5.46,5.47,5.38,5.24,4.82,4.36,4.09,3.97,4.0,4.02,4.33,4.18
|
||||||
|
01/08/2024,5.54,5.48,5.49,5.39,5.24,4.82,4.36,4.11,3.97,3.99,4.01,4.33,4.17
|
||||||
|
01/05/2024,5.54,5.48,5.47,5.41,5.24,4.84,4.4,4.17,4.02,4.04,4.05,4.37,4.21
|
||||||
|
01/04/2024,5.56,5.48,5.48,5.41,5.25,4.85,4.38,4.14,3.97,3.99,3.99,4.3,4.13
|
||||||
|
01/03/2024,5.54,5.54,5.48,5.41,5.25,4.81,4.33,4.07,3.9,3.92,3.91,4.21,4.05
|
||||||
|
01/02/2024,5.55,5.54,5.46,5.41,5.24,4.8,4.33,4.09,3.93,3.95,3.95,4.25,4.08
|
||||||
|
248
main.py
Normal file
248
main.py
Normal file
@ -0,0 +1,248 @@
|
|||||||
|
import io
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import pickle
|
||||||
|
import threading
|
||||||
|
import time
|
||||||
|
import uuid
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
import pandas as pd
|
||||||
|
import requests
|
||||||
|
from flask import Flask, jsonify, render_template, request, redirect, make_response
|
||||||
|
|
||||||
|
|
||||||
|
app = Flask(__name__)
|
||||||
|
|
||||||
|
TREASURY_URL = "https://home.treasury.gov/resource-center/data-chart-center/interest-rates/daily-treasury-rates.csv/2024/all?type=daily_treasury_yield_curve&field_tdr_date_value=2024&page&_format=csv"
|
||||||
|
|
||||||
|
CURRENT_DATE = 20240524 # Set to friday since market is closed
|
||||||
|
|
||||||
|
RATES = None
|
||||||
|
|
||||||
|
|
||||||
|
class Side(str, Enum):
|
||||||
|
BUY = "BUY"
|
||||||
|
SELL = "SELL"
|
||||||
|
|
||||||
|
|
||||||
|
# Since we know all the types of treasuries, just enumerate them
|
||||||
|
class Instrument(str, Enum):
|
||||||
|
I_1MO = "1 Mo"
|
||||||
|
I_2MO = "2 Mo"
|
||||||
|
I_3MO = "3 Mo"
|
||||||
|
I_4MO = "4 Mo"
|
||||||
|
I_6MO = "6 Mo"
|
||||||
|
I_1YR = "1 Yr"
|
||||||
|
I_2YR = "2 Yr"
|
||||||
|
I_3YR = "3 Yr"
|
||||||
|
I_5YR = "5 Yr"
|
||||||
|
I_7YR = "7 Yr"
|
||||||
|
I_10YR = "10 Yr"
|
||||||
|
I_20YR = "12 Yr"
|
||||||
|
I_30YR = "30 Yr"
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_text(cls, txt):
|
||||||
|
if txt == "1 Mo":
|
||||||
|
return Instrument.I_1MO
|
||||||
|
elif txt == "2 Mo":
|
||||||
|
return Instrument.I_2MO
|
||||||
|
elif txt == "3 Mo":
|
||||||
|
return Instrument.I_3MO
|
||||||
|
elif txt == "4 Mo":
|
||||||
|
return Instrument.I_4MO
|
||||||
|
elif txt == "6 Mo":
|
||||||
|
return Instrument.I_6MO
|
||||||
|
elif txt == "1 Yr":
|
||||||
|
return Instrument.I_1YR
|
||||||
|
elif txt == "2 Yr":
|
||||||
|
return Instrument.I_2YR
|
||||||
|
elif txt == "3 Yr":
|
||||||
|
return Instrument.I_3YR
|
||||||
|
elif txt == "5 Yr":
|
||||||
|
return Instrument.I_5YR
|
||||||
|
elif txt == "7 Yr":
|
||||||
|
return Instrument.I_7YR
|
||||||
|
elif txt == "10 Yr":
|
||||||
|
return Instrument.I_20YR
|
||||||
|
elif txt == "12 Yr":
|
||||||
|
return Instrument.I_20YR
|
||||||
|
return Instrument.I_30YR
|
||||||
|
|
||||||
|
|
||||||
|
class Transaction:
|
||||||
|
def __init__(
|
||||||
|
self, user: str, instrument: Instrument, quantity: int, date: int, side: Side, rate: float
|
||||||
|
):
|
||||||
|
self.user = user
|
||||||
|
self.instrument: str = str(instrument.value)
|
||||||
|
self.side: str = str(side.value)
|
||||||
|
self.quantity = quantity
|
||||||
|
self.date = date
|
||||||
|
self.rate = rate
|
||||||
|
|
||||||
|
|
||||||
|
class User:
|
||||||
|
def __init__(self, balance: int = 100000) -> None:
|
||||||
|
self.ID = str(uuid.uuid4())
|
||||||
|
self.balance = balance
|
||||||
|
|
||||||
|
|
||||||
|
USERS: dict[str, User] = {}
|
||||||
|
TRANSACTIONS: dict[str, list[Transaction]] = {}
|
||||||
|
|
||||||
|
|
||||||
|
def background_saver():
|
||||||
|
while True:
|
||||||
|
with open("users.p", "wb") as fp:
|
||||||
|
pickle.dump(USERS, fp)
|
||||||
|
with open("transactions.p", "wb") as fp:
|
||||||
|
pickle.dump(TRANSACTIONS, fp)
|
||||||
|
|
||||||
|
time.sleep(5)
|
||||||
|
|
||||||
|
|
||||||
|
def get_treasury_rates_for_year(year: int) -> pd.DataFrame | None:
|
||||||
|
global RATES
|
||||||
|
|
||||||
|
use_cache = False
|
||||||
|
if RATES is None:
|
||||||
|
if os.path.isfile("./cache.csv"):
|
||||||
|
if time.time() - os.path.getmtime("./cache.csv") < 60000:
|
||||||
|
use_cache = True
|
||||||
|
else:
|
||||||
|
use_cache = True
|
||||||
|
|
||||||
|
if use_cache:
|
||||||
|
csv: pd.DataFrame = pd.read_csv("./cache.csv")
|
||||||
|
RATES = csv
|
||||||
|
else:
|
||||||
|
response = requests.get(TREASURY_URL)
|
||||||
|
response.raise_for_status()
|
||||||
|
raw_csv = response.content.decode("UTF-8")
|
||||||
|
csv_io = io.StringIO(raw_csv)
|
||||||
|
csv: pd.DataFrame = pd.read_csv(csv_io)
|
||||||
|
csv.to_csv("./cache.csv", index=False)
|
||||||
|
RATES = csv
|
||||||
|
|
||||||
|
ret = {}
|
||||||
|
cols = list(RATES.columns.values)
|
||||||
|
for col in cols:
|
||||||
|
ret[col] = RATES[col].tolist()
|
||||||
|
|
||||||
|
dates = ret["Date"]
|
||||||
|
del ret["Date"]
|
||||||
|
return ret, dates
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/api/transactions/<id>")
|
||||||
|
def get_transactions_for_user(id):
|
||||||
|
t = TRANSACTIONS.get(id)
|
||||||
|
if t is None:
|
||||||
|
return jsonify([])
|
||||||
|
|
||||||
|
return jsonify([{"term": tx.instrument, "quantity": tx.quantity, "side": tx.side, "date": tx.date, "rate": tx.rate} for tx in t])
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/api/buy", methods=["POST"])
|
||||||
|
def buy():
|
||||||
|
content = request.json
|
||||||
|
print(content)
|
||||||
|
q = int(content["quantity"])
|
||||||
|
i = content["instrument"]
|
||||||
|
user_id = content["user_id"]
|
||||||
|
cost = q * 100
|
||||||
|
if USERS.get(user_id) is not None:
|
||||||
|
if cost > USERS[user_id].balance:
|
||||||
|
return "You do not have the funds for this transaction", 400
|
||||||
|
else:
|
||||||
|
return "User does not exist", 404
|
||||||
|
|
||||||
|
current_rate = RATES[i].tolist()[-1]
|
||||||
|
|
||||||
|
USERS[user_id].balance = USERS[user_id].balance - cost
|
||||||
|
t = Transaction(user_id, Instrument.from_text(i), q, time.time(), Side.BUY, current_rate)
|
||||||
|
|
||||||
|
if TRANSACTIONS.get(user_id) is None:
|
||||||
|
TRANSACTIONS[user_id] = [t]
|
||||||
|
else:
|
||||||
|
TRANSACTIONS[user_id].append(t)
|
||||||
|
|
||||||
|
return jsonify(t)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/api/login", methods=["POST"])
|
||||||
|
def login():
|
||||||
|
content = request.json
|
||||||
|
id = content.get("id")
|
||||||
|
if id is None:
|
||||||
|
return "ID cannot be empty", 400
|
||||||
|
u = USERS.get(id)
|
||||||
|
if u is None:
|
||||||
|
return "User does not exist", 404
|
||||||
|
resp = make_response(redirect("/"))
|
||||||
|
resp.set_cookie("userid", u.ID)
|
||||||
|
return resp
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/api/createuser", methods=["POST"])
|
||||||
|
def createuser():
|
||||||
|
u = User()
|
||||||
|
USERS[u.ID] = u
|
||||||
|
resp = make_response(redirect("/"))
|
||||||
|
resp.set_cookie("userid", u.ID)
|
||||||
|
return resp
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/register")
|
||||||
|
def register():
|
||||||
|
if "userid" in request.cookies:
|
||||||
|
return redirect("/")
|
||||||
|
return render_template("register.html")
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/logout")
|
||||||
|
def logout():
|
||||||
|
if "userid" not in request.cookies:
|
||||||
|
return redirect("/register")
|
||||||
|
resp = make_response(redirect("/register"))
|
||||||
|
resp.set_cookie("userid", "", expires=0)
|
||||||
|
return resp
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/")
|
||||||
|
def index():
|
||||||
|
if "userid" not in request.cookies:
|
||||||
|
return redirect("/register")
|
||||||
|
uid = request.cookies.get("userid")
|
||||||
|
if uid not in USERS:
|
||||||
|
resp = make_response(redirect("/register"))
|
||||||
|
resp.set_cookie("userid", "", expires=0)
|
||||||
|
return resp
|
||||||
|
rates, dates = get_treasury_rates_for_year(2024)
|
||||||
|
return render_template("index.html", rates=[rates], dates=dates, userid=uid, balance=USERS[uid].balance)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
# Load up files if they exist
|
||||||
|
if os.path.isfile("./users.p"):
|
||||||
|
print("Loading users")
|
||||||
|
with open("./users.p", "rb") as fp:
|
||||||
|
USERS = pickle.load(fp)
|
||||||
|
|
||||||
|
if os.path.isfile("./transactions.p"):
|
||||||
|
print("Loading transactions")
|
||||||
|
with open("./transactions.p", "rb") as fp:
|
||||||
|
TRANSACTIONS = pickle.load(fp)
|
||||||
|
|
||||||
|
# Start persistence
|
||||||
|
thread = threading.Thread(target=background_saver)
|
||||||
|
thread.start()
|
||||||
|
|
||||||
|
# Start Server
|
||||||
|
app.run(port=8008)
|
||||||
|
|
||||||
|
|
||||||
|
thread = threading.Thread(target=background_saver)
|
||||||
|
thread.start()
|
||||||
0
requirements.txt
Normal file
0
requirements.txt
Normal file
50
templates/base.html
Normal file
50
templates/base.html
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="en">
|
||||||
|
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||||
|
<title>DWSFi</title>
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/js/tabler.min.js"></script>
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/css/tabler.min.css">
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/css/tabler-flags.min.css">
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/css/tabler-payments.min.css">
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/css/tabler-vendors.min.css">
|
||||||
|
</head>
|
||||||
|
|
||||||
|
<body>
|
||||||
|
<div class="page">
|
||||||
|
<header class="navbar navbar-expand-sm navbar-light d-print-none">
|
||||||
|
<div class="container-xl">
|
||||||
|
<h1 class="navbar-brand navbar-brand-autodark d-none-navbar-horizontal pe-0 pe-md-3">
|
||||||
|
<a href="#">
|
||||||
|
<img src="https://images.rawpixel.com/image_png_800/czNmcy1wcml2YXRlL3Jhd3BpeGVsX2ltYWdlcy93ZWJzaXRlX2NvbnRlbnQvbHIvdjEwNDktMjIucG5n.png" width="110" height="32" alt="Tabler" class="navbar-brand-image" />
|
||||||
|
</a>
|
||||||
|
<span>DWSFi</span>
|
||||||
|
</h1>
|
||||||
|
|
||||||
|
<div class="navbar-nav flex-row order-md-last">
|
||||||
|
<div class="nav-item">
|
||||||
|
<a href="/logout" class="nav-link d-flex lh-1 text-reset">
|
||||||
|
<div class="d-xl-block ps-2">
|
||||||
|
{% if userid is not none %}
|
||||||
|
<div>{{userid}} - Logout</div>
|
||||||
|
{% endif %}
|
||||||
|
</div>
|
||||||
|
</a>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</header>
|
||||||
|
<div class="page-wrapper">
|
||||||
|
<div class="page-body">
|
||||||
|
<div class="container-xl">
|
||||||
|
{% block content %}{% endblock %}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
|
||||||
|
</html>
|
||||||
|
|
||||||
220
templates/index.html
Normal file
220
templates/index.html
Normal file
@ -0,0 +1,220 @@
|
|||||||
|
{% extends 'base.html' %}
|
||||||
|
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
<div class="row row-deck row-cards">
|
||||||
|
<div class="col-sm-6 col-lg-3">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-body">
|
||||||
|
<div class="d-flex align-items-center">
|
||||||
|
<div class="subheader">Account Balance</div>
|
||||||
|
<div class="ms-auto lh-1">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="d-flex align-items-baseline">
|
||||||
|
<div class="h1 mb-0 me-2">${{balance}}</div>
|
||||||
|
<div class="me-auto">
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="col-lg-12">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-header border-0">
|
||||||
|
<div class="card-title">Treasury Yields</div>
|
||||||
|
</div>
|
||||||
|
<div class="card-body">
|
||||||
|
|
||||||
|
<div id="chart-yields" class="chart-lg"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="col-lg-8">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-header">
|
||||||
|
<h3 class="card-title">Transactions</h3>
|
||||||
|
</div>
|
||||||
|
<div class="table-responsive">
|
||||||
|
<table class="table card-table table-vcenter text-nowrap datatable">
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th>Term</th>
|
||||||
|
<th>Quantity</th>
|
||||||
|
<th>Date</th>
|
||||||
|
<th>Side</th>
|
||||||
|
<th>Rate</th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody id="txbody">
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="col-lg-4">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-header">
|
||||||
|
<h3 class="card-title">Buy/Sell</h3>
|
||||||
|
</div>
|
||||||
|
<div class="card-body">
|
||||||
|
<div class="mb-3">
|
||||||
|
<div class="form-label">Term</div>
|
||||||
|
<select class="form-select" id="term">
|
||||||
|
<option value="1 Mo">1 Mo</option>
|
||||||
|
<option value="2 Mo">2 Mo</option>
|
||||||
|
<option value="3 Mo">3 Mo</option>
|
||||||
|
<option value="4 Mo">4 Mo</option>
|
||||||
|
<option value="6 Mo">6 Mo</option>
|
||||||
|
<option value="1 Yr">1 Yr</option>
|
||||||
|
<option value="2 Yr">2 Yr</option>
|
||||||
|
<option value="3 Yr">3 Yr</option>
|
||||||
|
<option value="5 Yr">5 Yr</option>
|
||||||
|
<option value="7 Yr">7 Yr</option>
|
||||||
|
<option value="10 Yr">10 Yr</option>
|
||||||
|
<option value="12 Yr">12 Yr</option>
|
||||||
|
<option value="30 Yr">30 Yr</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
<div class="mb-3">
|
||||||
|
<label class="form-label">Quantity</label>
|
||||||
|
<input type="number" class="form-control" name="quantity" id="quantity" placeholder="100" />
|
||||||
|
</div>
|
||||||
|
<div class="mb-3">
|
||||||
|
<label class="form-label">Side</label>
|
||||||
|
<select class="form-select">
|
||||||
|
<option value="BUY">Buy</option>
|
||||||
|
<option value="SELL">Sell</option>
|
||||||
|
</select>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button id="buybtn" class="btn btn-primary ms-auto">Submit Order</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/@tabler/core@1.0.0-beta17/dist/libs/apexcharts/dist/apexcharts.min.js" defer></script>
|
||||||
|
<script>
|
||||||
|
|
||||||
|
document.getElementById("buybtn").addEventListener('click', () => {
|
||||||
|
var r = new XMLHttpRequest();
|
||||||
|
r.open("POST", "/api/buy", true);
|
||||||
|
r.onload = function () {
|
||||||
|
window.location.reload();
|
||||||
|
}
|
||||||
|
|
||||||
|
r.setRequestHeader("Content-Type", "application/json")
|
||||||
|
// Send data in below way from JS
|
||||||
|
r.send(JSON.stringify({
|
||||||
|
"quantity": document.getElementById('quantity').value,
|
||||||
|
"instrument": document.getElementById("term").value,
|
||||||
|
"user_id": "{{userid}}"
|
||||||
|
}));
|
||||||
|
});
|
||||||
|
|
||||||
|
document.addEventListener("DOMContentLoaded", function() {
|
||||||
|
|
||||||
|
window.ApexCharts && (new ApexCharts(document.getElementById('chart-yields'), {
|
||||||
|
chart: {
|
||||||
|
type: "line",
|
||||||
|
fontFamily: 'inherit',
|
||||||
|
parentHeightOffset: 0,
|
||||||
|
height: 512,
|
||||||
|
toolbar: {
|
||||||
|
show: true,
|
||||||
|
},
|
||||||
|
animations: {
|
||||||
|
enabled: false
|
||||||
|
},
|
||||||
|
},
|
||||||
|
fill: {
|
||||||
|
opacity: 1,
|
||||||
|
},
|
||||||
|
stroke: {
|
||||||
|
width: 2,
|
||||||
|
lineCap: "round",
|
||||||
|
curve: "straight",
|
||||||
|
},
|
||||||
|
series: [
|
||||||
|
{% for rate in rates %}
|
||||||
|
{% for name, data in rate.items() %}
|
||||||
|
{
|
||||||
|
name: "{{name}}",
|
||||||
|
data: {{data}}
|
||||||
|
},
|
||||||
|
{% endfor %}
|
||||||
|
{% endfor %}
|
||||||
|
],
|
||||||
|
tooltip: {
|
||||||
|
theme: 'dark'
|
||||||
|
},
|
||||||
|
grid: {
|
||||||
|
padding: {
|
||||||
|
top: -20,
|
||||||
|
right: 0,
|
||||||
|
left: -4,
|
||||||
|
bottom: -4
|
||||||
|
},
|
||||||
|
strokeDashArray: 4,
|
||||||
|
},
|
||||||
|
xaxis: {
|
||||||
|
labels: {
|
||||||
|
padding: 0,
|
||||||
|
},
|
||||||
|
tooltip: {
|
||||||
|
enabled: false
|
||||||
|
},
|
||||||
|
type: 'datetime',
|
||||||
|
},
|
||||||
|
yaxis: {
|
||||||
|
labels: {
|
||||||
|
padding: 4
|
||||||
|
},
|
||||||
|
},
|
||||||
|
labels: [
|
||||||
|
{% for each in dates %}
|
||||||
|
"{{each}}",
|
||||||
|
{% endfor %}
|
||||||
|
],
|
||||||
|
|
||||||
|
legend: {
|
||||||
|
show: true,
|
||||||
|
position: 'bottom',
|
||||||
|
offsetY: 12,
|
||||||
|
markers: {
|
||||||
|
width: 10,
|
||||||
|
height: 10,
|
||||||
|
radius: 100,
|
||||||
|
},
|
||||||
|
itemMargin: {
|
||||||
|
horizontal: 8,
|
||||||
|
vertical: 8
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})).render();
|
||||||
|
|
||||||
|
let xhr = new XMLHttpRequest();
|
||||||
|
xhr.withCredentials = true;
|
||||||
|
xhr.open('GET', 'http://localhost:8008/api/transactions/{{userid}}');
|
||||||
|
|
||||||
|
xhr.onload = function() {
|
||||||
|
console.log(xhr.response);
|
||||||
|
var jsonResponse = JSON.parse(xhr.response);
|
||||||
|
for (let i = 0; i < jsonResponse.length; i++) {
|
||||||
|
var newRow="<tr><td>"+ jsonResponse[i].term + "</td><td>" + jsonResponse[i].quantity + "</td><td>" + jsonResponse[i].date + "</td><td>" + jsonResponse[i].side + "</td><td>"+ jsonResponse[i].rate +"</td></tr>";
|
||||||
|
document.getElementById('txbody').innerHTML += newRow;
|
||||||
|
}
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
xhr.send();
|
||||||
|
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
|
||||||
|
|
||||||
|
{% endblock %}
|
||||||
64
templates/register.html
Normal file
64
templates/register.html
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
|
||||||
|
{% extends 'base.html' %}
|
||||||
|
|
||||||
|
|
||||||
|
{% block content %}
|
||||||
|
<div class="row row-deck row-cards">
|
||||||
|
<h1>Registration</h1>
|
||||||
|
<div class="col-sm-6 col-lg-3">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-body">
|
||||||
|
<h4>Registration will assign you a UUID, please remember this UUID to login!</h4>
|
||||||
|
|
||||||
|
<button id="regbtn" class="btn btn-primary ms-auto">Register</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="col-sm-6 col-lg-3">
|
||||||
|
<div class="card">
|
||||||
|
<div class="card-body">
|
||||||
|
<h4>If you already have a UUID, enter it below to login</h4>
|
||||||
|
|
||||||
|
<div class="mb-3">
|
||||||
|
<label class="form-label">Enter ID</label>
|
||||||
|
<input type="text" class="form-control" placeholder="User ID" id="uuidid" />
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button id="loginbtn" class="btn btn-primary ms-auto">Login</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
document.getElementById("regbtn").addEventListener('click', () => {
|
||||||
|
var r = new XMLHttpRequest();
|
||||||
|
r.open("POST", "/api/createuser", true);
|
||||||
|
r.onreadystatechange = function () {
|
||||||
|
r.onload = function () {
|
||||||
|
window.location.reload();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
r.setRequestHeader("Content-Type", "application/json")
|
||||||
|
// Send data in below way from JS
|
||||||
|
r.send();
|
||||||
|
});
|
||||||
|
|
||||||
|
document.getElementById("loginbtn").addEventListener('click', () => {
|
||||||
|
var r = new XMLHttpRequest();
|
||||||
|
r.open("POST", "/api/login", true);
|
||||||
|
r.onreadystatechange = function () {
|
||||||
|
r.onload = function () {
|
||||||
|
window.location.reload();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
r.setRequestHeader("Content-Type", "application/json")
|
||||||
|
// Send data in below way from JS
|
||||||
|
r.send(JSON.stringify({
|
||||||
|
"id": document.getElementById('uuidid').value
|
||||||
|
}));
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
{% endblock %}
|
||||||
BIN
transactions.p
Normal file
BIN
transactions.p
Normal file
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user