ValueRaider
49 lines
1.7 KiB
Python
49 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import platformdirs as _ad
|
|
import datetime as _dt
|
|
import sys
|
|
import os
|
|
import yfinance
|
|
# from requests_ratelimiter import LimiterSession
|
|
# from pyrate_limiter import Duration, RequestRate, Limiter
|
|
|
|
_parent_dp = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
|
|
_src_dp = _parent_dp
|
|
sys.path.insert(0, _src_dp)
|
|
|
|
# Optional: see the exact requests that are made during tests:
|
|
# import logging
|
|
# logging.basicConfig(level=logging.DEBUG)
|
|
|
|
# Use adjacent cache folder for testing, delete if already exists and older than today
|
|
testing_cache_dirpath = os.path.join(_ad.user_cache_dir(), "py-yfinance-testing")
|
|
yfinance.set_tz_cache_location(testing_cache_dirpath)
|
|
if os.path.isdir(testing_cache_dirpath):
|
|
mtime = _dt.datetime.fromtimestamp(os.path.getmtime(testing_cache_dirpath))
|
|
if mtime.date() < _dt.date.today():
|
|
import shutil
|
|
shutil.rmtree(testing_cache_dirpath)
|
|
|
|
# Since switching to curl_cffi, the requests_ratelimiter|cache won't work.
|
|
session_gbl = None
|
|
|
|
# # Setup a session to only rate-limit
|
|
# history_rate = RequestRate(1, Duration.SECOND)
|
|
# limiter = Limiter(history_rate)
|
|
# session_gbl = LimiterSession(limiter=limiter)
|
|
|
|
# # Use this instead if you also want caching:
|
|
# from requests_cache import CacheMixin, SQLiteCache
|
|
# from requests_ratelimiter import LimiterMixin
|
|
# from requests import Session
|
|
# from pyrate_limiter import MemoryQueueBucket
|
|
# class CachedLimiterSession(CacheMixin, LimiterMixin, Session):
|
|
# pass
|
|
# cache_fp = os.path.join(testing_cache_dirpath, "unittests-cache")
|
|
# session_gbl = CachedLimiterSession(
|
|
# limiter=limiter,
|
|
# bucket_class=MemoryQueueBucket,
|
|
# backend=SQLiteCache(cache_fp, expire_after=_dt.timedelta(hours=1)),
|
|
# )
|