Add resource class
This commit is contained in:
parent
81293db8e9
commit
c20fab2594
1 changed files with 27 additions and 0 deletions
27
bulkredditdownloader/resource.py
Normal file
27
bulkredditdownloader/resource.py
Normal file
|
@ -0,0 +1,27 @@
|
|||
#!/usr/bin/env python3
|
||||
# coding=utf-8
|
||||
|
||||
import hashlib
|
||||
import re
|
||||
|
||||
from praw.models import Submission
|
||||
|
||||
|
||||
class Resource:
|
||||
def __init__(self, source_submission: Submission, url: str, content: bytes):
|
||||
self.source_submission = source_submission
|
||||
self.content = content
|
||||
self.url = url
|
||||
self.hash = hashlib.md5(content)
|
||||
self.extension = self._get_extension(url)
|
||||
|
||||
@staticmethod
|
||||
def _get_extension(url: str) -> str:
|
||||
pattern = re.compile(r'(\.(jpg|jpeg|png|mp4|webm|gif))')
|
||||
if results := re.search(pattern, url):
|
||||
if len(results.groups()) > 1:
|
||||
return results[0]
|
||||
if "v.redd.it" not in url:
|
||||
return '.jpg'
|
||||
else:
|
||||
return '.mp4'
|
Loading…
Reference in a new issue