From 38d29d89c7df2a5a888e874917501783dd5e2a98 Mon Sep 17 00:00:00 2001 From: blacktwin Date: Wed, 26 Oct 2016 22:11:39 -0400 Subject: [PATCH] Update notify_newip.py Spaces and clean up. --- notify_newip.py | 794 ++++++++++++++++++++++++++++++++++++++---------- 1 file changed, 632 insertions(+), 162 deletions(-) diff --git a/notify_newip.py b/notify_newip.py index 98d4489..65535dc 100644 --- a/notify_newip.py +++ b/notify_newip.py @@ -28,193 +28,663 @@ BODY_TEXT = """\ -

Hi!
-
Poster unavailable {p.user} has watched {p.media_type}:{p.title} from a new IP address: {p.ip_address}
-
On {p.platform}[{p.player}] in {g.city}, {g.country} {g.postal_code} at {p.timestamp} on {p.datestamp}
-

-
User email is: {u.email}
-

+

Hi!
+
Poster unavailable {p.user} has watched {p.media_type}:{p.title} from a new IP address: {p.ip_address}
+
On {p.platform}[{p.player}] in {g.city}, {g.country} {g.postal_code} at {p.timestamp} on {p.datestamp}
+

+
User email is: {u.email}
+

""" - + ##Geo Space## class GeoData(object): - def __init__(self, data=None): - data = data or {} - self.country = data.get('country', 'N/A') - self.city = data.get('city', 'N/A') - self.postal_code = data.get('postal_code', 'N/A') + def __init__(self, data=None): + data = data or {} + self.country = data.get('country', 'N/A') + self.city = data.get('city', 'N/A') + self.postal_code = data.get('postal_code', 'N/A') ##USER Space## class UserEmail(object): - def __init__(self, data=None): - data = data or {} - self.email = data.get('email', 'N/A') - self.user_id = data.get('user_id', 'N/A') - self.user_thumb = data.get('user_thumb', 'N/A') - + def __init__(self, data=None): + data = data or {} + self.email = data.get('email', 'N/A') + self.user_id = data.get('user_id', 'N/A') + self.user_thumb = data.get('user_thumb', 'N/A') + ##IP Space## class UserIPs(object): - def __init__(self, data=None): - data = data or {} - + def __init__(self, data=None): + data = data or {} + ##API Space## def get_user_ip_addresses(user_id='', ip_address=''): - # Get the user IP list from PlexPy - payload = {'apikey': PLEXPY_APIKEY, - 'cmd': 'get_user_ips', - 'user_id': user_id, - 'search': ip_address} - - try: - r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) - response = r.json() + # Get the user IP list from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_user_ips', + 'user_id': user_id, + 'search': ip_address} + + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() - if response['response']['result'] == 'success': - data = response['response']['data'] - if data.get('error'): - raise Exception(data['error']) - else: - sys.stdout.write("Successfully retrieved UserIPs data.") - if response['response']['data']['recordsFiltered'] == 0: - sys.stdout.write("IP has no history.") - return UserIPs(data=data) - else: - sys.stdout.write("IP has history, killing script.") - exit() - else: - raise Exception(response['response']['message']) - except Exception as e: - sys.stderr.write("PlexPy API 'get_user_ip_addresses' request failed: {0}.".format(e)) - return UserIPs() + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved UserIPs data.") + if response['response']['data']['recordsFiltered'] == 0: + sys.stdout.write("IP has no history.") + return UserIPs(data=data) + else: + sys.stdout.write("IP has history, killing script.") + exit() + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_user_ip_addresses' request failed: {0}.".format(e)) + return UserIPs() def get_geoip_info(ip_address=''): - # Get the geo IP lookup from PlexPy - payload = {'apikey': PLEXPY_APIKEY, - 'cmd': 'get_geoip_lookup', - 'ip_address': ip_address} + # Get the geo IP lookup from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_geoip_lookup', + 'ip_address': ip_address} - try: - r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) - response = r.json() + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() - if response['response']['result'] == 'success': - data = response['response']['data'] - if data.get('error'): - raise Exception(data['error']) - else: - sys.stdout.write("Successfully retrieved geolocation data.") - return GeoData(data=data) - else: - raise Exception(response['response']['message']) - except Exception as e: - sys.stderr.write("PlexPy API 'get_geoip_lookup' request failed: {0}.".format(e)) - return GeoData() + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved geolocation data.") + return GeoData(data=data) + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_geoip_lookup' request failed: {0}.".format(e)) + return GeoData() def get_user_email(user_id=''): - # Get the user email from PlexPy - payload = {'apikey': PLEXPY_APIKEY, - 'cmd': 'get_user', - 'user_id': user_id} + # Get the user email from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_user', + 'user_id': user_id} - try: - r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) - response = r.json() + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() - if response['response']['result'] == 'success': - data = response['response']['data'] - if data.get('error'): - raise Exception(data['error']) - else: - sys.stdout.write("Successfully retrieved user email data.") - return UserEmail(data=data) - else: - raise Exception(response['response']['message']) - except Exception as e: - sys.stderr.write("PlexPy API 'get_user' request failed: {0}.".format(e)) - return UserEmail() + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved user email data.") + return UserEmail(data=data) + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_user' request failed: {0}.".format(e)) + return UserEmail() -def send_notification(arguments=None, geodata=None, useremail=None, user_address=None): - # Format notification text - try: - subject = SUBJECT_TEXT.format(p=arguments, g=geodata, u=useremail, uip=user_address) - body = BODY_TEXT.format(p=arguments, g=geodata, u=useremail, uip=user_address) - except LookupError as e: - sys.stderr.write("Unable to substitute '{0}' in the notification subject or body".format(e)) - return None - # Send the notification through PlexPy - payload = {'apikey': PLEXPY_APIKEY, - 'cmd': 'notify', - 'agent_id': AGENT_ID, - 'subject': subject, - 'body': body} +def send_notification(arguments=None, geodata=None, useremail=None): + # Format notification text + try: + subject = SUBJECT_TEXT.format(p=arguments, g=geodata, u=useremail) + body = BODY_TEXT.format(p=arguments, g=geodata, u=useremail) + except LookupError as e: + sys.stderr.write("Unable to substitute '{0}' in the notification subject or body".format(e)) + return None + # Send the notification through PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'notify', + 'agent_id': AGENT_ID, + 'subject': subject, + 'body': body} - try: - r = requests.post(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) - response = r.json() - - if response['response']['result'] == 'success': - sys.stdout.write("Successfully sent PlexPy notification.") - else: - raise Exception(response['response']['message']) - except Exception as e: - sys.stderr.write("PlexPy API 'notify' request failed: {0}.".format(e)) - return None + try: + r = requests.post(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + if response['response']['result'] == 'success': + sys.stdout.write("Successfully sent PlexPy notification.") + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'notify' request failed: {0}.".format(e)) + return None if __name__ == '__main__': - # Parse arguments from PlexPy - parser = argparse.ArgumentParser() + # Parse arguments from PlexPy + parser = argparse.ArgumentParser() - parser.add_argument('-ip', '--ip_address', action='store', default='', - help='The IP address of the stream') - parser.add_argument('-us', '--user', action='store', default='', - help='Username of the person watching the stream') - parser.add_argument('-uid', '--user_id', action='store', default='', - help='User_ID of the person watching the stream') - parser.add_argument('-med', '--media_type', action='store', default='', - help='The media type of the stream') - parser.add_argument('-tt', '--title', action='store', default='', - help='The title of the media') - parser.add_argument('-pf', '--platform', action='store', default='', - help='The platform of the stream') - parser.add_argument('-pl', '--player', action='store', default='', - help='The player of the stream') - parser.add_argument('-da', '--datestamp', action='store', default='', - help='The date of the stream') - parser.add_argument('-ti', '--timestamp', action='store', default='', - help='The time of the stream') - parser.add_argument('-sn', '--show_name', action='store', default='', - help='The name of the TV show') - parser.add_argument('-ena', '--episode_name', action='store', default='', - help='The name of the episode') - parser.add_argument('-ssn', '--season_num', action='store', default='', - help='The season number of the TV show') - parser.add_argument('-enu', '--episode_num', action='store', default='', - help='The episode number of the TV show') - parser.add_argument('-srv', '--plex_server', action='store', default='', - help='The name of the Plex server') - parser.add_argument('-pos', '--poster', action='store', default='', - help='The poster url') - parser.add_argument('-sum', '--summary', action='store', default='', - help='The summary of the TV show') - parser.add_argument('-lbn', '--library_name', action='store', default='', - help='The name of the TV show') - - p = parser.parse_args() + parser.add_argument('-ip', '--ip_address', action='store', default='', + help='The IP address of the stream') + parser.add_argument('-us', '--user', action='store', default='', + help='Username of the person watching the stream') + parser.add_argument('-uid', '--user_id', action='store', default='', + help='User_ID of the person watching the stream') + parser.add_argument('-med', '--media_type', action='store', default='', + help='The media type of the stream') + parser.add_argument('-tt', '--title', action='store', default='', + help='The title of the media') + parser.add_argument('-pf', '--platform', action='store', default='', + help='The platform of the stream') + parser.add_argument('-pl', '--player', action='store', default='', + help='The player of the stream') + parser.add_argument('-da', '--datestamp', action='store', default='', + help='The date of the stream') + parser.add_argument('-ti', '--timestamp', action='store', default='', + help='The time of the stream') + parser.add_argument('-sn', '--show_name', action='store', default='', + help='The name of the TV show') + parser.add_argument('-ena', '--episode_name', action='store', default='', + help='The name of the episode') + parser.add_argument('-ssn', '--season_num', action='store', default='', + help='The season number of the TV show') + parser.add_argument('-enu', '--episode_num', action='store', default='', + help='The episode number of the TV show') + parser.add_argument('-srv', '--plex_server', action='store', default='', + help='The name of the Plex server') + parser.add_argument('-pos', '--poster', action='store', default='', + help='The poster url') + parser.add_argument('-sum', '--summary', action='store', default='', + help='The summary of the TV show') + parser.add_argument('-lbn', '--library_name', action='store', default='', + help='The name of the TV show') + + p = parser.parse_args() - # Check to make sure there is an IP address before proceeding - if p.ip_address: - if p.ip_address.startswith(LAN_SUBNET) and REPLACEMENT_WAN_IP: - ip_address = REPLACEMENT_WAN_IP - else: - ip_address = p.ip_address - - g = get_geoip_info(ip_address=ip_address) - u = get_user_email(user_id=p.user_id) - uip = get_user_ip_addresses(user_id=p.user_id, ip_address=p.ip_address) - send_notification(arguments=p, geodata=g, useremail=u, user_address=uip) - - else: - sys.stdout.write("No IP address passed from PlexPy.") + # Check to make sure there is an IP address before proceeding + if p.ip_address: + if p.ip_address.startswith(LAN_SUBNET) and REPLACEMENT_WAN_IP: + ip_address = REPLACEMENT_WAN_IP + else: + ip_address = p.ip_address + + g = get_geoip_info(ip_address=ip_address) + u = get_user_email(user_id=p.user_id) + get_user_ip_addresses(user_id=p.user_id, ip_address=p.ip_address) + send_notification(arguments=p, geodata=g, useremail=u) + + else: + sys.stdout.write("No IP address passed from PlexPy.") + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +""" +Pulling together User IP information and Email. + +""" +import argparse +import requests +import sys + +## -sn {show_name} -ena {episode_name} -ssn {season_num00} -enu {episode_num00} -srv {server_name} -med {media_type} -pos {poster_url} -tt {title} -sum {summary} -lbn {library_name} -ip {ip_address} -us {user} -uid {user_id} -pf {platform} -pl {player} -da {datestamp} -ti {timestamp} + +## EDIT THESE SETTINGS ## +PLEXPY_APIKEY = 'XXXX' # Your PlexPy API key +PLEXPY_URL = 'http://localhost:8181/' # Your PlexPy URL +AGENT_ID = 10 # The notification agent ID for PlexPy + +# Replace LAN IP addresses that start with the LAN_SUBNET with a WAN IP address +# to retrieve geolocation data. Leave REPLACEMENT_WAN_IP blank for no replacement. +LAN_SUBNET = '192.168.0' +REPLACEMENT_WAN_IP = '' + +# The notification subject and body +# - Use "{p.argument}" for script arguments +# - Use "{g.value}" for geolocation data +# - Use "{u.value}" for user data +SUBJECT_TEXT = "New IP has been detected using Plex." +BODY_TEXT = """\ + + + +

