Integrate new base_downloader class

This commit is contained in:
Serene-Arc
2021-02-25 20:40:08 +10:00
committed by Ali Parlakci
parent a75e94e43e
commit e0d321c785
13 changed files with 70 additions and 93 deletions

View File

@@ -1,9 +1,11 @@
#!/usr/bin/env python3
import logging
from typing import Optional
from praw.models import Submission
from bulkredditdownloader.authenticator import Authenticator
from bulkredditdownloader.resource import Resource
from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
@@ -14,8 +16,10 @@ class SelfPost(BaseDownloader):
def __init__(self, post: Submission):
super().__init__(post)
def download(self):
return Resource(self.post, self.post.url, bytes(self.export_to_string()))
def find_resources(self, authenticator: Optional[Authenticator] = None) -> list[Resource]:
out = Resource(self.post, self.post.url)
out.content = self.export_to_string()
return out
def export_to_string(self) -> str:
"""Self posts are formatted here"""