kj7rrv/stockquotes
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Semi-PEP8

Browse Source 
Make it PEP 8 compliant where it does not break the API
This commit is contained in:
scoopgracie 2020-04-20 07:24:37 -07:00
parent c2341d8989
commit f506c44cdf
1 changed files with 66 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

117
stockquotes/__init__.py
View File

@ -1,6 +1,6 @@
#!/usr/bin/env python3
#stockquotes - Python module to pull stock quotes from Yahoo! Finance
#Copyright 2019 ScoopGracie. All rights reversed.
#Copyright 2020 ScoopGracie. All rights reversed.
#This is free and unencumbered software released into the public domain.
#Anyone is free to copy, modify, publish, use, compile, sell, or
@ -25,52 +25,71 @@
#OTHER DEALINGS IN THE SOFTWARE.
from bs4 import BeautifulSoup as bs
import requests, datetime
class StockDoesNotExistError(Exception):
pass
class NetworkError(Exception):
pass
class Stock:
def __init__(self, ticker):
try:
r=requests.get('https://finance.yahoo.com/quote/' + ticker + '/history', headers={'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3941.4 Safari/537.36'})
except:
raise NetworkError()
if r.status_code is 302:
raise StockDoesNotExistError(ticker)
try:
soup = bs(r.text, features="lxml")
self.symbol = soup.h1.string.split('(')[1].split(')')[0]
self.name = soup.h1.string.split('(')[0].strip()
rows = soup.table.tbody.find_all('tr')
self.historical=[]
for i in rows:
row=i.find_all('td')
try:
parsed = {
"date" : datetime.datetime.strptime(
row[0].span.string,
'%b %d, %Y'
),
"open" : float(row[1].span.string.replace(',', '')),
"high" : float(row[2].span.string.replace(',', '')),
"low" : float(row[3].span.string.replace(',', '')),
"close" : float(row[4].span.string.replace(',', '')),
"adjClose": float(row[5].span.string.replace(',', '')),
"volume" : int(row[6].span.string.replace(',', ''))
}
except:
continue
self.historical.append(parsed)
topData = soup.find(id='quote-header-info')
try:
self.currentPrice = float(topData.findAll('span')[11].string.replace(',', ''))
rawChange = topData.findAll('span')[12].string
except IndexError:
self.currentPrice = float(topData.findAll('span')[3].string.replace(',', ''))
rawChange = topData.findAll('span')[4].string
import requests
import datetime
self.increaseDollars = float(rawChange.split(' ')[0].replace(',', ''))
self.increasePercent = float(rawChange.split(' ')[1].replace(',', '').replace('(', '').replace(')', '').replace('%', ''))
except AttributeError as error:
raise StockDoesNotExistError(ticker) from error
class StockDoesNotExistError(Exception):
pass
class NetworkError(Exception):
pass
class Stock:
def __init__(self, ticker):
try:
r=requests.get(
'https://finance.yahoo.com/quote/{}/history'\
.format(ticker),
headers={
'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64)' +
' AppleWebKit/537.36 (KHTML, like Gecko)' +
' Chrome/79.0.3941.4 Safari/537.36'
})
except:
raise NetworkError()
if r.status_code is 302:
raise StockDoesNotExistError(ticker)
try:
soup = bs(r.text, features="lxml")
self.symbol = soup.h1.string.split('(')[1].split(')')[0]
self.name = soup.h1.string.split('(')[0].strip()
rows = soup.table.tbody.find_all('tr')
self.historical=[]
for i in rows:
row=i.find_all('td')
try:
parsed = {
"date" : datetime.datetime.strptime(
row[0].span.string,
'%b %d, %Y'
),
"open" : float(row[1].span.string.replace(',', '')),
"high" : float(row[2].span.string.replace(',', '')),
"low" : float(row[3].span.string.replace(',', '')),
"close" : float(row[4].span.string.replace(',', '')),
"adjClose": float(row[5].span.string.replace(',', '')),
"volume" : int(row[6].span.string.replace(',', ''))
}
except:
continue
self.historical.append(parsed)
topData = soup.find(id='quote-header-info')
try:
self.currentPrice = float(
topData.findAll('span')[11].string.replace(',', ''))
rawChange = topData.findAll('span')[12].string
except IndexError:
self.currentPrice = float(
topData.findAll('span')[3].string.replace(',', ''))
rawChange = topData.findAll('span')[4].string
self.increaseDollars = float(
rawChange.split(' ')[0].replace(',', ''))
self.increasePercent = float(
rawChange.split(' ')[1]\
.replace(',', '')\
.replace('(', '')\
.replace(')', '')\
.replace('%', ''))
except AttributeError as error:
raise StockDoesNotExistError(ticker) from error