Mirrored_Repos/JBOPS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Decode script arguments

Browse Source
This commit is contained in:
JonnyWong16 2018-06-29 13:14:13 -07:00
parent ce66d95580
commit 1e8e96398b
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
killstream/kill_stream.py
View File

@ -52,6 +52,7 @@ TAUTULLI_URL = ''
TAUTULLI_APIKEY = '' TAUTULLI_APIKEY = ''
TAUTULLI_URL = os.getenv('TAUTULLI_URL', TAUTULLI_URL) TAUTULLI_URL = os.getenv('TAUTULLI_URL', TAUTULLI_URL)
TAUTULLI_APIKEY = os.getenv('TAUTULLI_APIKEY', TAUTULLI_APIKEY) TAUTULLI_APIKEY = os.getenv('TAUTULLI_APIKEY', TAUTULLI_APIKEY)
TAUTULLI_ENCODING = os.getenv('TAUTULLI_ENCODING', 'UTF-8')
SUBJECT_TEXT = "Tautulli has killed a stream." SUBJECT_TEXT = "Tautulli has killed a stream."
BODY_TEXT = "Killed session ID '{id}'. Reason: {message}" BODY_TEXT = "Killed session ID '{id}'. Reason: {message}"
@ -241,6 +242,10 @@ def terminate_long_pause(session_id, message, limit, interval, notify=None):
return return
def arg_decoding(arg):
return arg.decode(TAUTULLI_ENCODING)
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Killing Plex streams from Tautulli.") description="Killing Plex streams from Tautulli.")
@ -248,7 +253,7 @@ if __name__ == "__main__":
help='Kill selector.\nChoices: (%(choices)s)') help='Kill selector.\nChoices: (%(choices)s)')
parser.add_argument('--userId', type=int, parser.add_argument('--userId', type=int,
help='The unique identifier for the user.') help='The unique identifier for the user.')
parser.add_argument('--username', parser.add_argument('--username', type=arg_decoding,
help='The username of the person streaming.') help='The username of the person streaming.')
parser.add_argument('--sessionId', required=True, parser.add_argument('--sessionId', required=True,
help='The unique identifier for the stream.') help='The unique identifier for the stream.')
@ -259,7 +264,7 @@ if __name__ == "__main__":
help='The time session is allowed to remain paused.') help='The time session is allowed to remain paused.')
parser.add_argument('--interval', type=int, default=30, parser.add_argument('--interval', type=int, default=30,
help='The seconds between paused session checks.') help='The seconds between paused session checks.')
parser.add_argument('--killMessage', nargs='+', parser.add_argument('--killMessage', nargs='+', type=arg_decoding,
help='Message to send to user whose stream is killed.') help='Message to send to user whose stream is killed.')
opts = parser.parse_args() opts = parser.parse_args()