CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-06-27 18:40:34 +12:00
Code Issues Releases Wiki Activity

added progress bar to archiver and cloner

Browse source
This commit is contained in:
Piotr Migdal 2024-01-19 17:21:53 +01:00
parent 43bfc21cd4
commit f943078517
2 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
bdfr/archiver.py
View file

@ -19,6 +19,7 @@ from bdfr.archive_entry.submission_archive_entry import SubmissionArchiveEntry
from bdfr.configuration import Configuration
from bdfr.connector import RedditConnector
from bdfr.exceptions import ArchiverError
from bdfr.progress_bar import Progress
from bdfr.resource import Resource
logger = logging.getLogger(__name__)
@ -29,7 +30,9 @@ class Archiver(RedditConnector):
super().__init__(args, logging_handlers)
def download(self) -> None:
progress = Progress(self.args.progress_bar, len(self.reddit_lists))
for generator in self.reddit_lists:
progress.subreddit_new(generator)
try:
for submission in generator:
try:
@ -40,18 +43,22 @@ class Archiver(RedditConnector):
f"Submission {submission.id} in {submission.subreddit.display_name} skipped due to"
f" {submission.author.name if submission.author else 'DELETED'} being an ignored user"
)
progress.post_done(submission, False)
continue
if submission.id in self.excluded_submission_ids:
logger.debug(f"Object {submission.id} in exclusion list, skipping")
progress.post_done(submission, False)
continue
logger.debug(f"Attempting to archive submission {submission.id}")
self.write_entry(submission)
progress.post_done(submission, True)
except prawcore.PrawcoreException as e:
logger.error(f"Submission {submission.id} failed to be archived due to a PRAW exception: {e}")
except prawcore.PrawcoreException as e:
logger.error(f"The submission after {submission.id} failed to download due to a PRAW exception: {e}")
logger.debug("Waiting 60 seconds to continue")
sleep(60)
progress.subreddit_done()
def get_submissions_from_link(self) -> list[list[praw.models.Submission]]:
supplied_submissions = []

8
bdfr/cloner.py
View file

@ -9,6 +9,7 @@ import prawcore
from bdfr.archiver import Archiver
from bdfr.configuration import Configuration
from bdfr.downloader import RedditDownloader
from bdfr.progress_bar import Progress
logger = logging.getLogger(__name__)
@ -18,15 +19,20 @@ class RedditCloner(RedditDownloader, Archiver):
super().__init__(args, logging_handlers)
def download(self) -> None:
progress = Progress(self.args.progress_bar, len(self.reddit_lists))
for generator in self.reddit_lists:
progress.subreddit_new(generator)
try:
for submission in generator:
try:
self._download_submission(submission)
success = self._download_submission(submission)
self.write_entry(submission)
except prawcore.PrawcoreException as e:
logger.error(f"Submission {submission.id} failed to be cloned due to a PRAW exception: {e}")
success = False
progress.post_done(submission, success)
except prawcore.PrawcoreException as e:
logger.error(f"The submission after {submission.id} failed to download due to a PRAW exception: {e}")
logger.debug("Waiting 60 seconds to continue")
sleep(60)
progress.subreddit_done()