mp3tagbatchupdate/mp3tagbatchupdate.py at main · RyanDengo/mp3tagbatchupdate · GitHub

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
#! /usr/bin/env python
# encoding: utf-8
import glob
import eyed3
import sys
import os
import click


TAGS = ('album', 'album_artist', 'album_type', 'artist', \
'artist_origin', 'artist_url', 'audio_file_url', 'audio_source_url',\
 'composer', 'copyright', 'copyright_url', 'publisher', 'publisher_url', 'release_date', \
 'tagging_date', 'recording_date', 'title', 'track_num', 'disc_num')

@click.command()
@click.option('--directory', help='mp3所在目录')
@click.option('--album', help='专辑名称')
@click.option('--artist', help='艺人名称')
@click.option('--publisher', help='发行公司')
@click.option('--release_date', help='发行日期')
@click.option('--disc_num')
@click.option('--clear', help='clear all tags', is_flag=True)
@click.option('--renamefile', help='根据title标签重命名文件', is_flag=True)
@click.option('--filenameastitle', help='用文件名设置title标签', is_flag=True)
@click.option('--dirnameasalbum', help='用目录名设置album标签', is_flag=True)
@click.option('--verbose', is_flag=True)
def  go(directory, \
    clear, renamefile, filenameastitle, dirnameasalbum, verbose, **kwargs):

    if verbose:
        for k,v in locals().items():
            print(f'{k}: {v}')
    if directory is None:
        directory = os.getcwd()
    else:
        try:
            os.chdir(directory)
        except FileNotFoundError as e:
            print(e)
            sys.exit(1)

    mp3_filenames = glob.glob('*.[mM][pP]3')
    for f in mp3_filenames:
        print(f'Changing tag of {f}')
        mp3 = eyed3.load(f)
        tag_dict = {}
        if not mp3:
            print(f'无法打开{f}')
            continue

        if not mp3.tag:
            mp3.initTag()

        for i in TAGS:
            tg = getattr(mp3.tag, i)
            if not tg:
                tag_dict[i] = None
            else:
                try:
                    if isinstance(tg, str):
                        tag_dict[i] = tg.encode('latin1').decode('gbk')
                    else:
                        tag_dict[i] = tg
                except UnicodeEncodeError as e:
                    print(e)
                    tag_dict[i] = tg

        if dirnameasalbum:
            album = os.path.basename(directory)

        if filenameastitle:
            tag_dict['title'] = f[:-4] # strip '.mp3'

        if clear:
            #清空原来的标签,用新的标签覆盖
            mp3.tag.clear()
        else:
            mp3.initTag()

        for k, v in kwargs.items():
            if k in TAGS and  v:
                tag_dict[k] = v

        for k, v in tag_dict.items():
            if v:
                if verbose:
                    print(f'Setting mp3.tag.{k} to {v}')
                setattr(mp3.tag, k, v)

        mp3.tag.save()

        if renamefile:
            print(f'Renaming {f} to {mp3.tag.title}.mp3')
            os.rename(f, f"{mp3.tag.title}.mp3")

if __name__ == '__main__':
    go()