python-sponsorblock/python_sponsorblock/__init__.py

import requests
import logging
from urllib.parse import urlparse, urlunparse, parse_qs
import re
from typing import Optional

from .exceptions import SponsorBlockError, SponsorBlockIdNotFoundError


def _get_video_id(i: str, silent: bool = False) -> None:
    # check with regex if i is already an id like r1Fa1iWJVEA
    if re.match(r"^[a-zA-Z0-9_-]{11}$", i):
        return i.strip()

    url = urlparse(url=i)

    if url.netloc == "youtu.be":
        return url.path[1:]

    type_frag_list = url.path.split("/")
            
    query_stuff = parse_qs(url.query)
    if "v" not in query_stuff:
        if not silent:
            raise SponsorBlockIdNotFoundError("No video id found in the url")
        return None
    else:
        return query_stuff["v"][0]

class SponsorBlock:
    def __init__(self, session: requests.Session = None, base_url: str = "https://sponsor.ajay.app", silent: bool = False, _requests_logging_exists: bool = False):
        self.base_url: str = base_url
        self.session: requests.Session = session or requests.Session()

        self.silent: bool = silent
        self._requests_logging_exists: bool = _requests_logging_exists

        self.logger: logging.Logger = logging.Logger("SponsorBlock")

    def _request(self, method: str, endpoint: str) -> Optional[requests.Response]:
        error_message = ""
        url = self.base_url + endpoint

        r: requests.Response = None
        try:
            r = self.session.request(method="GET", url=url)
        except requests.exceptions.Timeout:
            error_message = f"Request timed out at \"{url}\""
        except requests.exceptions.ConnectionError:
            error_message = f"Couldn't connect to \"{url}\""

        if error_message != "":
            if not self._requests_logging_exists:
                self.logger.error(error_message)
            if not self.silent:
                raise exceptions.SponsorBlockConnectionError(error_message)

        return r