python上传图片限制格式怎么办 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。