added chromedriver and main.py

This commit is contained in:
cid
2023-12-23 09:00:11 +08:00
commit 96d8926ac8
2 changed files with 43 additions and 0 deletions

BIN
chromedriver Executable file

Binary file not shown.

43
main.py Normal file
View File

@ -0,0 +1,43 @@
import requests
from bs4 import BeautifulSoup
from urllib.parse import urlparse, urlunparse
def scrape_headings(url):
try:
# Check if the URL has a scheme (http/https), and add one if missing
parsed_url = urlparse(url)
if not parsed_url.scheme:
url = urlunparse(('http',) + parsed_url[1:])
# Send an HTTP GET request to the specified URL
response = requests.get(url)
# Check if the request was successful
if response.status_code == 200:
# Parse the HTML content using BeautifulSoup
soup = BeautifulSoup(response.text, 'html.parser')
# Find all the heading elements (h1, h2, h3, etc.)
headings = soup.find_all(['h1', 'h2', 'h3', 'h4', 'h5', 'h6'])
# Extract and print the text from the headings
for heading in headings:
print(heading.text)
else:
print(f"Failed to retrieve content from {url}. Status code: {response.status_code}")
except Exception as e:
print(f"An error occurred: {str(e)}")
def main_start():
if __name__ == "__main__":
url = input("Enter the URL: ")
scrape_headings(url)
search_again = input("Do you want to search again? y/n:").lower()
if search_again == 'y':
main_start()
else:
exit()
main_start()