mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 20:15:24 +00:00
133 lines
4.3 KiB
Python
133 lines
4.3 KiB
Python
# 18.06.24
|
|
|
|
import sys
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
# External libraries
|
|
import httpx
|
|
from googlesearch import search
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util.headers import get_headers
|
|
from Src.Util.console import console, msg
|
|
from Src.Util._jsonConfig import config_manager
|
|
|
|
|
|
def google_search(query):
|
|
"""
|
|
Perform a Google search and return the first result.
|
|
|
|
Args:
|
|
query (str): The search query to execute on Google.
|
|
|
|
Returns:
|
|
str: The first URL result from the search, or None if no result is found.
|
|
"""
|
|
# Perform the search on Google and limit to 1 result
|
|
search_results = search(query, num_results=1)
|
|
|
|
# Extract the first result
|
|
first_result = next(search_results, None)
|
|
|
|
if not first_result:
|
|
console.print("[red]No results found.[/red]")
|
|
|
|
return first_result
|
|
|
|
def get_final_redirect_url(initial_url):
|
|
"""
|
|
Follow redirects from the initial URL and return the final URL after all redirects.
|
|
|
|
Args:
|
|
initial_url (str): The URL to start with and follow redirects.
|
|
|
|
Returns:
|
|
str: The final URL after all redirects are followed.
|
|
"""
|
|
|
|
# Create a client with redirects enabled
|
|
with httpx.Client(follow_redirects=True) as client:
|
|
response = client.get(initial_url)
|
|
|
|
# Capture the final URL after all redirects
|
|
final_url = response.url
|
|
|
|
return final_url
|
|
|
|
def search_domain(site_name: str, base_url: str):
|
|
"""
|
|
Search for a valid domain for the given site name and base URL.
|
|
|
|
Parameters:
|
|
- site_name (str): The name of the site to search the domain for.
|
|
- base_url (str): The base URL to construct complete URLs.
|
|
- follow_redirects (bool): To follow redirect url or not.
|
|
|
|
Returns:
|
|
tuple: The found domain and the complete URL.
|
|
"""
|
|
|
|
# Extract config domain
|
|
domain = str(config_manager.get_dict("SITE", site_name)['domain'])
|
|
console.print(f"[cyan]Test site[white]: [red]{base_url}.{domain}")
|
|
|
|
try:
|
|
|
|
# Test the current domain
|
|
response_follow = httpx.get(f"{base_url}.{domain}", headers={'user-agent': get_headers()}, timeout=2)
|
|
console.print(f"[cyan]Response site[white]: [red]{response_follow.status_code}")
|
|
response_follow.raise_for_status()
|
|
|
|
except Exception as e:
|
|
console.print(f"[cyan]Change domain for site[white]: [red]{base_url}.{domain}, [cyan]error[white]: [red]{e}")
|
|
|
|
query = base_url.split("/")[-1]
|
|
first_url = google_search(query)
|
|
|
|
if first_url:
|
|
final_url = get_final_redirect_url(first_url)
|
|
console.print(f"\n[bold yellow]Suggestion:[/bold yellow] [white](Experimental)\n"
|
|
f"[cyan]New final URL[white]: [green]{final_url}")
|
|
|
|
def extract_domain(url):
|
|
parsed_url = urlparse(url)
|
|
domain = parsed_url.netloc
|
|
return domain.split(".")[-1]
|
|
|
|
new_domain_extract = extract_domain(str(final_url))
|
|
|
|
if msg.ask(f"[red]Do you want to auto update config.json - '[green]{site_name}[red]' with domain: [green]{new_domain_extract}", choices=["y", "n"], default="y").lower() == "y":
|
|
|
|
# Update domain in config.json
|
|
config_manager.config['SITE'][site_name]['domain'] = new_domain_extract
|
|
config_manager.write_config()
|
|
|
|
# Return config domain
|
|
console.print(f"[cyan]Return domain: [red]{new_domain_extract} \n")
|
|
return new_domain_extract, f"{base_url}.{new_domain_extract}"
|
|
|
|
else:
|
|
console.print("[bold red]\nManually change the domain in the JSON file.[/bold red]")
|
|
sys.exit(0)
|
|
|
|
else:
|
|
console.print("[bold red]No valid URL to follow redirects.[/bold red]")
|
|
sys.exit(0)
|
|
|
|
# Ensure the URL is in string format before parsing
|
|
parsed_url = urlparse(str(response_follow.url))
|
|
parse_domain = parsed_url.netloc
|
|
tld = parse_domain.split('.')[-1]
|
|
|
|
if tld is not None:
|
|
|
|
# Update domain in config.json
|
|
config_manager.config['SITE'][site_name]['domain'] = tld
|
|
config_manager.write_config()
|
|
|
|
# Return config domain
|
|
console.print(f"[cyan]Return domain: [red]{tld} \n")
|
|
return tld, f"{base_url}.{tld}"
|