walkman-utils/main.py at master · corenel/walkman-utils · 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
import os
import util
import setting

from tqdm import tqdm


def sync_playlist(playlists, walkman_dir, remove_unmatched=False):
    # Get music list
    util.status('Getting songs in playlists from iTunes')
    songs_in_playlists = util.get_files_in_playlist(playlists)
    itunes_folder, songs_in_playlists_rel = util.split_filepath(
        songs_in_playlists,
        setting.ITUNES_MEDIA_ROOT)
    util.status('Getting songs on Walkman')
    music_on_walkman = util.scan_directory(walkman_dir, setting.MUSIC_FILE_EXT)
    util.status('Comparing song lists')
    to_be_updated, to_be_removed, _ = util.compare_filelists(
        songs_in_playlists_rel, music_on_walkman,
        root_src=itunes_folder, root_dst=walkman_dir
    )

    # Update music
    util.status('Syncing songs from local to Walkman')
    util.sync_filelists(to_be_updated, to_be_removed,
                        src_dir=itunes_folder, dst_dir=walkman_dir,
                        remove_unmatched=remove_unmatched)


def create_m3u_playlist(playlists, walkman_dir, walkman_prefix):
    util.status('Creating m3u playlists on Walkman')
    progress = tqdm(playlists)
    for playlist in progress:
        progress.set_description('{}'.format(os.path.basename(playlist)))
        songs_in_playlists = util.get_files_in_playlist(playlist)
        songs_with_prefix = util.format_playlist_with_prefix(songs_in_playlists,
                                                             walkman_prefix)
        with open(os.path.join(walkman_dir, '{}.m3u'.format(playlist)), 'w') as f:
            f.write('\n'.join(songs_with_prefix))
            f.write('\n')


def create_local_lyrics(playlists, lyrics_dir, lyrics_source_dir):
    util.status('Creating local lyrics directory')
    tracks_in_playlist = util.get_tracks_in_playlist(playlists)
    util.struct_lyrics_dir(tracks_in_playlist,
                           src_dir=lyrics_source_dir,
                           dst_dir=lyrics_dir)


def sync_lyrics(playlists, lyrics_dir, walkman_dir, remove_unmatched=False):
    # Get lyrics list
    util.status('Getting list of lyrics files locally')
    lyrics_on_local = util.get_lyrics_files_in_playlist(playlists)
    _, lyrics_on_local_rel = util.split_filepath(lyrics_on_local)

    util.status('Getting list ofj lyrics files on Walkman')
    lyrics_on_walkman = util.scan_directory(walkman_dir, setting.LYRICS_FILE_EXT)

    # Update lyrics
    util.status('Comparing lyrics lists')
    to_be_updated, to_be_removed, _ = util.compare_filelists(
        lyrics_on_local_rel, lyrics_on_walkman,
        root_src=lyrics_dir, root_dst=walkman_dir
    )

    util.status('Syncing lyrics files from local to Walkman')
    util.sync_filelists(to_be_updated, to_be_removed,
                        src_dir=lyrics_dir, dst_dir=walkman_dir,
                        remove_unmatched=remove_unmatched)


if __name__ == '__main__':
    if util.confirm('Sync songs in playlist to Walkman?'):
        sync_playlist(playlists=setting.PLAYLISTS,
                      walkman_dir=setting.WALKMAN_DIR)

        create_m3u_playlist(playlists=setting.PLAYLISTS,
                            walkman_dir=setting.WALKMAN_DIR,
                            walkman_prefix=setting.WALKMAN_PLAYLIST_PREFIX)

    if util.confirm('Create local lyrics directory?'):
        create_local_lyrics(playlists=setting.PLAYLISTS,
                            lyrics_dir=setting.LYRICS_DIR,
                            lyrics_source_dir=setting.LYRICS_SOURCE_DIR)

    if util.confirm('Sync lyrics to Walkman?'):
        sync_lyrics(playlists=setting.PLAYLISTS,
                    lyrics_dir=setting.LYRICS_DIR,
                    walkman_dir=setting.WALKMAN_DIR)