Advertisement
srcveiga

Untitled

Apr 29th, 2023
18
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.01 KB | None | 0 0
  1. def update_main_ohlc_fig(selected_stock,stock_data,):
  2.  
  3. stock_index = stocks.index(selected_stock)
  4. open_values = np.array(list(stock_data['volume'][f'stock{stock_index}_volume_df']['open'].values()))
  5.  
  6. close_values = np.array(list(stock_data['volume'][f'stock{stock_index}_volume_df']['close'].values()))
  7. pos = open_values < close_values
  8. neg = open_values > close_values
  9. x_new_volume = np.array(list(stock_data['volume'][f'stock{stock_index}_volume_df']['time'].values()))
  10. volume_diff_values = np.array(list(stock_data['volume'][f'stock{stock_index}_volume_df']['volume_diff'].values()))
  11. new_volume_pos_x_axis = x_new_volume[pos] if not np.all(pos == False) else None
  12. new_volume_pos_y_axis = volume_diff_values[pos] if not np.all(pos == False) else None
  13. new_volume_neg_x_axis = x_new_volume[neg] if not np.all(neg == False) else None
  14. new_volume_neg_y_axis = volume_diff_values[neg] if not np.all(neg == False) else None
  15.  
  16. latest_price = stock_data['last_price'][f'stock{stock_index}_last_price_df'][stocks[stock_index]+'_price']
  17. latest_price_change = stock_data['last_price_change'][f'stock{stock_index}_last_price_change_df'][stocks[stock_index]+'_price_change']
  18.  
  19. main_fig_ohlc_tracer = {'x':list(stock_data['ohlc'][f'stock{stock_index}_ohlc_df']['time'].values()),
  20. 'open':list(stock_data['ohlc'][f'stock{stock_index}_ohlc_df']['open'].values()),
  21. 'high':list(stock_data['ohlc'][f'stock{stock_index}_ohlc_df']['high'].values()),
  22. 'low':list(stock_data['ohlc'][f'stock{stock_index}_ohlc_df']['low'].values()),
  23. 'close':list(stock_data['ohlc'][f'stock{stock_index}_ohlc_df']['close'].values())}
  24.  
  25. main_fig_ma5_tracer = { 'x':list(stock_data['ma5'][f'stock{stock_index}_ma5_df']['time'].values()),
  26. 'y':list(stock_data['ma5'][f'stock{stock_index}_ma5_df']['MA5'].values())}
  27.  
  28. main_fig_ma10_tracer = {'x':list(stock_data['ma10'][f'stock{stock_index}_ma10_df']['time'].values()),
  29. 'y':list(stock_data['ma10'][f'stock{stock_index}_ma10_df']['MA10'].values())}
  30.  
  31. new_ma20_data = {'x':list(stock_data['ma20'][f'stock{stock_index}_ma20_df']['time'].values()),
  32. 'y':list(stock_data['ma20'][f'stock{stock_index}_ma20_df']['MA20'].values())}
  33.  
  34. main_fig_volume_diff_pos_tracer = {'x':new_volume_pos_x_axis,
  35. 'y':new_volume_pos_y_axis}
  36.  
  37. main_fig_volume_diff_neg_tracer = {'x':new_volume_neg_x_axis,
  38. 'y':new_volume_neg_y_axis}
  39.  
  40. main_fig_rsi_tracer = {'x':new_volume_neg_x_axis,
  41. 'y':new_volume_neg_y_axis}
  42.  
  43.  
  44. return [main_fig_ohlc_tracer, main_fig_ma5_tracer, main_fig_ma10_tracer, new_ma20_data, main_fig_volume_diff_pos_tracer ,main_fig_volume_diff_neg_tracer, main_fig_rsi_tracer],\
  45. latest_price, latest_price_change,{'color': '#18b800' if latest_price_change[stock_index] == '+' else '#ff3503','fontSize': '20px'}
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement