playlistiterator.py

import random

class Song:
    def __init__(self, title, artist):
        self.title = title
        self.artist = artist

class Playlist:
    def __init__(self, name):
        self.name = name
        self.songs = []

    def add_song(self, song):
        self.songs.append(song)

    def __iter__(self):
        return PlaylistIterator(self.songs)

class PlaylistIterator:
    def __init__(self, songs ,  mode="Normal"): #Normal,Shuffle or Repeat
        self.songs = songs
        self.index = 0
        self.mode = mode

        if self.mode == "Shuffle":
                random.shuffle(self.songs)

    def __next__(self):
        if self.index < len(self.songs):
            song = self.songs[self.index]
            self.index += 1
            return song
        elif self.mode == "Repeat" and self.index >= len(self.songs):
            self.index = 0
            song = self.songs[self.index]
            self.index += 1
            return song
        else:
            raise StopIteration
   

Music1 = Song("Song 1", "Artist 1")
Music2 = Song("Song 2", "Artist 2")
Music3 = Song("Song 3", "Artist 3")
Music4 = Song("Song 4", "Artist 4")
my_playlist = Playlist("My Playlist")
my_playlist.add_song(Music1)
my_playlist.add_song(Music2)
my_playlist.add_song(Music3)
my_playlist.add_song(Music4)

for song in my_playlist:
    print(f"Title: {song.title}, Artist: {song.artist}")