else
totaldl 6
This commit is contained in:
parent
9668eec5c4
commit
9b6fd93f16
1 changed files with 14 additions and 5 deletions
|
@ -5,6 +5,10 @@ import time
|
||||||
|
|
||||||
HEADERS = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/42.0.2311.152 Safari/537.36'}
|
HEADERS = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/42.0.2311.152 Safari/537.36'}
|
||||||
|
|
||||||
|
DOWNLOAD_DIRECTORY = 'heyo'
|
||||||
|
# Save files to this folder
|
||||||
|
# If blank, it uses the local folder
|
||||||
|
|
||||||
IMGUR_ALBUM_INDV = '"og:image"content="htt'
|
IMGUR_ALBUM_INDV = '"og:image"content="htt'
|
||||||
# The HTML string which tells us that an image link is
|
# The HTML string which tells us that an image link is
|
||||||
# on this line.
|
# on this line.
|
||||||
|
@ -43,10 +47,14 @@ DO_GENERIC = True
|
||||||
|
|
||||||
last_request = 0
|
last_request = 0
|
||||||
|
|
||||||
|
if DOWNLOAD_DIRECTORY[-1] not in ['/', '\\']:
|
||||||
|
DOWNLOAD_DIRECTORY += '\\'
|
||||||
|
|
||||||
class StatusExc(Exception):
|
class StatusExc(Exception):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def download_file(url, localname):
|
def download_file(url, localname):
|
||||||
|
localname = DOWNLOAD_DIRECTORY + localname
|
||||||
if os.path.exists(localname):
|
if os.path.exists(localname):
|
||||||
print('\t%s already exists!!' % localname)
|
print('\t%s already exists!!' % localname)
|
||||||
return
|
return
|
||||||
|
@ -102,8 +110,9 @@ def handle_imgur(url, albumid='', customname=None):
|
||||||
if albumid and albumid != basename:
|
if albumid and albumid != basename:
|
||||||
|
|
||||||
if IMGUR_ALBUMFOLDERS:
|
if IMGUR_ALBUMFOLDERS:
|
||||||
if not os.path.exists(albumid):
|
|
||||||
os.makedirs(albumid)
|
if not os.path.exists(DOWNLOAD_DIRECTORY + albumid):
|
||||||
|
os.makedirs(DOWNLOAD_DIRECTORY + albumid)
|
||||||
localpath = '%s\\%s' % (albumid, name)
|
localpath = '%s\\%s' % (albumid, name)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
|
@ -318,9 +327,9 @@ def test(imgur=True, gfycat=True, vimeo=True, liveleak=True, youtube=True, gener
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
test(
|
test(
|
||||||
imgur=False,
|
imgur=True,
|
||||||
gfycat=False,
|
gfycat=False,
|
||||||
vimeo=False,
|
vimeo=False,
|
||||||
liveleak=True,
|
liveleak=False,
|
||||||
youtube=False,
|
youtube=False,
|
||||||
generic=False)
|
generic=True)
|
Loading…
Reference in a new issue