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

cleaning special characters from region, city, and friendly names.

Browse Source 
removed unused get_platform function
This commit is contained in:
Blacktwin 2018-05-24 13:29:38 -04:00
parent e660b8e69a
commit 6210e361f3
1 changed files with 10 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
maps/ips_to_maps.py
View File

@ -33,6 +33,7 @@ import argparse
import numpy as np import numpy as np
import time import time
import webbrowser import webbrowser
import re
## EDIT THESE SETTINGS ## ## EDIT THESE SETTINGS ##
TAUTULLI_APIKEY = '' # Your Tautulli API key TAUTULLI_APIKEY = '' # Your Tautulli API key
@ -71,13 +72,18 @@ PLATFORM_COLORS = {'Android': '#a4c639', # Green
title_string = "Location of Plex users based on ISP data" title_string = "Location of Plex users based on ISP data"
def clean_up_text(title):
cleaned = re.sub('\W+', ' ', title)
return cleaned
class GeoData(object): class GeoData(object):
def __init__(self, data=None): def __init__(self, data=None):
data = data or {} data = data or {}
self.continent = data.get('continent', 'N/A') self.continent = data.get('continent', 'N/A')
self.country = data.get('country', 'N/A') self.country = data.get('country', 'N/A')
self.region = data.get('region', 'N/A') self.region = clean_up_text(data.get('region', 'N/A'))
self.city = data.get('city', 'N/A') self.city = clean_up_text(data.get('city', 'N/A'))
self.postal_code = data.get('postal_code', 'N/A') self.postal_code = data.get('postal_code', 'N/A')
self.timezone = data.get('timezone', 'N/A') self.timezone = data.get('timezone', 'N/A')
self.latitude = data.get('latitude', 'N/A') self.latitude = data.get('latitude', 'N/A')
@ -89,7 +95,7 @@ class UserIPs(object):
def __init__(self, data=None): def __init__(self, data=None):
d = data or {} d = data or {}
self.ip_address = d['ip_address'] self.ip_address = d['ip_address']
self.friendly_name = d['friendly_name'] self.friendly_name = clean_up_text(d['friendly_name'])
self.play_count = d['play_count'] self.play_count = d['play_count']
self.platform = d['platform'] self.platform = d['platform']
@ -162,20 +168,6 @@ def get_geoip_info(ip_address=''):
pass pass
def get_stream_type_by_top_10_platforms():
# Get the user IP list from Tautulli
payload = {'apikey': TAUTULLI_APIKEY,
'cmd': 'get_stream_type_by_top_10_platforms'} # length is number of returns, default is 25
try:
r = requests.get(TAUTULLI_URL.rstrip('/') + '/api/v2', params=payload)
response = r.json()
res_data = response['response']['data']['categories']
return res_data
except Exception as e:
sys.stderr.write("Tautulli API 'get_stream_type_by_top_10_platforms' request failed: {0}.".format(e))
def add_to_dictlist(d, key, val): def add_to_dictlist(d, key, val):
if key not in d: if key not in d:
d[key] = [val] d[key] = [val]
@ -377,6 +369,7 @@ def draw_map(map_type, geo_dict, filename, headless, leg_choice):
mng.window.state('zoomed') mng.window.state('zoomed')
plt.show() plt.show()
if __name__ == '__main__': if __name__ == '__main__':
timestr = time.strftime("%Y%m%d-%H%M%S") timestr = time.strftime("%Y%m%d-%H%M%S")
@ -396,9 +389,6 @@ if __name__ == '__main__':
parser.add_argument('-i', '--ignore', nargs='+', type=str, default=None, choices=user_lst, metavar='', parser.add_argument('-i', '--ignore', nargs='+', type=str, default=None, choices=user_lst, metavar='',
help='Space separated list of case sensitive names to process. Allowed names are: \n' help='Space separated list of case sensitive names to process. Allowed names are: \n'
'%(choices)s \n(default: %(default)s)') '%(choices)s \n(default: %(default)s)')
parser.add_argument('-p', '--platform', nargs='+', type=str, default='all',
choices=get_stream_type_by_top_10_platforms(), metavar='',
help='Search by platform. Allowed platforms are:\n%(choices)s \n(default: %(default)s)')
parser.add_argument('-f', '--filename', nargs='+', type=str, default='Map_{}'.format(timestr), metavar='', parser.add_argument('-f', '--filename', nargs='+', type=str, default='Map_{}'.format(timestr), metavar='',
help='Filename of map. None will not save. \n(default: %(default)s)') help='Filename of map. None will not save. \n(default: %(default)s)')
parser.add_argument('-j', '--json', nargs='?', type=str, choices=json_check, metavar='', parser.add_argument('-j', '--json', nargs='?', type=str, choices=json_check, metavar='',