Mirrors/discord-rich-presence-plex
2
0
Fork 0
mirror of https://github.com/phin05/discord-rich-presence-plex synced 2024-11-22 01:23:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

Update discordRichPresencePlex.py

Browse source
This commit is contained in:
Phin 2018-04-27 11:33:25 +05:30
parent 473ffcc799
commit 642dd3f462
1 changed files with 76 additions and 61 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

137
discordRichPresencePlex.py
View file

@ -35,7 +35,6 @@ class discordRichPresence:
self.pipeWriter = None self.pipeWriter = None
self.process = None self.process = None
self.running = False self.running = False
self.resetNext = False
self.child = child self.child = child
async def read(self): async def read(self):
@ -43,8 +42,8 @@ class discordRichPresence:
data = await self.pipeReader.read(1024) data = await self.pipeReader.read(1024)
self.child.log("[READ] " + str(json.loads(data[8:].decode("utf-8")))) self.child.log("[READ] " + str(json.loads(data[8:].decode("utf-8"))))
except Exception as e: except Exception as e:
self.child.log("Error: " + str(e)) self.child.log("[READ] " + str(e))
self.resetNext = True self.stop()
def write(self, op, payload): def write(self, op, payload):
payload = json.dumps(payload) payload = json.dumps(payload)
@ -52,14 +51,17 @@ class discordRichPresence:
data = self.pipeWriter.write(struct.pack("<ii", op, len(payload)) + payload.encode("utf-8")) data = self.pipeWriter.write(struct.pack("<ii", op, len(payload)) + payload.encode("utf-8"))
async def handshake(self): async def handshake(self):
if (isLinux): try:
self.pipeReader, self.pipeWriter = await asyncio.open_unix_connection(self.IPCPipe, loop = self.loop) if (isLinux):
else: self.pipeReader, self.pipeWriter = await asyncio.open_unix_connection(self.IPCPipe, loop = self.loop)
self.pipeReader = asyncio.StreamReader(loop = self.loop) else:
self.pipeWriter, _ = await self.loop.create_pipe_connection(lambda: asyncio.StreamReaderProtocol(self.pipeReader, loop = self.loop), self.IPCPipe) self.pipeReader = asyncio.StreamReader(loop = self.loop)
self.write(0, {"v": 1, "client_id": self.clientID}) self.pipeWriter, _ = await self.loop.create_pipe_connection(lambda: asyncio.StreamReaderProtocol(self.pipeReader, loop = self.loop), self.IPCPipe)
await self.read() self.write(0, {"v": 1, "client_id": self.clientID})
self.running = True await self.read()
self.running = True
except Exception as e:
self.child.log("[HANDSHAKE] " + str(e))
def start(self): def start(self):
self.child.log("Opening Discord IPC Pipe") self.child.log("Opening Discord IPC Pipe")
@ -73,9 +75,13 @@ class discordRichPresence:
def stop(self): def stop(self):
self.child.log("Closing Discord IPC Pipe") self.child.log("Closing Discord IPC Pipe")
self.child.lastState, self.child.lastSessionKey, self.child.lastRatingKey = None, None, None
self.process.kill() self.process.kill()
self.pipeWriter.close() try:
self.loop.close() self.pipeWriter.close()
self.loop.close()
except:
pass
self.running = False self.running = False
def send(self, activity): def send(self, activity):
@ -95,6 +101,7 @@ class discordRichPresencePlex(discordRichPresence):
productName = "Plex Media Server" productName = "Plex Media Server"
plexAccount = None plexAccount = None
plexServer = None plexServer = None
plexAlertListener = None
lastState = None lastState = None
lastSessionKey = None lastSessionKey = None
lastRatingKey = None lastRatingKey = None
@ -111,32 +118,56 @@ class discordRichPresencePlex(discordRichPresence):
super().__init__("413407336082833418", self) super().__init__("413407336082833418", self)
def run(self): def run(self):
try: self.reset()
if (self.plexConfig.token): connected = False
self.plexAccount = plexapi.myplex.MyPlexAccount(self.plexConfig.username, token = self.plexConfig.token) while (not connected):
else: try:
self.plexAccount = plexapi.myplex.MyPlexAccount(self.plexConfig.username, self.plexConfig.password) if (self.plexConfig.token):
self.log("Logged in as Plex User \"" + self.plexAccount.username + "\"") self.plexAccount = plexapi.myplex.MyPlexAccount(self.plexConfig.username, token = self.plexConfig.token)
self.plexServer = None else:
for resource in self.plexAccount.resources(): self.plexAccount = plexapi.myplex.MyPlexAccount(self.plexConfig.username, self.plexConfig.password)
if (resource.product == self.productName and resource.name == self.plexConfig.serverName): self.log("Logged in as Plex User \"" + self.plexAccount.username + "\"")
self.plexServer = resource.connect() self.plexServer = None
self.plexServer.startAlertListener(self.onPlexServerAlert) for resource in self.plexAccount.resources():
self.log("Connected to " + self.productName + " \"" + self.plexConfig.serverName + "\"") if (resource.product == self.productName and resource.name == self.plexConfig.serverName):
self.log("Listening for PlaySessionStateNotification alerts from user \"" + self.plexConfig.listenForUser + "\"") self.plexServer = resource.connect()
if (self.checkConnectionTimer): self.log("Connected to " + self.productName + " \"" + self.plexConfig.serverName + "\"")
self.checkConnectionTimer.cancel() self.plexAlertListener = self.plexServer.startAlertListener(self.onPlexServerAlert)
self.checkConnectionTimer = None self.log("Listening for PlaySessionStateNotification alerts from user \"" + self.plexConfig.listenForUser + "\"")
self.checkConnectionTimer = threading.Timer(self.checkConnectionTimerInterval, self.checkConnection) if (self.checkConnectionTimer):
self.checkConnectionTimer.start() self.checkConnectionTimer.cancel()
self.checkConnectionTimer = None
self.checkConnectionTimer = threading.Timer(self.checkConnectionTimerInterval, self.checkConnection)
self.checkConnectionTimer.start()
connected = True
break
if (not self.plexServer):
self.log(self.productName + " \"" + self.plexConfig.serverName + "\" not found")
break break
if (not self.plexServer): except Exception as e:
self.log(self.productName + " \"" + self.plexConfig.serverName + "\" not found") self.log("Failed to connect to Plex: " + str(e))
except Exception as e: self.log("Reconnecting in 10 seconds")
self.log("Failed to connect to Plex: " + str(e)) time.sleep(10)
self.log("Reconnecting in 5 seconds")
time.sleep(5) def reset(self):
self.run() if (self.running):
self.stop()
self.plexAccount, self.plexServer = None, None
if (self.plexAlertListener):
try:
self.plexAlertListener.stop()
except:
pass
self.plexAlertListener = None
if (self.stopTimer):
self.stopTimer.cancel()
self.stopTimer = None
if (self.stopTimer2):
self.stopTimer2.cancel()
self.stopTimer2 = None
if (self.checkConnectionTimer):
self.checkConnectionTimer.cancel()
self.checkConnectionTimer = None
def checkConnection(self): def checkConnection(self):
try: try:
@ -144,16 +175,6 @@ class discordRichPresencePlex(discordRichPresence):
self.checkConnectionTimer = threading.Timer(self.checkConnectionTimerInterval, self.checkConnection) self.checkConnectionTimer = threading.Timer(self.checkConnectionTimerInterval, self.checkConnection)
self.checkConnectionTimer.start() self.checkConnectionTimer.start()
except Exception as e: except Exception as e:
self.plexAccount = None
self.plexServer = None
if (self.stopTimer):
self.stopTimer.cancel()
self.stopTimer = None
if (self.stopTimer2):
self.stopTimer2.cancel()
self.stopTimer2 = None
if (self.running):
self.stop()
self.log("Connection to Plex lost: " + str(e)) self.log("Connection to Plex lost: " + str(e))
self.log("Reconnecting") self.log("Reconnecting")
self.run() self.run()
@ -170,9 +191,9 @@ class discordRichPresencePlex(discordRichPresence):
lock.release() lock.release()
def onPlexServerAlert(self, data): def onPlexServerAlert(self, data):
if (not self.plexServer):
return
try: try:
if (not self.plexServer):
return
if (data["type"] == "playing" and "PlaySessionStateNotification" in data): if (data["type"] == "playing" and "PlaySessionStateNotification" in data):
sessionData = data["PlaySessionStateNotification"][0] sessionData = data["PlaySessionStateNotification"][0]
state = sessionData["state"] state = sessionData["state"]
@ -263,20 +284,17 @@ class discordRichPresencePlex(discordRichPresence):
if (state == "playing"): if (state == "playing"):
currentTimestamp = int(time.time()) currentTimestamp = int(time.time())
activity["timestamps"] = {"start": currentTimestamp - (viewOffset / 1000)} # "end": currentTimestamp + ((metadata.duration - viewOffset) / 1000) activity["timestamps"] = {"start": currentTimestamp - (viewOffset / 1000)} # "end": currentTimestamp + ((metadata.duration - viewOffset) / 1000)
if (self.resetNext):
self.resetNext = False
self.stop()
if (not self.running): if (not self.running):
self.start() self.start()
self.send(activity) if (self.running):
self.send(activity)
else:
self.stop()
except Exception as e: except Exception as e:
self.log("onPlexServerAlert Error: " + str(e)) self.log("onPlexServerAlert Error: " + str(e))
if (self.process):
self.process.kill()
def stopOnNoUpdate(self): def stopOnNoUpdate(self):
self.log("No updates from session key " + str(self.lastSessionKey) + ", stopping", "red", True) self.log("No updates from session key " + str(self.lastSessionKey) + ", stopping", "red", True)
self.lastState, self.lastSessionKey, self.lastRatingKey = None, None, None
self.stop() self.stop()
isLinux = sys.platform in ["linux", "darwin"] isLinux = sys.platform in ["linux", "darwin"]
@ -316,9 +334,6 @@ try:
time.sleep(3600) time.sleep(3600)
except KeyboardInterrupt: except KeyboardInterrupt:
for discordRichPresencePlexInstance in discordRichPresencePlexInstances: for discordRichPresencePlexInstance in discordRichPresencePlexInstances:
if (discordRichPresencePlexInstance.running): discordRichPresencePlexInstance.reset()
discordRichPresencePlexInstance.stop()
if (discordRichPresencePlexInstance.checkConnectionTimer):
discordRichPresencePlexInstance.checkConnectionTimer.cancel()
except Exception as e: except Exception as e:
print("Error: " + str(e)) print("Error: " + str(e))