溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
終于忙完有空更新了，這次說下一個比較簡單的東西，限制上傳圖片格式問題。
先上代碼！
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 上傳圖片限制格式問題,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對億速云網站的支持！
如果你覺得本文對你有幫助，歡迎轉載，煩請注明出處，謝謝！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
