public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Less collision-prone file names for downloads

Browse source
This commit is contained in:
Grant Sanderson 2024-11-25 09:31:15 -07:00
parent 0ad5a0e76e
commit e80b9d0e47
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
manimlib/utils/file_ops.py
View file

@ -1,6 +1,8 @@
from __future__ import annotations from __future__ import annotations
import os import os
from pathlib import Path
import hashlib
import numpy as np import numpy as np
import validators import validators
@ -35,9 +37,11 @@ def find_file(
if validators.url(file_name): if validators.url(file_name):
import urllib.request import urllib.request
from manimlib.utils.directories import get_downloads_dir from manimlib.utils.directories import get_downloads_dir
stem, name = os.path.split(file_name) suffix = Path(file_name).suffix
file_hash = hashlib.sha256(file_name.encode('utf-8')).hexdigest()[:32]
folder = get_downloads_dir() folder = get_downloads_dir()
path = os.path.join(folder, name)
path = Path(folder, file_hash).with_suffix(suffix)
urllib.request.urlretrieve(file_name, path) urllib.request.urlretrieve(file_name, path)
return path return path