From 10cbc378ad0daf0e80f5ceed92d70fdbf573df88 Mon Sep 17 00:00:00 2001 From: gabrielgio Date: Sun, 18 Jul 2021 19:56:59 +0200 Subject: ref: Move to OO implementation Heavily inspired by the `youtube-dl` implementation I moved to OO implementation where now every source type has its own class, making easy to add new providers. Also new it has a fallback back, where if no provider is chose it will try to download with `YoutubeDlProvideBase`. Add `_TEST` to each class to make it easy to add test to new providers. --- importer/providers/g_reddit.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 importer/providers/g_reddit.py (limited to 'importer/providers/g_reddit.py') diff --git a/importer/providers/g_reddit.py b/importer/providers/g_reddit.py new file mode 100644 index 0000000..53ee5df --- /dev/null +++ b/importer/providers/g_reddit.py @@ -0,0 +1,19 @@ +from praw import Reddit + +from importer.providers.raw_image_base import RawImageProviderBase + + +class GReddit(RawImageProviderBase): + regex = "^.*www.reddit.com/gallery.*$" + + def __init__(self, url: str, reddit: Reddit): + super(GReddit, self).__init__(url) + self.reddit = reddit + + def download(self): + submission = self.reddit.submission(url=self.url) + for key in submission.media_metadata: + value = submission.media_metadata[key] + url = value['s']['u'] + path = self._download_raw_file(url) + self.paths.append(path) -- cgit v1.2.3