93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
import csv
|
|
import os
|
|
import tomllib
|
|
|
|
import flask_apscheduler
|
|
from flask import Flask
|
|
from flask_bcrypt import Bcrypt
|
|
from flask_login import LoginManager
|
|
from flask_migrate import Migrate
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_wtf.csrf import CSRFProtect
|
|
from whoosh.fields import *
|
|
from whoosh.index import create_in
|
|
from whoosh.qparser import QueryParser
|
|
|
|
from application.csvparser import CsvParser
|
|
|
|
db = SQLAlchemy()
|
|
migrate = Migrate()
|
|
bcrypt = Bcrypt()
|
|
login_manager = LoginManager()
|
|
|
|
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"] = os.urandom(24)
|
|
APP.config["UPLOAD_FOLDER"] = "tmpupload"
|
|
APP.config["IMAGE_FOLDER"] = "static/images"
|
|
|
|
APP.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///libraryusers.db"
|
|
APP.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = True
|
|
|
|
APP.config["LIBRARY_FILENAME"] = settings["libaryfilename"]
|
|
APP.config["TITLE"] = settings["title"]
|
|
|
|
csrf = CSRFProtect(APP)
|
|
csrf.init_app(APP)
|
|
login_manager.init_app(APP)
|
|
db.init_app(APP)
|
|
migrate.init_app(APP, db, render_as_batch=True)
|
|
bcrypt.init_app(APP)
|
|
|
|
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"])
|
|
|
|
@APP.context_processor
|
|
def inject_title():
|
|
return dict(title=APP.config["TITLE"])
|
|
|
|
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 = {}
|
|
if os.path.isfile("settings_development.toml"):
|
|
with open("settings_development.toml", "rb") as settings_file:
|
|
return tomllib.load(settings_file)
|
|
with open("settings.toml", "rb") as settings_file:
|
|
return tomllib.load(settings_file)
|