1
0
Fork 0
mirror of synced 2024-06-02 18:34:37 +12:00

Simplify errors

This commit is contained in:
Serene-Arc 2021-02-15 16:30:39 +10:00 committed by Ali Parlakci
parent dd2804eb97
commit efffc3ee3f
4 changed files with 11 additions and 19 deletions

View file

@ -4,25 +4,17 @@ class BulkDownloaderException(Exception):
pass
class NotADownloadableLinkError(BulkDownloaderException):
pass
class RedditAuthenticationError(BulkDownloaderException):
pass
class InvalidJSONFile(BulkDownloaderException):
class SiteDownloaderError(BulkDownloaderException):
pass
class FailedToDownload(BulkDownloaderException):
class NotADownloadableLinkError(SiteDownloaderError):
pass
class ImageNotFound(BulkDownloaderException):
pass
class ExtensionError(BulkDownloaderException):
class ResourceNotFound(SiteDownloaderError):
pass

View file

@ -7,7 +7,7 @@ from abc import ABC, abstractmethod
import requests
from praw.models import Submission
from bulkredditdownloader.errors import FailedToDownload
from bulkredditdownloader.errors import SiteDownloaderError
from bulkredditdownloader.resource import Resource
logger = logging.getLogger(__name__)
@ -38,7 +38,7 @@ class BaseDownloader(ABC):
try:
download_content = requests.get(resource_url, headers=headers).content
except ConnectionResetError:
raise FailedToDownload
raise SiteDownloaderError
return Resource(self.post, resource_url, download_content)
raise FailedToDownload
raise SiteDownloaderError

View file

@ -6,7 +6,7 @@ import logging
import requests
from praw.models import Submission
from bulkredditdownloader.errors import ImageNotFound, NotADownloadableLinkError
from bulkredditdownloader.errors import ResourceNotFound, NotADownloadableLinkError
from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
logger = logging.getLogger(__name__)
@ -44,7 +44,7 @@ class Gallery(BaseDownloader):
}
res = requests.get(link, headers=headers)
if res.status_code != 200:
raise ImageNotFound(f"Server responded with {res.status_code} to {link}")
raise ResourceNotFound(f"Server responded with {res.status_code} to {link}")
page_source = res.text
starting_string = "_r = {"

View file

@ -6,7 +6,7 @@ import logging
import requests
from praw.models import Submission
from bulkredditdownloader.errors import ExtensionError, ImageNotFound, NotADownloadableLinkError
from bulkredditdownloader.errors import NotADownloadableLinkError, ResourceNotFound, SiteDownloaderError
from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
from bulkredditdownloader.site_downloaders.direct import Direct
@ -63,7 +63,7 @@ class Imgur(BaseDownloader):
cookies = {"over18": "1", "postpagebeta": "0"}
res = requests.get(link, cookies=cookies)
if res.status_code != 200:
raise ImageNotFound(f"Server responded with {res.status_code} to {link}")
raise ResourceNotFound(f"Server responded with {res.status_code} to {link}")
page_source = requests.get(link, cookies=cookies).text
starting_string = "image : "
@ -94,4 +94,4 @@ class Imgur(BaseDownloader):
if extension in extension_suffix:
return extension
else:
raise ExtensionError(f'"{extension_suffix}" is not recognized as a valid extension for Imgur')
raise SiteDownloaderError(f'"{extension_suffix}" is not recognized as a valid extension for Imgur')