main.py

# -*- coding: utf-8 -*-

"""Muzik analyser"""
import os.path
import pickle
from collections import defaultdict

from models.Song import Song
from utils.utils import Utils

__author__ = "Filip Koprivec"


def main(export=False):
    if export:
        utils = Utils()

        print("Exporting database")

        try:
            utils.export_database()
        except FileNotFoundError:
            print("Database not found, using git exported files")

    print("Importing songs...")

    songs = Song.list_all_songs_from_analyzing_servers()

    print("All songs:", len(songs))

    similar = defaultdict(list)

    for song in songs:
        similar[song.comparable_name].append(song)

    duplicates = []
    for name, duplicate_songs in similar.items():
        if len(duplicate_songs) > 2 and name:
            duplicates.append(duplicate_songs)

    pickle.dump(songs, open(os.path.join("data", "output", "songs.out"), "wb"))
    pickle.dump(duplicates, open(os.path.join("data", "output", "duplicates.out"), "wb"))
    print("Duplicates:", len(duplicates))


if __name__ == '__main__':
    main()