60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
import os
|
||
|
||
#rtdir = '/srv/ftp/hummingbird/2021/07/21'
|
||
rtdir = os.getcwd()
|
||
|
||
|
||
files_read = list()
|
||
for cdir, _, files in os.walk(rtdir):
|
||
for f in files:
|
||
if f.endswith('.json'):
|
||
files_read.append(os.path.join(cdir, f))
|
||
|
||
|
||
|
||
import json
|
||
|
||
fcontents = dict()
|
||
|
||
for f in files_read:
|
||
with open(f) as ff:
|
||
fcontents[f] = json.load(ff)
|
||
|
||
|
||
|
||
to_purge = dict()
|
||
for fname, contents in fcontents.items():
|
||
if len(contents) > 0:
|
||
nscores = 0
|
||
for x in contents:
|
||
if len(x['scores']) > 0:
|
||
nscores+= max(x['scores']) >= 0.6
|
||
# nscores = sum([max(x['scores']) for x in contents])
|
||
# nscores = sum([len(x['scores']) for x in contents])
|
||
if nscores == 0:
|
||
to_purge[fname] = nscores
|
||
|
||
|
||
|
||
delete_list = set()
|
||
for f_json, n_scores in to_purge.items():
|
||
vid_path = f_json.rstrip('.json') + '.mp4'
|
||
img_path = vid_path.rstrip('.mp4')+'.jpg'
|
||
if not os.path.exists(img_path):
|
||
img_path_sp = img_path.split('_')
|
||
if 'trimmed' in img_path:
|
||
img_path_sp[-2] = str(int(img_path_sp[-2].rstrip('.jpg'))+1)
|
||
|
||
else:
|
||
img_path_sp[-1] = str(int(img_path_sp[-1].rstrip('.jpg'))+1) + '.jpg'
|
||
img_path = '_'.join(img_path_sp)
|
||
|
||
delete_list.update({vid_path,img_path, f_json})
|
||
|
||
|
||
print(f'Deleting {len(delete_list)} files')
|
||
for f in delete_list:
|
||
if os.path.exists(f):
|
||
# pass
|
||
os.remove(f)
|