Уроки Python 14 - Скачивание файлов

Уроки Python 14 - Скачивание файлов

Сегодня мы научимся скачивать файлы из интернета двумя разными способами. Первый способ предполагает использование модуля под названием wget, и он достаточно хорош для больших файлов. Второй способ использует библиотеку requests, и применяется в основном для сохранения картинок с сайтов. Дело в том, что некоторые сайты ограничивают скачивание картинок с помощью библиотеки wget. Именно поэтому при сохранении html страниц с картинками, приходится использовать вместо wget библиотеку requests. Также понадобится модуль shutil который не нужно ставить, он входит в стандартную поставку модулей Python.

Давайте установим оба модуля:

pip install wget

pip install requests

Теперь давайте попробуем скачать какой-либо файл с помощью модуля wget. Пусть это будет например картинка с моего сайта.

Код программы (обратите внимание что в коде я заменил расширение картинки jpg на jbg - при копировании кода замените обратно):

import wget, os

s='pic/20170413081151825.jbg'

filename = wget.download(s)

os.rename(filename, u''+os.getcwd()+'/'+filename)

Для начала мы поместили адрес скачиваемого файла в переменную s. Далее с помощью команды filename = wget.download(s) мы скачали файл и сохранили его под временным именем, которое поместилось в переменную filename. Теперь, чтобы сохранить в файл под тем именем, которое нам нужно, задействуем модуль os, и дадим команду os.rename(filename, u''+os.getcwd()+filename), которая переместит временный файл в папку, где лежит наш скрипт, сохранив при этом имя файла. Если такой файл там уже существует (например мы запустили скрипт два раза подряд), может возникнуть ошибка, связанная с попыткой перезаписи уже существующего файла, имейте это ввиду.

Теперь скачаем тот же самый файл с помощью библиотеки requests.

import requests, shutil, os

s='pic/20170413081151825.jbg'

(dirname, filename) = os.path.split(s)

r = requests.get(s, stream=True)

if r.status_code == 200:

with open(filename, 'wb') as f:

r.raw.decode_content = True

shutil.copyfileobj(r.raw, f)

Разберем данный код. Сначала мы как обычно помещаем адрес файла, который нужно скачать в переменную s. Далее нам нужно выделить из этого адреса имя файла, под которым мы будем его сохранять. Для этого дадим команду (dirname, filename) = os.path.split(s), которая разделит адрес файла на путь к нему и собственно имя файла, которое нам и нужно. Теперь с помощью get запроса из модуля requests получим файл в переменную r, указав в параметрах запроса stream=true, что означает потоковое сохранение принимаемых данных в файл. Если сервер вернет нам статус 200, что свидетельствует об успешном соединении, с помощью функции open откроем на запись файл с именем filename в режиме записи байтов - 'wb'. И далее с помощью команды shutil.copyfileobj(r.raw, f) скопируем принимаемый поток собственно в файл. Как видим, такой способ немного сложнее чем вариант с wget, но зато он срабатывает в большинстве случаев.