github/plexpy
1
0
Fork 0
mirror of https://github.com/Tautulli/Tautulli.git synced 2025-07-15 17:52:59 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Remove tokens from downloaded database

Browse source
This commit is contained in:
JonnyWong16 2022-01-15 16:13:53 -08:00
parent 6742ce613a
commit ce0cdf9ce6
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
2 changed files with 19 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

14
plexpy/webserve.py
View file

@ -4852,16 +4852,26 @@ class WebInterface(object):
def download_database(self, **kwargs):
""" Download the Tautulli database file. """
database_file = database.FILENAME
database_copy = os.path.join(plexpy.CONFIG.CACHE_DIR, database_file)
try:
db = database.MonitorDatabase()
db.connection.execute('begin immediate')
shutil.copyfile(plexpy.DB_FILE, os.path.join(plexpy.CONFIG.CACHE_DIR, database_file))
shutil.copyfile(plexpy.DB_FILE, database_copy)
db.connection.rollback()
except:
pass
return serve_download(os.path.join(plexpy.CONFIG.CACHE_DIR, database_file), name=database_file)
# Remove tokens
db = database.MonitorDatabase(database_copy)
try:
db.action('UPDATE users SET user_token = NULL, server_token = NULL')
except:
logger.error('Failed to remove tokens from downloaded database.')
cherrypy.response.status = 500
return 'Error downloading database. Check the logs.'
return serve_download(database_copy, name=database_file)
@cherrypy.expose
@requireAuth(member_of("admin"))