This repository has been archived by the owner on Apr 18, 2021. It is now read-only.
forked from dskrad/mp3chaps
/
mp3chaps.py
executable file
·104 lines (90 loc) · 2.92 KB
/
mp3chaps.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python
# encoding: utf-8
"""
Usage:
mp3chaps.py (-l | -i | -r) <filename>
Options:
-h Show this help text
-l List chapters in <filename>
-i Import chapters from <filename>.chapters.txt
-r Remove chapters from <filename>
"""
from eyed3.id3 import Tag
from eyed3 import core
from docopt import docopt
import os
def list_chaps(tag):
"list chapters in tag"
print "Chapters:"
for chap in tag.chapters:
print chap.sub_frames.get("TIT2")[0]._text
def remove_chaps(tag):
"remove all the chapters and save tag to file"
chaps = [chap for chap in tag.chapters]
for chap in chaps:
print "removing {}".format(chap.sub_frames.get("TIT2")[0]._text)
tag.chapters.remove(chap.element_id)
tag.save()
def parse_chapters_file(fname):
filename, ext = os.path.splitext(fname)
chapters_fname = "{}.chapters.txt".format(filename)
chaps = []
with open(chapters_fname, "r") as f:
for line in f.readlines():
time, title = line.split()[0], " ".join(line.split()[1:])
chaps.append((to_millisecs(time), title))
return chaps
def add_chapters(tag, fname):
chaps = parse_chapters_file(fname)
audioFile = core.load(fname)
total_length = audioFile.info.time_secs * 1000
chaps_ = []
for i, chap in enumerate(chaps):
if i < (len(chaps)-1):
chaps_.append( ((chap[0], chaps[i+1][0]), chap[1]) )
chaps_.append( ((chaps[-1][0], total_length), chaps[-1][1]) )
index = 0
child_ids = []
for chap in chaps_:
element_id = "ch{}".format(index)
times, title = chap
new_chap = tag.chapters.set(element_id, times)
new_chap.sub_frames.setTextFrame("TIT2", u"{}".format(title))
child_ids.append(element_id)
index += 1
tag.table_of_contents.set("toc", child_ids=child_ids)
list_chaps(tag)
tag.save()
def to_millisecs(time):
h, m, s = [float(x) for x in time.split(":")]
return int(1000 * (s + m*60 + h*60*60))
#print to_millisecs("00:10:25.055"))
def main():
"Entry point"
args = docopt(__doc__, version="mp3chaps 0.1")
tag = Tag()
tag.parse(args["<filename>"])
if args["-l"]:
list_chaps(tag)
elif args["-i"]:
add_chapters(tag, args["<filename>"])
elif args["-r"]:
remove_chaps(tag)
if __name__ == '__main__':
main()
#first we will set chapters with element_id and times tuple (start, end)
#times are in milliseconds
#tag.chapters.set("ch1", (0,10000))
#tag.chapters.set("ch2", (10000, 360000))
#tag.chapters.set("ch3", (360000, 1800000))
#now we will set titles for each chapter
#chap1 = tag.chapters.get("ch1")
#chap1.sub_frames.setTextFrame("TIT2", u"Here is my first chapter title")
#chap2 = tag.chapters.get("ch2")
#chap2.sub_frames.setTextFrame("TIT2", u"Here is my second chapter title")
#chap3 = tag.chapters.get("ch3")
#chap3.sub_frames.setTextFrame("TIT2", u"Here is my third chapter title")
#don't forget to add chapters to the toc
#tag.table_of_contents.set("toc", child_ids=["ch1", "ch2", "ch3"])
#last but not least, save our tag
#tag.save()