diff --git a/tests/test_site/__init__.py b/tests/test_site/__init__.py index 32a90ab..941af29 100644 --- a/tests/test_site/__init__.py +++ b/tests/test_site/__init__.py @@ -91,9 +91,9 @@ def create_app(is_testing: bool = False) -> Flask: return user.id can_view: t.Callable[[], bool] = lambda: auth.current_user() is not None \ - and auth.current_user().username in ["viewer", "editor"] + and auth.current_user().username in ["viewer", "editor", "editor2"] can_edit: t.Callable[[], bool] = lambda: auth.current_user() is not None \ - and auth.current_user().username == "editor" + and auth.current_user().username in ["editor", "editor2"] accounting.init_app(app, user_utils=UserUtils(), can_view_func=can_view, can_edit_func=can_edit) @@ -106,7 +106,7 @@ def init_db_command() -> None: """Initializes the database.""" db.create_all() from .auth import User - for username in ["viewer", "editor", "nobody"]: + for username in ["viewer", "editor", "editor2", "nobody"]: if User.query.filter(User.username == username).first() is None: db.session.add(User(username=username)) db.session.commit() diff --git a/tests/test_site/auth.py b/tests/test_site/auth.py index 44fbeab..de8374c 100644 --- a/tests/test_site/auth.py +++ b/tests/test_site/auth.py @@ -58,7 +58,8 @@ def login() -> redirect: :return: The redirection to the home page. """ - if request.form.get("username") not in ["viewer", "editor", "nobody"]: + if request.form.get("username") not in ["viewer", "editor", "editor2", + "nobody"]: return redirect(url_for("auth.login")) session["user"] = request.form.get("username") return redirect(url_for("home.home")) diff --git a/tests/test_site/templates/login.html b/tests/test_site/templates/login.html index 0b9db75..edf20eb 100644 --- a/tests/test_site/templates/login.html +++ b/tests/test_site/templates/login.html @@ -29,6 +29,7 @@ First written: 2023/1/27 + diff --git a/tests/test_site/translations/zh_Hant/LC_MESSAGES/messages.po b/tests/test_site/translations/zh_Hant/LC_MESSAGES/messages.po index 2bc323f..51553cf 100644 --- a/tests/test_site/translations/zh_Hant/LC_MESSAGES/messages.po +++ b/tests/test_site/translations/zh_Hant/LC_MESSAGES/messages.po @@ -9,8 +9,8 @@ msgid "" msgstr "" "Project-Id-Version: Mia! Accounting Flask Demonstration 0.0.0\n" "Report-Msgid-Bugs-To: imacat@mail.imacat.idv.tw\n" -"POT-Creation-Date: 2023-01-28 13:42+0800\n" -"PO-Revision-Date: 2023-01-28 13:42+0800\n" +"POT-Creation-Date: 2023-02-06 23:25+0800\n" +"PO-Revision-Date: 2023-02-06 23:26+0800\n" "Last-Translator: imacat \n" "Language: zh_Hant\n" "Language-Team: zh_Hant \n" @@ -51,6 +51,10 @@ msgid "Editor" msgstr "記帳者" #: tests/test_site/templates/login.html:32 +msgid "Editor2" +msgstr "記帳者2" + +#: tests/test_site/templates/login.html:33 msgid "Nobody" msgstr "沒有權限者"