1
0
mirror of https://git.sr.ht/~cadence/NewLeaf synced 2024-11-14 12:07:30 +00:00
NewLeaf/tools/files.py
2021-04-03 14:57:51 +13:00

19 lines
555 B
Python

import os
def get_created_files(id):
if id[0] == "-":
id = "_" + id[1:] # youtube-dl changes - to _ at the start, presumably to not accidentally trigger switches with * in shell
id += "_"
# youtube-dl thinks it's a really good idea to do this, for some reason.
trim_id = id.lstrip("_")
if trim_id.startswith("-"):
trim_id = "_" + trim_id[len("-"):]
return (f for f in os.listdir() if f.startswith(id) or f.startswith(trim_id))
def clean_up_temp_files(id):
created_files = get_created_files(id)
for file in created_files:
os.unlink(file)