python 上傳圖片限制格式問題

python 上傳圖片限制格式問題

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

		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。

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

關注一下吧，畢竟我是個新人。哈哈哈