mirror of
https://github.com/simon987/Simple-Incremental-Search-Tool.git
synced 2025-04-19 10:16:41 +00:00
136 lines
3.3 KiB
Python
136 lines
3.3 KiB
Python
from flask import Flask, render_template, send_file, request
|
|
import pysolr
|
|
import mimetypes
|
|
import requests
|
|
import json
|
|
from PIL import Image
|
|
import os
|
|
|
|
SOLR_URL = "http://localhost:8983/solr/test/"
|
|
|
|
solr = pysolr.Solr(SOLR_URL, timeout=10)
|
|
|
|
app = Flask(__name__)
|
|
|
|
#
|
|
# class Document:
|
|
# def __init__(self, doc_id, name, path, size, md5):
|
|
# self.doc_id = doc_id
|
|
# self.name = name
|
|
# self.path = path
|
|
# self.size = size
|
|
# self.md5 = md5
|
|
#
|
|
#
|
|
# class ImageDocument(Document):
|
|
# def __init__(self, doc_id, name, path, size, md5):
|
|
# super().__init__(doc_id, name, path, size, md5)
|
|
# self.type = "image"
|
|
#
|
|
#
|
|
# class AudioClipDocument(Document):
|
|
# def __init__(self, doc_id, name, path, size, md5):
|
|
# super().__init__(doc_id, name, path, size, md5)
|
|
# self.type = "audio"
|
|
#
|
|
#
|
|
# def get_document(id):
|
|
#
|
|
# response = requests.get(SOLR_URL + "get?id=" + id)
|
|
#
|
|
# return json.loads(response.text)["doc"]
|
|
#
|
|
#
|
|
# def make_thumb(doc):
|
|
# size = (1024, 1024)
|
|
#
|
|
# thumb_path = "thumbnails/" + doc["id"]
|
|
#
|
|
# if not os.path.exists(thumb_path):
|
|
#
|
|
# file_path = doc["path"][0] + "/" + doc["name"][0]
|
|
#
|
|
# if doc["width"][0] > size[0]:
|
|
#
|
|
# image = Image.open(file_path)
|
|
# image.thumbnail(size, Image.ANTIALIAS)
|
|
#
|
|
# if image.mode == "RGB":
|
|
# image.save(thumb_path, "JPEG")
|
|
# elif image.mode == "RGBA":
|
|
# image.save(thumb_path, "PNG")
|
|
# else:
|
|
# image = image.convert("RGB")
|
|
# image.save(thumb_path, "JPEG")
|
|
# else:
|
|
# print("Skipping thumbnail")
|
|
# os.symlink(file_path, thumb_path)
|
|
#
|
|
# return "thumbnails/" + doc["id"]
|
|
#
|
|
#
|
|
# @app.route("/search/")
|
|
# def search():
|
|
#
|
|
# query = request.args.get("query")
|
|
# page = int(request.args.get("page"))
|
|
# per_page = int(request.args.get("per_page"))
|
|
#
|
|
# results = solr.search(query, None, rows=per_page, start=per_page * page)
|
|
#
|
|
# docs = []
|
|
# for r in results:
|
|
#
|
|
# if "mime" in r:
|
|
# mime_type = r["mime"][0]
|
|
# else:
|
|
# mime_type = ""
|
|
#
|
|
# if mime_type.startswith("image"):
|
|
# docs.append(ImageDocument(r["id"], r["name"][0], r["path"][0], r["size"], r["md5"]))
|
|
#
|
|
# elif mime_type.startswith("audio"):
|
|
# docs.append(AudioClipDocument(r["id"], r["name"][0], r["path"][0], r["size"], r["md5"]))
|
|
#
|
|
# return render_template("search.html", docs=docs)
|
|
#
|
|
#
|
|
# @app.route("/")
|
|
# def index():
|
|
# return render_template("index.html")
|
|
#
|
|
#
|
|
# @app.route("/files/<id>/")
|
|
# def files(id):
|
|
#
|
|
# doc = get_document(id)
|
|
#
|
|
# if doc is not None:
|
|
# file_path = doc["path"][0] + "/" + doc["name"][0]
|
|
# return send_file(file_path, mimetype=mimetypes.guess_type(file_path)[0])
|
|
# else:
|
|
# return "File not found"
|
|
#
|
|
#
|
|
# @app.route("/thumbs/<doc_id>/")
|
|
# def thumbs(doc_id):
|
|
#
|
|
# doc = get_document(doc_id)
|
|
#
|
|
# if doc is not None:
|
|
#
|
|
# thumb_path = make_thumb(doc)
|
|
#
|
|
# return send_file("thumbnails/" + doc_id, mimetype=mimetypes.guess_type(thumb_path)[0])
|
|
# else:
|
|
# return "File not found"
|
|
|
|
|
|
@app.route("/")
|
|
def tmp_route():
|
|
return "test"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run("0.0.0.0", 8080)
|