前提・実現したいこと
pythonで指定のURLを入力するとそのURL内にある画像を全てダウンロードするプログラムを作っています。
発生している問題・エラーメッセージ
問題としては画像のダウンロードはできるのですが、拡張子がなくなっています
該当のソースコード
python3
1import os,requests,sys,bs4,datetime 2 3#画像のダウンロード 4def download_image(url, timeout = 10): 5 response = requests.get(url, allow_redirects = False, timeout=timeout) 6 if response.status_code != 200: 7 raise Exception("HTTP status: " + response.status_code) 8 9 content_type = response.headers["content-type"] 10 if 'image' not in content_type: 11 raise Exception("content-type: " + content_type) 12 13 return response.content 14 15#画像のファイル名を決める 16def make_filename(base_dir, number, url): 17 ext = os.path.splitext(url)[1] #拡張子の取得 18 filename = number + ext #番号に拡張子をつけてファイル名にする 19 20 fullpath = os.path.join(base_dir, filename) 21 return fullpath 22 23#画像を保存する 24def save_image(filename,image): 25 with open(filename, "wb") as fout: 26 fout.write(image) 27try: 28 os.mkdir("./images") 29except: 30 pass 31 32#指定したurlのHTMLから画像のurlを抽出し、ダウンロードする 33def main(): 34 url = input() 35 urls = [] 36 res = requests.get(url) 37 soup = bs4.BeautifulSoup(res.text) 38 for tag in soup.find_all('img'): 39 urls.append(tag.get("src")) 40 images_dir = '/Users/user/Downloads' 41 for idx,url in enumerate(urls): 42 filename = make_filename(images_dir, str(idx) , url) 43 try: 44 image = download_image(url) 45 except: 46 pass 47 else: 48 save_image(filename, image) 49 50if __name__ == '__main__': 51 main() 52 53
補足情報(FW/ツールのバージョンなど)
python3
macOS
make_filenameが誤った文字列を返してる気がするのですが、そこは問題ないですか?パット見た感じではおかしいところは見つけられなかったのですが。
回答1件
あなたの回答
tips
プレビュー