Umphreys Songs and frequencies

an_oop_umph_concert_simulator.py

import random

import pandas as pd
import numpy as np
# import requests
# from bs4 import BeautifulSoup as bs

# base_setlist_url = "https://allthings.umphreys.com/setlists/2020.html"

df= pd.read_csv('https://gist.githubusercontent.com/duncangh/f732e79528c809c6affcfcbfaa087cba/raw/393a5861d749b0f1b60879038ce70eca99fe3517/songs.csv')

weights = df.times_played_live
weights = weights / sum(weights)
songs = df.song_name


class Concert:
	songs = df.song_name
	weights = df.times_played_live / sum(df.times_played_live)

	def __init__(self):
		self.total_songs = random.choice(range(13, 20))
		self.total_covers = random.choices(range(4), weights=[.45, .40, .10, .05])

		self.concert_songs_arr = None
		self.set_1 = None
		self.set_2 = None
		self.encore = None


		self.main()

		repr(self)


	def main(self):
		self.concert_songs_arr = np.random.choice(self.songs,
		                                          p=self.weights,
		                                          size=(self.total_songs,),
		                                          replace=False)

		slice_1, slice_2 = self._generate_random_setlist_slices()
		self._split_songs_into_sets(slice_1=slice_1, slice_2=slice_2)

		return

	def _generate_random_setlist_slices(self):
		"""get the random numbers needed for context. E.g. concert song length, setlist length and how many covers to play"""
		_setlist_len_range = range(
			round(self.total_songs / 3) + 1,
			self.total_songs-round(self.total_songs / 3) + 1
		)
		self.setlist_1_len = random.choice(_setlist_len_range)

		slice_1 = slice(self.setlist_1_len)
		slice_2 = slice(self.setlist_1_len, self.total_songs - 1)
		return slice_1, slice_2

	def _split_songs_into_sets(self, slice_1, slice_2):
		self.set_1 = self.concert_songs_arr[slice_1]
		self.set_2 = self.concert_songs_arr[slice_2]

		self.encore = self.concert_songs_arr[-1]
		return

	def __repr__(self):
		return f"""Umphrey's McGee at Simulated Stadium\nSet 1:\n\t{', '.join(self.set_1)}\nSet 2:\n\t{', '.join(self.set_2)}\nEncore:\n\t{self.encore}"""

c = Concert()

print(c)

"""
Umphrey's McGee at Simulated Stadium
Set 1:
	Phil's Farm, Roulette, 1348, In The Kitchen, So What, Mullet (Over), Glory, 40's Theme, Pay the Snucka
Set 2:
	Preamble, Muff II: The Revenge, Plunger, It's About That Time
Encore:
	Nothing Too Fancy
"""

songs.csv

umphrey's mcgee concert simulator.py

import random

import pandas as pd
import numpy as np
# import requests
# from bs4 import BeautifulSoup as bs

# base_setlist_url = "https://allthings.umphreys.com/setlists/2020.html"

df= pd.read_csv('https://gist.githubusercontent.com/duncangh/f732e79528c809c6affcfcbfaa087cba/raw/393a5861d749b0f1b60879038ce70eca99fe3517/songs.csv')

weights = df.times_played_live
weights = weights / sum(weights)
songs = df.song_name



def generate_concert(songs=songs, weights=songs, total_songs=16):
	concert_sonngs_arr = np.random.choice(songs,
	                                      p=weights,
	                                      size=(16,),
	                                      replace=False)

	set_1, set_2, encore = concert_sonngs_arr[0:8], concert_sonngs_arr[8:15], concert_sonngs_arr[15]

	print(f"""Umphrey's McGee at Simulated Stadium
Set 1: 
	{', '.join(set_1)}

Set 2:
	{', '.join(set_2)}
Encore:		
	{encore}""")

generate_concert(songs=songs, weights=weights)
"""Umphrey's McGee at Simulated Stadium
Set 1: 
	Plunger, Muff II: The Revenge, Get In The Van, In The Kitchen, Tribute to the Spinal Shaft, Kimble, Smell the Mitten, Mulche's Odyssey

Set 2:
	13 Days, Gurgle, Great American, Kula, The Floor, August, The Bottom Half
Encore:		
	Bittersweet Haj
"""