mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-22 18:53:21 +01:00
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2014, 2015 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Common classes and constants used by downloader modules."""
|
|
|
|
import os
|
|
|
|
class BasicDownloader():
|
|
"""Base class for downlaoder modules"""
|
|
|
|
max_tries = 5
|
|
|
|
def download(self, url, path):
|
|
"""Download the resource at 'url' and write it to a file given by 'path'"""
|
|
with open(path, "wb") as file:
|
|
try:
|
|
return self.download_impl(url, file)
|
|
except:
|
|
# make sure to remove file if download failed
|
|
os.unlink(path)
|
|
raise
|
|
|
|
def download_impl(self, url, file_handle):
|
|
"""Actual implementaion of the download process"""
|
|
pass
|
|
|
|
@staticmethod
|
|
def print_error(file, error, tries, max_tries=5):
|
|
"""Print a message indicating an error during download"""
|
|
if tries == 1 and hasattr(file, "name"):
|
|
print("\r\033[1;31m", file.name, sep="")
|
|
print("\033[0;31m[Error]\033[0m ", error, " (", tries, "/", max_tries, ")", sep="")
|