| import logo from './logo.svg'; | |
| import './App.css'; | |
| import {useState, useEffect} from 'react'; | |
| import axios from 'axios'; | |
| import {BarChart, Bar, YAxis} from 'recharts'; | |
| function App() { | |
| const [history, setHistory] = useState([]) | |
| const [symbol, setSymbol] = useState('MSFT') | |
| const getData = async () => { | |
| const {data} = await axios | |
| .get(`http://localhost:8000/history/${symbol}`) | |
| setHistory(data) | |
| console.log(data) | |
| } | |
| useEffect(() => { | |
| getData(); | |
| }, []); | |
| return ( | |
| <div className="App"> | |
| <> | |
| <input type="text" onChange={event => setSymbol(event.target.value)} /> | |
| <button onClick={event => getData()}>Get History</button> | |
| <BarChart width={800} height={600} data={history}> | |
| <YAxis /> | |
| <Bar dataKey="close" fill="#ff0000" /> | |
| </BarChart> | |
| </> | |
| </div> | |
| ); | |
| } | |
| export default App; |
| from fastapi import FastAPI | |
| from fastapi.middleware.cors import CORSMiddleware | |
| from pydantic import BaseModel | |
| from mongita import MongitaClientDisk | |
| import yfinance as yf | |
| class PortfolioItem(BaseModel): | |
| name: str | |
| symbol: str | |
| no_shares: int | |
| app = FastAPI() | |
| client = MongitaClientDisk() | |
| db = client.db | |
| portfolio = db.portfolio | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=["*"], # DON'T DO THIS IN PRODUCTION! | |
| allow_credentials=True, | |
| allow_headers=["*"], | |
| allow_methods=["*"] | |
| ) | |
| def get_current(symbol, shares): | |
| stock_info = yf.Ticker(symbol).info | |
| return {'current_price': stock_info['currentPrice'] * shares} | |
| @app.get('/') | |
| async def home(): | |
| return {'home': 'sweet home'} | |
| @app.get('/current/{symbol}') | |
| async def current(symbol: str, shares: int = 1): | |
| return get_current(symbol, shares) | |
| @app.get('/history/{symbol}') | |
| async def history(symbol: str): | |
| stock_history = yf.Ticker(symbol).history('1mo') | |
| stock_close = stock_history['Close'] | |
| return [ | |
| {"name": ts, "close": closing} for ts, closing in zip( | |
| [tsp.strftime('%Y-%m-%d') for tsp in stock_close.index], | |
| [float(c) for c in stock_close.values] | |
| ) | |
| ] | |
| @app.get('/portfolio') | |
| async def get_portfolio(): | |
| trades = portfolio.find({}) | |
| return [ | |
| { | |
| "name": trade["name"], | |
| "symbol": trade["symbol"], | |
| "no_shares": trade["no_shares"], | |
| "value": get_current(trade["symbol"], trade["no_shares"]) | |
| } for trade in trades | |
| ] | |
| @app.post('/portfolio') | |
| async def post_portfolio(item: PortfolioItem): | |
| portfolio.insert_one(item.dict()) | |
| return item |