def get_session():
CACHE_FOLDER.mkdir(exist_ok=True)
cache = FileCache(str(CACHE_FOLDER), forever=True)
cache.set("foo", b"bar")
assert cache.get("foo") == b"bar"
session = RateLimitingSession()
# session.headers.update({"x-api-key": "something-something-darkside"})
session.mount(
"https://www.metlink.org.nz/",
CacheControlAdapter(heuristic=BetterExpiresAfter(days=7), cache=cache),
)
session.mount(
METLINK_API_URL_PREFIX,
CacheControlAdapter(heuristic=BetterExpiresAfter(days=1), cache=cache),
)
session.mount(
METLINK_API_URL_PREFIX + "ServiceLocation/",
CacheControlAdapter(heuristic=BetterExpiresAfter(seconds=90),
cache=cache),
)
return session
