Python3 初学实践案例（14）打造一个私人图床服务器

时间：2022-05-04 11:30:49

Python3 初学实践案例（14）打造一个私人图床服务器

近来工作压力颇大，一直都在从事项目的开发工作，所以自学 python 的进度也拉下来一截，哎。由于进来很多项目用 react 编写，所以准备些一些相关的博客文章。但是博客的图片管理颇让人烦恼。之前尝试过把新浪微博当成图床，但是上传的稳定性一直是一个问题。使用 csdn 的博客图片上传功能吧，一直也比较不方便。因此，昨天突发奇想，决定写一个私人的图床服务器，放在我自己的服务器上。

需求分析

要干什么，首先要明确，到底要TM干什么。列出需求如下：

  1. 这是一个图片上传服务，需要随时接收我的图片文件。
  2. 只能接收我上传文件，因此需要认证机制。
  3. 只能在我允许的站点显示图片，其他站点都应该是无权访问，所谓防盗链机制。
  4. 图片不允许重复上传，毕竟硬盘不大。
  5. 因为只是自己使用，所以文件校验和图片大小校验等，可以放松。

综上，是我的需求。将他们详细分析一下，得到：

  1. 我需要一个 http 的接口服务，为 Post 接口，用来接受我的图片上传请求。
  2. 图片不校验大小，采用 nginx 的默认配置，图片文件格式校验采用后缀名。
  3. 该接口必须使用密钥验证，没有密钥的人无法访问。嗯，随机一个加密字符串就可以。
  4. 图片命名采用 md5 的方式，我不能保证自己不上传重复文件，但要保证服务器不会重复浪费磁盘空间。
  5. 因为要防盗链，虽然可以使用 nginx 来进行配置，但是，还是在 python 中实现比较好。
  6. 所以需要一个 get 接口，通过接口判断是否具有权限，返回不同图片。

开发图床

技术选型

首先，是使用 python3.6 为开发语言，这个是确定了的。但是如何开发这个，我花了5分钟考虑了一下。第一种呢，是全部自己实现。也不是不可以，但是我感觉耗费的时间有点不值当子。第二种呢，是使用公司目前采用的自研开发框架，这个是最顺手的，但问题是，我就是干这么个小活儿，不至于杀鸡用牛刀。嗯，那就采用第三方的框架，并迅速的锁定了 sanic 框架。

sanic 是一个比较新的，但是发展比较快的框架。其特征是速度非常快。据他们官方网站自己说，sanic 是最快的 python 框架。

sanic 官方文档地址：http://sanic.readthedocs.io/en/latest/

开干。

最终代码

#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
from sanic import Sanic
from sanic.response import json, text, file
import os, sys
import hashlib
import base64

app = Sanic()
# 图片存储目录
baseDir = sys.path[0] + '/image/'
# 校验 Token 写死就成，反正自己用的嘛
token = 'SheIsABeautifulGirl'
# 允许的域名列表
allowHost = [
            'localhost',
            'ilovethisword',
            'i.fengcss.com',
            'blog.csdn.net'
        ]

# 成功返回方法
def ok(data):
    if type(data) == list:
        return json({"data": {"list": data}, "status": 0})
    else:
        return json({"data": data, "status": 0})
# 失败返回方法
def fail(data):
    return json({"data": data, "status": 1})

# 获取图片后缀名
def getSuffix(filename):
    tempArr = filename.split('.')
    suffix = tempArr[-1]
    fileType = ['jpg', 'jpeg', 'gif', 'png']
    if len(tempArr) < 2:
        return 'error name'
    elif suffix not in fileType:
        return 'error type'
    else:
        return suffix
# 检查请求地址是否授权
def checkHost(host):
    for i in allowHost:
        if i in host:
            return True
    return False

# 上传图片文件接口
@app.route('/api/v1/upimg', methods=['POST'])
async def upimg(request):
    # 判断用户是否具有上传权限
    if request.headers.get('token') != token:
        return fail('您没有使用本服务的权限')
    image = request.files.get('file').body
    # 判断文件是否支持
    imageName = request.files.get('file').name
    imageSuffix = getSuffix(imageName)
    if 'error' in imageSuffix:
        return fail(imageSuffix)
    # 组织图片存储路径
    m1 = hashlib.md5()
    m1.update(image)
    md5Name = m1.hexdigest()

    # 用 md5 的前两位来建文件夹，防止单个文件夹下图片过多，又或者根目录下建立太多的文件夹
    saveDir = baseDir + md5Name[0:2] + '/'
    savePath = saveDir + md5Name[2:] + '.' + imageSuffix
    resPath = '/' + md5Name[0:2] + '/' + md5Name[2:] + '.' + imageSuffix

    # 如果文件夹不存在，就创建文件夹
    if not os.path.exists(saveDir):
        os.makedirs(saveDir)

    # 将文件写入到硬盘
    tempFile = open(savePath, 'wb')
    tempFile.write(image)
    tempFile.close()

    # 给客户端返回结果
    return ok({"path": resPath})

