import plistlib

with open(filename, 'rb') as plist_file:
    plist = plistlib.load(plist_file)

import re, argparse
import sys
from matplotlib import pyplot
import plistlib
import numpy as np
# 找到重復的音樂
def find_duplicates(file_name):
    print("Find duplicate tracks in %s..." % file_name)
    with open(file_name, 'rb') as f:
        plist = plistlib.load(f)
    # 讀取playlist
    # 獲取音軌目錄
    tracks = plist['Tracks']
    # 創(chuàng)建音軌字典
    track_names = {}
    # 結構：{name:(duration,count)}
    # 遍歷 添加
    for track_id, track in tracks.items():
        try:
            name = track['name']
            duration = track['Total Time']
            # 檢查是不是以及在在字典里面了
            if name in track_names:
            # 匹配 歌曲名稱和時長
                if duration // 1000 == track_names['name'][0] // 1000:
                    count = track_names[name][1]
                    track_names[name] = (duration, count + 1)
            else:
                # 不匹配的情況下
                track_names[name] = (duration, count)
        except:
            # ignore
            pass
    # 保存重復的音軌， name, count
    dups = []
    for k, v in track_names.items():
        if v[1] > 1:
            dups.append(k, v[1])
    # 保存到一個文件
    if len(dups) > 0:
        print("發(fā)現(xiàn)一共有%d個重復的文件， 以及保存到了dup.txt中" % len(dups))
    else:
        print("沒有發(fā)現(xiàn)任何重復的文件")
    f = open("dups.txt", "w")
    for val in dups:
        f.writable("[%d] %s \n" % (val[0], val[1]))
    f.close()
# 查找多個播放列表中的共同的音軌
def find_common_tracks(file_names):
    # a list of sets of track names
    track_name_sets = []
    # 遍歷讀取多個plist文件
    for file_name in file_names:
        track_names = set()
        with open(file_name, 'rb') as f:
            plist = plistlib.load(f)
        # 獲取音軌節(jié)點
        tracks = plist.get("Tracks", {})
        # 遍歷迭代
        for track_id, track in tracks.items():
            try:
                track_names.add(track['Name'])
            except:
                # ignore
                pass
    # 添加到track_name_sets中
    track_name_sets.append(track_names)
    # 交集處理
    common_tracks = set.intersection(*track_name_sets)
    # 寫文件
    if len(common_tracks) > 0:
        f = open("common.txt", "w")
        for val in common_tracks:
            # s = "%s\n" % val
            f.write(f"{val}\n")
        f.close()
        print(f"Track names wirte to common.txt { len(common_tracks)}")
    else:
        print("No common tracks!")
# 收集歌曲評分和時長
def plot_stats(file_name):
    # 讀取播放列表
    with open(file_name, 'rb') as f:
        plist = plistlib.load(f)
    tracks = plist['Tracks']
    # 創(chuàng)建音軌排序和時長
    ratings = []
    durations = []
    for track_id, track in tracks.items():
        try:
            ratings.append(track['Album Rating'])
            durations.append(track['Total Time'])
        except:
            # ignore
            pass
    # 確保是有效數(shù)據
    if ratings == [] or durations == []:
        print(f"在文件中%s沒有有效的Album Rating/Total Time  數(shù)據 {file_name}")
        return
     # scatter plot 
    x= np.array(durations, np.int32) 
    # convert to minutes 
    x = x/60000.0 
    y = np.array(ratings, np.int32) 
    pyplot.subplot(2, 1, 1) 
    pyplot.plot(x, y, 'o') 
    pyplot.axis([0, 1.05*np.max(x), -1, 110]) 
    pyplot.xlabel('Track duration') 
    pyplot.ylabel('Track rating') 
    # plot histogram 
    pyplot.subplot(2, 1, 2) 
    pyplot.hist(x, bins=20) 
    pyplot.xlabel('Track duration') 
    pyplot.ylabel('Count') 
    # show plot 
    pyplot.show() 
def main():
    # create parser
    descStr = """
    This program analyzes playlist files (.xml) exported from iTunes.
    """
    parser = argparse.ArgumentParser(description=descStr)
    # add a mutually exclusive group of arguments
    group = parser.add_mutually_exclusive_group()
    # add expected arguments
    group.add_argument('--common', nargs = '*', dest='plFiles', required=False)
    group.add_argument('--stats', dest='plFile', required=False)
    group.add_argument('--dup', dest='plFileD', required=False)
    # parse args
    args = parser.parse_args()
    if args.plFiles:
        # find common tracks
        find_common_tracks(args.plFiles)
    elif args.plFile:
        # plot stats
        plot_stats(args.plFile)
    elif args.plFileD:
        # find duplicate tracks
        find_duplicates(args.plFileD)
    else:
        print("These are not the tracks you are looking for.")
if __name__ == "__main__":
    main()