diff --git a/scraper/server/__main__.py b/scraper/server/__main__.py index b34f18f..1a4c756 100644 --- a/scraper/server/__main__.py +++ b/scraper/server/__main__.py @@ -2,7 +2,7 @@ from scraper.config import config from flask import Flask, Response, request, render_template, url_for from pathlib import Path import py34 -from .block import BLOCK_SIZE, list_blocks, load_blocks, load_block, load_block_stats, save_block, loads as parse_block +from .block import BLOCK_SIZE, list_blocks, load_blocks, load_block, load_block_stats, save_block, loads as parse_block, enttype2ext from .job import assign_job, working_on, any_job, jobs @@ -107,7 +107,7 @@ def get_image(post_id: int = None): return Response( status = 307, headers = { - "Location": str(py34.url.ImageURL(entry.dir, entry.image, entry.ext)) + "Location": str(py34.url.ImageURL(entry.dir, entry.image, enttype2ext(entry.type))) } )