Move to inheritance system for downloaders

This commit is contained in:
Serene-Arc
2021-02-07 14:46:20 +10:00
committed by Ali Parlakci
parent 69e21e46a2
commit f573038a21
11 changed files with 253 additions and 280 deletions

View File

@@ -1,4 +1,5 @@
import os
#!/usr/bin/env python3
import pathlib
from bulkredditdownloader.downloaders.base_downloader import BaseDownloader
@@ -8,11 +9,11 @@ from bulkredditdownloader.utils import GLOBAL
class Direct(BaseDownloader):
def __init__(self, directory: pathlib.Path, post: dict):
super().__init__(directory, post)
post['EXTENSION'] = self.getExtension(post['CONTENTURL'])
if not os.path.exists(directory):
os.makedirs(directory)
self.download()
filename = GLOBAL.config['filename'].format(**post) + post["EXTENSION"]
short_filename = post['POSTID'] + post['EXTENSION']
def download(self):
self.post['EXTENSION'] = self._get_extension(self.post['CONTENTURL'])
self.directory.mkdir(exist_ok=True)
self.getFile(filename, short_filename, directory, post['CONTENTURL'])
filename = GLOBAL.config['filename'].format(**self.post) + self.post["EXTENSION"]
self._download_resource(pathlib.Path(filename), self.directory, self.post['CONTENTURL'])