Better imports in the pipeline.
This commit is contained in:
@@ -98,6 +98,7 @@ def main() -> int:
|
|||||||
input_file=Path(args.input_file),
|
input_file=Path(args.input_file),
|
||||||
output_file=Path(args.output_file),
|
output_file=Path(args.output_file),
|
||||||
parts=parts,
|
parts=parts,
|
||||||
|
framerate = args.framerate,
|
||||||
verbose=args.verbose,
|
verbose=args.verbose,
|
||||||
coarse=args.coarse,
|
coarse=args.coarse,
|
||||||
threshold=args.threshold,
|
threshold=args.threshold,
|
||||||
@@ -108,6 +109,10 @@ def main() -> int:
|
|||||||
all_optional_tools = all_optional_tools)
|
all_optional_tools = all_optional_tools)
|
||||||
|
|
||||||
logger.debug("Configuration: %s", config)
|
logger.debug("Configuration: %s", config)
|
||||||
|
try:
|
||||||
process_recording(config)
|
process_recording(config)
|
||||||
|
except TSCutError as exc:
|
||||||
|
logger.error("%s", exc)
|
||||||
|
return 1
|
||||||
|
|
||||||
return 0
|
return 0
|
||||||
|
|||||||
+2
-1
@@ -4,12 +4,13 @@
|
|||||||
|
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
from datetime import timedelta
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class ProcessingOptions:
|
class ProcessingOptions:
|
||||||
input_file: Path
|
input_file: Path
|
||||||
output_file: Path
|
output_file: Path
|
||||||
parts: list
|
parts: list[tuple[timedelta, timedelta]]
|
||||||
tools_paths: dict[str, str]
|
tools_paths: dict[str, str]
|
||||||
all_optional_tools: bool
|
all_optional_tools: bool
|
||||||
framerate: int|None = None
|
framerate: int|None = None
|
||||||
|
|||||||
+32
-22
@@ -11,34 +11,44 @@ import logging
|
|||||||
|
|
||||||
import hexdump
|
import hexdump
|
||||||
|
|
||||||
from .config import ProcessingOptions
|
from tscut.config import ProcessingOptions
|
||||||
from .tscut import (
|
from tscut.tools.mkvtoolnix import (
|
||||||
SupportedFormat,
|
|
||||||
change_codec_private_data,
|
|
||||||
concatenate_h264_parts,
|
|
||||||
concatenate_h264_ts_parts,
|
|
||||||
do_ocr,
|
|
||||||
dump_codec_private_data,
|
|
||||||
extract_all_streams,
|
|
||||||
extract_mkv_part,
|
|
||||||
extract_srt,
|
|
||||||
extract_track_from_mkv,
|
|
||||||
ffmpeg_convert,
|
|
||||||
find_subtitles_tracks,
|
|
||||||
get_avc_config_from_h264,
|
|
||||||
get_codec_private_data_from_mkv,
|
get_codec_private_data_from_mkv,
|
||||||
|
change_codec_private_data,
|
||||||
|
extract_mkv_part,
|
||||||
|
extract_track_from_mkv,
|
||||||
|
merge_mkvs,
|
||||||
|
remove_video_tracks_from_mkv,
|
||||||
|
remux_srt_subtitles
|
||||||
|
)
|
||||||
|
from tscut.tools.ffmpeg import (
|
||||||
|
ffmpeg_convert,
|
||||||
|
extract_all_streams
|
||||||
|
)
|
||||||
|
from tscut.tools.ffprobe import (
|
||||||
|
find_subtitles_tracks,
|
||||||
get_format,
|
get_format,
|
||||||
get_frame_rate,
|
get_frame_rate,
|
||||||
get_movie_duration,
|
get_movie_duration,
|
||||||
get_nearest_iframe,
|
get_nearest_iframe,
|
||||||
get_streams,
|
get_streams
|
||||||
get_tesseract_supported_lang,
|
|
||||||
get_ts_frame,
|
|
||||||
merge_mkvs,
|
|
||||||
parse_codec_private,
|
|
||||||
remove_video_tracks_from_mkv,
|
|
||||||
remux_srt_subtitles,
|
|
||||||
)
|
)
|
||||||
|
from tscut.h264.avc import (
|
||||||
|
parse_codec_private,
|
||||||
|
get_avc_config_from_h264
|
||||||
|
)
|
||||||
|
from tscut.matroska.codec import dump_codec_private_data
|
||||||
|
from tscut.tscut import (
|
||||||
|
concatenate_h264_parts,
|
||||||
|
concatenate_h264_ts_parts,
|
||||||
|
SupportedFormat
|
||||||
|
)
|
||||||
|
from tscut.subtitles.ocr import (
|
||||||
|
do_ocr,
|
||||||
|
extract_srt,
|
||||||
|
get_tesseract_supported_lang
|
||||||
|
)
|
||||||
|
from tscut.tools.timeframe import get_ts_frame
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|||||||
+3
-2
@@ -46,9 +46,10 @@ from tscut.tools.timeframe import (
|
|||||||
parse_time_interval
|
parse_time_interval
|
||||||
)
|
)
|
||||||
from tscut.tools.discovery import check_required_tools
|
from tscut.tools.discovery import check_required_tools
|
||||||
from tscut.ocr.subtitles import (
|
from tscut.subtitles.ocr import (
|
||||||
do_ocr,
|
do_ocr,
|
||||||
extract_srt,get_tesseract_supported_lang
|
extract_srt,
|
||||||
|
get_tesseract_supported_lang
|
||||||
)
|
)
|
||||||
from tscut.tools.mkvtoolnix import (
|
from tscut.tools.mkvtoolnix import (
|
||||||
extract_mkv_part,
|
extract_mkv_part,
|
||||||
|
|||||||
Reference in New Issue
Block a user