parent
5545624da1
commit
28fa7c53b4
@ -1,21 +1,37 @@
|
|||||||
|
# -*- 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
|
import os
|
||||||
|
|
||||||
class BasicDownloader():
|
class BasicDownloader():
|
||||||
|
"""Base class for downlaoder modules"""
|
||||||
|
|
||||||
max_tries = 5
|
max_tries = 5
|
||||||
|
|
||||||
def download(self, url, path):
|
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:
|
with open(path, "wb") as file:
|
||||||
try:
|
try:
|
||||||
return self.download_impl(url, file)
|
return self.download_impl(url, file)
|
||||||
file.close()
|
|
||||||
except:
|
except:
|
||||||
# make sure to remove file if download failed
|
# make sure to remove file if download failed
|
||||||
os.unlink(path)
|
os.unlink(path)
|
||||||
raise
|
raise
|
||||||
|
|
||||||
|
def download_impl(self, url, file_handle):
|
||||||
|
"""Actual implementaion of the download process"""
|
||||||
|
pass
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def print_error(file, error, tries, max_tries=5):
|
def print_error(file, error, tries, max_tries=5):
|
||||||
|
"""Print a message indicating an error during download"""
|
||||||
if tries == 1 and hasattr(file, "name"):
|
if tries == 1 and hasattr(file, "name"):
|
||||||
print("\r\033[1;31m", file.name, sep="")
|
print("\r\033[1;31m", file.name, sep="")
|
||||||
print("\033[0;31m[Error]\033[0m ", error, " (", tries, "/", max_tries, ")", sep="")
|
print("\033[0;31m[Error]\033[0m ", error, " (", tries, "/", max_tries, ")", sep="")
|
||||||
|
Loading…
Reference in new issue