1
0
Fork 0
mirror of synced 2024-06-13 15:54:37 +12:00

Rename function

This commit is contained in:
Serene-Arc 2021-04-06 10:48:21 +10:00 committed by Ali Parlakci
parent 1768096b85
commit 9cb4dd4cf3
7 changed files with 7 additions and 7 deletions

View file

@ -26,7 +26,7 @@ class BaseDownloader(ABC):
raise NotImplementedError
@staticmethod
def get_link(url: str, cookies: dict = None, headers: dict = None) -> requests.Response:
def retrieve_url(url: str, cookies: dict = None, headers: dict = None) -> requests.Response:
res = requests.get(url, cookies=cookies, headers=headers)
if res.status_code != 200:
raise ResourceNotFound(f'Server responded with {res.status_code} to {url}')

View file

@ -33,7 +33,7 @@ class Erome(BaseDownloader):
@staticmethod
def _get_links(url: str) -> set[str]:
page = Erome.get_link(url)
page = Erome.retrieve_url(url)
soup = bs4.BeautifulSoup(page.text, 'html.parser')
front_images = soup.find_all('img', attrs={'class': 'lasyload'})
out = [im.get('data-src') for im in front_images]

View file

@ -32,7 +32,7 @@ class Gallery(BaseDownloader):
' Chrome/67.0.3396.87 Safari/537.36 OPR/54.0.2952.64',
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8',
}
page = Gallery.get_link(url, headers=resource_headers)
page = Gallery.retrieve_url(url, headers=resource_headers)
soup = bs4.BeautifulSoup(page.text, 'html.parser')
links = soup.findAll('a', attrs={'target': '_blank', 'href': re.compile(r'https://preview\.redd\.it.*')})

View file

@ -24,7 +24,7 @@ class Gfycat(GifDeliveryNetwork):
gfycat_id = re.match(r'.*/(.*?)/?$', url).group(1)
url = 'https://gfycat.com/' + gfycat_id
response = Gfycat.get_link(url)
response = Gfycat.retrieve_url(url)
if 'gifdeliverynetwork' in response.url:
return GifDeliveryNetwork._get_link(url)

View file

@ -21,7 +21,7 @@ class GifDeliveryNetwork(BaseDownloader):
@staticmethod
def _get_link(url: str) -> str:
page = GifDeliveryNetwork.get_link(url)
page = GifDeliveryNetwork.retrieve_url(url)
soup = BeautifulSoup(page.text, 'html.parser')
content = soup.find('source', attrs={'id': 'mp4Source', 'type': 'video/mp4'})

View file

@ -41,7 +41,7 @@ class Imgur(BaseDownloader):
link = link.replace('i.imgur', 'imgur')
link = link.rstrip('.gifv')
res = Imgur.get_link(link, cookies={'over18': '1', 'postpagebeta': '0'})
res = Imgur.retrieve_url(link, cookies={'over18': '1', 'postpagebeta': '0'})
soup = bs4.BeautifulSoup(res.text, 'html.parser')
scripts = soup.find_all('script', attrs={'type': 'text/javascript'})

View file

@ -30,7 +30,7 @@ class Redgifs(GifDeliveryNetwork):
' Chrome/67.0.3396.87 Safari/537.36 OPR/54.0.2952.64',
}
page = Redgifs.get_link(url, headers=headers)
page = Redgifs.retrieve_url(url, headers=headers)
soup = BeautifulSoup(page.text, 'html.parser')
content = soup.find('script', attrs={'data-react-helmet': 'true', 'type': 'application/ld+json'})