終于忙完有空更新了，這次說下一個比較簡單的東西，限制上傳圖片格式問題。

先上代碼！

img_file = 'D:\\image\\test.jpg'

# uuid生成文件名

key = shortuuid.ShortUUID().random(length=9)

if img_file.size > 2097152:

raise exceptions.APIException('圖片大小超出限制')

# file_postfix = img_file.name.split('.')[1]

suffix = img_file.name.rfind('.')

if suffix == -1:

raise exceptions.APIException('圖片不正確')

file_postfix = img_file.name[suffix + 1:]

if file_postfix not in ['jpg', 'png', 'gif', 'bmp', 'jpeg', 'JPG', 'PNG',

'BMP', 'JPEG', ]:

raise exceptions.APIException('圖片格式不正確')

file_name = key + '.' + file_postfix

以前項目里總是再用file_postfix = img_file.name.split('.')[1]這種方法再取文件后綴進行判斷格式，這種方法對普通的文件名還是沒問題的例如：圖片1.jpg、圖片2.jpg、圖片3.jpg。但是遇到www.圖片1.jpg這種的圖片名稱就會爆炸，因為圖片名稱中存在兩個.

suffix = img_file.name.rfind('.') 而str的rfind方法就可以避免這種問題，因為這是反響尋找。舉個例子

a = '123456'

print('下標', a.rfind('5'))

輸出結果：下標 4

a = '123456'

print('下標', a.rfind('7'))

輸出結果: 下標 -1

rfind方法從字符串最后以為開始找，找到返回下標，找不到返回-1。

所以 www.test.jpg 這種格式的圖片他會從最后位置開始找.找到之后在截取就只剩jpg了，也就好判斷了。

要注意的是rfind的下標是從0開始的，所以截取的時候記得加1。

行了 其他的基本沒啥了。這篇文章就到這吧，最近會更新上傳圖片到微信服務器。

總結

以上所述是小編給大家介紹的解決python 上傳圖片限制格式問題,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！

如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！

總結

以上是生活随笔為你收集整理的python脚本限制_解决python 上传图片限制格式问题的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。