diff --git a/decorators.py b/decorators.py index 473728d..4af96d3 100644 --- a/decorators.py +++ b/decorators.py @@ -1,5 +1,6 @@ import time import logging +import tempfile from lockfile import FileLock, AlreadyLocked, LockTimeout @@ -39,7 +40,7 @@ def wrapper(self, *args, **options): logging.debug("-" * 72) lock_name = self.__module__.split('.').pop() - lock = FileLock('/tmp/pdk_lock_' + lock_name) + lock = FileLock(tempfile.gettempdir() + '/pdk_lock_' + lock_name) logging.debug("%s - acquiring lock..." % lock_name) try: diff --git a/management/commands/compile_reports.py b/management/commands/compile_reports.py index 9247286..29e7b80 100644 --- a/management/commands/compile_reports.py +++ b/management/commands/compile_reports.py @@ -4,6 +4,7 @@ import json import os import pytz +import tempdir import traceback import importlib @@ -56,7 +57,7 @@ def handle(self, *args, **options): if ('raw_data' in report.parameters) and report.parameters['raw_data'] is True: raw_json = True - filename = '/tmp/pdk_export_' + str(report.pk) + '.zip' + filename = tempfile.gettempdir() + '/pdk_export_' + str(report.pk) + '.zip' with ZipFile(filename, 'w') as export_file: for generator in generators: diff --git a/pdk_api.py b/pdk_api.py index 9fe348d..d508b61 100644 --- a/pdk_api.py +++ b/pdk_api.py @@ -2,6 +2,7 @@ import codecs import csv import json +import tempdir from django.template.loader import render_to_string @@ -30,7 +31,7 @@ # return None def compile_report(generator, sources): - filename = '/tmp/pdk_' + generator + '.txt' + filename = tempfile.gettempdir() + '/pdk_' + generator + '.txt' with open(filename, 'w') as outfile: writer = csv.writer(outfile, delimiter='\t')