KankenOnline/kanken_online/__init__.py

import json
import os
from flask import Flask, redirect, render_template, g, request, session, url_for
from pathlib import Path

from kanken_online.api import get_kanji_by_character
from kanken_online.database import get_database
from .auth import login_required


DATABASE_NAME = "kanken_online.sqlite"
JITEN_DB_NAME = "kanken.db"
def create_app(test_config=None):
    app = Flask(__name__, instance_relative_config=True)
    app.config.from_mapping(
        SECRET_KEY="dev",
        DATABASE=str(Path(app.instance_path) / DATABASE_NAME),
        JITEN_DB=str(Path(app.instance_path) / JITEN_DB_NAME)
    )

    if test_config is None:
        app.config.from_pyfile("config.py", silent=True)
    else:
        app.config.from_mapping(test_config)
    
    # Ensure instance path exists
    os.makedirs(app.instance_path, exist_ok=True)

    @app.route("/hello")
    def hello():
        return "Hello, World!"
    
    @app.route("/")
    def index():
        return render_template("index.html")
    
    @app.route("/about")
    def about_page():
        return render_template("about.html")

    def update_settings(form):
        db = get_database()

        if session["user_id"]:
            # Set values in the database
            settings = db.execute("SELECT * FROM user_settings WHERE user_id = ?", 
                        (session["user_id"],)
            ).fetchone()
            if settings:
                db.execute("UPDATE user_settings SET lang = ?, theme = ? WHERE user_id = ?",
                           (form["language"], form["theme"], session["user_id"]))
            else:
                db.execute("INSERT INTO user_settings VALUES (?, ?, ?)",
                           (session["user_id"], form["language"], form["theme"]))
            db.commit()
        
        # Set values directly in the session
        session["language"] = form["language"]
        session["theme"] = form["theme"]

        return redirect("/options")
    
    @app.route("/options", methods=["GET", "POST"])
    def options():
        if request.method == "GET":
            return render_template("options.html")
        else:
            return update_settings(request.form)
    
    @app.route("/kanji/<kanji>")
    def kanji_page(kanji: str):
        kanji_obj = get_kanji_by_character(kanji)

        class Kanji():
            pass

        out = Kanji()
        out.character = kanji_obj.character
        out.is_joyo = "常用" if kanji_obj.level not in ["1", "準1"] else "表外"
        out.level = kanji_obj.level
        out.strokes = kanji_obj.stroke_count
        out.radical = kanji_obj.radical
        out.added_strokes = kanji_obj.radical_added_stroke_count
        out.goon = " ・ ".join(str(obj.reading) for obj in kanji_obj.goon)
        out.kanon = ""
        out.toon = ""
        out.soon = ""
        out.kanyoon = ""
        out.kun = ""
        out.meanings = kanji_obj.meanings
        out.glyph_origin = kanji_obj.glyph_origin
        

        return render_template("kanji.html", kanji=out)

    @app.route("/kotoba/<kotoba>")
    def kotoba_page(kotoba: str):
        return render_template("kotoba.html", kotoba=kotoba)
    
    from . import database
    database.initialize_app(app)

    from . import auth, api, forum, search, indices
    app.register_blueprint(auth.blueprint)
    app.register_blueprint(api.blueprint)
    app.register_blueprint(forum.blueprint)
    app.register_blueprint(search.blueprint)
    app.register_blueprint(indices.blueprint)

    from . import lang
    lang.add_translation_commands(app)

    @app.route("/translations", methods=["GET", "POST"])
    def strings_translation():
        if request.method == "GET":
            strings = {}
            for language, language_data in lang.LANGUAGES.items():
                strings[language] = {
                    string: translation for string, translation in language_data.items()
                }

            return render_template("translations.html", strings=strings)
        else:
            strings = {}
            for i, language in enumerate(lang.LANGUAGES):
                strings[language] = {
                    string: request.form.getlist(string)[i] for string in request.form
                }
                with open(Path(app.root_path, "static", "lang", f"{language}.json"), mode="w") as f:
                    json.dump(strings[language], f, ensure_ascii=False, indent=0)
            
            lang.update_languages()
            return redirect("/translations")

    @app.route("/data")
    def data_page():
        return render_template("data.html")

    # def use_english(text_id: str):
    #     return lang.localize(text_id, lang.JAPANESE)
    app.jinja_env.globals.update(localize=lang.localize)
    

    return app