fix: added a check in cleanup.py to see if the assets folder exists before clearing it
This commit is contained in:
1 parent
f8c340f612
commit
80653eab7c
1 file changed
+17
-14
+17
-14
@@ -1,20 +1,23 @@
|
||||
import os
|
||||
from os.path import exists
|
||||
|
||||
|
||||
def cleanup() -> int:
|
||||
count = 0
|
||||
files = [f for f in os.listdir('.') if f.endswith('.mp4') and 'temp' in f.lower()]
|
||||
count += len(files)
|
||||
for f in files:
|
||||
os.remove(f)
|
||||
try:
|
||||
for file in os.listdir('./assets/temp/mp4'):
|
||||
if exists('./assets/temp'):
|
||||
count = 0
|
||||
files = [f for f in os.listdir('.') if f.endswith('.mp4') and 'temp' in f.lower()]
|
||||
count += len(files)
|
||||
for f in files:
|
||||
os.remove(f)
|
||||
try:
|
||||
for file in os.listdir('./assets/temp/mp4'):
|
||||
count += 1
|
||||
os.remove('./assets/temp/mp4/' + file)
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
for file in os.listdir('./assets/temp/mp3'):
|
||||
count += 1
|
||||
os.remove('./assets/temp/mp4/' + file)
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
for file in os.listdir('./assets/temp/mp3'):
|
||||
count += 1
|
||||
os.remove('./assets/temp/mp3/' + file)
|
||||
return count
|
||||
os.remove('./assets/temp/mp3/' + file)
|
||||
return count
|
||||
return 0
|
||||
|
||||
Reference in new issue
Block a user