如何优化这段搬运图片的Python代码?

使用CN2/CN2GIA顶级线路,支持Shadowsocks/V2ray科学上网,支持支付宝付款,每月仅需 5 美元
## 加入品葱精选 Telegram Channel ##

品葱用户 Cyberspace 提问于 9/19/2020

代码是抄网上后魔改的,如何减少大量的if语句和优化正则表达所在的那行代码?
顺代一提正在用Python汇总翻车新闻和微博谈到一个文件里,自己边学边写。
也请教下品葱的大佬。

import os  
import re  
import time  
from base64 import b64encode  
import pyperclip  
import requests  
  
url = 'https://api.imgur.com/3/image' #备用https://api.mashiro.top/imgur-api/3/image  
  
  
def upload(url1):  
    return requests.post(url, headers={'Authorization': 'Client-ID 98cd21cdfc58130'},  
                         data={'image': b64encode(open(url1, 'rb').read()), 'type': 'base64'}).text  
  
  
PicDir = r'C:\\1'  #本地图片存储目录  
a = \[\]  
for Dir in os.listdir(PicDir):  
    if Dir.split('.')\[1\] == 'jpeg':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'png':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'gif':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'apng':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'jpg':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'tif':  
        a.append(PicDir + '\\\\' + Dir)  
    elif Dir.split('.')\[1\] == 'bmp':  
        a.append(PicDir + '\\\\' + Dir)  
a.append(a\[1\])   #搬运微博谈调换图片顺序  
del a\[1\]  
b = ''  
for img in a:  
    print(img)  
    res = 'https://' + repr(re.findall(re.compile('i.imgur.com.\*?(?="})'), upload(img))).replace('\\\\', '')\[2:-2\]  
    print(res)  
    b += '\[img\]' + res + '\[/img\]\\n'  
    time.sleep(0.5)  
print('\\n' + b)  
pyperclip.copy(b)

点击品葱原文参与讨论

最简单好用的 VPS,没有之一,注册立得 100 美金
comments powered by Disqus

See Also

〖微博谈搬运2020/9/18〗“习惯”太可怕了

搬运自博谈网 原地址:https://botanwang.com/articles/202009/〖微博谈〗“习惯”太可怕了.html 微博谈部分条目有较高娱乐性与魔幻现实主义色彩,故而加上“翻车新闻”标签。 上一 …

〖微博谈搬运2020/9/17〗恐怖分子才会这么做

搬运自博谈网 原地址:https://botanwang.com/articles/202009/〖微博谈〗恐怖分子才会这么做.html 微博谈部分条目有较高娱乐性与魔幻现实主义色彩,故而加上“翻车新闻”标签。 上一 …

〖微博谈搬运2020/9/16〗眼看他楼塌了

搬运自博谈网 原地址:https://botanwang.com/articles/202009/〖微博谈〗眼看他楼塌了.html 微博谈部分条目有较高娱乐性与魔幻现实主义色彩,故而加上“翻车新闻”标签。 上一 …