1
0
Fork 0
mirror of synced 2024-06-14 16:24:45 +12:00

Update extension regex to match URI fragments (#264)

This commit is contained in:
Nathan Spicer-Davis 2021-04-12 23:51:40 -04:00 committed by Ali Parlakci
parent ab7a0f6a51
commit 59ab5d8777
2 changed files with 3 additions and 1 deletions

View file

@ -64,7 +64,7 @@ class Resource:
self.hash = hashlib.md5(self.content)
def _determine_extension(self) -> Optional[str]:
extension_pattern = re.compile(r'.*(\..{3,5})(?:\?.*)?$')
extension_pattern = re.compile(r'.*(\..{3,5})(?:\?.*)?(?:#.*)?$')
match = re.search(extension_pattern, self.url)
if match:
return match.group(1)

View file

@ -15,6 +15,8 @@ from bdfr.resource import Resource
('https://www.resource.com/test/example.jpg', '.jpg'),
('hard.png.mp4', '.mp4'),
('https://preview.redd.it/7zkmr1wqqih61.png?width=237&format=png&auto=webp&s=19de214e634cbcad99', '.png'),
('test.jpg#test','.jpg'),
('test.jpg?width=247#test','.jpg'),
))
def test_resource_get_extension(test_url: str, expected: str):
test_resource = Resource(MagicMock(), test_url)