aboutsummaryrefslogtreecommitdiff
path: root/importer/providers/g_reddit.py
blob: 53ee5dfe14abe51e8b345ab4f8feab733386f6ed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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)