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)