Forum: General Discussion

import os
import random
from datetime import datetime, timedelta

def read_m3u_list(m3u_file):
    """Read an M3U playlist and return a set of files."""
    excluded_files = set()
    if os.path.isfile(m3u_file):
        with open(m3u_file, "r", encoding="utf-8") as file:
            for line in file:
                line = line.strip()
                if line and not line.startswith("#"):
                    excluded_files.add(line)
    return excluded_files

def read_history_files_by_name(history_folder, days_to_exclude=1):
    """Read M3U playlists from the History folder by filename date and return a set of excluded files."""
    excluded_files = set()
    if not os.path.isdir(history_folder):
        print(f"The history folder does not exist: {history_folder}")
        return excluded_files

    cutoff_date = datetime.now() - timedelta(days=days_to_exclude)
    for file in os.listdir(history_folder):
        file_path = os.path.join(history_folder, file)
        # Check for M3U files and parse date from filename
        if file.endswith(".m3u") and os.path.isfile(file_path):
            try:
                file_date = datetime.strptime(file[:10], "%Y-%m-%d")  # Extract date from filename
                if file_date >= cutoff_date:
                    excluded_files.update(read_m3u_list(file_path))
            except ValueError:
                print(f"Skipping file with invalid date format: {file}")
    return excluded_files

def generate_m3u_playlists(folder_path, special_folder, output_folder, history_folder, num_playlists=24, days_to_exclude=4):
    # Supported audio file extensions
    supported_extensions = {".mp3", ".wav", ".flac", ".aac", ".ogg", ".m4a"}

    # Check if folders exist
    if not os.path.isdir(folder_path):
        print(f"The specified folder does not exist: {folder_path}")
        return

    if not os.path.isdir(special_folder):
        print(f"The special folder does not exist: {special_folder}")
        return

    if not os.path.isdir(output_folder):
        os.makedirs(output_folder)

    if not os.path.isdir(history_folder):
        print(f"The history folder does not exist: {history_folder}")
        return

    # Read files from the History folder to exclude by filename date
    excluded_files = read_history_files_by_name(history_folder, days_to_exclude=days_to_exclude)

    # Collect all audio files from the main folder
    audio_files = [
        os.path.join(folder_path, file)
        for file in os.listdir(folder_path)
        if os.path.splitext(file)[1].lower() in supported_extensions
           and os.path.join(folder_path, file) not in excluded_files
    ]

    # Collect all audio files from the special folder (do not analyze history for these)
    special_files = [
        os.path.join(special_folder, file)
        for file in os.listdir(special_folder)
        if os.path.splitext(file)[1].lower() in supported_extensions
    ]

    if not special_files:
        print("No special audio files found in the special folder.")
        return

    # Always use the same jingle as the special track (first file in the folder)
    special_track = special_files[0]  # Always pick the first file from the special folder

    if not audio_files:
        print("No audio files found in the specified folder.")
        return

    # Shuffle the audio files
    random.shuffle(audio_files)

    # Distribute files across playlists
    chunk_size = len(audio_files) // num_playlists
    for i in range(num_playlists):
        # Distribute the remaining tracks across playlists
        playlist_files = audio_files[i * chunk_size: (i + 1) * chunk_size]
        if i == num_playlists - 1:  # Add leftover tracks to the last playlist
            playlist_files += audio_files[(i + 1) * chunk_size:]

        # Add the special track at the top
        playlist_files.insert(0, special_track)

        # Generate the playlist name
        playlist_name = os.path.join(output_folder, f"playlist_{i + 1}.m3u")

        # Write the playlist to a file
        with open(playlist_name, "w", encoding="utf-8") as m3u:
            m3u.write("#EXTM3U\n")  # M3U header
            for file in playlist_files:
                m3u.write(f"{file}\n")

        print(f"Playlist generated: {playlist_name}")

# Parameters
folder_path = r"C:\Muziek\POP"  # Folder with audio files
special_folder = r"C:\Muziek\r2 jingles"  # Folder with special audio files
output_folder = r"C:\Muziek\Playlist VDJ"  # Folder to save playlists
history_folder = r"C:\Users\xxxx\AppData\Local\VirtualDJ\History"  # Folder with previous playlists

# Generate playlists
generate_m3u_playlists(folder_path, special_folder, output_folder, history_folder)