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