def get6Months(self, ticker):
# 1 day
history = p.chart(symbol=ticker,
timeframe='6m',
date=None,
token=self.secret,
version='v1',
filter='')
PATH = f'./stock/6m/{ticker}.json'
data = {}
for chunk in history:
formatted = {}
formatted['open'] = chunk['open']
formatted['close'] = chunk['close']
formatted['high'] = chunk['high']
formatted['low'] = chunk['low']
formatted['volume'] = chunk['volume']
data[chunk['date']] = formatted
with open(PATH, 'w') as JSON:
json.dump(data, JSON)
print('Done 6m.')
def get5Days(self, ticker):
# 10 min
history = p.chart(ticker,
timeframe='5dm',
date=None,
token=self.secret,
version='v1',
filter='')
PATH = f'./stock/5d/{ticker}.json'
data = {}
for chunk in history:
formatted = {}
formatted['price'] = chunk['average']
formatted['volume'] = chunk['volume']
date = chunk['date']
try:
data[date][chunk['minute']] = formatted
except:
data[date] = {chunk['minute']: formatted}
with open(PATH, 'w') as JSON:
json.dump(data, JSON)
print('Done 5d.')
def test_chart(self):
from pyEX import chart
from pyEX.common import PyEXception
with patch('requests.get') as mock:
mock.return_value = MagicMock()
mock.return_value.status_code = 200
chart('test')
try:
chart('test', '5')
assert False
except PyEXception:
pass
chart('test', None)
chart('test', None, '20150707')
def chart(stock): #downloads the price chart for each stock
close_label = stock + 'Close'
vwap_label = stock + 'vWAP'
chart = p.chart(stock, '5y')
print("STOCK DATA DOWNLOADING..", stock)
global chartlen
chartlen = len(chart)
daycount = 0
for day in reversed(chart):
df.loc[daycount, 'Date'] = day['date']
df.loc[daycount, close_label] = day['close']
df.loc[daycount, vwap_label] = day['vwap']
daycount += 1
def pyEXChart(query):
# chartData = pyEX.chart(symbol=query, timeframe='6m')
chartData = pyEX.chart(symbol=query)
price = []
date = []
for x in chartData:
price.append(x['close'])
date.append(x['date'])
return price, date
def test_chart(self):
from pyEX import chart
from pyEX.common import PyEXception
with patch("requests.get") as mock, patch("pickle.dump"):
mock.return_value = MagicMock()
mock.return_value.status_code = 200
chart("test")
try:
chart("test", "5")
assert False
except PyEXception:
pass
chart("test", None)
chart("test", None, "20150707")
def test_chart(self):
from pyEX import chart
chart(C)
def test_chart(self):
from pyEX import chart
with patch('requests.get') as mock:
mock.return_value = MagicMock()
mock.return_value.status_code = 200
chart('test')
