dlflannery commited on
Commit
8416c88
·
verified ·
1 Parent(s): 81859a5

Update app.py

Browse files

time zone correction for stock pull

Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -20,6 +20,7 @@ import pprint
20
  import pandas as pd
21
  import yfinance as yf
22
  from datetime import datetime, timedelta
 
23
 
24
 
25
  load_dotenv(override=True)
@@ -62,6 +63,12 @@ class MathReasoning(BaseModel):
62
  def Client():
63
  return OpenAI(api_key = key)
64
 
 
 
 
 
 
 
65
  def get_stock_report():
66
  try:
67
  stock_data = {}
@@ -94,7 +101,7 @@ def get_stock_report():
94
 
95
  def get_last_closing(symbol, timeout=10):
96
  try:
97
- today = datetime.today()
98
  if today.hour >= 16:
99
  today = today + timedelta(days=1)
100
  five_days_ago = today - timedelta(days=6)
 
20
  import pandas as pd
21
  import yfinance as yf
22
  from datetime import datetime, timedelta
23
+ from zoneinfo import ZoneInfo
24
 
25
 
26
  load_dotenv(override=True)
 
63
  def Client():
64
  return OpenAI(api_key = key)
65
 
66
+ def etz_today():
67
+ utime = datetime.utcnow()
68
+ aware_utime = utime.replace(tzinfo=ZoneInfo('UTC'))
69
+ etime = aware_utime.astimezone(ZoneInfo('America/New_York'))
70
+ return etime.today()
71
+
72
  def get_stock_report():
73
  try:
74
  stock_data = {}
 
101
 
102
  def get_last_closing(symbol, timeout=10):
103
  try:
104
+ today = etz_today()
105
  if today.hour >= 16:
106
  today = today + timedelta(days=1)
107
  five_days_ago = today - timedelta(days=6)