[featuresession]

featuresession.py

import requests
from concurrent.futures import ThreadPoolExecutor
from threading import local

class FuturesSession(requests.Session):
  def __init__(self, max_workers=2, session_factory=requests.Session, *args, **kwargs):
    super().__init__(*args, **kwargs)
    self.session_factory = session_factory
    self.session_args = args
    self.session_kwargs = kwargs
    self.sessions = []
    self.executor = ThreadPoolExecutor(max_workers=max_workers)
    self.tls = local()

  def request(self, *args, **kwargs):
    def func(*args, **kwargs):
      session = getattr(self.tls, "session", None)
      if session is None:
        session = self.session_factory(*self.session_args, **self.session_kwargs)
        self.tls.session = session
        self.sessions.append(session)
      return session.request(*args, **kwargs)

    return self.executor.submit(func, *args, **kwargs)

  def close(self):
    self.executor.shutdown(wait=True)
    for session in self.sessions:
      session.close()
    super().close()