Hi!
+
Poster unavailable {p.user} has watched {p.media_type}:{p.title} from a new IP address: {p.ip_address}
+
On {p.platform}[{p.player}] in {g.city}, {g.country} {g.postal_code} at {p.timestamp} on {p.datestamp}
+

+
User email is: {u.email}
+

+ + +""" + +##Geo Space## +class GeoData(object): + def __init__(self, data=None): + data = data or {} + self.country = data.get('country', 'N/A') + self.city = data.get('city', 'N/A') + self.postal_code = data.get('postal_code', 'N/A') + +##USER Space## +class UserEmail(object): + def __init__(self, data=None): + data = data or {} + self.email = data.get('email', 'N/A') + self.user_id = data.get('user_id', 'N/A') + self.user_thumb = data.get('user_thumb', 'N/A') + +##IP Space## +class UserIPs(object): + def __init__(self, data=None): + data = data or {} + +##API Space## +def get_user_ip_addresses(user_id='', ip_address=''): + # Get the user IP list from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_user_ips', + 'user_id': user_id, + 'search': ip_address} + + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved UserIPs data.") + if response['response']['data']['recordsFiltered'] == 0: + sys.stdout.write("IP has no history.") + return UserIPs(data=data) + else: + sys.stdout.write("IP has history, killing script.") + exit() + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_user_ip_addresses' request failed: {0}.".format(e)) + return UserIPs() + +def get_geoip_info(ip_address=''): + # Get the geo IP lookup from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_geoip_lookup', + 'ip_address': ip_address} + + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved geolocation data.") + return GeoData(data=data) + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_geoip_lookup' request failed: {0}.".format(e)) + return GeoData() + + +def get_user_email(user_id=''): + # Get the user email from PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'get_user', + 'user_id': user_id} + + try: + r = requests.get(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + if response['response']['result'] == 'success': + data = response['response']['data'] + if data.get('error'): + raise Exception(data['error']) + else: + sys.stdout.write("Successfully retrieved user email data.") + return UserEmail(data=data) + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'get_user' request failed: {0}.".format(e)) + return UserEmail() + +def send_notification(arguments=None, geodata=None, useremail=None, user_address=None): + # Format notification text + try: + subject = SUBJECT_TEXT.format(p=arguments, g=geodata, u=useremail, uip=user_address) + body = BODY_TEXT.format(p=arguments, g=geodata, u=useremail, uip=user_address) + except LookupError as e: + sys.stderr.write("Unable to substitute '{0}' in the notification subject or body".format(e)) + return None + # Send the notification through PlexPy + payload = {'apikey': PLEXPY_APIKEY, + 'cmd': 'notify', + 'agent_id': AGENT_ID, + 'subject': subject, + 'body': body} + + try: + r = requests.post(PLEXPY_URL.rstrip('/') + '/api/v2', params=payload) + response = r.json() + + if response['response']['result'] == 'success': + sys.stdout.write("Successfully sent PlexPy notification.") + else: + raise Exception(response['response']['message']) + except Exception as e: + sys.stderr.write("PlexPy API 'notify' request failed: {0}.".format(e)) + return None + +if __name__ == '__main__': + # Parse arguments from PlexPy + parser = argparse.ArgumentParser() + + parser.add_argument('-ip', '--ip_address', action='store', default='', + help='The IP address of the stream') + parser.add_argument('-us', '--user', action='store', default='', + help='Username of the person watching the stream') + parser.add_argument('-uid', '--user_id', action='store', default='', + help='User_ID of the person watching the stream') + parser.add_argument('-med', '--media_type', action='store', default='', + help='The media type of the stream') + parser.add_argument('-tt', '--title', action='store', default='', + help='The title of the media') + parser.add_argument('-pf', '--platform', action='store', default='', + help='The platform of the stream') + parser.add_argument('-pl', '--player', action='store', default='', + help='The player of the stream') + parser.add_argument('-da', '--datestamp', action='store', default='', + help='The date of the stream') + parser.add_argument('-ti', '--timestamp', action='store', default='', + help='The time of the stream') + parser.add_argument('-sn', '--show_name', action='store', default='', + help='The name of the TV show') + parser.add_argument('-ena', '--episode_name', action='store', default='', + help='The name of the episode') + parser.add_argument('-ssn', '--season_num', action='store', default='', + help='The season number of the TV show') + parser.add_argument('-enu', '--episode_num', action='store', default='', + help='The episode number of the TV show') + parser.add_argument('-srv', '--plex_server', action='store', default='', + help='The name of the Plex server') + parser.add_argument('-pos', '--poster', action='store', default='', + help='The poster url') + parser.add_argument('-sum', '--summary', action='store', default='', + help='The summary of the TV show') + parser.add_argument('-lbn', '--library_name', action='store', default='', + help='The name of the TV show') + + p = parser.parse_args() + + # Check to make sure there is an IP address before proceeding + if p.ip_address: + if p.ip_address.startswith(LAN_SUBNET) and REPLACEMENT_WAN_IP: + ip_address = REPLACEMENT_WAN_IP + else: + ip_address = p.ip_address + + g = get_geoip_info(ip_address=ip_address) + u = get_user_email(user_id=p.user_id) + uip = get_user_ip_addresses(user_id=p.user_id, ip_address=p.ip_address) + send_notification(arguments=p, geodata=g, useremail=u, user_address=uip) + + else: + sys.stdout.write("No IP address passed from PlexPy.")