# 请求图片接口
@app.route('/', methods=['GET'])
async def img(request):
    # 判断是否为网站请求，否则就加上自定义的字符串（允许本地访问）
    host = request.headers.get('referer') or 'ilovethisword'
    # 判断请求接口是否带参数，否则加上自定义字符串（没有这个文件夹，返回404）
    args = request.args.get('path') or 'ilovemywife'
    # 拼接文件地址
    path = baseDir + args
    # 如果不在允许列表，则展示 401 图片
    if not checkHost(host):
        path = baseDir + '/7b/e49a54f761da42174d6121fa13e0b3.png'
    # 如果文件不存在，则展示 404 图片
    if not os.path.exists(path):
        path = baseDir + '/b4/74335c3944f42275e3caa13930a9b9.png'
    # 返回文件
    return await file(path)
# 启动服务
if __name__ == "__main__":
    app.run(host="127.0.0.1", port=8000)

部署图床

程序开发完成后，通过本地测试，确定没啥问题，决定上生产环境。（就是放到我的小破驴上去）

centos 7 系统 yum 源中 python 最高版本为 3.4 ，嗯，安装一下。不详述了，详情请查看How to Install Python 3.6.4 on CentOS 7

搞好环境之后，先找个地方把文件村上。这里我的存储位置是 /srv/py-tuchuang/run.py

配置 python 服务

首先，我们使用 chmod +x /srv/py-tuchuang/run.py 命令，让程序具有执行权限。

然后，我们用 vim /usr/lib/systemd/system/py-tuchuang.service 命令创建一个程序文件，然后输入以下内容

[Unit]
Description=Python image upload and download Service
After=network.target
Wants=network.target

[Service]
Type=simple
# 用户要配置有文件读写权限的，实在不行就root
User=root
Group=root
# 重要的就是下面的这句话
ExecStart=/srv/py-tuchuang/run.py

[Install]
WantedBy=multi-user.target

保存退出之后，我们使用下面的命令，分别设置立即启动和开机启动

# 启动服务
systemctl start py-tuchuang.service
# 将服务设置为开机启动
systemctl enable py-tuchuang.service

好，以上，我们的服务就已经配置好了。但是，我们程序中设定我们的服务是跑在 127.0.0.1 这个 ip 上的。所以，他只能跑在本地，你不能访问到，所以，我们需要一层代理。

配置 nginx 代理

nginx 是我服务器上现成的，毕竟服务器不是为了专门干这么一件破事儿嘛。

vim /etc/nginx/nginx.conf 编辑配置文件，在合适位置添加如下内容

server {
  server_name __YOU_DOMAIN_IN_HERE__;
  location / {
    proxy_pass  http://127.0.0.1:8000;
  }
}

然后保存退出，重启 nginx 服务。然后就可以啦。

使用图床

图床已经搭建好了，我们如何使用呢？方法有很多哦！

命令行上传图片

其实非常简单，我们在命令行中输入以下命令就可以上传文件了。

curl http://__YOU_DOMAIN_IN_HERE__/api/v1/upimg -F "file=@__UPFILE__PATH__" -H "token: SheIsABeautifulGirl" -v

将上面的 __YOU_DOMAIN_IN_HERE__ 替换为你的服务域名， __UPFILE__PATH__ 修改为你要上传的文件路径，然后，就可以上传文件啦~

再写一个脚本，完全可以啦。不过还是破烦。

自己写一个前端页面，专门用来上传

…省略

用 jquery 或者 vue 或者其他什么，写一个上传文件的前端脚本还是不难的，不过不在本文讨论范围之内。

配置 MWeb 编辑器图床

如下图配置 MWeb 编辑器，然后在里面写文章的时候，就太TM方便啦。随便一个屏幕截图，也不需要保存，直接到编辑器中粘贴，就会自动上传，并获取地址，变成 markdown 代码存好。

Python3 初学实践案例（14）打造一个私人图床服务器

本地图片也是非常方便，直接往编辑器中一拖动，就可以了。

好，这就是本人自己搞一个私人图床服务器的全部过程，希望对大家有所帮助。

本文由 FungLeo 原创，允许转载，但转载必须保留首发链接。

标签：

相关文章