github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-08-14 02:26:58 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add option to uninstall the GeoLite2 database

Browse source
This commit is contained in:
JonnyWong16 2016-07-01 09:33:13 -07:00
commit 8b787e4ae0
4 changed files with 47 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
plexpy/helpers.py
View file

@ -577,6 +577,19 @@ def install_geoip_db():
return True
def uninstall_geoip_db():
logger.debug(u"PlexPy Helpers :: Uninstalling the GeoLite2 database...")
try:
os.remove(plexpy.CONFIG.GEOIP_DB)
plexpy.CONFIG.__setattr__('GEOIP_DB', '')
plexpy.CONFIG.write()
except Exception as e:
logger.error(u"PlexPy Helpers :: Failed to uninstall the GeoLite2 database: %s" % e)
return False
logger.debug(u"PlexPy Helpers :: GeoLite2 database uninstalled successfully.")
return True
def geoip_lookup(ip_address):
if not plexpy.CONFIG.GEOIP_DB:
return 'GeoLite2 database not installed. Please install from the ' \

21
plexpy/webserve.py
View file

@ -2802,15 +2802,30 @@ class WebInterface(object):
@cherrypy.expose
@cherrypy.tools.json_out()
@requireAuth(member_of("admin"))
@addtoapi()
def install_geoip_db(self):
""" Downloads and installs the GeoIP database """
""" Downloads and installs the GeoLite2 database """
result = helpers.install_geoip_db()
if result:
return {'result': 'success', 'message': 'GeoIP database installed successful.'}
return {'result': 'success', 'message': 'GeoLite2 database installed successful.'}
else:
return {'result': 'error', 'message': 'GeoIP database install failed.'}
return {'result': 'error', 'message': 'GeoLite2 database install failed.'}
@cherrypy.expose
@cherrypy.tools.json_out()
@requireAuth(member_of("admin"))
@addtoapi()
def uninstall_geoip_db(self):
""" Uninstalls the GeoLite2 database """
result = helpers.uninstall_geoip_db()
if result:
return {'result': 'success', 'message': 'GeoLite2 database uninstalled successfully.'}
else:
return {'result': 'error', 'message': 'GeoLite2 database uninstall failed.'}
@cherrypy.expose
@requireAuth(member_of("admin"))