58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
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 |