black / add artist_genre_import script
This commit is contained in:
@@ -9,99 +9,111 @@ import random
|
||||
import discord_presence_priv
|
||||
import catbox
|
||||
from setproctitle import setproctitle
|
||||
|
||||
from typing import Optional
|
||||
from pypresence import Presence, ActivityType
|
||||
|
||||
setproctitle("disc-presence")
|
||||
|
||||
|
||||
class ImageUpload:
|
||||
def __init__(self) -> None:
|
||||
self.upload_uri = "https://api.codey.lol/misc/upload_activity_image"
|
||||
|
||||
|
||||
def upload(self, file: bytes) -> None:
|
||||
try:
|
||||
r = requests.post(self.upload_uri, files=
|
||||
{'image': file})
|
||||
r = requests.post(self.upload_uri, files={"image": file})
|
||||
print(f"Response {r.status_code}:\n{r.text}")
|
||||
r.raise_for_status()
|
||||
except Exception as e:
|
||||
print(f"Exception: {str(e)}")
|
||||
|
||||
|
||||
class DBus:
|
||||
def __init__(self) -> None:
|
||||
self.session_bus = dbus.SessionBus()
|
||||
self.session_bus = dbus.SessionBus()
|
||||
self.catbox = catbox.Catbox()
|
||||
self.uploader = ImageUpload()
|
||||
try:
|
||||
self.player_dbus_proxy = self.session_bus.get_object('org.mpris.MediaPlayer2.playerctld',
|
||||
'/org/mpris/MediaPlayer2')
|
||||
self.player_interface = dbus.Interface(self.player_dbus_proxy,
|
||||
'org.freedesktop.DBus.Properties')
|
||||
self.player_dbus_proxy = self.session_bus.get_object(
|
||||
"org.mpris.MediaPlayer2.playerctld", "/org/mpris/MediaPlayer2"
|
||||
)
|
||||
self.player_interface = dbus.Interface(
|
||||
self.player_dbus_proxy, "org.freedesktop.DBus.Properties"
|
||||
)
|
||||
self.interface_metadata = None
|
||||
self.plex_last = None
|
||||
self.plex_np = {
|
||||
'name': None,
|
||||
'details': None,
|
||||
'art': None,
|
||||
'elapsed': None,
|
||||
'duration': None,
|
||||
"name": None,
|
||||
"details": None,
|
||||
"art": None,
|
||||
"elapsed": None,
|
||||
"duration": None,
|
||||
}
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
def is_plexing(self) -> bool:
|
||||
try:
|
||||
self.interface_metadata = self.player_interface.GetAll('org.mpris.MediaPlayer2.Player').get('Metadata')
|
||||
if "app.plex.tv" in self.interface_metadata.get('xesam:url'):
|
||||
self.interface_metadata = self.player_interface.GetAll(
|
||||
"org.mpris.MediaPlayer2.Player"
|
||||
).get("Metadata")
|
||||
if "app.plex.tv" in self.interface_metadata.get("xesam:url"):
|
||||
return True
|
||||
print(f"WTF, NO PLEX?")
|
||||
except:
|
||||
print(traceback.format_exc())
|
||||
return False
|
||||
|
||||
|
||||
def get_now_playing(self):
|
||||
try:
|
||||
self.interface_details = self.player_interface.GetAll('org.mpris.MediaPlayer2.Player')
|
||||
self.interface_metadata = self.interface_details.get('Metadata')
|
||||
original_art = str(self.interface_metadata.get('mpris:artUrl', ''))
|
||||
self.interface_details = self.player_interface.GetAll(
|
||||
"org.mpris.MediaPlayer2.Player"
|
||||
)
|
||||
self.interface_metadata = self.interface_details.get("Metadata")
|
||||
original_art = str(self.interface_metadata.get("mpris:artUrl", ""))
|
||||
self.plex_np = {
|
||||
'name': str(" ".join(self.interface_metadata.get('xesam:artist'))),
|
||||
'details': f'{str(self.interface_metadata.get('xesam:album', ''))} - {str(self.interface_metadata.get('xesam:title'))}',
|
||||
"name": str(" ".join(self.interface_metadata.get("xesam:artist"))),
|
||||
"details": f"{str(self.interface_metadata.get('xesam:album', ''))} - {str(self.interface_metadata.get('xesam:title'))}",
|
||||
# 'elapsed': time.time() - int(self.interface_details.get('Position')/1000000),
|
||||
# 'duration': int(self.interface_metadata.get('mpris:length')/1000000),
|
||||
}
|
||||
if not self.interface_metadata.get('xesam:album'):
|
||||
self.plex_np['details'] = self.plex_np['details'][2:] # Fix for "'- ' prefix"
|
||||
if not self.plex_last or\
|
||||
(self.plex_np.get('name') == self.plex_last.get('name') and self.plex_np.get('details') == self.plex_last.get('details')):
|
||||
if not self.interface_metadata.get("xesam:album"):
|
||||
self.plex_np["details"] = self.plex_np["details"][
|
||||
2:
|
||||
] # Fix for "'- ' prefix"
|
||||
if not self.plex_last or (
|
||||
self.plex_np.get("name") == self.plex_last.get("name")
|
||||
and self.plex_np.get("details") == self.plex_last.get("details")
|
||||
):
|
||||
if original_art:
|
||||
original_art = original_art.split("file://", maxsplit=1)[1]
|
||||
bytes_art = None
|
||||
try:
|
||||
with open(original_art, 'rb') as f:
|
||||
with open(original_art, "rb") as f:
|
||||
bytes_art = f.read()
|
||||
if isinstance(bytes_art, bytes):
|
||||
# self.uploader.upload(bytes_art)
|
||||
self.plex_np['art'] = 'https://api.codey.lol/misc/get_activity_image'
|
||||
self.plex_np["art"] = (
|
||||
"https://api.codey.lol/misc/get_activity_image"
|
||||
)
|
||||
else:
|
||||
print(f"bytes art: {type(bytes_art)}")
|
||||
self.plex_np['art'] = None
|
||||
self.plex_np["art"] = None
|
||||
except:
|
||||
traceback.print_exc()
|
||||
self.plex_np['art'] = None
|
||||
self.plex_np["art"] = None
|
||||
else:
|
||||
# print("No original art for this file")
|
||||
self.plex_np['art'] = None
|
||||
self.plex_last = self.plex_np
|
||||
self.plex_np["art"] = None
|
||||
self.plex_last = self.plex_np
|
||||
return self.plex_np
|
||||
except:
|
||||
self.plex_np = {
|
||||
'name': None,
|
||||
'details': None,
|
||||
'art': None,
|
||||
'elapsed': None,
|
||||
'duration': None,
|
||||
"name": None,
|
||||
"details": None,
|
||||
"art": None,
|
||||
"elapsed": None,
|
||||
"duration": None,
|
||||
}
|
||||
print(traceback.format_exc())
|
||||
return False
|
||||
@@ -115,31 +127,35 @@ class DiscordPresence:
|
||||
self.client_id = discord_presence_priv.CLIENT_ID
|
||||
self.client_id_plex = discord_presence_priv.CLIENT_ID_PLEX
|
||||
self.api_key = discord_presence_priv.API_KEY
|
||||
self.api_url = 'https://api.codey.lol'
|
||||
self.album_art_url = 'https://api.codey.lol/radio/album_art'
|
||||
self.api_url = "https://api.codey.lol"
|
||||
self.album_art_url = "https://api.codey.lol/radio/album_art"
|
||||
self.api_req_data = {
|
||||
'bid': 0,
|
||||
'cmd': 'radio_metadata',
|
||||
'key': f'Bearer {self.api_key}',
|
||||
"bid": 0,
|
||||
"cmd": "radio_metadata",
|
||||
"key": f"Bearer {self.api_key}",
|
||||
}
|
||||
self.last_uuid: Optional[str] = None
|
||||
|
||||
def loop(self):
|
||||
try:
|
||||
lyr_start_idx = 0
|
||||
lyr_end_idx = 1
|
||||
lyr_end_idx = 1
|
||||
rand_msg = None
|
||||
lyrics_content = None
|
||||
lyrics_current_iteration = None
|
||||
chosen_image = random.choice([
|
||||
{
|
||||
'image': "qu",
|
||||
'label': "quietscheentchen",
|
||||
},
|
||||
# {
|
||||
# 'image': "rooster",
|
||||
# 'label': ":3",
|
||||
# }
|
||||
])
|
||||
RPC = Presence(self.client_id, pipe=0)
|
||||
chosen_image = random.choice(
|
||||
[
|
||||
{
|
||||
"image": "qu",
|
||||
"label": "quietscheentchen",
|
||||
},
|
||||
# {
|
||||
# 'image': "rooster",
|
||||
# 'label': ":3",
|
||||
# }
|
||||
]
|
||||
)
|
||||
RPC = Presence(self.client_id, pipe=0)
|
||||
RPC.connect()
|
||||
while True:
|
||||
try:
|
||||
@@ -151,11 +167,10 @@ class DiscordPresence:
|
||||
if plex_current == plex_new:
|
||||
continue
|
||||
|
||||
|
||||
if self.last_updated and (now - self.last_updated < 15):
|
||||
time.sleep(0.5)
|
||||
continue
|
||||
|
||||
|
||||
RPC.client_id = self.client_id
|
||||
# if RPC.client_id != self.client_id:
|
||||
# try:5
|
||||
@@ -163,43 +178,60 @@ class DiscordPresence:
|
||||
# except:
|
||||
# pass
|
||||
# RPC = Presence(self.client_id, pipe=0)
|
||||
# RPC.connect()
|
||||
print(RPC.update(
|
||||
details=plex_new.get('name', '\u2064') if plex_new.get('name') else plex_new.get('details'),
|
||||
state=plex_new.get('details', None) if plex_new.get('details') else None,
|
||||
large_image=plex_new.get('art', 'https://www.plex.tv/wp-content/themes/plex/assets/img/favicons/plex-192.png'),
|
||||
# large_image="https://codey.lol/images/cat.png",
|
||||
# large_text=f"{rand_msg}" if rand_msg else None,k
|
||||
large_text=lyrics_current_iteration if lyrics_current_iteration else None,
|
||||
# small_image=f"https://codey.lol/images/{chosen_image.get('image')}.png",
|
||||
# small_text=chosen_image.get('label'),
|
||||
activity_type=ActivityType.WATCHING,
|
||||
# start=plex_new.get('elapsed'),
|
||||
# end=plex_new.get('duration')/1000,
|
||||
))
|
||||
# RPC.connect()
|
||||
print(
|
||||
RPC.update(
|
||||
details=(
|
||||
plex_new.get("name", "\u2064")
|
||||
if plex_new.get("name")
|
||||
else plex_new.get("details")
|
||||
),
|
||||
state=(
|
||||
plex_new.get("details", None)
|
||||
if plex_new.get("details")
|
||||
else None
|
||||
),
|
||||
large_image=plex_new.get(
|
||||
"art",
|
||||
"https://www.plex.tv/wp-content/themes/plex/assets/img/favicons/plex-192.png",
|
||||
),
|
||||
# large_image="https://codey.lol/images/cat.png",
|
||||
# large_text=f"{rand_msg}" if rand_msg else None,k
|
||||
large_text=(
|
||||
lyrics_current_iteration
|
||||
if lyrics_current_iteration
|
||||
else None
|
||||
),
|
||||
# small_image=f"https://codey.lol/images/{chosen_image.get('image')}.png",
|
||||
# small_text=chosen_image.get('label'),
|
||||
activity_type=ActivityType.WATCHING,
|
||||
# start=plex_new.get('elapsed'),
|
||||
# end=plex_new.get('duration')/1000,
|
||||
)
|
||||
)
|
||||
time.sleep(1)
|
||||
continue
|
||||
|
||||
|
||||
|
||||
request = requests.post(f'{self.api_url}/radio/np',
|
||||
headers={
|
||||
'content-type': 'application/json; charset=utf-8'
|
||||
}, timeout=(2, 10))
|
||||
request = requests.post(
|
||||
f"{self.api_url}/radio/np",
|
||||
headers={"content-type": "application/json; charset=utf-8"},
|
||||
timeout=(2, 10),
|
||||
)
|
||||
request.raise_for_status()
|
||||
data = request.json()
|
||||
track = data.get('artistsong')
|
||||
track_artist = data.get('artist')[0:127]
|
||||
track_title = data.get('song')[0:127]
|
||||
track_album = data.get('album', '')[0:127]
|
||||
track_genre = data.get('genre', '')
|
||||
track_id = data.get('id')
|
||||
start_time = data.get('start')
|
||||
end_time = data.get('end')
|
||||
|
||||
if self.last_track == track:
|
||||
track = data.get("artistsong")
|
||||
track_artist = data.get("artist")[0:127]
|
||||
track_title = data.get("song")[0:127]
|
||||
track_album = data.get("album", "")[0:127]
|
||||
track_genre = data.get("genre", "")
|
||||
track_id = data.get("id")
|
||||
track_uuid = data.get("uuid")
|
||||
start_time = data.get("start")
|
||||
end_time = data.get("end")
|
||||
|
||||
if self.last_uuid == track_uuid:
|
||||
continue
|
||||
|
||||
|
||||
# if self.last_updated and (now - self.last_updated < 15):
|
||||
# time.sleep(0.7)
|
||||
# continue
|
||||
@@ -211,9 +243,6 @@ class DiscordPresence:
|
||||
# lyr_start_idx = 0
|
||||
# lyr_end_idx = 1
|
||||
|
||||
|
||||
|
||||
|
||||
# try:
|
||||
# # rand_msg_request = requests.post(f'{self.api_url}/randmsg/',
|
||||
# # headers={
|
||||
@@ -247,8 +276,6 @@ class DiscordPresence:
|
||||
# if len(lyrics_current_iteration) > 128:
|
||||
# lyrics_current_iteration = lyrics_current_iteration.split(" / ", maxsplit=1)[0].strip()
|
||||
|
||||
|
||||
|
||||
# except:
|
||||
# try:
|
||||
# lyrics_current_iteration = " / ".join(lyrics_content.split("<br>")[0]).strip()
|
||||
@@ -259,38 +286,39 @@ class DiscordPresence:
|
||||
|
||||
self.last_updated = now
|
||||
self.last_track = track
|
||||
lyrics_current_iteration = None # disable
|
||||
self.last_uuid = track_uuid
|
||||
lyrics_current_iteration = None # disable
|
||||
|
||||
print(
|
||||
RPC.update(
|
||||
details=track_title,
|
||||
state=f"{track_artist[0:100]} [{track_genre}]"[0:127],
|
||||
large_image=f"{self.album_art_url}?t={now}&track_id={track_id}",
|
||||
# large_image="https://codey.lol/images/cat.png",
|
||||
# large_text=f"{rand_msg}" if rand_msg else None,k
|
||||
large_text=track_album if track_album else None,
|
||||
small_image=f"https://codey.lol/images/{chosen_image.get('image')}.png",
|
||||
small_text=chosen_image.get("label"),
|
||||
start=start_time,
|
||||
end=end_time,
|
||||
activity_type=ActivityType.LISTENING,
|
||||
buttons=[
|
||||
{"label": "Listen", "url": "https://codey.lol/radio"},
|
||||
],
|
||||
)
|
||||
)
|
||||
|
||||
print(RPC.update(
|
||||
details=track_title,
|
||||
state=f"{track_artist[0:100]} [{track_genre}]"[0:127],
|
||||
large_image=f"{self.album_art_url}?t={now}&track_id={track_id}",
|
||||
# large_image="https://codey.lol/images/cat.png",
|
||||
# large_text=f"{rand_msg}" if rand_msg else None,k
|
||||
large_text=track_album if track_album else None,
|
||||
# small_image=f"https://codey.lol/images/{chosen_image.get('image')}.png",
|
||||
# small_text=chosen_image.get('label'),
|
||||
start=start_time,
|
||||
end=end_time,
|
||||
activity_type=ActivityType.LISTENING,
|
||||
buttons=[
|
||||
{
|
||||
"label": "Listen",
|
||||
"url": "https://codey.lol/radio"
|
||||
},]
|
||||
))
|
||||
|
||||
# match chosen_image.get('image'):
|
||||
# case "rooster":
|
||||
# chosen_image = {
|
||||
# 'image': "qu",
|
||||
# 'label': "rubber duck :3"
|
||||
# }
|
||||
# case "qu":
|
||||
# chosen_image = {
|
||||
# 'image': "rooster",
|
||||
# 'label': ":3",
|
||||
# }
|
||||
# case "rooster":
|
||||
# chosen_image = {
|
||||
# 'image': "qu",
|
||||
# 'label': "rubber duck :3"
|
||||
# }
|
||||
# case "qu":
|
||||
# chosen_image = {
|
||||
# 'image': "rooster",
|
||||
# 'label': ":3",
|
||||
# }
|
||||
except:
|
||||
print(traceback.format_exc())
|
||||
time.sleep(0.7)
|
||||
@@ -303,11 +331,10 @@ class DiscordPresence:
|
||||
pass
|
||||
|
||||
|
||||
|
||||
def __init__():
|
||||
discord_presence = DiscordPresence()
|
||||
discord_presence.loop()
|
||||
discord_presence.loop()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
__init__()
|
||||
|
||||
Reference in New Issue
Block a user