Import python venv for stability
This commit is contained in:
@@ -0,0 +1,159 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# yfinance - market data downloader
|
||||
# https://github.com/ranaroussi/yfinance
|
||||
#
|
||||
# Copyright 2017-2019 Ran Aroussi
|
||||
#
|
||||
# Licensed under the Apache License, Version 2.0 (the "License");
|
||||
# you may not use this file except in compliance with the License.
|
||||
# You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing, software
|
||||
# distributed under the License is distributed on an "AS IS" BASIS,
|
||||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
#
|
||||
|
||||
import json as _json
|
||||
|
||||
from . import utils
|
||||
from .config import YfConfig
|
||||
from .const import _BASE_URL_
|
||||
from .data import YfData
|
||||
from .exceptions import YFDataException
|
||||
|
||||
|
||||
class Search:
|
||||
def __init__(self, query, max_results=8, news_count=8, lists_count=8, include_cb=True, include_nav_links=False,
|
||||
include_research=False, include_cultural_assets=False, enable_fuzzy_query=False, recommended=8,
|
||||
session=None, timeout=30, raise_errors=True):
|
||||
"""
|
||||
Fetches and organizes search results from Yahoo Finance, including stock quotes and news articles.
|
||||
|
||||
Args:
|
||||
query: The search query (ticker symbol or company name).
|
||||
max_results: Maximum number of stock quotes to return (default 8).
|
||||
news_count: Number of news articles to include (default 8).
|
||||
lists_count: Number of lists to include (default 8).
|
||||
include_cb: Include the company breakdown (default True).
|
||||
include_nav_links: Include the navigation links (default False).
|
||||
include_research: Include the research reports (default False).
|
||||
include_cultural_assets: Include the cultural assets (default False).
|
||||
enable_fuzzy_query: Enable fuzzy search for typos (default False).
|
||||
recommended: Recommended number of results to return (default 8).
|
||||
session: Custom HTTP session for requests (default None).
|
||||
timeout: Request timeout in seconds (default 30).
|
||||
raise_errors: Raise exceptions on error (default True).
|
||||
"""
|
||||
self.session = session
|
||||
self._data = YfData(session=self.session)
|
||||
|
||||
self.query = query
|
||||
self.max_results = max_results
|
||||
self.enable_fuzzy_query = enable_fuzzy_query
|
||||
self.news_count = news_count
|
||||
self.timeout = timeout
|
||||
self.raise_errors = raise_errors
|
||||
|
||||
self.lists_count = lists_count
|
||||
self.include_cb = include_cb
|
||||
self.nav_links = include_nav_links
|
||||
self.enable_research = include_research
|
||||
self.enable_cultural_assets = include_cultural_assets
|
||||
self.recommended = recommended
|
||||
|
||||
self._logger = utils.get_yf_logger()
|
||||
|
||||
self._response = {}
|
||||
self._all = {}
|
||||
self._quotes = []
|
||||
self._news = []
|
||||
self._lists = []
|
||||
self._research = []
|
||||
self._nav = []
|
||||
|
||||
self.search()
|
||||
|
||||
def search(self) -> 'Search':
|
||||
"""Search using the query parameters defined in the constructor."""
|
||||
url = f"{_BASE_URL_}/v1/finance/search"
|
||||
params = {
|
||||
"q": self.query,
|
||||
"quotesCount": self.max_results,
|
||||
"enableFuzzyQuery": self.enable_fuzzy_query,
|
||||
"newsCount": self.news_count,
|
||||
"quotesQueryId": "tss_match_phrase_query",
|
||||
"newsQueryId": "news_cie_vespa",
|
||||
"listsCount": self.lists_count,
|
||||
"enableCb": self.include_cb,
|
||||
"enableNavLinks": self.nav_links,
|
||||
"enableResearchReports": self.enable_research,
|
||||
"enableCulturalAssets": self.enable_cultural_assets,
|
||||
"recommendedCount": self.recommended
|
||||
}
|
||||
|
||||
self._logger.debug(f'{self.query}: Yahoo GET parameters: {str(dict(params))}')
|
||||
|
||||
data = self._data.cache_get(url=url, params=params, timeout=self.timeout)
|
||||
if data is None or "Will be right back" in data.text:
|
||||
raise YFDataException("*** YAHOO! FINANCE IS CURRENTLY DOWN! ***")
|
||||
try:
|
||||
data = data.json()
|
||||
except _json.JSONDecodeError:
|
||||
if not YfConfig.debug.hide_exceptions:
|
||||
raise
|
||||
self._logger.error(f"{self.query}: 'search' fetch received faulty data")
|
||||
data = {}
|
||||
|
||||
self._response = data
|
||||
# Filter quotes to only include symbols
|
||||
self._quotes = [quote for quote in data.get("quotes", []) if "symbol" in quote]
|
||||
self._news = data.get("news", [])
|
||||
self._lists = data.get("lists", [])
|
||||
self._research = data.get("researchReports", [])
|
||||
self._nav = data.get("nav", [])
|
||||
|
||||
self._all = {"quotes": self._quotes, "news": self._news, "lists": self._lists, "research": self._research,
|
||||
"nav": self._nav}
|
||||
|
||||
return self
|
||||
|
||||
@property
|
||||
def quotes(self) -> 'list':
|
||||
"""Get the quotes from the search results."""
|
||||
return self._quotes
|
||||
|
||||
@property
|
||||
def news(self) -> 'list':
|
||||
"""Get the news from the search results."""
|
||||
return self._news
|
||||
|
||||
@property
|
||||
def lists(self) -> 'list':
|
||||
"""Get the lists from the search results."""
|
||||
return self._lists
|
||||
|
||||
@property
|
||||
def research(self) -> 'list':
|
||||
"""Get the research reports from the search results."""
|
||||
return self._research
|
||||
|
||||
@property
|
||||
def nav(self) -> 'list':
|
||||
"""Get the navigation links from the search results."""
|
||||
return self._nav
|
||||
|
||||
@property
|
||||
def all(self) -> 'dict[str,list]':
|
||||
"""Get all the results from the search results: filtered down version of response."""
|
||||
return self._all
|
||||
|
||||
@property
|
||||
def response(self) -> 'dict':
|
||||
"""Get the raw response from the search results."""
|
||||
return self._response
|
||||
Reference in New Issue
Block a user