Added commercial support. Currently seems to be working ok, but needs adjusting / testing. Currently am using it with a time-shift of '5'.

src/PseudoChannelCommercial.py

@@ -0,0 +1,133 @@
+"""Commercial Functionality
+"""
+from random import shuffle
+import random
+import copy
+import datetime
+from src import Commercial
+
+class PseudoChannelCommercial():
+
+	MIN_DURATION_FOR_COMMERCIAL = 10 #seconds
+	daily_schedule = []
+
+	def __init__(self, commercials):
+
+		self.commercials = commercials
+
+	def get_commercials_to_inject(self):
+
+		self.go()
+
+		return None
+
+	def get_random_commercial(self):
+
+		random_commercial = random.choice(self.commercials)
+
+		random_commercial_dur_seconds = (int(random_commercial[4])/1000)%60
+
+		while random_commercial_dur_seconds < self.MIN_DURATION_FOR_COMMERCIAL:
+
+			 random_commercial = random.choice(self.commercials)
+
+			 random_commercial_dur_seconds = (int(random_commercial[4])/1000)%60
+
+		return random_commercial
+
+	def go(self):
+
+		shuffled_commercial_list = copy.deepcopy(self.commercials)
+
+		random.shuffle(self.commercials, random.random)
+
+		#print shuffled_commercial_list
+
+		prev_item = None
+
+		for entry in self.daily_schedule:
+
+			"""First Episode"""
+			if prev_item == None:
+
+				prev_item = entry
+
+			else:
+
+				prev_item_end_time = datetime.datetime.strptime(prev_item[9], '%Y-%m-%d %H:%M:%S.%f')
+
+				curr_item_start_time = datetime.datetime.strptime(entry[8], '%I:%M:%S %p')
+
+				time_diff = (curr_item_start_time - prev_item_end_time)
+
+				days, hours, minutes = time_diff.days, time_diff.seconds // 3600, time_diff.seconds // 60 % 60
+
+				count = 0
+
+				commercial_list = []
+
+				commercial_dur_sum = 0
+
+				while int(time_diff.total_seconds()) >= commercial_dur_sum and count < len(self.commercials):
+
+
+					random_commercial = self.get_random_commercial()
+
+					commercial_list.append(random_commercial)
+
+					commercial_dur_sum += int(random_commercial[4])
+
+				print commercial_list
+
+				prev_item = entry
+
+	def get_commercials_to_place_between_media(self, last_ep, now_ep):
+
+		print last_ep.end_time, now_ep.start_time
+
+		prev_item_end_time = datetime.datetime.strptime(last_ep.end_time.strftime('%Y-%m-%d %H:%M:%S.%f'), '%Y-%m-%d %H:%M:%S.%f')
+
+		curr_item_start_time = datetime.datetime.strptime(now_ep.start_time, '%I:%M:%S %p')
+
+		time_diff = (curr_item_start_time - prev_item_end_time)
+
+		days, hours, minutes = time_diff.days, time_diff.seconds // 3600, time_diff.seconds // 60 % 60
+
+		count = 0
+
+		commercial_list = []
+
+		commercial_dur_sum = 0
+
+		while int(time_diff.total_seconds()) >= commercial_dur_sum and count < len(self.commercials):
+
+			random_commercial = self.get_random_commercial()
+
+			new_commercial_seconds = (int(random_commercial[4])/1000)%60
+
+			commercial_dur_sum += new_commercial_seconds
+
+			new_commercial_start_time = prev_item_end_time + datetime.timedelta(seconds=commercial_dur_sum)
+
+			new_commercial_end_time = new_commercial_start_time + datetime.timedelta(seconds=int(new_commercial_seconds))
+
+			formatted_time_for_new_commercial = new_commercial_start_time.strftime('%I:%M:%S %p')
+
+			new_commercial = Commercial(
+				"Commercials",
+				random_commercial[3],
+				formatted_time_for_new_commercial, # natural_start_time
+				new_commercial_end_time,
+				random_commercial[4],
+				"everyday", # day_of_week
+	            "true", # is_strict_time
+	            "1", # time_shift 
+	            "0", # overlap_max
+	            "", # plex_media_id
+			)
+
+			commercial_list.append(new_commercial)
+
+			#print "here!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
+
+		return commercial_list