28 lines
744 B
Python
28 lines
744 B
Python
|
#!/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'
|