You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

使用flask_testing测试Flask REST API时静态文件无法访问的解决方法

问题:Flask-RESTX API测试时PDF生成找不到静态资源的解决方法

问题场景

使用Flask-RESTX开发Flask REST API,部分接口生成PDF文档时依赖static/目录下的字体文件和图片。API运行时一切正常,但测试调用相同接口时,出现文件找不到的错误:

OSError: Cannot open resource "static/logo.png"

目录结构如下:

/src
  /application
    /main
      app.py
    /test
      test.py
  /static
    logo.png

当前应用中访问资源的方式为硬编码相对路径:static/logo.png

核心原因

硬编码的相对路径依赖当前工作目录:

  • 开发时通常在src/application/main目录启动服务,相对路径static/logo.png实际指向../../static/logo.png,能正确找到资源;
  • 测试时可能在src/application/test目录运行测试脚本,此时相对路径static/logo.png会指向../static/logo.png,而实际静态资源在src/static,因此找不到文件。

优雅解决方案:使用绝对路径访问静态资源

不需要编写环境切换逻辑,通过计算项目根目录的方式生成静态资源的绝对路径,确保所有环境下路径一致。

步骤1:在app.py中统一静态路径配置

利用__file__变量计算项目根目录,再拼接静态文件夹路径:

from flask import Flask
from flask_restx import Api, Resource
import os

# 计算项目根目录(src/文件夹的绝对路径)
PROJECT_ROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
# 静态资源文件夹的绝对路径
STATIC_DIR = os.path.join(PROJECT_ROOT, 'static')

# 初始化Flask应用(可选:指定static_folder属性,符合Flask规范)
app = Flask(__name__, static_folder=STATIC_DIR)
api = Api(app)

@api.route('/generate-pdf')
class PDFGenerator(Resource):
    def get(self):
        # 构建具体资源的绝对路径
        logo_path = os.path.join(STATIC_DIR, 'logo.png')
        # 后续PDF生成逻辑使用logo_path即可
        return {"status": "success", "logo_path": logo_path}

if __name__ == '__main__':
    app.run(debug=True)

步骤2:测试代码中复用统一路径

测试脚本直接导入应用中定义的STATIC_DIR,确保路径一致:

import pytest
from application.main.app import app, STATIC_DIR
import os

def test_pdf_resource_exists():
    # 验证静态资源是否存在
    assert os.path.exists(os.path.join(STATIC_DIR, 'logo.png'))

def test_pdf_generation_api():
    client = app.test_client()
    response = client.get('/generate-pdf')
    assert response.status_code == 200
    assert 'logo_path' in response.json

额外优化:利用Flask内置属性

如果初始化Flask应用时指定了static_folder,可以直接使用app.static_folder属性获取静态目录的绝对路径,无需手动计算:

logo_path = os.path.join(app.static_folder, 'logo.png')

这种方式更符合Flask的设计规范,同时保证路径的一致性。

内容的提问来源于stack exchange,提问作者Denis Vitez

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.06 21:25:35