from bs4 import BeautifulSoup
import requests
from selenium import webdriver
import urllib.request
import os
import shutil
from mimetypes import guess_extension
img_folder = "c:/test"
if os.path.exists(img_folder):
    shutil.rmtree(img_folder)
path = r"C:\Users\qpslt\Desktop\py\chromedriver_win32\chromedriver.exe"
driver = webdriver.Chrome(path)
site_url = "https://gall.dcinside.com/board/view/?id=baseball_new8&no=10131338&exception_mode=recommend&page=1"
driver.get(site_url)
images = driver.find_elements_by_xpath('//div[@class="writing_view_box"]//img')
for i, img in enumerate(images, 1):
    img_url = img.get_attribute('src')
    print(i, img_url)
    r = requests.get(img_url, headers={'Referer': site_url})
    try:   #文件夹创建
        if not os.path.exists(img_folder):
            os.makedirs(img_folder)
    except Exception as er:
        print("{}发生错误。".format(er))
        break;
    break;
    with open("c:/test/{}.jpg".format(i), 'wb') as f:
        f.write(r.content)
我不总是知道正在下载的文件的扩展名。
您如何知道要下载的文件的扩展名？
<details>
<summary>英文:</summary>
This is my jpg image download source:
    from bs4 import BeautifulSoup
    import requests
    from selenium import webdriver
    import urllib.request
    import os
    import shutil
    from mimetypes import guess_extension
    
    img_folder = (&quot;c:/test&quot;)
    if os.path.exists(img_folder):
        shutil.rmtree(img_folder)
    
    path = (r&quot;C:\Users\qpslt\Desktop\py\chromedriver_win32\chromedriver.exe&quot;)
    driver = webdriver.Chrome(path)
    site_url = (&quot;https://gall.dcinside.com/board/view/?id=baseball_new8&amp;no=10131338&amp;exception_mode=recommend&amp;page=1&quot;)
    driver.get(site_url)
    images = driver.find_elements_by_xpath(&#39;//div[@class=&quot;writing_view_box&quot;]//img&#39;)
    
    for i, img in enumerate(images, 1):
        img_url = img.get_attribute(&#39;src&#39;)
        print(i, img_url)
        r = requests.get(img_url, headers={&#39;Referer&#39;: site_url})
        try:   #폴더 만들기
            if not os.path.exists(img_folder):
                os.makedirs(img_folder)
        except Exception as er:
            print(&quot;{}에러가 발생했습니다.&quot;.format(er))
            break;
        break;
        with open(&quot;c:/test/{}.jpg&quot;.format(i), &#39;wb&#39;) as f:
            f.write(r.content)
I don&#39;t always know the extension of the image.
How do you know the extension of the file you are downloading?
</details>
# 答案1
**得分**: 5
如果图片链接没有扩展名（例如，如果图片是通过php脚本动态生成的），那么您可以使用[`mimetypes.guess_extension()`](https://docs.python.org/3/library/mimetypes.html#mimetypes.guess_extension)将图片响应的`content-type`标头映射到文件扩展名。
例如：
```python
import mimetypes
...
r = requests.get(img_url, headers={'Referer': site_url})
extension = mimetypes.guess_extension(r.headers.get('content-type', '').split(';')[0])
...
with open("c:/test/{}{}".format(i, extension or '.jpg'), 'wb') as f:

img_url = 'path/to/your/picture.jpg'
split1 = img_url.split('/') # 返回 [‘path’, ‘to’, ‘your’, ‘picture.jpg’]
file = split1[-1]
filename, extension = file.split('.') # file 是 ‘picture’，extension 是 ‘jpg’