def test_walkfiles():
basedir = tempfile.mkdtemp()
subdir = tempfile.mkdtemp(dir=basedir)
file1, file1path = tempfile.mkstemp(dir=basedir, suffix="_foo")
file2, file2path = tempfile.mkstemp(dir=subdir, suffix="_foo")
file3, file3path = tempfile.mkstemp(dir=subdir, suffix="_bar")
eq_(sorted(walkfiles(basedir, suffix="_foo")), sorted([file1path, file2path]))
eq_(sorted(walkfiles(basedir)), sorted([file1path, file3path, file2path]))
def test_walkfiles():
basedir = tempfile.mkdtemp()
subdir = tempfile.mkdtemp(dir=basedir)
file1, file1path = tempfile.mkstemp(dir=basedir, suffix='_foo')
file2, file2path = tempfile.mkstemp(dir=subdir, suffix='_foo')
file3, file3path = tempfile.mkstemp(dir=subdir, suffix='_bar')
eq_(sorted(walkfiles(basedir, suffix='_foo')),
sorted([file1path, file2path]))
eq_(sorted(walkfiles(basedir)), sorted([file1path, file3path, file2path]))
def unhide_disabled_files():
# Files are getting stuck in /guarded-addons for some reason. This job
# makes sure guarded add-ons are supposed to be disabled.
log = logging.getLogger('z.files.disabled')
q = (Q(version__addon__status=mkt.STATUS_DISABLED) |
Q(version__addon__disabled_by_user=True))
files = set(File.objects.filter(q | Q(status=mkt.STATUS_DISABLED))
.values_list('version__addon', 'filename'))
for filepath in walkfiles(settings.GUARDED_ADDONS_PATH):
addon, filename = filepath.split('/')[-2:]
if tuple([int(addon), filename]) not in files:
log.warning('File that should not be guarded: %s.' % filepath)
try:
file_ = (File.objects.select_related('version__addon')
.get(version__addon=addon, filename=filename))
file_.unhide_disabled_file()
except File.DoesNotExist:
log.warning('File object does not exist for: %s.' % filepath)
except Exception:
log.error('Could not unhide file: %s.' % filepath,
exc_info=True)
def unhide_disabled_files():
# Files are getting stuck in /guarded-addons for some reason. This job
# makes sure guarded add-ons are supposed to be disabled.
log = logging.getLogger('z.files.disabled')
q = (Q(version__addon__status=mkt.STATUS_DISABLED)
| Q(version__addon__disabled_by_user=True))
files = set(
File.objects.filter(q | Q(status=mkt.STATUS_DISABLED)).values_list(
'version__addon', 'filename'))
for filepath in walkfiles(settings.GUARDED_ADDONS_PATH):
addon, filename = filepath.split('/')[-2:]
if tuple([int(addon), filename]) not in files:
log.warning('File that should not be guarded: %s.' % filepath)
try:
file_ = (File.objects.select_related('version__addon').get(
version__addon=addon, filename=filename))
file_.unhide_disabled_file()
except File.DoesNotExist:
log.warning('File object does not exist for: %s.' % filepath)
except Exception:
log.error('Could not unhide file: %s.' % filepath,
exc_info=True)
