Jonathan Hodgson
7 years ago
parent
b0bf4612f0
commit
55473e31fa
2 changed files with 698 additions and 0 deletions
@ -0,0 +1,696 @@ |
|||||||
|
#!/usr/bin/env python3 |
||||||
|
# coding=utf-8 |
||||||
|
|
||||||
|
""" |
||||||
|
Rip DVDs quickly and easily from the commandline. |
||||||
|
|
||||||
|
Features: |
||||||
|
- With minimal configuration: |
||||||
|
- Encodes videos in mkv files with h.264 video and aac audio. |
||||||
|
(compatible with a wide variety of media players without |
||||||
|
additional transcoding, including PS3, Roku, and most smart |
||||||
|
phones, smart TVs and tablets). |
||||||
|
- Preserves all audio tracks, all subtitle tracks, and chapter |
||||||
|
markers. |
||||||
|
- Intelligently chooses output filename based on a provided prefix. |
||||||
|
- Generates one video file per DVD title, or optionally one per |
||||||
|
chapter. |
||||||
|
- Easy to read "scan" mode tells you what you need need to know about |
||||||
|
a disk to decide on how to rip it. |
||||||
|
|
||||||
|
Why I wrote this: |
||||||
|
This script exists because I wanted a simple way to back up DVDs with |
||||||
|
reasonably good compression and quality settings, and in a format I could |
||||||
|
play on the various media players I own including PS3, Roku, smart TVs, |
||||||
|
smartphones and tablets. Using mkv files with h.264 video and aac audio seems |
||||||
|
to be the best fit for these constraints. |
||||||
|
|
||||||
|
I also wanted it to preserve as much as possible: chapter markers, subtitles, |
||||||
|
and (most of all) *all* of the audio tracks. My kids have a number of |
||||||
|
bilingual DVDs, and I wanted to back these up so they don't have to handle |
||||||
|
the physical disks, but can still watch their shows in either language. For |
||||||
|
some reason HandBrakeCLI doesn't have a simple “encode all audio tracks” |
||||||
|
option. |
||||||
|
|
||||||
|
This script also tries to be smart about the output name. You just tell it |
||||||
|
the pathname prefix, eg: "/tmp/AwesomeVideo", and it'll decide whether to |
||||||
|
produce a single file, "/tmp/AwesomeVideo.mkv", or a directory |
||||||
|
"/tmp/AwesomeVideo/" which will contain separate files for each title, |
||||||
|
depending on whether you're ripping a single title or multiple titles. |
||||||
|
|
||||||
|
|
||||||
|
Using it, Step 1: |
||||||
|
|
||||||
|
The first step is to scan your DVD and decide whether or not you want |
||||||
|
to split chapters. Here's an example of a disc with 6 episodes of a TV |
||||||
|
show, plus a "bump", all stored as a single title. |
||||||
|
|
||||||
|
$ dvdrip --scan /dev/cdrom |
||||||
|
Reading from '/media/EXAMPLE1' |
||||||
|
Title 1/ 1: 02:25:33 720×576 4:3 25 fps |
||||||
|
audio 1: Chinese (5.1ch) [48000Hz, 448000bps] |
||||||
|
chapter 1: 00:24:15 ◖■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 2: 00:24:15 ◖‥‥‥‥‥‥‥‥■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 3: 00:24:14 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 4: 00:24:15 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 5: 00:24:15 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 6: 00:24:14 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■◗ |
||||||
|
chapter 7: 00:00:05 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■◗ |
||||||
|
|
||||||
|
Knowing that this is 6 episodes of a TV show, I'd choose to split the |
||||||
|
chapters. If it was a movie with 6 chapters, I would choose to not |
||||||
|
split it. |
||||||
|
|
||||||
|
Here's a disc with 3 2-segment episodes of a show, plus two "bumps", |
||||||
|
stored as 8 titles. |
||||||
|
|
||||||
|
Reading from '/media/EXAMPLE2' |
||||||
|
Title 1/ 5: 00:23:22 720×576 4:3 25 fps |
||||||
|
audio 1: Chinese (2.0ch) [48000Hz, 192000bps] |
||||||
|
audio 2: English (2.0ch) [48000Hz, 192000bps] |
||||||
|
sub 1: English [(Bitmap)(VOBSUB)] |
||||||
|
chapter 1: 00:11:41 ◖■■■■■■■■■■■■■■■■■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 2: 00:11:41 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■■■■■■■■■■■■■■■■■■◗ |
||||||
|
|
||||||
|
Title 2/ 5: 00:22:40 720×576 4:3 25 fps |
||||||
|
audio 1: Chinese (2.0ch) [48000Hz, 192000bps] |
||||||
|
audio 2: English (2.0ch) [48000Hz, 192000bps] |
||||||
|
sub 1: English [(Bitmap)(VOBSUB)] |
||||||
|
chapter 1: 00:11:13 ◖■■■■■■■■■■■■■■■■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 2: 00:11:28 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■■■■■■■■■■■■■■■■■◗ |
||||||
|
|
||||||
|
Title 3/ 5: 00:22:55 720×576 4:3 25 fps |
||||||
|
audio 1: Chinese (2.0ch) [48000Hz, 192000bps] |
||||||
|
audio 2: English (2.0ch) [48000Hz, 192000bps] |
||||||
|
sub 1: English [(Bitmap)(VOBSUB)] |
||||||
|
chapter 1: 00:15:56 ◖■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■■‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥◗ |
||||||
|
chapter 2: 00:06:59 ◖‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥‥■■■■■■■■■■■■■■■■◗ |
||||||
|
|
||||||
|
Title 4/ 5: 00:00:08 720×576 4:3 25 fps |
||||||
|
audio 1: English (2.0ch) [None] |
||||||
|
chapter 1: 00:00:08 ◖◗ |
||||||
|
|
||||||
|
Title 5/ 5: 00:00:05 720×576 4:3 25 fps |
||||||
|
chapter 1: 00:00:05 ◖◗ |
||||||
|
|
||||||
|
Given that these are 2-segment episodes (it's pretty common for kids' |
||||||
|
shows to have two segments per episode -- essentially 2 "mini-episodes") you |
||||||
|
can choose whether to do the default one video per title (episodes) or |
||||||
|
split by chapter (segments / mini-episodes). |
||||||
|
|
||||||
|
Using it, Step 2: |
||||||
|
|
||||||
|
If you've decided to split by chapter, execute: |
||||||
|
|
||||||
|
dvdrip.py -c /dev/cdrom -o Output_Name |
||||||
|
|
||||||
|
Otherwise, leave out the -c flag. |
||||||
|
|
||||||
|
If there is only one video being ripped, it will be named Output_Name.mkv. If |
||||||
|
there are multiple files, they will be placed in a new directory called |
||||||
|
Output_Name. |
||||||
|
|
||||||
|
Limitations: |
||||||
|
|
||||||
|
This script has been tested on both Linux and Mac OS X with Python 3, |
||||||
|
HandBrakeCLI and VLC installed (and also MacPorts in the case of OS X). |
||||||
|
""" |
||||||
|
|
||||||
|
# TODO: Detect if HandBrakeCLI is burning in vobsubs. |
||||||
|
|
||||||
|
import argparse |
||||||
|
import os |
||||||
|
import re |
||||||
|
import stat |
||||||
|
import subprocess |
||||||
|
import sys |
||||||
|
import time |
||||||
|
|
||||||
|
from pprint import pprint |
||||||
|
from collections import namedtuple |
||||||
|
from fractions import gcd |
||||||
|
|
||||||
|
|
||||||
|
class UserError(Exception): |
||||||
|
def __init__(self, message): |
||||||
|
self.message = message |
||||||
|
|
||||||
|
CHAR_ENCODING = 'UTF-8' |
||||||
|
|
||||||
|
def check_err(*popenargs, **kwargs): |
||||||
|
process = subprocess.Popen(stderr=subprocess.PIPE, *popenargs, **kwargs) |
||||||
|
_, stderr = process.communicate() |
||||||
|
retcode = process.poll() |
||||||
|
if retcode: |
||||||
|
cmd = kwargs.get("args") |
||||||
|
if cmd is None: |
||||||
|
cmd = popenargs[0] |
||||||
|
raise subprocess.CalledProcessError(retcode, cmd, output=stderr) |
||||||
|
return stderr.decode(CHAR_ENCODING, 'replace') |
||||||
|
|
||||||
|
def check_output(*args, **kwargs): |
||||||
|
return subprocess.check_output(*args, **kwargs).decode(CHAR_ENCODING) |
||||||
|
|
||||||
|
HANDBRAKE = 'HandBrakeCLI' |
||||||
|
|
||||||
|
TITLE_COUNT_REGEXES = [ |
||||||
|
re.compile(r'^Scanning title 1 of (\d+)\.\.\.$'), |
||||||
|
re.compile(r'^\[\d\d:\d\d:\d\d] scan: DVD has (\d+) title\(s\)$'), |
||||||
|
] |
||||||
|
|
||||||
|
def FindTitleCount(scan, verbose): |
||||||
|
for regex in TITLE_COUNT_REGEXES: |
||||||
|
for line in scan: |
||||||
|
m = regex.match(line) |
||||||
|
if m: break |
||||||
|
if m: |
||||||
|
return int(m.group(1)) |
||||||
|
if verbose: |
||||||
|
for line in scan: |
||||||
|
print(line) |
||||||
|
raise AssertionError("Can't find TITLE_COUNT_REGEX in scan") |
||||||
|
|
||||||
|
|
||||||
|
STRUCTURED_LINE_RE = re.compile(r'( *)\+ (([a-z0-9 ]+):)?(.*)') |
||||||
|
|
||||||
|
def ExtractTitleScan(scan): |
||||||
|
result = [] |
||||||
|
in_title_scan = False |
||||||
|
for line in scan: |
||||||
|
if not in_title_scan: |
||||||
|
if line.startswith('+'): |
||||||
|
in_title_scan = True |
||||||
|
if in_title_scan: |
||||||
|
m = STRUCTURED_LINE_RE.match(line) |
||||||
|
if m: |
||||||
|
result.append(line) |
||||||
|
else: |
||||||
|
break |
||||||
|
return tuple(result) |
||||||
|
|
||||||
|
|
||||||
|
TRACK_VALUE_RE = re.compile(r'(\d+), (.*)') |
||||||
|
|
||||||
|
def MassageTrackData(node, key): |
||||||
|
if key in node: |
||||||
|
track_data = node[key] |
||||||
|
if type(track_data) is list: |
||||||
|
new_track_data = {} |
||||||
|
for track in track_data: |
||||||
|
k, v = TRACK_VALUE_RE.match(track).groups() |
||||||
|
new_track_data[k] = v |
||||||
|
node[key] = new_track_data |
||||||
|
|
||||||
|
def ParseTitleScan(scan): |
||||||
|
pos, result = ParseTitleScanHelper(scan, pos=0, indent=0) |
||||||
|
|
||||||
|
# HandBrakeCLI inexplicably uses a comma instead of a colon to |
||||||
|
# separate the track identifier from the track data in the "audio |
||||||
|
# tracks" and "subtitle tracks" nodes, so we "massage" these parsed |
||||||
|
# nodes to get a consistent parsed reperesentation. |
||||||
|
for value in result.values(): |
||||||
|
MassageTrackData(value, 'audio tracks') |
||||||
|
MassageTrackData(value, 'subtitle tracks') |
||||||
|
return result |
||||||
|
|
||||||
|
def ParseTitleScanHelper(scan, pos, indent): |
||||||
|
result = {} |
||||||
|
cruft = [] |
||||||
|
while True: |
||||||
|
pos, node = ParseNode(scan, pos=pos, indent=indent) |
||||||
|
if node: |
||||||
|
if type(node) is tuple: |
||||||
|
k, v = node |
||||||
|
result[k] = v |
||||||
|
else: |
||||||
|
cruft.append(node) |
||||||
|
result[None] = cruft |
||||||
|
else: |
||||||
|
break |
||||||
|
if len(result) == 1 and None in result: |
||||||
|
result = result[None] |
||||||
|
return pos, result |
||||||
|
|
||||||
|
def ParseNode(scan, pos, indent): |
||||||
|
if pos >= len(scan): |
||||||
|
return pos, None |
||||||
|
line = scan[pos] |
||||||
|
spaces, colon, name, value = STRUCTURED_LINE_RE.match(line).groups() |
||||||
|
spaces = len(spaces) / 2 |
||||||
|
if spaces < indent: |
||||||
|
return pos, None |
||||||
|
assert spaces == indent, '%d <> %r' % (indent, line) |
||||||
|
pos += 1 |
||||||
|
if colon: |
||||||
|
if value: |
||||||
|
node = (name, value) |
||||||
|
else: |
||||||
|
pos, children = ParseTitleScanHelper(scan, pos, indent + 1) |
||||||
|
node = (name, children) |
||||||
|
else: |
||||||
|
node = value |
||||||
|
return pos, node |
||||||
|
|
||||||
|
def only(iterable): |
||||||
|
""" |
||||||
|
Return the one and only element in iterable. |
||||||
|
|
||||||
|
Raises an ValueError if iterable does not have exactly one item. |
||||||
|
""" |
||||||
|
result, = iterable |
||||||
|
return result |
||||||
|
|
||||||
|
Title = namedtuple('Title', ['number', 'info']) |
||||||
|
Task = namedtuple('Task', ['title', 'chapter']) |
||||||
|
|
||||||
|
TOTAL_EJECT_SECONDS = 5 |
||||||
|
EJECT_ATTEMPTS_PER_SECOND = 10 |
||||||
|
|
||||||
|
class DVD: |
||||||
|
def __init__(self, mountpoint, verbose, mount_timeout=0): |
||||||
|
if stat.S_ISBLK(os.stat(mountpoint).st_mode): |
||||||
|
mountpoint = FindMountPoint(mountpoint, mount_timeout) |
||||||
|
if not os.path.isdir(mountpoint): |
||||||
|
raise UserError('%r is not a directory' % mountpoint) |
||||||
|
self.mountpoint = mountpoint |
||||||
|
self.verbose = verbose |
||||||
|
|
||||||
|
def RipTitle(self, task, output, dry_run, verbose): |
||||||
|
if verbose: |
||||||
|
print('Title Scan:') |
||||||
|
pprint(task.title.info) |
||||||
|
print('-' * 78) |
||||||
|
|
||||||
|
audio_tracks = task.title.info['audio tracks'].keys() |
||||||
|
audio_encoders = ['faac'] * len(audio_tracks) |
||||||
|
subtitles = task.title.info['subtitle tracks'].keys() |
||||||
|
|
||||||
|
args = [ |
||||||
|
HANDBRAKE, |
||||||
|
'--title', str(task.title.number), |
||||||
|
'--preset', "High Profile", |
||||||
|
'--encoder', 'x264', |
||||||
|
'--audio', ','.join(audio_tracks), |
||||||
|
'--aencoder', ','.join(audio_encoders), |
||||||
|
] |
||||||
|
if task.chapter is not None: |
||||||
|
args += [ |
||||||
|
'--chapters', str(task.chapter), |
||||||
|
] |
||||||
|
if subtitles: |
||||||
|
args += [ |
||||||
|
'--subtitle', ','.join(subtitles), |
||||||
|
] |
||||||
|
args += [ |
||||||
|
'--markers', |
||||||
|
'--optimize', |
||||||
|
#'--no-dvdnav', # TODO: turn this on as a fallback |
||||||
|
'--input', self.mountpoint, |
||||||
|
'--output', output, |
||||||
|
] |
||||||
|
if verbose: |
||||||
|
print(' '.join(('\n ' + a) |
||||||
|
if a.startswith('-') else a for a in args)) |
||||||
|
print('-' * 78) |
||||||
|
if not dry_run: |
||||||
|
if verbose: |
||||||
|
subprocess.call(args) |
||||||
|
else: |
||||||
|
check_err(args) |
||||||
|
|
||||||
|
def ScanTitle(self, i): |
||||||
|
for line in check_err([ |
||||||
|
HANDBRAKE, |
||||||
|
#'--no-dvdnav', # TODO: turn this on as a fallback |
||||||
|
'--scan', |
||||||
|
'--title', str(i), |
||||||
|
'-i', |
||||||
|
self.mountpoint], stdout=subprocess.PIPE).split('\n'): |
||||||
|
if self.verbose: |
||||||
|
print('< %s' % line.rstrip()) |
||||||
|
yield line |
||||||
|
|
||||||
|
def ScanTitles(self, title_numbers, verbose): |
||||||
|
""" |
||||||
|
Returns an iterable of parsed titles. |
||||||
|
""" |
||||||
|
first = title_numbers[0] if title_numbers else 1 |
||||||
|
raw_scan = tuple(self.ScanTitle(first)) |
||||||
|
title_count = FindTitleCount(raw_scan, verbose) |
||||||
|
print('Disc claims to have %d titles.' % title_count) |
||||||
|
title_name, title_info = only( |
||||||
|
ParseTitleScan(ExtractTitleScan(raw_scan)).items()) |
||||||
|
del raw_scan |
||||||
|
|
||||||
|
def MakeTitle(name, number, info): |
||||||
|
assert ('title %d' % number) == name |
||||||
|
info['duration'] = ExtractDuration('duration ' + info['duration']) |
||||||
|
return Title(number, info) |
||||||
|
|
||||||
|
yield MakeTitle(title_name, first, title_info) |
||||||
|
|
||||||
|
to_scan = [x for x in range(1, title_count + 1) |
||||||
|
if x != first |
||||||
|
and ((not title_numbers) |
||||||
|
or x in title_numbers)] |
||||||
|
for i in to_scan: |
||||||
|
try: |
||||||
|
scan = ExtractTitleScan(self.ScanTitle(i)) |
||||||
|
except subprocess.CalledProcessError as exc: |
||||||
|
warn("Cannot scan title %d." % i) |
||||||
|
else: |
||||||
|
title_info_names = ParseTitleScan(scan).items() |
||||||
|
if title_info_names: |
||||||
|
title_name, title_info = only(title_info_names) |
||||||
|
yield MakeTitle(title_name, i, title_info) |
||||||
|
else: |
||||||
|
warn("Cannot parse scan of title %d." % i) |
||||||
|
|
||||||
|
def Eject(self): |
||||||
|
# TODO: this should really be a while loop that terminates once a |
||||||
|
# deadline is met. |
||||||
|
for i in range(TOTAL_EJECT_SECONDS * EJECT_ATTEMPTS_PER_SECOND): |
||||||
|
if not subprocess.call(['eject', self.mountpoint]): |
||||||
|
return |
||||||
|
time.sleep(1.0 / EJECT_ATTEMPTS_PER_SECOND) |
||||||
|
|
||||||
|
def ParseDuration(s): |
||||||
|
result = 0 |
||||||
|
for field in s.strip().split(':'): |
||||||
|
result *= 60 |
||||||
|
result += int(field) |
||||||
|
return result |
||||||
|
|
||||||
|
def FindMountPoint(dev, timeout): |
||||||
|
regex = re.compile(r'^' + re.escape(os.path.realpath(dev)) + r'\b') |
||||||
|
|
||||||
|
now = time.time() |
||||||
|
end_time = now + timeout |
||||||
|
while end_time >= now: |
||||||
|
for line in check_output(['df', '-P']).split('\n'): |
||||||
|
m = regex.match(line) |
||||||
|
if m: |
||||||
|
line = line.split(None, 5) |
||||||
|
if len(line) > 1: |
||||||
|
return line[-1] |
||||||
|
time.sleep(0.1) |
||||||
|
now = time.time() |
||||||
|
raise UserError('%r not mounted.' % dev) |
||||||
|
|
||||||
|
def FindMainFeature(titles, verbose=False): |
||||||
|
if verbose: |
||||||
|
print('Attempting to determine main feature of %d titles...' |
||||||
|
% len(titles)) |
||||||
|
main_feature = max(titles, |
||||||
|
key=lambda title: ParseDuration(title.info['duration'])) |
||||||
|
if verbose: |
||||||
|
print('Selected %r as main feature.' % main_feature.number) |
||||||
|
print() |
||||||
|
|
||||||
|
def ConstructTasks(titles, chapter_split): |
||||||
|
for title in titles: |
||||||
|
num_chapters = len(title.info['chapters']) |
||||||
|
if chapter_split and num_chapters > 1: |
||||||
|
for chapter in range(1, num_chapters + 1): |
||||||
|
yield Task(title, chapter) |
||||||
|
else: |
||||||
|
yield Task(title, None) |
||||||
|
|
||||||
|
def TaskFilenames(tasks, output, dry_run=False): |
||||||
|
if (len(tasks) > 1): |
||||||
|
def ComputeFileName(task): |
||||||
|
if task.chapter is None: |
||||||
|
return os.path.join(output, |
||||||
|
'Title%02d.mkv' % task.title.number) |
||||||
|
else: |
||||||
|
return os.path.join(output, |
||||||
|
'Title%02d_%02d.mkv' |
||||||
|
% (task.title.number, task.chapter)) |
||||||
|
if not dry_run: |
||||||
|
os.makedirs(output) |
||||||
|
else: |
||||||
|
def ComputeFileName(task): |
||||||
|
return '%s.mkv' % output |
||||||
|
result = [ComputeFileName(task) for task in tasks] |
||||||
|
if len(set(result)) != len(result): |
||||||
|
raise UserError("multiple tasks use same filename") |
||||||
|
return result |
||||||
|
|
||||||
|
def PerformTasks(dvd, tasks, title_count, filenames, |
||||||
|
dry_run=False, verbose=False): |
||||||
|
for task, filename in zip(tasks, filenames): |
||||||
|
print('=' * 78) |
||||||
|
if task.chapter is None: |
||||||
|
print('Title %s / %s => %r' |
||||||
|
% (task.title.number, title_count, filename)) |
||||||
|
else: |
||||||
|
num_chapters = len(task.title.info['chapters']) |
||||||
|
print('Title %s / %s , Chapter %s / %s=> %r' |
||||||
|
% (task.title.number, title_count, task.chapter, |
||||||
|
num_chapters, filename)) |
||||||
|
print('-' * 78) |
||||||
|
dvd.RipTitle(task, filename, dry_run, verbose) |
||||||
|
|
||||||
|
Size = namedtuple('Size', |
||||||
|
['width', 'height', 'pix_aspect_width', 'pix_aspect_height', 'fps']) |
||||||
|
|
||||||
|
SIZE_REGEX = re.compile( |
||||||
|
r'^\s*(\d+)x(\d+),\s*' |
||||||
|
r'pixel aspect: (\d+)/(\d+),\s*' |
||||||
|
r'display aspect: (?:\d+(?:\.\d+)),\s*' |
||||||
|
r'(\d+(?:\.\d+)) fps\s*$') |
||||||
|
|
||||||
|
SIZE_CTORS = [int] * 4 + [float] |
||||||
|
|
||||||
|
def ParseSize(s): |
||||||
|
return Size(*(f(x) |
||||||
|
for f, x in zip(SIZE_CTORS, SIZE_REGEX.match(s).groups()))) |
||||||
|
|
||||||
|
def ComputeAspectRatio(size): |
||||||
|
w = size.width * size.pix_aspect_width |
||||||
|
h = size.height * size.pix_aspect_height |
||||||
|
d = gcd(w, h) |
||||||
|
return (w // d, h // d) |
||||||
|
|
||||||
|
DURATION_REGEX = re.compile( |
||||||
|
r'^(?:.*,)?\s*duration\s+(\d\d):(\d\d):(\d\d)\s*(?:,.*)?$') |
||||||
|
|
||||||
|
class Duration(namedtuple('Duration', 'hours minutes seconds')): |
||||||
|
def __str__(self): |
||||||
|
return '%02d:%02d:%02d' % (self) |
||||||
|
|
||||||
|
def in_seconds(self): |
||||||
|
return 60 * (60 * self.hours + self.minutes) + self.seconds |
||||||
|
|
||||||
|
def ExtractDuration(s): |
||||||
|
return Duration(*map(int, DURATION_REGEX.match(s).groups())) |
||||||
|
|
||||||
|
Chapter = namedtuple('Chapter', 'number duration') |
||||||
|
|
||||||
|
def ParseChapters(d): |
||||||
|
""" |
||||||
|
Parses dictionary of (str) chapter numbers to chapter. |
||||||
|
|
||||||
|
Result will be an iterable of Chapter objects, sorted by number. |
||||||
|
""" |
||||||
|
for number, info in sorted(((int(n), info) for (n, info) in d.items())): |
||||||
|
yield Chapter(number, ExtractDuration(info)) |
||||||
|
|
||||||
|
AUDIO_TRACK_REGEX = re.compile( |
||||||
|
r'^(\S+)\s*((?:\([^)]*\)\s*)*)(?:,\s*(.*))?$') |
||||||
|
|
||||||
|
AUDIO_TRACK_FIELD_REGEX = re.compile( |
||||||
|
r'^\(([^)]*)\)\s*\(([^)]*?)\s*ch\)\s*' + |
||||||
|
r'((?:\([^()]*\)\s*)*)\(iso639-2:\s*([^)]+)\)$') |
||||||
|
|
||||||
|
AudioTrack = namedtuple('AudioTrack', |
||||||
|
'number lang codec channels iso639_2 extras') |
||||||
|
|
||||||
|
def ParseAudioTracks(d): |
||||||
|
for number, info in sorted(((int(n), info) for (n, info) in d.items())): |
||||||
|
m = AUDIO_TRACK_REGEX.match(info) |
||||||
|
if m: |
||||||
|
lang, field_string, extras = m.groups() |
||||||
|
m2 = AUDIO_TRACK_FIELD_REGEX.match(field_string) |
||||||
|
if m2: |
||||||
|
codec, channels, more_extras, iso639_2 = m2.groups() |
||||||
|
if more_extras: |
||||||
|
extras = more_extras + extras |
||||||
|
yield AudioTrack(number, lang, codec, channels, |
||||||
|
iso639_2, extras) |
||||||
|
else: |
||||||
|
warn('Cannot parse audio track fields %r' % field_string) |
||||||
|
else: |
||||||
|
warn('Cannot parse audio track info %r' % info) |
||||||
|
|
||||||
|
SUB_TRACK_REGEX = re.compile( |
||||||
|
r'^(\S(?:.*\S)?)\s+\(iso639-2:\s*([^)]+)\)\s*((?:\S(?:.*\S)?)?)$') |
||||||
|
|
||||||
|
SubtitleTrack = namedtuple('SubtitleTrack', |
||||||
|
'number name iso639_2 extras') |
||||||
|
|
||||||
|
def ParseSubtitleTracks(d): |
||||||
|
for number, info in sorted(((int(n), info) for (n, info) in d.items())): |
||||||
|
m = SUB_TRACK_REGEX.match(info) |
||||||
|
assert m, 'UNMATCHED %r' % info |
||||||
|
name, iso639_2, extras = m.groups() |
||||||
|
yield SubtitleTrack(number, name, iso639_2, extras) |
||||||
|
|
||||||
|
def RenderBar(start, length, total, width): |
||||||
|
end = start + length |
||||||
|
start = int(round(start * (width - 1) / total)) |
||||||
|
length = int(round(end * (width - 1) / total)) - start + 1 |
||||||
|
return ('‥' * start + |
||||||
|
'■' * length + |
||||||
|
'‥' * (width - start - length)) |
||||||
|
|
||||||
|
MAX_BAR_WIDTH = 50 |
||||||
|
|
||||||
|
def DisplayScan(titles): |
||||||
|
max_title_seconds = max( |
||||||
|
title.info['duration'].in_seconds() |
||||||
|
for title in titles) |
||||||
|
|
||||||
|
for title in titles: |
||||||
|
info = title.info |
||||||
|
size = ParseSize(info['size']) |
||||||
|
xaspect, yaspect = ComputeAspectRatio(size) |
||||||
|
duration = info['duration'] |
||||||
|
title_seconds = duration.in_seconds() |
||||||
|
print('Title % 3d/% 3d: %s %d×%d %d:%d %3g fps' % |
||||||
|
(title.number, len(titles), duration, size.width, |
||||||
|
size.height, xaspect, yaspect, size.fps)) |
||||||
|
for at in ParseAudioTracks(info['audio tracks']): |
||||||
|
print(' audio % 3d: %s (%sch) [%s]' % |
||||||
|
(at.number, at.lang, at.channels, at.extras)) |
||||||
|
for sub in ParseSubtitleTracks(info['subtitle tracks']): |
||||||
|
print(' sub % 3d: %s [%s]' % |
||||||
|
(sub.number, sub.name, sub.extras)) |
||||||
|
position = 0 |
||||||
|
if title_seconds > 0: |
||||||
|
for chapter in ParseChapters(info['chapters']): |
||||||
|
seconds = chapter.duration.in_seconds() |
||||||
|
bar_width = int(round( |
||||||
|
MAX_BAR_WIDTH * title_seconds / max_title_seconds)) |
||||||
|
bar = RenderBar(position, seconds, title_seconds, bar_width) |
||||||
|
print(' chapter % 3d: %s ◖%s◗' |
||||||
|
% (chapter.number, chapter.duration, bar)) |
||||||
|
position += seconds |
||||||
|
print() |
||||||
|
|
||||||
|
def ParseArgs(): |
||||||
|
description, epilog = __doc__.strip().split('\n', 1) |
||||||
|
parser = argparse.ArgumentParser(description=description, epilog=epilog, |
||||||
|
formatter_class=argparse.RawDescriptionHelpFormatter) |
||||||
|
parser.add_argument('-v', '--verbose', |
||||||
|
action='store_true', |
||||||
|
help="Increase verbosity.") |
||||||
|
parser.add_argument('-c', '--chapter_split', |
||||||
|
action='store_true', |
||||||
|
help="Split each chapter out into a separate file.") |
||||||
|
parser.add_argument('-n', '--dry-run', |
||||||
|
action='store_true', |
||||||
|
help="Don't actually write anything.") |
||||||
|
parser.add_argument('--scan', |
||||||
|
action='store_true', |
||||||
|
help="Display scan of disc; do not rip.") |
||||||
|
parser.add_argument('--main-feature', |
||||||
|
action='store_true', |
||||||
|
help="Rip only the main feature title.") |
||||||
|
parser.add_argument('-t', '--titles', |
||||||
|
default="*", |
||||||
|
help="""Comma-separated list of title numbers to consider |
||||||
|
(starting at 1) or * for all titles.""") |
||||||
|
parser.add_argument('-i', '--input', |
||||||
|
help="Volume to rip (must be a directory).") |
||||||
|
parser.add_argument('-o', '--output', |
||||||
|
help="""Output location. Extension is added if only one title |
||||||
|
being ripped, otherwise, a directory will be created to contain |
||||||
|
ripped titles.""") |
||||||
|
parser.add_argument('--mount-timeout', |
||||||
|
default=15, |
||||||
|
help="Amount of time to wait for a mountpoint to be mounted", |
||||||
|
type=float) |
||||||
|
args = parser.parse_args() |
||||||
|
if not args.scan and args.output is None: |
||||||
|
raise UserError("output argument is required") |
||||||
|
return args |
||||||
|
|
||||||
|
# TODO: make it possible to have ranges with no end (meaning they end at last |
||||||
|
# title) |
||||||
|
NUM_RANGE_REGEX = re.compile(r'^(\d*)-(\d+)|(\d+)$') |
||||||
|
def parse_titles_arg(titles_arg): |
||||||
|
if titles_arg == '*': |
||||||
|
return None # all titles |
||||||
|
else: |
||||||
|
def str_to_ints(s): |
||||||
|
m = NUM_RANGE_REGEX.match(s) |
||||||
|
if not m : |
||||||
|
raise UserError( |
||||||
|
"--titles must be * or list of integer ranges, found %r" % |
||||||
|
titles_arg) |
||||||
|
else: |
||||||
|
start,end,only = m.groups() |
||||||
|
if only is not None: |
||||||
|
return [int(only)] |
||||||
|
else: |
||||||
|
start = int(start) if start else 1 |
||||||
|
end = int(end) |
||||||
|
return range(start, end + 1) |
||||||
|
result = set() |
||||||
|
for s in titles_arg.split(','): |
||||||
|
result.update(str_to_ints(s)) |
||||||
|
result = sorted(list(result)) |
||||||
|
return result |
||||||
|
|
||||||
|
def main(): |
||||||
|
args = ParseArgs() |
||||||
|
dvd = DVD(args.input, args.verbose, args.mount_timeout) |
||||||
|
print('Reading from %r' % dvd.mountpoint) |
||||||
|
title_numbers = parse_titles_arg(args.titles) |
||||||
|
titles = tuple(dvd.ScanTitles(title_numbers, args.verbose)) |
||||||
|
|
||||||
|
if args.scan: |
||||||
|
DisplayScan(titles) |
||||||
|
else: |
||||||
|
if args.main_feature and len(titles) > 1: |
||||||
|
# TODO: make this affect scan as well |
||||||
|
titles = [FindMainFeature(titles, args.verbose)] |
||||||
|
|
||||||
|
if not titles: |
||||||
|
raise UserError("No titles to rip") |
||||||
|
else: |
||||||
|
if not args.output: |
||||||
|
raise UserError("No output specified") |
||||||
|
print('Writing to %r' % args.output) |
||||||
|
tasks = tuple(ConstructTasks(titles, args.chapter_split)) |
||||||
|
|
||||||
|
filenames = TaskFilenames(tasks, args.output, dry_run=args.dry_run) |
||||||
|
# Don't stomp on existing files |
||||||
|
for filename in filenames: |
||||||
|
if os.path.exists(filename): |
||||||
|
raise UserError('%r already exists' % filename) |
||||||
|
|
||||||
|
PerformTasks(dvd, tasks, len(titles), filenames, |
||||||
|
dry_run=args.dry_run, verbose=args.verbose) |
||||||
|
|
||||||
|
print('=' * 78) |
||||||
|
if not args.dry_run: |
||||||
|
dvd.Eject() |
||||||
|
|
||||||
|
def warn(msg): |
||||||
|
print('warning: %s' % (msg,), file=sys.stderr) |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
error = None |
||||||
|
try: |
||||||
|
main() |
||||||
|
except FileExistsError as exc: |
||||||
|
error = '%s: %r' % (exc.strerror, exc.filename) |
||||||
|
except UserError as exc: |
||||||
|
error = exc.message |
||||||
|
|
||||||
|
if error is not None: |
||||||
|
print('%s: error: %s' |
||||||
|
% (os.path.basename(sys.argv[0]), error), file=sys.stderr) |
||||||
|
sys.exit(1) |
||||||
Loading…
Reference in new issue