Python搭建Gitee图床的示例代码

git clone https://gitee.com/xxx/xxx.git

pip install --upgrade tinify
pip install gitpython

import random
import time
import os
import shutil
from git import Repo
import tinify

repo = Repo('./res') #创建版本库对象
tinify.key = '****************' #在此粘贴刚刚复制的API Key

exts = ['.png','.jpg','.bmp'] #支持的图像格式
compression = ['.png','.jpg'] #支持压缩的图像格式
srcdir = './temp' #源文件夹
dstdir = './res/upload_images' #目标文件夹
url = 'https://xxx.gitee.io/res/upload_images/' #图床路径（末尾必须加“/”），将xxx替换成自己的用户名

def random_hex(length):
    result = hex(random.randint(0,16**length)).replace('0x','').lower()
    if(len(result)<length):
        result = '0'*(length-len(result))+result
    return result

def auto_code(ext):
    while True:
        name = random_hex(8) #随机8位16进制编码
        result = os.path.join(dstdir,name + ext)
        if not os.path.exists(result):
            break #目标路径不存在则可以移动图片
    return result

def main():
    f = open('./output.txt','w') #打开输出文件
    list = os.listdir(srcdir) #列出文件夹下所有的目录与文件
    for i in range(0,len(list)):
        srcpath = os.path.join(srcdir,list[i])
        if not os.path.isfile(srcpath):
            continue #不是文件则跳过
        ext=os.path.splitext(srcpath)[-1].lower() #获取文件扩展名
        if ext not in exts:
            continue #不是支持的图像格式则跳过
        dstpath = auto_code(ext)
        if ext in compression:
            tinify.from_file(srcpath).to_file(srcpath) #压缩文件
            shutil.move(srcpath,dstpath) #移动文件
            print('成功压缩并移动：' + os.path.basename(srcpath))
        else:
            shutil.move(srcpath,dstpath) #移动文件
            print('成功移动：' + os.path.basename(srcpath))
        f.write(os.path.basename(srcpath) + ':![](' + url + os.path.basename(dstpath) + ')\n') #将原始文件名和与之对应的图片网址写入txt文件
    f.close()
    print('输出文件output.txt已生成')
    print(repo.git.add('--all')) #添加全部更改
    print(repo.git.commit('-m upload images')) #提交
    print(repo.remote().push('master')) #推送
    print('已推送至远程仓库，python即将退出')
    time.sleep(1)

if __name__ == '__main__':
    main()