csv-library-website/library/app.py

import csv
import os
import tomllib

import flask_apscheduler
from flask import Flask
from whoosh.fields import *
from whoosh.index import create_in
from whoosh.qparser import QueryParser

from application.csvparser import CsvParser

SCRIPT_DIR = os.path.dirname(__file__)
DATA_DIR = os.path.abspath(os.path.join(SCRIPT_DIR, "data"))


def create_app():
    settings = settings_from_file()

    APP = Flask(__name__, static_folder="static")
    APP.config["SECRET_KEY"] = "ty4425hk54a21eee5719b9s9df7sdfklx"
    APP.config["UPLOAD_FOLDER"] = "tmpupload"
    APP.config["LIBRARY_FILENAME"] = settings["libaryfilename"]
    APP.config["IMAGE_FOLDER"] = "static/images"
    APP.config["TITLE"] = settings["title"]
    scheduler = flask_apscheduler.APScheduler()
    scheduler.api_enabled = False
    scheduler.init_app(APP)
    scheduler.start()
    index_books(APP.config["LIBRARY_FILENAME"], APP.config["IMAGE_FOLDER"])

    @scheduler.task("interval", id="update", minutes=10)
    def update():
        index_books(APP.config["LIBRARY_FILENAME"], APP.config["IMAGE_FOLDER"])

    return APP


def get_app():
    APP = Flask(__name__, static_folder="static")
    return APP


def index_books(filename: str, image_folder: str):
    csvparser = CsvParser(filename, image_folder)
    filename = os.path.join(DATA_DIR, csvparser.csv_file)
    schema = Schema(
        title=TEXT(stored=True), path=ID(stored=True), content=TEXT
    )
    ix = create_in(DATA_DIR, schema)
    writer = ix.writer()
    with open(filename, "r", encoding="utf_8_sig") as libcsv:
        csv_as_dict = csv.DictReader(libcsv)
        for row in csv_as_dict:
            rowcontent = csvparser.concatenate_csv_row(row)
            writer.add_document(title=row["Id"], path="/a", content=rowcontent)
    writer.commit()


def settings_from_file():
    settings = {}
    with open("settings.toml", "rb") as settings_file:
        settings = tomllib.load(settings_file)
    return settings