import os import json import requests import time import sqlite3 from sqlite3 import Error from secrets import familyfeed_id, familyqueue_id, spotify_user_id from refresh import Refresh refreshCaller = Refresh() class cynthia: def __init__(self): self.user_id = spotify_user_id self.spotify_token = "" self.here_i_am = os.path.dirname(os.path.abspath(__file__)) self.playlist_id = familyfeed_id self.queue_id = familyqueue_id self.queue_list = [] self.queue_track_name_list = [] self.queue_track_list = [] self.rowid_list = [] self.urimatch_list = [] self.namematch_list = [] self.played_track = "" self.uri_match = 0 self.name_match = 0 def call_refresh(self): # NOTE Euan is king global refreshCaller while True: self.spotify_token = refreshCaller.refresh() if self.spotify_token == -1: print("==============================\n[!] Error refreshing token, retrying in 30 Seconds\n==============================") time.sleep(30) else: break def repopulate_the_queue(self): # NOTE List of functions os.system('cls') print("Cynthia You Spawn of Satan v.6.6.6. 15/03/2022 Viking Utilities") print("And Finally...........") self.get_endless_playlist() self.database = self.here_i_am + r"\\" + self.playlist_name + ".db" print("Getting Queue Count") self.get_queue_count() print("Getting Queue Items") self.get_the_queue() print("Getting Last 50 Played Tracks") self.check_played_list() print("Getting New Tracks For Queue") self.get_new_tracks() print("Remaining Un-Played Tracks in Database = " + str(self.remain_records)) print(str(self.uri_match) + " URI Matches") #print(*self.urimatch_list, sep='\n') print(str(self.name_match) + " Name Matches") #print(*self.namematch_list, sep='\n') print(str(self.record_wanted) + " Tracks added from database records:- ") print(*self.rowid_list, sep=', ') def get_queue_count(self): # NOTE Get how many tracks in Endless Queue query = "https://api.spotify.com/v1/playlists/{}".format(self.queue_id) response = requests.request("get", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}) json_data = json.loads(response.text) self.queue_total = (json_data["tracks"]["total"]) self.queue_name = (json_data["name"]) return def get_the_queue_again(self): q = 0 while q < self.queue_total: try: for q in range(self.queue_total): self.offset = q query = "https://api.spotify.com/v1/playlists/{}/tracks".format(self.queue_id) response = requests.request("get", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}, params={"limit": 1, "offset": self.offset}) json_data = json.loads(response.text) self.track_uri = json_data["items"][0]["track"]["uri"] self.track_name = json_data["items"][0]["track"]["name"] self.artist_name = json_data["items"][0]["track"]["artists"][0]["name"] self.sort_out_the_queue_track_name() if self.track_artist_name == self.played_artist_name or self.track_name == self.played_name: self.played_track = self.track_uri return if response.status_code == 200: q = q + 1 continue except: print("Track " + self.played_name + " Not Present in " + self.queue_name + " Playlist! Please Investigate Playlist & Database!!") return def get_the_queue(self): i = 0 while i < self.queue_total: for i in range(self.queue_total): self.offset = i query = "https://api.spotify.com/v1/playlists/{}/tracks".format(self.queue_id) response = requests.request("get", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}, params={"limit": 1, "offset": self.offset}) json_data = json.loads(response.text) self.track_uri = json_data["items"][0]["track"]["uri"] self.track_name = json_data["items"][0]["track"]["name"] self.artist_name = json_data["items"][0]["track"]["artists"][0]["name"] self.sort_out_the_queue_track_name() self.queue_list.append(self.track_uri) self.queue_track_list.append(self.track_name) self.queue_track_name_list.append(self.track_artist_name) if response.status_code == 200: i = i + 1 continue def sort_out_the_queue_track_name(self): self.track_name = str(self.track_name).upper() self.artist_name = str(self.artist_name).upper() self.hyphen = str(self.track_name).find(" - ") if self.hyphen != -1: self.track_name = self.track_name[:self.hyphen] self.track_name = self.track_name.strip() self.artist_name = self.artist_name.strip() self.track_artist_name = self.track_name + " - " + self.artist_name def check_played_list(self): # NOTE Get Last 50 played tracks. Checking to see if each one is in the queue list last_track = False query = "https://api.spotify.com/v1/me/player/recently-played" response = requests.request("get", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}, params={"limit": 50}) response_json = response.json() for k in response_json["items"]: found = False self.played_track = (k["track"]["uri"]) self.played_name = (k["track"]["name"]) self.played_artist = (k["track"]["album"]["artists"][0]["name"]) self.sort_out_the_last_played_track_name() if last_track == False: print("Saving Last Track Played Which Was:- " + (k["track"]["name"])) last_track = True continue if self.played_track in self.queue_list: self.uri_match = self.uri_match + 1 self.urimatch_list.append(self.played_name) self.mark_played_and_remove() found = True continue if found == False: if self.played_artist_name in self.queue_track_name_list or self.played_name in self.queue_track_list: self.get_the_queue_again() self.name_match = self.name_match + 1 self.namematch_list.append(self.played_name) self.mark_played_and_remove() continue def sort_out_the_last_played_track_name(self): self.played_name = str(self.played_name).upper() self.played_artist = str(self.played_artist).upper() self.hyphen = str(self.played_name).find(" - ") if self.hyphen != -1: self.played_name = self.played_name[:self.hyphen] self.played_name = self.played_name.strip() self.played_artist = self.played_artist.strip() self.played_artist_name = self.played_name + " - " + self.played_artist def mark_played_and_remove(self): # NOTE If it is in the queue and has been played then deal with it self.mark_played_in_db() self.remove_from_playlist() return def get_endless_playlist(self): # NOTE Get the Endless Playlist Name query = "https://api.spotify.com/v1/playlists/{}".format(self.playlist_id) response = requests.request("get", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}) json_data = json.loads(response.text) self.playlist_name = (json_data["name"]) return def mark_played_in_db(self): # NOTE Mark the played tracks in the database global conn conn = self.create_connection(self.database) cur = conn.cursor() try: with conn: try: cur.execute("SELECT * FROM Track_List WHERE URI = '" + self.played_track + "'") row = cur.fetchone() play_count = row[5] play_count = play_count + 1 cur.execute("UPDATE Track_List SET PLAY_COUNT = '" + str(play_count) + "' WHERE URI = '" + self.played_track + "'") conn.commit() return except: print("Track " + str(self.played_name) + " May Not Be Present in Database. Please Investigate Playlist & Database!!") return except Error as e2: print("E2") print(e2) return def create_connection(self, db_file): # NOTE Connect to the database conn = None try: conn = sqlite3.connect(db_file) return conn except Error as e1: print("E1") print(e1) return conn def remove_from_playlist(self): # NOTE Remove played tracks from the Endless Queue playlist query = "https://api.spotify.com/v1/playlists/{}/tracks".format(self.queue_id) response = requests.request("delete", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}, data=json.dumps({"tracks": [{"uri": self.played_track}]})) if response.status_code != 200: print(response.status_code) self.played_track = "" return def get_new_tracks(self): # NOTE Get how many new tracks are needed self.get_queue_count() self.remaining_queue = self.queue_total self.get_tracks() return def get_tracks(self): # NOTE Get the new tracks from the Endless Playlist self.needed = 50 - self.remaining_queue self.record_wanted = self.needed print(str(self.needed) + " New Tracks Required!") print("Performing Record Check") self.check_played_count() if self.needed == 0: return conn = self.create_connection(self.database) cur = conn.cursor() try: cur.execute("SELECT * FROM Track_List WHERE PLAYED = 0 order by RANDOM()") except Error as e: print(e) return rows = cur.fetchall() for row in rows: self.newtrack = row[0] self.add_track_to_queue() self.set_played_flag() self.needed = self.needed - 1 if self.needed == 0: return return def set_played_flag(self): conn = self.create_connection(self.database) cur = conn.cursor() cur.execute("SELECT * FROM Track_List WHERE URI = '" + self.newtrack + "'") cur.execute("SELECT rowid, * FROM Track_List WHERE URI = '" + self.newtrack +"'") toe = cur.fetchone() self.rowid_list.append(toe[0]) cur.execute("UPDATE Track_List SET PLAYED = True WHERE URI = '" + self.newtrack + "'") conn.commit() return def add_track_to_queue(self): # NOTE Add the new tracks to the Endless Queue Playlist query = "https://api.spotify.com/v1/playlists/{}/tracks?uris={}".format(self.queue_id, self.newtrack) response = requests.request("post", url=query, headers={"Content-type": "application/json", "Authorization": "Bearer {}".format(self.spotify_token)}) return def check_played_count(self): # NOTE See how many more tracks are un-played conn = self.create_connection(self.database) cur = conn.cursor() try: cur.execute("SELECT COUNT (*) FROM Track_List") total_records = cur.fetchone() cur.execute("SELECT COUNT (*) FROM Track_List WHERE PLAYED = 1") played_records = cur.fetchone() self.remain_records = total_records[0] - played_records[0] if self.remain_records < self.needed: self.reset_played_flag() return except Error as e: print(e) return def reset_played_flag(self): # NOTE If no tracks left to play then reset the whole database so they are all available print("Reseting Database '" + self.playlist_name + "'") conn = self.create_connection(self.database) with conn: cur = conn.cursor() try: cur.execute("SELECT PLAYED FROM Track_List") except Error as e: print(e) return rows = cur.fetchall() for row in rows: cur.execute("UPDATE Track_List SET PLAYED = 0") return a = cynthia() a.call_refresh() a.repopulate_the_queue()