fix flatpagesroot

This commit is contained in:
2021-11-20 08:30:51 +00:00
parent 126e4a070d
commit b207d85629
2 changed files with 10 additions and 6 deletions

View File

@@ -2,7 +2,7 @@
from flask import Flask, Blueprint,render_template, send_from_directory,jsonify, url_for from flask import Flask, Blueprint,render_template, send_from_directory,jsonify, url_for
from flatpages_index import FlatPagesIndex from flatpages_index import FlatPagesIndex
import flatpages_index import flatpages_index
import os
def create_bp(app): def create_bp(app):
flatpages = FlatPagesIndex(app) flatpages = FlatPagesIndex(app)
@@ -32,8 +32,8 @@ def create_bp(app):
return render_template(page["template"], post=page, return render_template(page["template"], post=page,
pth=page["dirpath"]) pth=page["dirpath"])
if os.path.exists(os.path.join(FLATPAGES_ROOT,name)): if os.path.exists(os.path.join(app.config['FLATPAGES_ROOT'],name)):
return send_from_directory(FLATPAGES_ROOT,name) return send_from_directory(app.config['FLATPAGES_ROOT'],name)
else: else:
return send_from_directory('static',name) return send_from_directory('static',name)
@@ -49,8 +49,8 @@ def create_bp(app):
# page.links.file_url=lambda n: url_for('intern.post', name=n) # page.links.file_url=lambda n: url_for('intern.post', name=n)
return jsonify(page=dict(page)) return jsonify(page=dict(page))
if os.path.exists(u'{}/{}'.format(FLATPAGES_ROOT,path)): if os.path.exists(u'{}/{}'.format(app.config['FLATPAGES_ROOT'],path)):
return send_from_directory(FLATPAGES_ROOT,path) return send_from_directory(app.config['FLATPAGES_ROOT'],path)
else: else:
return send_from_directory('static',path) return send_from_directory('static',path)
return page_blueprint return page_blueprint

View File

@@ -30,3 +30,7 @@ def test_links_in_index(index_html):
def test_text_in_index(index_html): def test_text_in_index(index_html):
assert 'asdfasdfasdf' in index_html assert 'asdfasdfasdf' in index_html
def test_jpg(client):
resp = client.get('/topic1/test.jpg/')
assert resp.status_code == 200