Spaces:
No application file
No application file
File size: 2,089 Bytes
1f8f99f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
# import os
# css_file = "newFile.css"
# print(f'file:{os.getcwd()}{css_file}')
import requests
from bs4 import BeautifulSoup
import shutil
def download_images(url, selector):
response = requests.get(url, stream=True, headers={'User-agent': 'Mozilla/6.0'})
soup = BeautifulSoup(response.text, 'html.parser')
# print(soup)
img_tags = []
modifiedSelector = selector[1:]
if selector[0] == '.':
img_tags += soup.select_one(f'div{selector}').find_all('img')
else:
img_tags += soup.find(id = modifiedSelector).find_all('img')
try:
print(img_tags[0]['data-src'])
except Exception as e:
print(e)
# if(img_tags[0]['data-src']):
# print(img_tags[0]['data-src'])
# else:
# print('Yo')
# for img in img_tags:
# # Split URL by the last backslash
# split_url = img['src'].rsplit('/', 1)
# file_name_with_query = split_url[1]
# # Search for the file extension
# extensions = ['.jpg', '.png', '.webp']
# img_name = file_name_with_query
# for extension in extensions:
# if extension in file_name_with_query:
# img_name = file_name_with_query.split(extension)[0] + extension
# break
# r = requests.get(img['src'], stream=True, headers={'User-agent': 'Mozilla/6.0'})
# if r.status_code == 200:
# with open(img_name, 'wb') as f:
# r.raw.decode_content = True
# shutil.copyfileobj(r.raw, f)
# else:
# print(r.status_code)
# try:
# urlretrieve(img['src'], img_name)
# # print(f"Downloaded {img_name}")
# except HTTPError:
# urlretrieve(img_url, img_name)
# # except Exception as e:
# # print(f"Failed to download {img_name}: {str(e)}")
url = input("URL: ")
selector = input("Selector: ")
download_images(url,selector)
# # Usage example
# download_images('https://www.elixirsolutions.com/', element_class='columncontainer')
|