Skip to content

Commit 762d994

Browse files
committed
mp4: Fix output already exists
1 parent 9c84eb6 commit 762d994

File tree

1 file changed

+13
-17
lines changed

1 file changed

+13
-17
lines changed

StreamingCommunity/Lib/Downloader/MP4/downloader.py

+13-17
Original file line numberDiff line numberDiff line change
@@ -84,31 +84,27 @@ def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = No
8484
console.log("[red]Output file already exists.")
8585
if TELEGRAM_BOT:
8686
bot.send_message(f"Contenuto già scaricato!", None)
87-
return 400
87+
return None, False
8888

8989
if GET_ONLY_LINK:
9090
return {'path': path, 'url': url}
9191

9292
if not (url.lower().startswith('http://') or url.lower().startswith('https://')):
9393
logging.error(f"Invalid URL: {url}")
9494
console.print(f"[bold red]Invalid URL: {url}[/bold red]")
95-
return None
95+
return None, False
9696

97-
try:
98-
headers = {}
99-
if referer:
100-
headers['Referer'] = referer
101-
102-
if headers_:
103-
headers.update(headers_)
104-
else:
105-
headers['User-Agent'] = get_userAgent()
106-
107-
except Exception as header_err:
108-
logging.error(f"Error preparing headers: {header_err}")
109-
console.print(f"[bold red]Error preparing headers: {header_err}[/bold red]")
110-
return None
97+
# Set headers
98+
headers = {}
99+
if referer:
100+
headers['Referer'] = referer
111101

102+
if headers_:
103+
headers.update(headers_)
104+
else:
105+
headers['User-Agent'] = get_userAgent()
106+
107+
# Set interrupt handler
112108
temp_path = f"{path}.temp"
113109
interrupt_handler = InterruptHandler()
114110
original_handler = signal.signal(signal.SIGINT, partial(signal_handler, interrupt_handler=interrupt_handler, original_handler=signal.getsignal(signal.SIGINT)))
@@ -123,7 +119,7 @@ def MP4_downloader(url: str, path: str, referer: str = None, headers_: dict = No
123119

124120
if total == 0:
125121
console.print("[bold red]No video stream found.[/bold red]")
126-
return None
122+
return None, False
127123

128124
progress_bar = tqdm(
129125
total=total,

0 commit comments

Comments
 (0)