Sunil Mohan Adapa 9368504da5
*.py: Use SPDX license identifier
Reviewed-by: Veiko Aasa <veiko17@disroot.org>
2020-02-19 14:38:55 +02:00

28 lines
658 B
Python

# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Decorators for the backup views.
"""
import functools
import os
from . import SESSION_PATH_VARIABLE
def delete_tmp_backup_file(function):
"""Decorator to delete uploaded backup files.
XXX: Implement a better way to delete uploaded files.
"""
@functools.wraps(function)
def wrapper(request, *args, **kwargs):
path = request.session.get(SESSION_PATH_VARIABLE, None)
if path:
if os.path.isfile(path):
os.remove(path)
del request.session[SESSION_PATH_VARIABLE]
return function(request, *args, **kwargs)
return wrapper