@@ -165,7 +165,8 @@ def get_type_stream(self, segments) -> str:
165165 elif self .is_stream_aac :
166166 return "aac"
167167 else :
168- return None
168+ console .log ("[yellow]Warning: Unable to determine stream type." )
169+ return "ts" # Default to ts
169170
170171 def get_info (self ) -> None :
171172 """
@@ -275,7 +276,7 @@ async def _download_single_segment(self, client: httpx.AsyncClient, ts_url: str,
275276
276277 except Exception :
277278 if attempt + 1 == max_retry :
278- console .print (f" -- [red]Final retry failed for segment: { index } " )
279+ console .print (f" -- [red]Failed request for segment: { index } " )
279280 return index , False , max_retry , 0
280281
281282 sleep_time = 0.5 + attempt * 0.5 if attempt < 2 else min (3.0 , 1.02 ** attempt )
@@ -437,9 +438,6 @@ async def download_segments_async(self, description: str, type: str):
437438 finally :
438439 self ._cleanup_resources (temp_dir , progress_bar )
439440
440- if not self .download_interrupted :
441- self ._verify_download_completion ()
442-
443441 return self ._generate_results (type )
444442
445443 else :
@@ -524,13 +522,6 @@ def _generate_results(self, stream_type: str, output_path: str = None) -> Dict:
524522 'output_path' : output_path if output_path else self .final_output_path
525523 }
526524
527- def _verify_download_completion (self ) -> None :
528- """Validate final download integrity."""
529- total = len (self .segments )
530- if len (self .downloaded_segments ) / total < 0.999 :
531- missing = sorted (set (range (total )) - self .downloaded_segments )
532- raise RuntimeError (f"Download incomplete ({ len (self .downloaded_segments )/ total :.1%} ). Missing segments: { missing } " )
533-
534525 def _cleanup_resources (self , temp_dir : str , progress_bar : tqdm ) -> None :
535526 """Ensure resource cleanup and final reporting."""
536527 progress_bar .close ()
@@ -550,7 +541,7 @@ def _cleanup_resources(self, temp_dir: str, progress_bar: tqdm) -> None:
550541
551542 def _display_error_summary (self ) -> None :
552543 """Generate final error report."""
553- console .print (f" [cyan]Max retries: [red]{ self .info_maxRetry } [white] | "
544+ console .log (f"[cyan]Max retries: [red]{ self .info_maxRetry } [white] | "
554545 f"[cyan]Total retries: [red]{ self .info_nRetry } [white] | "
555546 f"[cyan]Failed segments: [red]{ self .info_nFailed } " )
556547
0 commit comments