def create_stock(input: StockRequest, db: Session = Depends(get_db)):
"""
Create a new stock and save in the database.
"""
stock = StockItem()
stock.ticker = input.ticker
db.add(stock)
db.commit()
fetch_stock_data(stock.id)
return {
"code": "success",
"message": "stock created"
}
def update_table(input: StockRequest, db: Session = Depends(get_db)):
"""
Replace existing stocks data with the latest data from yfinance
"""
engine.execute('DELETE FROM Stocks') # delete all data from table
for tick in eval(input.ticker):
stock = StockItem()
stock.ticker = tick
db.add(stock)
db.commit()
fetch_stock_data(stock.id)
return {
"code": "success",
"message": "stock table updated"
}
