Compare commits
No commits in common. "main" and "feature/pricelist" have entirely different histories.
main
...
feature/pr
|
@ -67,8 +67,7 @@ instance/
|
||||||
|
|
||||||
# Sphinx documentation
|
# Sphinx documentation
|
||||||
docs/_build/
|
docs/_build/
|
||||||
# pdoc
|
docs/
|
||||||
docs/html
|
|
||||||
|
|
||||||
# PyBuilder
|
# PyBuilder
|
||||||
target/
|
target/
|
||||||
|
@ -122,7 +121,6 @@ dmypy.json
|
||||||
*.swo
|
*.swo
|
||||||
.vscode/
|
.vscode/
|
||||||
*.log
|
*.log
|
||||||
.fleet/
|
|
||||||
|
|
||||||
data/
|
data/
|
||||||
|
|
||||||
|
|
|
@ -1,6 +0,0 @@
|
||||||
pipeline:
|
|
||||||
lint:
|
|
||||||
image: python:slim
|
|
||||||
commands:
|
|
||||||
- pip install black
|
|
||||||
- black --check --line-length 120 --target-version=py39 .
|
|
|
@ -1,19 +0,0 @@
|
||||||
pipeline:
|
|
||||||
install:
|
|
||||||
image: python:${PYTHON}-slim
|
|
||||||
commands:
|
|
||||||
- python -m venv --clear venv
|
|
||||||
- export PATH=venv/bin:$PATH
|
|
||||||
- python -m pip install --upgrade pip
|
|
||||||
- pip install -v ".[tests]"
|
|
||||||
test:
|
|
||||||
image: python:${PYTHON}-slim
|
|
||||||
commands:
|
|
||||||
- export PATH=venv/bin:$PATH
|
|
||||||
- python -m pytest
|
|
||||||
|
|
||||||
|
|
||||||
matrix:
|
|
||||||
PYTHON:
|
|
||||||
- 3.10
|
|
||||||
- 3.9
|
|
116
README.md
116
README.md
|
@ -1,116 +0,0 @@
|
||||||
# Flaschengeist
|
|
||||||
![status-badge](https://ci.os-sc.org/api/badges/Flaschengeist/flaschengeist/status.svg)
|
|
||||||
|
|
||||||
This is the backend of the Flaschengeist.
|
|
||||||
|
|
||||||
## Installation
|
|
||||||
### Requirements
|
|
||||||
- `mysql` or `mariadb`
|
|
||||||
- maybe `libmariadb` development files[1]
|
|
||||||
- python 3.9+
|
|
||||||
- pip 21.0+
|
|
||||||
|
|
||||||
*[1] By default Flaschengeist uses mysql as database backend, if you are on Windows Flaschengeist uses `PyMySQL`, but on
|
|
||||||
Linux / Mac the faster `mysqlclient` is used, if it is not already installed installing from pypi requires the
|
|
||||||
development files for `libmariadb` to be present on your system.*
|
|
||||||
|
|
||||||
### Install python files
|
|
||||||
It is recommended to upgrade pip to the latest version before installing:
|
|
||||||
|
|
||||||
python -m pip install --upgrade pip
|
|
||||||
|
|
||||||
Default installation with *mariadb*/*mysql* support:
|
|
||||||
|
|
||||||
pip3 install --user ".[mysql]"
|
|
||||||
|
|
||||||
or with ldap support
|
|
||||||
|
|
||||||
pip3 install --user ".[ldap]"
|
|
||||||
|
|
||||||
or if you want to also run the tests:
|
|
||||||
|
|
||||||
pip3 install --user ".[ldap,tests]"
|
|
||||||
|
|
||||||
You will also need a MySQL driver, by default one of this is installed:
|
|
||||||
- `mysqlclient` (non Windows)
|
|
||||||
- `PyMySQL` (on Windows)
|
|
||||||
|
|
||||||
#### Hint on MySQL driver on Windows:
|
|
||||||
If you want to use `mysqlclient` instead of `PyMySQL` (performance?) you have to follow [this guide](https://www.radishlogic.com/coding/python-3/installing-mysqldb-for-python-3-in-windows/)
|
|
||||||
|
|
||||||
### Install database
|
|
||||||
The user needs to have full permissions to the database.
|
|
||||||
If not you need to create user and database manually do (or similar on Windows):
|
|
||||||
|
|
||||||
(
|
|
||||||
echo "CREATE DATABASE flaschengeist;"
|
|
||||||
echo "CREATE USER 'flaschengeist'@'localhost' IDENTIFIED BY 'flaschengeist';"
|
|
||||||
echo "GRANT ALL PRIVILEGES ON flaschengeist.* TO 'flaschengeist'@'localhost';"
|
|
||||||
echo "FLUSH PRIVILEGES;"
|
|
||||||
) | sudo mysql
|
|
||||||
|
|
||||||
Then you can install the database tables, this will update all tables from core + all enabled plugins.
|
|
||||||
And also install all enabled plugins:
|
|
||||||
|
|
||||||
$ flaschengeist install
|
|
||||||
|
|
||||||
*Hint:* To only install the database tables, or upgrade the database after plugins or core are updated later
|
|
||||||
you can use this command:
|
|
||||||
|
|
||||||
$ flaschengeist db upgrade heads
|
|
||||||
|
|
||||||
|
|
||||||
## Plugins
|
|
||||||
To only upgrade one plugin (for example the `events` plugin):
|
|
||||||
|
|
||||||
$ flaschengeist db upgrade events@head
|
|
||||||
|
|
||||||
## Configuration
|
|
||||||
Configuration is done within the a `flaschengeist.toml`file, you can copy the one located inside the module path
|
|
||||||
(where flaschegeist is installed) or create an empty one and place it inside either:
|
|
||||||
1. `~/.config/`
|
|
||||||
2. A custom path and set environment variable `FLASCHENGEIST_CONF`
|
|
||||||
|
|
||||||
Uncomment and change at least all the database parameters!
|
|
||||||
|
|
||||||
#### CRON
|
|
||||||
Some functionality used by some plugins rely on regular updates,
|
|
||||||
but as flaschengeists works as an WSGI app it can not controll when it gets called.
|
|
||||||
|
|
||||||
So you have to configure one of the following options to call flaschengeists CRON tasks:
|
|
||||||
|
|
||||||
1. Passive Web-CRON: Every time an users calls flaschengeist a task is scheduled (**NOT RECOMMENDED**)
|
|
||||||
- Pros: No external configuration needed
|
|
||||||
- Cons: Slower user experience, no guaranteed execution time of tasks
|
|
||||||
2. Active Web-CRON: You configure a webworker to call `<flaschengeist>/cron`
|
|
||||||
- Pros: Guaranteed execution interval, no impact on user experience (at least if you do not limit wsgi worker threads)
|
|
||||||
- Cons: Uses one of the webserver threads while executing
|
|
||||||
|
|
||||||
### Run
|
|
||||||
Flaschengeist provides a CLI, based on the flask CLI, respectivly called `flaschengeist`.
|
|
||||||
|
|
||||||
⚠️ When using the CLI for running Flaschengeist, please note that logging will happen as configured,
|
|
||||||
with the difference of the main logger will be forced to output to `stderr` and the logging level
|
|
||||||
of the CLI will override the logging level you have configured for the main logger.
|
|
||||||
|
|
||||||
$ flaschengeist run
|
|
||||||
|
|
||||||
or with debug messages:
|
|
||||||
|
|
||||||
$ flaschengeist run --debug
|
|
||||||
|
|
||||||
This will run the backend on http://localhost:5000
|
|
||||||
|
|
||||||
## Tests
|
|
||||||
$ pip install '.[test]'
|
|
||||||
$ pytest
|
|
||||||
Run with coverage report:
|
|
||||||
|
|
||||||
$ coverage run -m pytest
|
|
||||||
$ coverage report
|
|
||||||
Or with html output (open `htmlcov/index.html` in a browser):
|
|
||||||
|
|
||||||
$ coverage html
|
|
||||||
|
|
||||||
## Development
|
|
||||||
Please refer to our [development wiki](https://flaschengeist.dev/Flaschengeist/flaschengeist/wiki/Development).
|
|
|
@ -1,57 +0,0 @@
|
||||||
# Plugin Development
|
|
||||||
|
|
||||||
## File Structure
|
|
||||||
|
|
||||||
- your_plugin/
|
|
||||||
- __init__.py
|
|
||||||
- ...
|
|
||||||
- migrations/ (optional)
|
|
||||||
- ...
|
|
||||||
- setup.cfg
|
|
||||||
|
|
||||||
The basic layout of a plugin is quite simple, you will only need the `setup.cfg` or `setup.py` and
|
|
||||||
the package containing your plugin code, at lease a `__init__.py` file with your `Plugin` class.
|
|
||||||
|
|
||||||
If you use custom database tables you need to provide a `migrations` directory within your package,
|
|
||||||
see next section.
|
|
||||||
|
|
||||||
## Database Tables / Migrations
|
|
||||||
To allow upgrades of installed plugins, the database is versioned and handled
|
|
||||||
through [Alembic](https://alembic.sqlalchemy.org/en/latest/index.html) migrations.
|
|
||||||
Each plugin, which uses custom database tables, is represented as an other base.
|
|
||||||
So you could simply follow the Alembic tutorial on [how to work with multiple bases](https://alembic.sqlalchemy.org/en/latest/branches.html#creating-a-labeled-base-revision).
|
|
||||||
|
|
||||||
A quick overview on how to work with migrations for your plugin:
|
|
||||||
|
|
||||||
$ flaschengeist db revision -m "Create my super plugin" \
|
|
||||||
--head=base --branch-label=myplugin_name --version-path=your/plugin/migrations
|
|
||||||
|
|
||||||
This would add a new base named `myplugin_name`, which should be the same as the pypi name of you plugin.
|
|
||||||
If your tables depend on an other plugin or a specific base version you could of cause add
|
|
||||||
|
|
||||||
--depends-on=VERSION
|
|
||||||
|
|
||||||
or
|
|
||||||
|
|
||||||
--depends-on=other_plugin
|
|
||||||
|
|
||||||
|
|
||||||
### Plugin Removal and Database Tables
|
|
||||||
As generic downgrades are most often hard to write, your plugin is not required to provide such functionallity.
|
|
||||||
For Flaschengeist only instable versions provide meaningful downgrade migrations down to the latest stable version.
|
|
||||||
|
|
||||||
So this means if you do not provide downgrades you must at lease provide a series of migrations toward removal of
|
|
||||||
the database tables in case the users wants to delete the plugin.
|
|
||||||
|
|
||||||
(base) ----> 1.0 <----> 1.1 <----> 1.2
|
|
||||||
|
|
|
||||||
--> removal
|
|
||||||
|
|
||||||
After the removal step the database is stamped to to "remove" your
|
|
||||||
|
|
||||||
## Useful Hooks
|
|
||||||
There are some predefined hooks, which might get handy for you.
|
|
||||||
|
|
||||||
For more information, please refer to
|
|
||||||
- `flaschengeist.utils.hook.HookBefore` and
|
|
||||||
- `flaschengeist.utils.hook.HookAfter`
|
|
|
@ -1,9 +1,14 @@
|
||||||
"""Flaschengeist"""
|
""" Server-package
|
||||||
|
|
||||||
|
Initialize app, CORS, database and add it to the application.
|
||||||
|
|
||||||
|
"""
|
||||||
import logging
|
import logging
|
||||||
from importlib.metadata import version
|
import pkg_resources
|
||||||
|
from pathlib import Path
|
||||||
|
from werkzeug.local import LocalProxy
|
||||||
|
|
||||||
__version__ = version("flaschengeist")
|
__version__ = pkg_resources.get_distribution("flaschengeist").version
|
||||||
__pdoc__ = {}
|
_module_path = Path(__file__).parent
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger: logging.Logger = LocalProxy(lambda: logging.getLogger(__name__))
|
||||||
__pdoc__["logger"] = "Flaschengeist's logger instance (`werkzeug.local.LocalProxy`)"
|
|
||||||
|
|
|
@ -1,5 +0,0 @@
|
||||||
from pathlib import Path
|
|
||||||
|
|
||||||
|
|
||||||
alembic_migrations_path = str(Path(__file__).resolve().parent / "migrations")
|
|
||||||
alembic_script_path = str(Path(__file__).resolve().parent)
|
|
|
@ -1,53 +0,0 @@
|
||||||
# A generic, single database configuration.
|
|
||||||
# No used by flaschengeist
|
|
||||||
|
|
||||||
[alembic]
|
|
||||||
# template used to generate migration files
|
|
||||||
# file_template = %%(rev)s_%%(slug)s
|
|
||||||
|
|
||||||
# set to 'true' to run the environment during
|
|
||||||
# the 'revision' command, regardless of autogenerate
|
|
||||||
# revision_environment = false
|
|
||||||
|
|
||||||
version_path_separator = os
|
|
||||||
version_locations = %(here)s/migrations
|
|
||||||
|
|
||||||
# Logging configuration
|
|
||||||
[loggers]
|
|
||||||
keys = root,sqlalchemy,alembic,flask_migrate
|
|
||||||
|
|
||||||
[handlers]
|
|
||||||
keys = console
|
|
||||||
|
|
||||||
[formatters]
|
|
||||||
keys = generic
|
|
||||||
|
|
||||||
[logger_root]
|
|
||||||
level = WARN
|
|
||||||
handlers = console
|
|
||||||
qualname =
|
|
||||||
|
|
||||||
[logger_sqlalchemy]
|
|
||||||
level = WARN
|
|
||||||
handlers =
|
|
||||||
qualname = sqlalchemy.engine
|
|
||||||
|
|
||||||
[logger_alembic]
|
|
||||||
level = INFO
|
|
||||||
handlers =
|
|
||||||
qualname = alembic
|
|
||||||
|
|
||||||
[logger_flask_migrate]
|
|
||||||
level = INFO
|
|
||||||
handlers =
|
|
||||||
qualname = flask_migrate
|
|
||||||
|
|
||||||
[handler_console]
|
|
||||||
class = StreamHandler
|
|
||||||
args = (sys.stderr,)
|
|
||||||
level = NOTSET
|
|
||||||
formatter = generic
|
|
||||||
|
|
||||||
[formatter_generic]
|
|
||||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
|
||||||
datefmt = %H:%M:%S
|
|
|
@ -1,74 +0,0 @@
|
||||||
import logging
|
|
||||||
from logging.config import fileConfig
|
|
||||||
from pathlib import Path
|
|
||||||
from flask import current_app
|
|
||||||
from alembic import context
|
|
||||||
|
|
||||||
# this is the Alembic Config object, which provides
|
|
||||||
# access to the values within the .ini file in use.
|
|
||||||
config = context.config
|
|
||||||
|
|
||||||
# Interpret the config file for Python logging.
|
|
||||||
# This line sets up loggers basically.
|
|
||||||
fileConfig(Path(config.get_main_option("script_location")) / config.config_file_name.split("/")[-1])
|
|
||||||
logger = logging.getLogger("alembic.env")
|
|
||||||
|
|
||||||
config.set_main_option("sqlalchemy.url", str(current_app.extensions["migrate"].db.get_engine().url).replace("%", "%%"))
|
|
||||||
target_metadata = current_app.extensions["migrate"].db.metadata
|
|
||||||
|
|
||||||
|
|
||||||
def run_migrations_offline():
|
|
||||||
"""Run migrations in 'offline' mode.
|
|
||||||
|
|
||||||
This configures the context with just a URL
|
|
||||||
and not an Engine, though an Engine is acceptable
|
|
||||||
here as well. By skipping the Engine creation
|
|
||||||
we don't even need a DBAPI to be available.
|
|
||||||
|
|
||||||
Calls to context.execute() here emit the given string to the
|
|
||||||
script output.
|
|
||||||
|
|
||||||
"""
|
|
||||||
url = config.get_main_option("sqlalchemy.url")
|
|
||||||
context.configure(url=url, target_metadata=target_metadata, literal_binds=True)
|
|
||||||
|
|
||||||
with context.begin_transaction():
|
|
||||||
context.run_migrations()
|
|
||||||
|
|
||||||
|
|
||||||
def run_migrations_online():
|
|
||||||
"""Run migrations in 'online' mode.
|
|
||||||
|
|
||||||
In this scenario we need to create an Engine
|
|
||||||
and associate a connection with the context.
|
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
# this callback is used to prevent an auto-migration from being generated
|
|
||||||
# when there are no changes to the schema
|
|
||||||
# reference: http://alembic.zzzcomputing.com/en/latest/cookbook.html
|
|
||||||
def process_revision_directives(context, revision, directives):
|
|
||||||
if getattr(config.cmd_opts, "autogenerate", False):
|
|
||||||
script = directives[0]
|
|
||||||
if script.upgrade_ops.is_empty():
|
|
||||||
directives[:] = []
|
|
||||||
logger.info("No changes in schema detected.")
|
|
||||||
|
|
||||||
connectable = current_app.extensions["migrate"].db.get_engine()
|
|
||||||
|
|
||||||
with connectable.connect() as connection:
|
|
||||||
context.configure(
|
|
||||||
connection=connection,
|
|
||||||
target_metadata=target_metadata,
|
|
||||||
process_revision_directives=process_revision_directives,
|
|
||||||
**current_app.extensions["migrate"].configure_args,
|
|
||||||
)
|
|
||||||
|
|
||||||
with context.begin_transaction():
|
|
||||||
context.run_migrations()
|
|
||||||
|
|
||||||
|
|
||||||
if context.is_offline_mode():
|
|
||||||
run_migrations_offline()
|
|
||||||
else:
|
|
||||||
run_migrations_online()
|
|
|
@ -1,154 +0,0 @@
|
||||||
"""Initial core db
|
|
||||||
|
|
||||||
Revision ID: 20482a003db8
|
|
||||||
Revises:
|
|
||||||
Create Date: 2022-08-25 15:13:34.900996
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
import flaschengeist
|
|
||||||
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = "20482a003db8"
|
|
||||||
down_revision = None
|
|
||||||
branch_labels = ("flaschengeist",)
|
|
||||||
depends_on = None
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.create_table(
|
|
||||||
"image",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("filename", sa.String(length=255), nullable=False),
|
|
||||||
sa.Column("mimetype", sa.String(length=127), nullable=False),
|
|
||||||
sa.Column("thumbnail", sa.String(length=255), nullable=True),
|
|
||||||
sa.Column("path", sa.String(length=255), nullable=True),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_image")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"plugin",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=127), nullable=False),
|
|
||||||
sa.Column("version", sa.String(length=30), nullable=False),
|
|
||||||
sa.Column("enabled", sa.Boolean(), nullable=True),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_plugin")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"role",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=True),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_role")),
|
|
||||||
sa.UniqueConstraint("name", name=op.f("uq_role_name")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"permission",
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=True),
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("plugin", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["plugin"], ["plugin.id"], name=op.f("fk_permission_plugin_plugin")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_permission")),
|
|
||||||
sa.UniqueConstraint("name", name=op.f("uq_permission_name")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"plugin_setting",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("plugin", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.Column("name", sa.String(length=127), nullable=False),
|
|
||||||
sa.Column("value", sa.PickleType(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["plugin"], ["plugin.id"], name=op.f("fk_plugin_setting_plugin_plugin")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_plugin_setting")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"user",
|
|
||||||
sa.Column("userid", sa.String(length=30), nullable=False),
|
|
||||||
sa.Column("display_name", sa.String(length=30), nullable=True),
|
|
||||||
sa.Column("firstname", sa.String(length=50), nullable=False),
|
|
||||||
sa.Column("lastname", sa.String(length=50), nullable=False),
|
|
||||||
sa.Column("deleted", sa.Boolean(), nullable=True),
|
|
||||||
sa.Column("birthday", sa.Date(), nullable=True),
|
|
||||||
sa.Column("mail", sa.String(length=60), nullable=True),
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("avatar", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["avatar"], ["image.id"], name=op.f("fk_user_avatar_image")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_user")),
|
|
||||||
sa.UniqueConstraint("userid", name=op.f("uq_user_userid")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"notification",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("text", sa.Text(), nullable=True),
|
|
||||||
sa.Column("data", sa.PickleType(), nullable=True),
|
|
||||||
sa.Column("time", flaschengeist.database.types.UtcDateTime(), nullable=False),
|
|
||||||
sa.Column("user", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("plugin", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.ForeignKeyConstraint(["plugin"], ["plugin.id"], name=op.f("fk_notification_plugin_plugin")),
|
|
||||||
sa.ForeignKeyConstraint(["user"], ["user.id"], name=op.f("fk_notification_user_user")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_notification")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"password_reset",
|
|
||||||
sa.Column("user", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("token", sa.String(length=32), nullable=True),
|
|
||||||
sa.Column("expires", flaschengeist.database.types.UtcDateTime(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["user"], ["user.id"], name=op.f("fk_password_reset_user_user")),
|
|
||||||
sa.PrimaryKeyConstraint("user", name=op.f("pk_password_reset")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"role_x_permission",
|
|
||||||
sa.Column("role_id", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.Column("permission_id", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["permission_id"], ["permission.id"], name=op.f("fk_role_x_permission_permission_id_permission")
|
|
||||||
),
|
|
||||||
sa.ForeignKeyConstraint(["role_id"], ["role.id"], name=op.f("fk_role_x_permission_role_id_role")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"session",
|
|
||||||
sa.Column("expires", flaschengeist.database.types.UtcDateTime(), nullable=True),
|
|
||||||
sa.Column("token", sa.String(length=32), nullable=True),
|
|
||||||
sa.Column("lifetime", sa.Integer(), nullable=True),
|
|
||||||
sa.Column("browser", sa.String(length=127), nullable=True),
|
|
||||||
sa.Column("platform", sa.String(length=64), nullable=True),
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("user_id", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["user_id"], ["user.id"], name=op.f("fk_session_user_id_user")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_session")),
|
|
||||||
sa.UniqueConstraint("token", name=op.f("uq_session_token")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"user_attribute",
|
|
||||||
sa.Column("id", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("user", flaschengeist.database.types.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=True),
|
|
||||||
sa.Column("value", sa.PickleType(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["user"], ["user.id"], name=op.f("fk_user_attribute_user_user")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_user_attribute")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"user_x_role",
|
|
||||||
sa.Column("user_id", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.Column("role_id", flaschengeist.database.types.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["role_id"], ["role.id"], name=op.f("fk_user_x_role_role_id_role")),
|
|
||||||
sa.ForeignKeyConstraint(["user_id"], ["user.id"], name=op.f("fk_user_x_role_user_id_user")),
|
|
||||||
)
|
|
||||||
# ### end Alembic commands ###
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.drop_table("user_x_role")
|
|
||||||
op.drop_table("user_attribute")
|
|
||||||
op.drop_table("session")
|
|
||||||
op.drop_table("role_x_permission")
|
|
||||||
op.drop_table("password_reset")
|
|
||||||
op.drop_table("notification")
|
|
||||||
op.drop_table("user")
|
|
||||||
op.drop_table("plugin_setting")
|
|
||||||
op.drop_table("permission")
|
|
||||||
op.drop_table("role")
|
|
||||||
op.drop_table("plugin")
|
|
||||||
op.drop_table("image")
|
|
||||||
# ### end Alembic commands ###
|
|
|
@ -1,25 +0,0 @@
|
||||||
"""${message}
|
|
||||||
|
|
||||||
Revision ID: ${up_revision}
|
|
||||||
Revises: ${down_revision | comma,n}
|
|
||||||
Create Date: ${create_date}
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
import flaschengeist
|
|
||||||
${imports if imports else ""}
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = ${repr(up_revision)}
|
|
||||||
down_revision = ${repr(down_revision)}
|
|
||||||
branch_labels = ${repr(branch_labels)}
|
|
||||||
depends_on = ${repr(depends_on)}
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade():
|
|
||||||
${upgrades if upgrades else "pass"}
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
|
||||||
${downgrades if downgrades else "pass"}
|
|
|
@ -1,21 +1,17 @@
|
||||||
import enum
|
import enum
|
||||||
import json
|
|
||||||
|
|
||||||
from flask import Flask
|
import pkg_resources
|
||||||
|
from flask import Flask, current_app
|
||||||
from flask_cors import CORS
|
from flask_cors import CORS
|
||||||
from datetime import datetime, date
|
from datetime import datetime, date
|
||||||
from flask.json import jsonify
|
from flask.json import JSONEncoder, jsonify
|
||||||
from json import JSONEncoder
|
|
||||||
from flask.json.provider import JSONProvider
|
|
||||||
from sqlalchemy.exc import OperationalError
|
from sqlalchemy.exc import OperationalError
|
||||||
from werkzeug.exceptions import HTTPException
|
from werkzeug.exceptions import HTTPException
|
||||||
|
|
||||||
from flaschengeist import logger
|
from . import logger
|
||||||
from flaschengeist.controller import pluginController
|
from .plugins import AuthPlugin
|
||||||
from flaschengeist.utils.hook import Hook
|
from flaschengeist.config import config, configure_app
|
||||||
from flaschengeist.config import configure_app
|
from flaschengeist.controller import roleController
|
||||||
|
|
||||||
from flaschengeist.database import db
|
|
||||||
|
|
||||||
|
|
||||||
class CustomJSONEncoder(JSONEncoder):
|
class CustomJSONEncoder(JSONEncoder):
|
||||||
|
@ -39,65 +35,78 @@ class CustomJSONEncoder(JSONEncoder):
|
||||||
return JSONEncoder.default(self, o)
|
return JSONEncoder.default(self, o)
|
||||||
|
|
||||||
|
|
||||||
class CustomJSONProvider(JSONProvider):
|
def __load_plugins(app):
|
||||||
ensure_ascii: bool = True
|
logger.info("Search for plugins")
|
||||||
sort_keys: bool = True
|
|
||||||
|
|
||||||
def dumps(self, obj, **kwargs):
|
|
||||||
kwargs.setdefault("ensure_ascii", self.ensure_ascii)
|
|
||||||
kwargs.setdefault("sort_keys", self.sort_keys)
|
|
||||||
return json.dumps(obj, **kwargs, cls=CustomJSONEncoder)
|
|
||||||
|
|
||||||
def loads(self, s: str | bytes, **kwargs):
|
|
||||||
return json.loads(s, **kwargs)
|
|
||||||
|
|
||||||
|
|
||||||
@Hook("plugins.loaded")
|
|
||||||
def load_plugins(app: Flask):
|
|
||||||
app.config["FG_PLUGINS"] = {}
|
app.config["FG_PLUGINS"] = {}
|
||||||
|
for entry_point in pkg_resources.iter_entry_points("flaschengeist.plugin"):
|
||||||
|
logger.debug("Found plugin: >{}<".format(entry_point.name))
|
||||||
|
plugin = None
|
||||||
|
if entry_point.name in config and config[entry_point.name].get("enabled", False):
|
||||||
|
try:
|
||||||
|
logger.info(f"Load plugin {entry_point.name}")
|
||||||
|
plugin = entry_point.load()
|
||||||
|
if not hasattr(plugin, "name"):
|
||||||
|
setattr(plugin, "name", entry_point.name)
|
||||||
|
plugin = plugin(config[entry_point.name])
|
||||||
|
if hasattr(plugin, "blueprint") and plugin.blueprint is not None:
|
||||||
|
app.register_blueprint(plugin.blueprint)
|
||||||
|
except:
|
||||||
|
logger.error(
|
||||||
|
f"Plugin {entry_point.name} was enabled, but could not be loaded due to an error.",
|
||||||
|
exc_info=True,
|
||||||
|
)
|
||||||
|
del plugin
|
||||||
|
continue
|
||||||
|
if isinstance(plugin, AuthPlugin):
|
||||||
|
logger.debug(f"Found authentication plugin: {entry_point.name}")
|
||||||
|
if entry_point.name == config["FLASCHENGEIST"]["auth"]:
|
||||||
|
app.config["FG_AUTH_BACKEND"] = plugin
|
||||||
|
else:
|
||||||
|
del plugin
|
||||||
|
continue
|
||||||
|
if plugin:
|
||||||
|
app.config["FG_PLUGINS"][entry_point.name] = plugin
|
||||||
|
if "FG_AUTH_BACKEND" not in app.config:
|
||||||
|
logger.error("No authentication plugin configured or authentication plugin not found")
|
||||||
|
raise RuntimeError("No authentication plugin configured or authentication plugin not found")
|
||||||
|
|
||||||
for plugin in pluginController.get_enabled_plugins():
|
|
||||||
logger.debug(f"Searching for enabled plugin {plugin.name}")
|
def install_all():
|
||||||
try:
|
from flaschengeist.database import db
|
||||||
# Load class
|
|
||||||
cls = plugin.entry_point.load()
|
db.create_all()
|
||||||
# plugin = cls.query.get(plugin.id) if plugin.id is not None else plugin
|
db.session.commit()
|
||||||
# plugin = db.session.query(cls).get(plugin.id) if plugin.id is not None else plugin
|
installed = []
|
||||||
plugin = db.session.get(cls, plugin.id) if plugin.id is not None else plugin
|
for name, plugin in current_app.config["FG_PLUGINS"].items():
|
||||||
# Custom loading tasks
|
if not plugin:
|
||||||
plugin.load()
|
logger.debug(f"Skip disabled plugin: {name}")
|
||||||
# Register blueprint
|
|
||||||
if hasattr(plugin, "blueprint") and plugin.blueprint is not None:
|
|
||||||
app.register_blueprint(plugin.blueprint)
|
|
||||||
except:
|
|
||||||
logger.error(
|
|
||||||
f"Plugin {plugin.name} was enabled, but could not be loaded due to an error.",
|
|
||||||
exc_info=True,
|
|
||||||
)
|
|
||||||
continue
|
continue
|
||||||
logger.info(f"Loaded plugin: {plugin.name}")
|
logger.info(f"Install plugin {name}")
|
||||||
app.config["FG_PLUGINS"][plugin.name] = plugin
|
plugin.install()
|
||||||
|
installed.append(plugin)
|
||||||
|
if plugin.permissions:
|
||||||
|
roleController.create_permissions(plugin.permissions)
|
||||||
|
for plugin in installed:
|
||||||
|
plugin.post_install()
|
||||||
|
|
||||||
|
|
||||||
def create_app(test_config=None, cli=False):
|
def create_app(test_config=None):
|
||||||
app = Flask("flaschengeist")
|
app = Flask(__name__)
|
||||||
app.json_provider_class = CustomJSONProvider
|
app.json_encoder = CustomJSONEncoder
|
||||||
app.json = CustomJSONProvider(app)
|
|
||||||
CORS(app)
|
CORS(app)
|
||||||
|
|
||||||
with app.app_context():
|
with app.app_context():
|
||||||
from flaschengeist.database import db, migrate
|
from flaschengeist.database import db
|
||||||
|
|
||||||
configure_app(app, test_config)
|
configure_app(app, test_config)
|
||||||
db.init_app(app)
|
db.init_app(app)
|
||||||
migrate.init_app(app, db, compare_type=True)
|
__load_plugins(app)
|
||||||
load_plugins(app)
|
|
||||||
|
|
||||||
@app.route("/", methods=["GET"])
|
@app.route("/", methods=["GET"])
|
||||||
def __get_state():
|
def __get_state():
|
||||||
from . import __version__ as version
|
from . import __version__ as version
|
||||||
|
|
||||||
return jsonify({"plugins": pluginController.get_loaded_plugins(), "version": version})
|
return jsonify({"plugins": app.config["FG_PLUGINS"], "version": version})
|
||||||
|
|
||||||
@app.errorhandler(Exception)
|
@app.errorhandler(Exception)
|
||||||
def handle_exception(e):
|
def handle_exception(e):
|
||||||
|
|
|
@ -1,123 +0,0 @@
|
||||||
import io
|
|
||||||
import sys
|
|
||||||
import inspect
|
|
||||||
import logging
|
|
||||||
|
|
||||||
|
|
||||||
class InterfaceGenerator:
|
|
||||||
known = []
|
|
||||||
classes = {}
|
|
||||||
mapper = {
|
|
||||||
"str": "string",
|
|
||||||
"int": "number",
|
|
||||||
"float": "number",
|
|
||||||
"date": "Date",
|
|
||||||
"datetime": "Date",
|
|
||||||
"NoneType": "null",
|
|
||||||
"bool": "boolean",
|
|
||||||
}
|
|
||||||
|
|
||||||
def __init__(self, namespace, filename, logger=logging.getLogger()):
|
|
||||||
self.basename = ""
|
|
||||||
self.namespace = namespace
|
|
||||||
self.filename = filename
|
|
||||||
self.this_type = None
|
|
||||||
self.logger = logger
|
|
||||||
|
|
||||||
def pytype(self, cls):
|
|
||||||
a = self._pytype(cls)
|
|
||||||
return a
|
|
||||||
|
|
||||||
def _pytype(self, cls):
|
|
||||||
import typing
|
|
||||||
|
|
||||||
origin = typing.get_origin(cls)
|
|
||||||
arguments = typing.get_args(cls)
|
|
||||||
|
|
||||||
if origin is typing.ForwardRef: # isinstance(cls, typing.ForwardRef):
|
|
||||||
return "", "this" if cls.__forward_arg__ == self.this_type else cls.__forward_arg__
|
|
||||||
if origin is typing.Union:
|
|
||||||
if len(arguments) == 2 and arguments[1] is type(None):
|
|
||||||
return "?", self.pytype(arguments[0])[1]
|
|
||||||
else:
|
|
||||||
return "", "|".join([self.pytype(pt)[1] for pt in arguments])
|
|
||||||
if origin is list:
|
|
||||||
return "", "Array<{}>".format("|".join([self.pytype(a_type)[1] for a_type in arguments]))
|
|
||||||
if cls is typing.Any:
|
|
||||||
return "", "any"
|
|
||||||
|
|
||||||
name = cls.__name__ if hasattr(cls, "__name__") else cls if isinstance(cls, str) else None
|
|
||||||
if name is not None:
|
|
||||||
if name in self.mapper:
|
|
||||||
return "", self.mapper[name]
|
|
||||||
else:
|
|
||||||
return "", name
|
|
||||||
self.logger.warning(f"This python version might not detect all types (try >= 3.9). Could not identify >{cls}<")
|
|
||||||
return "?", "any"
|
|
||||||
|
|
||||||
def walker(self, module):
|
|
||||||
if sys.version_info < (3, 9):
|
|
||||||
raise RuntimeError("Python >= 3.9 is required to export API")
|
|
||||||
import typing
|
|
||||||
|
|
||||||
if (
|
|
||||||
inspect.ismodule(module[1])
|
|
||||||
and module[1].__name__.startswith(self.basename)
|
|
||||||
and module[1].__name__ not in self.known
|
|
||||||
):
|
|
||||||
self.known.append(module[1].__name__)
|
|
||||||
for cls in inspect.getmembers(module[1], lambda x: inspect.isclass(x) or inspect.ismodule(x)):
|
|
||||||
self.walker(cls)
|
|
||||||
elif (
|
|
||||||
inspect.isclass(module[1])
|
|
||||||
and module[1].__module__.startswith(self.basename)
|
|
||||||
and module[0] not in self.classes
|
|
||||||
and not module[0].startswith("_")
|
|
||||||
and hasattr(module[1], "__annotations__")
|
|
||||||
):
|
|
||||||
self.this_type = module[0]
|
|
||||||
|
|
||||||
d = {}
|
|
||||||
for param, ptype in typing.get_type_hints(module[1], globalns=None, localns=None).items():
|
|
||||||
if not param.startswith("_") and not param.endswith("_"):
|
|
||||||
d[param] = self.pytype(ptype)
|
|
||||||
|
|
||||||
if len(d) == 1:
|
|
||||||
key, value = d.popitem()
|
|
||||||
self.classes[module[0]] = value[1]
|
|
||||||
else:
|
|
||||||
self.classes[module[0]] = d
|
|
||||||
|
|
||||||
def run(self, models):
|
|
||||||
self.basename = models.__name__
|
|
||||||
self.walker(("models", models))
|
|
||||||
|
|
||||||
def _write_types(self):
|
|
||||||
TYPE = "type {name} = {alias};\n"
|
|
||||||
INTERFACE = "interface {name} {{\n{properties}}}\n"
|
|
||||||
PROPERTY = "\t{name}{modifier}: {type};\n"
|
|
||||||
|
|
||||||
buffer = io.StringIO()
|
|
||||||
for cls, props in self.classes.items():
|
|
||||||
if isinstance(props, str):
|
|
||||||
buffer.write(TYPE.format(name=cls, alias=props))
|
|
||||||
else:
|
|
||||||
buffer.write(
|
|
||||||
INTERFACE.format(
|
|
||||||
name=cls,
|
|
||||||
properties="".join(
|
|
||||||
[PROPERTY.format(name=name, modifier=props[name][0], type=props[name][1]) for name in props]
|
|
||||||
),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
return buffer
|
|
||||||
|
|
||||||
def write(self):
|
|
||||||
with open(self.filename, "w") if self.filename else sys.stdout as file:
|
|
||||||
if self.namespace:
|
|
||||||
file.write(f"declare namespace {self.namespace} {{\n")
|
|
||||||
for line in self._write_types().getvalue().split("\n"):
|
|
||||||
file.write(f"\t{line}\n")
|
|
||||||
file.write("}\n")
|
|
||||||
else:
|
|
||||||
file.write(self._write_types().getvalue())
|
|
|
@ -1,103 +0,0 @@
|
||||||
from os import environ
|
|
||||||
import sys
|
|
||||||
import click
|
|
||||||
import logging
|
|
||||||
|
|
||||||
from flask.cli import FlaskGroup, with_appcontext
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.app import create_app
|
|
||||||
|
|
||||||
LOGGING_MIN = 5 # TRACE (custom)
|
|
||||||
LOGGING_MAX = logging.ERROR
|
|
||||||
|
|
||||||
|
|
||||||
def get_version(ctx, param, value):
|
|
||||||
if not value or ctx.resilient_parsing:
|
|
||||||
return
|
|
||||||
|
|
||||||
import platform
|
|
||||||
from werkzeug import __version__ as werkzeug_version
|
|
||||||
from flask import __version__ as flask_version
|
|
||||||
from flaschengeist import __version__
|
|
||||||
|
|
||||||
click.echo(
|
|
||||||
f"Python {platform.python_version()}\n"
|
|
||||||
f"Flask {flask_version}\n"
|
|
||||||
f"Werkzeug {werkzeug_version}\n"
|
|
||||||
f"Flaschengeist {__version__}",
|
|
||||||
color=ctx.color,
|
|
||||||
)
|
|
||||||
ctx.exit()
|
|
||||||
|
|
||||||
|
|
||||||
def configure_logger(level):
|
|
||||||
"""Reconfigure main logger"""
|
|
||||||
global logger
|
|
||||||
|
|
||||||
# Handle TRACE -> meaning enable debug even for werkzeug
|
|
||||||
if level == 5:
|
|
||||||
level = 10
|
|
||||||
logging.getLogger("werkzeug").setLevel(level)
|
|
||||||
|
|
||||||
logger.setLevel(level)
|
|
||||||
environ["FG_LOGGING"] = logging.getLevelName(level)
|
|
||||||
for h in logger.handlers:
|
|
||||||
if isinstance(h, logging.StreamHandler) and h.name == "wsgi":
|
|
||||||
h.setLevel(level)
|
|
||||||
h.setStream(sys.stderr)
|
|
||||||
|
|
||||||
|
|
||||||
@with_appcontext
|
|
||||||
def verbosity(ctx, param, value):
|
|
||||||
"""Callback: Toggle verbosity between ERROR <-> TRACE"""
|
|
||||||
|
|
||||||
if not value or ctx.resilient_parsing:
|
|
||||||
return
|
|
||||||
configure_logger(LOGGING_MAX - max(LOGGING_MIN, min(value * 10, LOGGING_MAX - LOGGING_MIN)))
|
|
||||||
|
|
||||||
|
|
||||||
@click.group(
|
|
||||||
cls=FlaskGroup,
|
|
||||||
add_version_option=False,
|
|
||||||
add_default_commands=False,
|
|
||||||
create_app=create_app,
|
|
||||||
)
|
|
||||||
@click.option(
|
|
||||||
"--version",
|
|
||||||
help="Show the flask version",
|
|
||||||
expose_value=False,
|
|
||||||
callback=get_version,
|
|
||||||
is_flag=True,
|
|
||||||
is_eager=True,
|
|
||||||
)
|
|
||||||
@click.option(
|
|
||||||
"--verbose",
|
|
||||||
"-v",
|
|
||||||
help="Increase logging level",
|
|
||||||
callback=verbosity,
|
|
||||||
count=True,
|
|
||||||
expose_value=False,
|
|
||||||
)
|
|
||||||
def cli():
|
|
||||||
"""Management script for the Flaschengeist application."""
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
def main(*args, **kwargs):
|
|
||||||
from .plugin_cmd import plugin
|
|
||||||
from .export_cmd import export
|
|
||||||
from .docs_cmd import docs
|
|
||||||
from .run_cmd import run
|
|
||||||
from .install_cmd import install
|
|
||||||
from .docker_cmd import docker
|
|
||||||
|
|
||||||
# Override logging level
|
|
||||||
environ.setdefault("FG_LOGGING", logging.getLevelName(LOGGING_MAX))
|
|
||||||
|
|
||||||
cli.add_command(export)
|
|
||||||
cli.add_command(docs)
|
|
||||||
cli.add_command(install)
|
|
||||||
cli.add_command(plugin)
|
|
||||||
cli.add_command(run)
|
|
||||||
cli.add_command(docker)
|
|
||||||
cli(*args, **kwargs)
|
|
|
@ -1,54 +0,0 @@
|
||||||
import click
|
|
||||||
from click.decorators import pass_context
|
|
||||||
from flask.cli import with_appcontext
|
|
||||||
from os import environ
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.controller import pluginController
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
import traceback
|
|
||||||
|
|
||||||
|
|
||||||
@click.group()
|
|
||||||
def docker():
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
@docker.command()
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
def setup(ctx):
|
|
||||||
"""Setup flaschengesit in docker container"""
|
|
||||||
click.echo("Setup docker")
|
|
||||||
|
|
||||||
plugins = environ.get("FG_ENABLE_PLUGINS")
|
|
||||||
|
|
||||||
if not plugins:
|
|
||||||
click.secho("no evironment variable is set for 'FG_ENABLE_PLUGINS'", fg="yellow")
|
|
||||||
click.secho("set 'FG_ENABLE_PLUGINS' to 'auth_ldap', 'mail', 'balance', 'pricelist_old', 'events'")
|
|
||||||
plugins = ("auth_ldap", "mail", "pricelist_old", "events", "balance")
|
|
||||||
else:
|
|
||||||
plugins = plugins.split(" ")
|
|
||||||
|
|
||||||
print(plugins)
|
|
||||||
|
|
||||||
for name in plugins:
|
|
||||||
click.echo(f"Installing {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
try:
|
|
||||||
pluginController.install_plugin(name)
|
|
||||||
except Exception as e:
|
|
||||||
click.secho(" failed", fg="red")
|
|
||||||
if logger.getEffectiveLevel() > 10:
|
|
||||||
ctx.fail(f"[{e.__class__.__name__}] {e}")
|
|
||||||
else:
|
|
||||||
ctx.fail(traceback.format_exc())
|
|
||||||
else:
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
|
|
||||||
for name in plugins:
|
|
||||||
click.echo(f"Enabling {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
try:
|
|
||||||
pluginController.enable_plugin(name)
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
except NotFound:
|
|
||||||
click.secho(" not installed / not found", fg="red")
|
|
|
@ -1,38 +0,0 @@
|
||||||
import click
|
|
||||||
import pathlib
|
|
||||||
import subprocess
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@click.option(
|
|
||||||
"--output",
|
|
||||||
"-o",
|
|
||||||
help="Documentation output path",
|
|
||||||
default="./docs/html",
|
|
||||||
type=click.Path(file_okay=False, path_type=pathlib.Path),
|
|
||||||
)
|
|
||||||
@click.pass_context
|
|
||||||
def docs(ctx: click.Context, output: pathlib.Path):
|
|
||||||
"""Generate and export API documentation using pdoc"""
|
|
||||||
import pkg_resources
|
|
||||||
|
|
||||||
try:
|
|
||||||
pkg_resources.get_distribution("pdoc>=8.0.1")
|
|
||||||
except pkg_resources.DistributionNotFound:
|
|
||||||
click.echo(
|
|
||||||
f"Error: pdoc was not found, maybe you need to install it. Try:\n" "\n" '$ pip install "pdoc>=8.0.1"\n'
|
|
||||||
)
|
|
||||||
ctx.exit(1)
|
|
||||||
output.mkdir(parents=True, exist_ok=True)
|
|
||||||
command = [
|
|
||||||
"python",
|
|
||||||
"-m",
|
|
||||||
"pdoc",
|
|
||||||
"--docformat",
|
|
||||||
"google",
|
|
||||||
"--output-directory",
|
|
||||||
str(output),
|
|
||||||
"flaschengeist",
|
|
||||||
]
|
|
||||||
click.echo(f"Running command: {command}")
|
|
||||||
subprocess.check_call(command)
|
|
|
@ -1,29 +0,0 @@
|
||||||
import click
|
|
||||||
from importlib.metadata import entry_points
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@click.option("--output", "-o", help="Output file, default is stdout", type=click.Path())
|
|
||||||
@click.option("--namespace", "-n", help="TS namespace for the interfaces", type=str, show_default=True)
|
|
||||||
@click.option("--plugin", "-p", help="Also export types for a plugin (even if disabled)", multiple=True, type=str)
|
|
||||||
@click.option("--no-core", help="Skip models / types from flaschengeist core", is_flag=True)
|
|
||||||
def export(namespace, output, no_core, plugin):
|
|
||||||
from flaschengeist import logger, models
|
|
||||||
from flaschengeist.cli.InterfaceGenerator import InterfaceGenerator
|
|
||||||
|
|
||||||
gen = InterfaceGenerator(namespace, output, logger)
|
|
||||||
if not no_core:
|
|
||||||
gen.run(models)
|
|
||||||
if plugin:
|
|
||||||
for entry_point in entry_points(group="flaschengeist.plugins"):
|
|
||||||
if len(plugin) == 0 or entry_point.name in plugin:
|
|
||||||
try:
|
|
||||||
plugin = entry_point.load()
|
|
||||||
gen.run(plugin.models)
|
|
||||||
except:
|
|
||||||
logger.error(
|
|
||||||
f"Plugin {entry_point.name} could not be loaded due to an error.",
|
|
||||||
exc_info=True,
|
|
||||||
)
|
|
||||||
continue
|
|
||||||
gen.write()
|
|
|
@ -1,23 +0,0 @@
|
||||||
import click
|
|
||||||
from click.decorators import pass_context
|
|
||||||
from flask.cli import with_appcontext
|
|
||||||
from flask_migrate import upgrade
|
|
||||||
|
|
||||||
from flaschengeist.controller import pluginController
|
|
||||||
from flaschengeist.utils.hook import Hook
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
@Hook("plugins.installed")
|
|
||||||
def install(ctx: click.Context):
|
|
||||||
plugins = pluginController.get_enabled_plugins()
|
|
||||||
|
|
||||||
# Install database
|
|
||||||
upgrade(revision="flaschengeist@head")
|
|
||||||
|
|
||||||
# Install plugins
|
|
||||||
for plugin in plugins:
|
|
||||||
plugin = pluginController.install_plugin(plugin.name)
|
|
||||||
pluginController.enable_plugin(plugin.id)
|
|
|
@ -1,144 +0,0 @@
|
||||||
import traceback
|
|
||||||
import click
|
|
||||||
from click.decorators import pass_context
|
|
||||||
from flask import current_app
|
|
||||||
from flask.cli import with_appcontext
|
|
||||||
from importlib.metadata import EntryPoint, entry_points
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.controller import pluginController
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
|
|
||||||
|
|
||||||
@click.group()
|
|
||||||
def plugin():
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
@plugin.command()
|
|
||||||
@click.argument("plugin", nargs=-1, required=True, type=str)
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
def enable(ctx, plugin):
|
|
||||||
"""Enable one or more plugins"""
|
|
||||||
for name in plugin:
|
|
||||||
click.echo(f"Enabling {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
try:
|
|
||||||
pluginController.enable_plugin(name)
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
except NotFound:
|
|
||||||
click.secho(" not installed / not found", fg="red")
|
|
||||||
|
|
||||||
|
|
||||||
@plugin.command()
|
|
||||||
@click.argument("plugin", nargs=-1, required=True, type=str)
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
def disable(ctx, plugin):
|
|
||||||
"""Disable one or more plugins"""
|
|
||||||
for name in plugin:
|
|
||||||
click.echo(f"Disabling {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
try:
|
|
||||||
pluginController.disable_plugin(name)
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
except NotFound:
|
|
||||||
click.secho(" not installed / not found", fg="red")
|
|
||||||
|
|
||||||
|
|
||||||
@plugin.command()
|
|
||||||
@click.argument("plugin", nargs=-1, type=str)
|
|
||||||
@click.option("--all", help="Install all enabled plugins", is_flag=True)
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
def install(ctx: click.Context, plugin, all):
|
|
||||||
"""Install one or more plugins"""
|
|
||||||
all_plugins = entry_points(group="flaschengeist.plugins")
|
|
||||||
if all:
|
|
||||||
plugins = [ep.name for ep in all_plugins]
|
|
||||||
elif len(plugin) > 0:
|
|
||||||
plugins = plugin
|
|
||||||
for name in plugin:
|
|
||||||
if not all_plugins.select(name=name):
|
|
||||||
ctx.fail(f"Invalid plugin name, could not find >{name}<")
|
|
||||||
else:
|
|
||||||
ctx.fail("At least one plugin must be specified, or use `--all` flag.")
|
|
||||||
|
|
||||||
for name in plugins:
|
|
||||||
click.echo(f"Installing {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
try:
|
|
||||||
pluginController.install_plugin(name)
|
|
||||||
except Exception as e:
|
|
||||||
click.secho(" failed", fg="red")
|
|
||||||
if logger.getEffectiveLevel() > 10:
|
|
||||||
ctx.fail(f"[{e.__class__.__name__}] {e}")
|
|
||||||
else:
|
|
||||||
ctx.fail(traceback.format_exc())
|
|
||||||
else:
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
|
|
||||||
|
|
||||||
@plugin.command()
|
|
||||||
@click.argument("plugin", nargs=-1, required=True, type=str)
|
|
||||||
@with_appcontext
|
|
||||||
@pass_context
|
|
||||||
def uninstall(ctx: click.Context, plugin):
|
|
||||||
"""Uninstall one or more plugins"""
|
|
||||||
|
|
||||||
plugins = {plg.name: plg for plg in pluginController.get_installed_plugins() if plg.name in plugin}
|
|
||||||
try:
|
|
||||||
for name in plugin:
|
|
||||||
pluginController.disable_plugin(plugins[name])
|
|
||||||
if (
|
|
||||||
click.prompt(
|
|
||||||
"You are going to uninstall:\n\n"
|
|
||||||
f"\t{', '.join([plugin_name for plugin_name in plugins.keys()])}\n\n"
|
|
||||||
"Are you sure?",
|
|
||||||
default="n",
|
|
||||||
show_choices=True,
|
|
||||||
type=click.Choice(["y", "N"], False),
|
|
||||||
).lower()
|
|
||||||
!= "y"
|
|
||||||
):
|
|
||||||
ctx.exit()
|
|
||||||
click.echo(f"Uninstalling {name}{'.'*(20-len(name))}", nl=False)
|
|
||||||
pluginController.uninstall_plugin(plugins[name])
|
|
||||||
click.secho(" ok", fg="green")
|
|
||||||
except KeyError:
|
|
||||||
ctx.fail(f"Invalid plugin ID, could not find >{name}<")
|
|
||||||
|
|
||||||
|
|
||||||
@plugin.command()
|
|
||||||
@click.option("--enabled", "-e", help="List only enabled plugins", is_flag=True)
|
|
||||||
@click.option("--no-header", "-n", help="Do not show header", is_flag=True)
|
|
||||||
@with_appcontext
|
|
||||||
def ls(enabled, no_header):
|
|
||||||
def plugin_version(p):
|
|
||||||
if isinstance(p, EntryPoint):
|
|
||||||
return p.dist.version
|
|
||||||
return p.version
|
|
||||||
|
|
||||||
plugins = entry_points(group="flaschengeist.plugins")
|
|
||||||
installed_plugins = {plg.name: plg for plg in pluginController.get_installed_plugins()}
|
|
||||||
loaded_plugins = current_app.config["FG_PLUGINS"].keys()
|
|
||||||
|
|
||||||
if not no_header:
|
|
||||||
print(f"{' '*13}{'name': <20}| version | {' ' * 8} state")
|
|
||||||
print("-" * 63)
|
|
||||||
for plugin in plugins:
|
|
||||||
is_installed = plugin.name in installed_plugins.keys()
|
|
||||||
is_enabled = is_installed and installed_plugins[plugin.name].enabled
|
|
||||||
if enabled and is_enabled:
|
|
||||||
continue
|
|
||||||
print(f"{plugin.name: <33}|{plugin_version(plugin): >12} | ", end="")
|
|
||||||
if is_enabled:
|
|
||||||
if plugin.name in loaded_plugins:
|
|
||||||
print(click.style(" enabled", fg="green"))
|
|
||||||
else:
|
|
||||||
print(click.style("(failed to load)", fg="red"))
|
|
||||||
elif is_installed:
|
|
||||||
print(click.style(" disabled", fg="yellow"))
|
|
||||||
else:
|
|
||||||
print("not installed")
|
|
||||||
for name, plugin in installed_plugins.items():
|
|
||||||
if plugin.enabled and name not in loaded_plugins:
|
|
||||||
print(f"{name: <33}|{'': >12} |" f"{click.style(' failed to load', fg='red')}")
|
|
|
@ -1,37 +0,0 @@
|
||||||
import click
|
|
||||||
from os import environ
|
|
||||||
from flask import current_app
|
|
||||||
from flask.cli import with_appcontext, run_command
|
|
||||||
|
|
||||||
|
|
||||||
class PrefixMiddleware(object):
|
|
||||||
def __init__(self, app, prefix=""):
|
|
||||||
self.app = app
|
|
||||||
self.prefix = prefix
|
|
||||||
|
|
||||||
def __call__(self, environ, start_response):
|
|
||||||
if environ["PATH_INFO"].startswith(self.prefix):
|
|
||||||
environ["PATH_INFO"] = environ["PATH_INFO"][len(self.prefix) :]
|
|
||||||
environ["SCRIPT_NAME"] = self.prefix
|
|
||||||
return self.app(environ, start_response)
|
|
||||||
else:
|
|
||||||
start_response("404", [("Content-Type", "text/plain")])
|
|
||||||
return ["This url does not belong to the app.".encode()]
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@click.option("--host", help="set hostname to listen on", default="127.0.0.1", show_default=True)
|
|
||||||
@click.option("--port", help="set port to listen on", type=int, default=5000, show_default=True)
|
|
||||||
@click.option("--debug", help="run in debug mode", is_flag=True)
|
|
||||||
@with_appcontext
|
|
||||||
@click.pass_context
|
|
||||||
def run(ctx, host, port, debug):
|
|
||||||
"""Run Flaschengeist using a development server."""
|
|
||||||
from flaschengeist.config import config
|
|
||||||
|
|
||||||
current_app.wsgi_app = PrefixMiddleware(current_app.wsgi_app, prefix=config["FLASCHENGEIST"].get("root", ""))
|
|
||||||
if debug:
|
|
||||||
environ["FLASK_DEBUG"] = "1"
|
|
||||||
environ["FLASK_ENV"] = "development"
|
|
||||||
|
|
||||||
ctx.invoke(run_command, reload=True, host=host, port=port, debugger=debug)
|
|
|
@ -1,12 +1,12 @@
|
||||||
import os
|
import os
|
||||||
import toml
|
import toml
|
||||||
|
import logging.config
|
||||||
import collections.abc
|
import collections.abc
|
||||||
|
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from logging.config import dictConfig
|
|
||||||
from werkzeug.middleware.proxy_fix import ProxyFix
|
from werkzeug.middleware.proxy_fix import ProxyFix
|
||||||
|
from flaschengeist import _module_path, logger
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
|
|
||||||
# Default config:
|
# Default config:
|
||||||
config = {"DATABASE": {"engine": "mysql", "port": 3306}}
|
config = {"DATABASE": {"engine": "mysql", "port": 3306}}
|
||||||
|
@ -23,17 +23,17 @@ def update_dict(d, u):
|
||||||
|
|
||||||
def read_configuration(test_config):
|
def read_configuration(test_config):
|
||||||
global config
|
global config
|
||||||
paths = [Path(__file__).parent]
|
paths = [_module_path]
|
||||||
|
|
||||||
if not test_config:
|
if not test_config:
|
||||||
paths.append(Path.home() / ".config")
|
paths.append(Path.home() / ".config")
|
||||||
if "FLASCHENGEIST_CONF" in os.environ:
|
if "FLASCHENGEIST_CONF" in os.environ:
|
||||||
paths.append(Path(str(os.environ.get("FLASCHENGEIST_CONF"))))
|
paths.append(Path(os.environ.get("FLASCHENGEIST_CONF")))
|
||||||
|
|
||||||
for loc in paths:
|
for loc in paths:
|
||||||
try:
|
try:
|
||||||
with (loc / "flaschengeist.toml").open() as source:
|
with (loc / "flaschengeist.toml").open() as source:
|
||||||
logger.warning(f"Reading config file from >{loc}<") # default root logger, goes to stderr
|
print("Reading config file from >{}<".format(loc))
|
||||||
update_dict(config, toml.load(source))
|
update_dict(config, toml.load(source))
|
||||||
except IOError:
|
except IOError:
|
||||||
pass
|
pass
|
||||||
|
@ -42,33 +42,26 @@ def read_configuration(test_config):
|
||||||
|
|
||||||
|
|
||||||
def configure_logger():
|
def configure_logger():
|
||||||
"""Configure the logger
|
global config
|
||||||
|
|
||||||
force_console: Force a console handler
|
|
||||||
"""
|
|
||||||
|
|
||||||
def set_level(level):
|
|
||||||
# TRACE means even with werkzeug's request traces
|
|
||||||
if isinstance(level, str) and level.lower() == "trace":
|
|
||||||
level = "DEBUG"
|
|
||||||
logger_config["loggers"]["werkzeug"] = {"level": level}
|
|
||||||
logger_config["loggers"]["flaschengeist"] = {"level": level}
|
|
||||||
logger_config["handlers"]["wsgi"]["level"] = level
|
|
||||||
|
|
||||||
# Read default config
|
# Read default config
|
||||||
logger_config = toml.load(Path(__file__).parent / "logging.toml")
|
logger_config = toml.load(_module_path / "logging.toml")
|
||||||
|
|
||||||
if "LOGGING" in config:
|
if "LOGGING" in config:
|
||||||
# Override with user config
|
# Override with user config
|
||||||
update_dict(logger_config, config.get("LOGGING"))
|
update_dict(logger_config, config.get("LOGGING"))
|
||||||
# Check for shortcuts
|
# Check for shortcuts
|
||||||
if "level" in config["LOGGING"]:
|
if "level" in config["LOGGING"]:
|
||||||
set_level(config["LOGGING"]["level"])
|
logger_config["loggers"]["flaschengeist"] = {"level": config["LOGGING"]["level"]}
|
||||||
|
logger_config["handlers"]["console"]["level"] = config["LOGGING"]["level"]
|
||||||
# Override logging, used e.g. by CLI
|
logger_config["handlers"]["file"]["level"] = config["LOGGING"]["level"]
|
||||||
if "FG_LOGGING" in os.environ:
|
if not config["LOGGING"].get("console", True):
|
||||||
set_level(os.environ.get("FG_LOGGING", "CRITICAL"))
|
logger_config["handlers"]["console"]["level"] = "CRITICAL"
|
||||||
|
if "file" in config["LOGGING"]:
|
||||||
dictConfig(logger_config)
|
logger_config["root"]["handlers"].append("file")
|
||||||
|
logger_config["handlers"]["file"]["filename"] = config["LOGGING"]["file"]
|
||||||
|
path = Path(config["LOGGING"]["file"])
|
||||||
|
path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
logging.config.dictConfig(logger_config)
|
||||||
|
|
||||||
|
|
||||||
def configure_app(app, test_config=None):
|
def configure_app(app, test_config=None):
|
||||||
|
@ -77,11 +70,21 @@ def configure_app(app, test_config=None):
|
||||||
|
|
||||||
configure_logger()
|
configure_logger()
|
||||||
|
|
||||||
if "secret_key" not in config["FLASCHENGEIST"]:
|
# Always enable this builtin plugins!
|
||||||
logger.critical("No secret key was configured, please configure one for production systems!")
|
update_dict(
|
||||||
raise RuntimeError("No secret key was configured")
|
config,
|
||||||
|
{
|
||||||
|
"auth": {"enabled": True},
|
||||||
|
"roles": {"enabled": True},
|
||||||
|
"users": {"enabled": True},
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
app.config["SECRET_KEY"] = config["FLASCHENGEIST"]["secret_key"]
|
if "secret_key" not in config["FLASCHENGEIST"]:
|
||||||
|
logger.warning("No secret key was configured, please configure one for production systems!")
|
||||||
|
app.config["SECRET_KEY"] = "0a657b97ef546da90b2db91862ad4e29"
|
||||||
|
else:
|
||||||
|
app.config["SECRET_KEY"] = config["FLASCHENGEIST"]["secret_key"]
|
||||||
|
|
||||||
if test_config is not None:
|
if test_config is not None:
|
||||||
config["DATABASE"]["engine"] = "sqlite"
|
config["DATABASE"]["engine"] = "sqlite"
|
||||||
|
|
|
@ -1 +0,0 @@
|
||||||
"""Basic controllers for interaction with the Flaschengeist core"""
|
|
|
@ -1,14 +1,15 @@
|
||||||
from datetime import date
|
from datetime import date
|
||||||
from pathlib import Path
|
|
||||||
from flask import send_file
|
from flask import send_file
|
||||||
|
from pathlib import Path
|
||||||
from PIL import Image as PImage
|
from PIL import Image as PImage
|
||||||
from werkzeug.utils import secure_filename
|
|
||||||
from werkzeug.datastructures import FileStorage
|
|
||||||
from werkzeug.exceptions import NotFound, UnprocessableEntity
|
|
||||||
|
|
||||||
from ..models import Image
|
from werkzeug.exceptions import NotFound, UnprocessableEntity
|
||||||
from ..database import db
|
from werkzeug.datastructures import FileStorage
|
||||||
from ..config import config
|
from werkzeug.utils import secure_filename
|
||||||
|
|
||||||
|
from flaschengeist.models.image import Image
|
||||||
|
from flaschengeist.database import db
|
||||||
|
from flaschengeist.config import config
|
||||||
|
|
||||||
|
|
||||||
def check_mimetype(mime: str):
|
def check_mimetype(mime: str):
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
from ..utils.hook import Hook
|
from flaschengeist.utils.hook import Hook
|
||||||
from ..models import User, Role
|
from flaschengeist.models.user import User, Role
|
||||||
|
|
||||||
|
|
||||||
class Message:
|
class Message:
|
||||||
|
|
|
@ -1,172 +0,0 @@
|
||||||
"""Controller for Plugin logic
|
|
||||||
|
|
||||||
Used by plugins for setting and notification functionality.
|
|
||||||
"""
|
|
||||||
|
|
||||||
from typing import Union, List
|
|
||||||
from flask import current_app
|
|
||||||
from werkzeug.exceptions import NotFound, BadRequest
|
|
||||||
from sqlalchemy.exc import OperationalError, ProgrammingError
|
|
||||||
from flask_migrate import upgrade as database_upgrade
|
|
||||||
from importlib.metadata import entry_points
|
|
||||||
|
|
||||||
from flaschengeist import version as flaschengeist_version
|
|
||||||
|
|
||||||
from .. import logger
|
|
||||||
from ..database import db
|
|
||||||
from ..utils.hook import Hook
|
|
||||||
from ..plugins import Plugin, AuthPlugin
|
|
||||||
from ..models import Notification
|
|
||||||
|
|
||||||
|
|
||||||
__required_plugins = ["users", "roles", "scheduler", "auth"]
|
|
||||||
|
|
||||||
|
|
||||||
def get_authentication_provider():
|
|
||||||
return [
|
|
||||||
current_app.config["FG_PLUGINS"][plugin.name]
|
|
||||||
for plugin in get_loaded_plugins().values()
|
|
||||||
if isinstance(plugin, AuthPlugin)
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def get_loaded_plugins(plugin_name: str = None):
|
|
||||||
"""Get loaded plugin(s)"""
|
|
||||||
plugins = current_app.config["FG_PLUGINS"]
|
|
||||||
if plugin_name is not None:
|
|
||||||
plugins = [plugins[plugin_name]]
|
|
||||||
return {name: db.session.merge(plugins[name], load=False) for name in plugins}
|
|
||||||
|
|
||||||
|
|
||||||
def get_installed_plugins() -> list[Plugin]:
|
|
||||||
"""Get all installed plugins"""
|
|
||||||
return Plugin.query.all()
|
|
||||||
|
|
||||||
|
|
||||||
def get_enabled_plugins() -> list[Plugin]:
|
|
||||||
"""Get all installed and enabled plugins"""
|
|
||||||
try:
|
|
||||||
enabled_plugins = Plugin.query.filter(Plugin.enabled == True).all()
|
|
||||||
except (OperationalError, ProgrammingError) as e:
|
|
||||||
logger.error("Could not connect to database or database not initialized! No plugins enabled!")
|
|
||||||
logger.debug("Can not query enabled plugins", exc_info=True)
|
|
||||||
# Fake load required plugins so the database can at least be installed
|
|
||||||
enabled_plugins = [
|
|
||||||
entry_points(group="flaschengeist.plugins", name=name)[0].load()(
|
|
||||||
name=name, enabled=True, installed_version=flaschengeist_version
|
|
||||||
)
|
|
||||||
for name in __required_plugins
|
|
||||||
]
|
|
||||||
return enabled_plugins
|
|
||||||
|
|
||||||
|
|
||||||
def notify(plugin_id: int, user, text: str, data=None):
|
|
||||||
"""Create a new notification for an user
|
|
||||||
|
|
||||||
Args:
|
|
||||||
plugin_id: ID of the plugin
|
|
||||||
user: `flaschengeist.models.user.User` to notify
|
|
||||||
text: Visibile notification text
|
|
||||||
data: Optional data passed to the notificaton
|
|
||||||
Returns:
|
|
||||||
ID of the created `flaschengeist.models.notification.Notification`
|
|
||||||
|
|
||||||
Hint: use the data for frontend actions.
|
|
||||||
"""
|
|
||||||
if not user.deleted:
|
|
||||||
n = Notification(text=text, data=data, plugin_id_=plugin_id, user_=user)
|
|
||||||
db.session.add(n)
|
|
||||||
db.session.commit()
|
|
||||||
return n.id
|
|
||||||
|
|
||||||
|
|
||||||
def get_notifications(plugin_id) -> List[Notification]:
|
|
||||||
"""Get all notifications for a plugin
|
|
||||||
|
|
||||||
Args:
|
|
||||||
plugin_id: ID of the plugin
|
|
||||||
Returns:
|
|
||||||
List of `flaschengeist.models.notification.Notification`
|
|
||||||
"""
|
|
||||||
return db.session.execute(db.select(Notification).where(Notification.plugin_id_ == plugin_id)).scalars().all()
|
|
||||||
|
|
||||||
|
|
||||||
@Hook("plugins.installed")
|
|
||||||
def install_plugin(plugin_name: str):
|
|
||||||
logger.debug(f"Installing plugin {plugin_name}")
|
|
||||||
entry_point = entry_points(group="flaschengeist.plugins", name=plugin_name)
|
|
||||||
if not entry_point:
|
|
||||||
raise NotFound
|
|
||||||
|
|
||||||
cls = entry_point[0].load()
|
|
||||||
plugin: Plugin = cls.query.filter(Plugin.name == plugin_name).one_or_none()
|
|
||||||
if plugin is None:
|
|
||||||
plugin = cls(name=plugin_name, installed_version=entry_point[0].dist.version)
|
|
||||||
db.session.add(plugin)
|
|
||||||
db.session.commit()
|
|
||||||
# Custom installation steps
|
|
||||||
plugin.install()
|
|
||||||
# Check migrations
|
|
||||||
directory = entry_point[0].dist.locate_file("")
|
|
||||||
logger.debug(f"Checking for migrations in {directory}")
|
|
||||||
for loc in entry_point[0].module.split(".") + ["migrations"]:
|
|
||||||
directory /= loc
|
|
||||||
logger.debug(f"Checking for migrations with loc in {directory}")
|
|
||||||
if directory.exists():
|
|
||||||
logger.debug(f"Found migrations in {directory}")
|
|
||||||
database_upgrade(revision=f"{plugin_name}@head")
|
|
||||||
db.session.commit()
|
|
||||||
return plugin
|
|
||||||
|
|
||||||
|
|
||||||
@Hook("plugin.uninstalled")
|
|
||||||
def uninstall_plugin(plugin_id: Union[str, int, Plugin]):
|
|
||||||
plugin = disable_plugin(plugin_id)
|
|
||||||
logger.debug(f"Uninstall plugin {plugin.name}")
|
|
||||||
plugin.uninstall()
|
|
||||||
db.session.delete(plugin)
|
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
|
|
||||||
@Hook("plugins.enabled")
|
|
||||||
def enable_plugin(plugin_id: Union[str, int]) -> Plugin:
|
|
||||||
logger.debug(f"Enabling plugin {plugin_id}")
|
|
||||||
plugin = Plugin.query
|
|
||||||
if isinstance(plugin_id, str):
|
|
||||||
plugin = plugin.filter(Plugin.name == plugin_id).one_or_none()
|
|
||||||
elif isinstance(plugin_id, int):
|
|
||||||
plugin = plugin.get(plugin_id)
|
|
||||||
else:
|
|
||||||
raise TypeError
|
|
||||||
if plugin is None:
|
|
||||||
raise NotFound
|
|
||||||
plugin.enabled = True
|
|
||||||
db.session.commit()
|
|
||||||
plugin = plugin.entry_point.load().query.get(plugin.id)
|
|
||||||
current_app.config["FG_PLUGINS"][plugin.name] = plugin
|
|
||||||
return plugin
|
|
||||||
|
|
||||||
|
|
||||||
@Hook("plugins.disabled")
|
|
||||||
def disable_plugin(plugin_id: Union[str, int, Plugin]):
|
|
||||||
logger.debug(f"Disabling plugin {plugin_id}")
|
|
||||||
plugin: Plugin = Plugin.query
|
|
||||||
if isinstance(plugin_id, str):
|
|
||||||
plugin = plugin.filter(Plugin.name == plugin_id).one_or_none()
|
|
||||||
elif isinstance(plugin_id, int):
|
|
||||||
plugin = plugin.get(plugin_id)
|
|
||||||
elif isinstance(plugin_id, Plugin):
|
|
||||||
plugin = plugin_id
|
|
||||||
else:
|
|
||||||
raise TypeError
|
|
||||||
if plugin is None:
|
|
||||||
raise NotFound
|
|
||||||
if plugin.name in __required_plugins:
|
|
||||||
raise BadRequest
|
|
||||||
plugin.enabled = False
|
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
if plugin.name in current_app.config["FG_PLUGINS"].keys():
|
|
||||||
del current_app.config["FG_PLUGINS"][plugin.name]
|
|
||||||
|
|
||||||
return plugin
|
|
|
@ -1,32 +1,23 @@
|
||||||
from typing import Union
|
|
||||||
from sqlalchemy.exc import IntegrityError
|
from sqlalchemy.exc import IntegrityError
|
||||||
from werkzeug.exceptions import BadRequest, Conflict, NotFound
|
from werkzeug.exceptions import BadRequest, NotFound
|
||||||
|
|
||||||
from .. import logger
|
from flaschengeist.models.user import Role, Permission
|
||||||
from ..models import Role, Permission
|
from flaschengeist.database import db, case_sensitive
|
||||||
from ..database import db, case_sensitive
|
from flaschengeist import logger
|
||||||
from ..utils.hook import Hook
|
from flaschengeist.utils.hook import Hook
|
||||||
|
|
||||||
|
|
||||||
def get_all():
|
def get_all():
|
||||||
return Role.query.all()
|
return Role.query.all()
|
||||||
|
|
||||||
|
|
||||||
def get(role_name: Union[int, str]) -> Role:
|
def get(role_name):
|
||||||
"""Get role by ID or name
|
|
||||||
Args:
|
|
||||||
role_name: Name or ID of the role
|
|
||||||
Returns:
|
|
||||||
Matching role
|
|
||||||
Raises:
|
|
||||||
NotFound
|
|
||||||
"""
|
|
||||||
if type(role_name) is int:
|
if type(role_name) is int:
|
||||||
role = Role.query.get(role_name)
|
role = Role.query.get(role_name)
|
||||||
else:
|
else:
|
||||||
role = Role.query.filter(Role.name == role_name).one_or_none()
|
role = Role.query.filter(Role.name == role_name).one_or_none()
|
||||||
if not role:
|
if not role:
|
||||||
raise NotFound("no such role")
|
raise NotFound
|
||||||
return role
|
return role
|
||||||
|
|
||||||
|
|
||||||
|
@ -36,8 +27,15 @@ def get_permissions():
|
||||||
|
|
||||||
@Hook
|
@Hook
|
||||||
def update_role(role, new_name):
|
def update_role(role, new_name):
|
||||||
if new_name is None or not isinstance(new_name, str):
|
if new_name is None:
|
||||||
raise BadRequest("Invalid new name")
|
try:
|
||||||
|
logger.debug(f"Hallo, dies ist die {role.serialize()}")
|
||||||
|
db.session.delete(role)
|
||||||
|
logger.debug(f"Hallo, dies ist die {role.serialize()}")
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
logger.debug("IntegrityError: Role might still be in use", exc_info=True)
|
||||||
|
raise BadRequest("Role still in use")
|
||||||
else:
|
else:
|
||||||
if role.name == new_name or db.session.query(db.exists().where(Role.name == case_sensitive(new_name))).scalar():
|
if role.name == new_name or db.session.query(db.exists().where(Role.name == case_sensitive(new_name))).scalar():
|
||||||
raise BadRequest("Name already used")
|
raise BadRequest("Name already used")
|
||||||
|
@ -46,10 +44,11 @@ def update_role(role, new_name):
|
||||||
|
|
||||||
|
|
||||||
def set_permissions(role, permissions):
|
def set_permissions(role, permissions):
|
||||||
perms = Permission.query.filter(Permission.name.in_(permissions)).all()
|
for name in permissions:
|
||||||
if len(perms) < len(permissions):
|
p = Permission.query.filter(Permission.name.in_(permissions)).all()
|
||||||
raise BadRequest("Invalid permission name")
|
if not p or len(p) < len(permissions):
|
||||||
role.permissions = list(perms)
|
raise BadRequest("Invalid permission name >{}<".format(name))
|
||||||
|
role.permissions = list(p)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
@ -64,22 +63,14 @@ def create_permissions(permissions):
|
||||||
|
|
||||||
def create_role(name: str, permissions=[]):
|
def create_role(name: str, permissions=[]):
|
||||||
logger.debug(f"Create new role with name: {name}")
|
logger.debug(f"Create new role with name: {name}")
|
||||||
try:
|
role = Role(name=name)
|
||||||
role = Role(name=name)
|
db.session.add(role)
|
||||||
db.session.add(role)
|
set_permissions(role, permissions)
|
||||||
set_permissions(role, permissions)
|
db.session.commit()
|
||||||
db.session.commit()
|
logger.debug(f"Created role: {role.serialize()}")
|
||||||
logger.debug(f"Created role: {role.serialize()}")
|
|
||||||
except IntegrityError:
|
|
||||||
raise BadRequest("role already exists")
|
|
||||||
return role
|
return role
|
||||||
|
|
||||||
|
|
||||||
def delete(role):
|
def delete(role):
|
||||||
role.permissions.clear()
|
role.permissions.clear()
|
||||||
try:
|
update_role(role, None)
|
||||||
db.session.delete(role)
|
|
||||||
db.session.commit()
|
|
||||||
except IntegrityError:
|
|
||||||
logger.debug("IntegrityError: Role might still be in use", exc_info=True)
|
|
||||||
raise Conflict("Role still in use")
|
|
||||||
|
|
|
@ -1,22 +1,14 @@
|
||||||
import secrets
|
import secrets
|
||||||
|
from flaschengeist.models.session import Session
|
||||||
from datetime import datetime, timezone
|
from flaschengeist.database import db
|
||||||
|
from flaschengeist import logger
|
||||||
from werkzeug.exceptions import Forbidden, Unauthorized
|
from werkzeug.exceptions import Forbidden, Unauthorized
|
||||||
from ua_parser import user_agent_parser
|
from datetime import datetime, timezone
|
||||||
|
|
||||||
from .. import logger
|
|
||||||
from ..models import Session
|
|
||||||
from ..database import db
|
|
||||||
|
|
||||||
|
|
||||||
lifetime = 1800
|
lifetime = 1800
|
||||||
|
|
||||||
|
|
||||||
def get_user_agent(request_headers):
|
def validate_token(token, user_agent, permission):
|
||||||
return user_agent_parser.Parse(request_headers.get("User-Agent", "") if request_headers else "")
|
|
||||||
|
|
||||||
|
|
||||||
def validate_token(token, request_headers, permission):
|
|
||||||
"""Verify session
|
"""Verify session
|
||||||
|
|
||||||
Verify a Session and Roles so if the User has permission or not.
|
Verify a Session and Roles so if the User has permission or not.
|
||||||
|
@ -24,7 +16,7 @@ def validate_token(token, request_headers, permission):
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
token: Token to verify.
|
token: Token to verify.
|
||||||
request_headers: Headers to validate user agent of browser
|
user_agent: User agent of browser to check
|
||||||
permission: Permission needed to access restricted routes
|
permission: Permission needed to access restricted routes
|
||||||
Returns:
|
Returns:
|
||||||
A Session for this given Token
|
A Session for this given Token
|
||||||
|
@ -36,12 +28,8 @@ def validate_token(token, request_headers, permission):
|
||||||
session = Session.query.filter_by(token=token).one_or_none()
|
session = Session.query.filter_by(token=token).one_or_none()
|
||||||
if session:
|
if session:
|
||||||
logger.debug("token found, check if expired or invalid user agent differs")
|
logger.debug("token found, check if expired or invalid user agent differs")
|
||||||
user_agent = get_user_agent(request_headers)
|
|
||||||
platform = user_agent["os"]["family"]
|
|
||||||
browser = user_agent["user_agent"]["family"]
|
|
||||||
|
|
||||||
if session.expires >= datetime.now(timezone.utc) and (
|
if session.expires >= datetime.now(timezone.utc) and (
|
||||||
session.browser == browser and session.platform == platform
|
session.browser == user_agent.browser and session.platform == user_agent.platform
|
||||||
):
|
):
|
||||||
if not permission or session.user_.has_permission(permission):
|
if not permission or session.user_.has_permission(permission):
|
||||||
session.refresh()
|
session.refresh()
|
||||||
|
@ -56,26 +44,24 @@ def validate_token(token, request_headers, permission):
|
||||||
raise Unauthorized
|
raise Unauthorized
|
||||||
|
|
||||||
|
|
||||||
def create(user, request_headers=None) -> Session:
|
def create(user, user_agent=None) -> Session:
|
||||||
"""Create a Session
|
"""Create a Session
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
user: For which User is to create a Session
|
user: For which User is to create a Session
|
||||||
request_headers: Headers to validate user agent of browser
|
user_agent: User agent to identify session
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
Session: A created Token for User
|
Session: A created Token for User
|
||||||
"""
|
"""
|
||||||
logger.debug("create access token")
|
logger.debug("create access token")
|
||||||
token_str = secrets.token_hex(16)
|
token_str = secrets.token_hex(16)
|
||||||
user_agent = get_user_agent(request_headers)
|
|
||||||
logger.debug(f"platform: {user_agent['os']['family']}, browser: {user_agent['user_agent']['family']}")
|
|
||||||
session = Session(
|
session = Session(
|
||||||
token=token_str,
|
token=token_str,
|
||||||
user_=user,
|
user_=user,
|
||||||
lifetime=lifetime,
|
lifetime=lifetime,
|
||||||
platform=user_agent["os"]["family"],
|
browser=user_agent.browser,
|
||||||
browser=user_agent["user_agent"]["family"],
|
platform=user_agent.platform,
|
||||||
)
|
)
|
||||||
session.refresh()
|
session.refresh()
|
||||||
db.session.add(session)
|
db.session.add(session)
|
||||||
|
|
|
@ -1,36 +1,16 @@
|
||||||
import re
|
|
||||||
import secrets
|
import secrets
|
||||||
import hashlib
|
from flask import current_app
|
||||||
|
from datetime import datetime, timedelta, timezone
|
||||||
from io import BytesIO
|
|
||||||
from typing import Optional, Union
|
|
||||||
from flask import make_response
|
|
||||||
from flask.json import provider
|
|
||||||
from sqlalchemy import exc
|
|
||||||
from sqlalchemy_utils import merge_references
|
|
||||||
from datetime import datetime, timedelta, timezone, date
|
|
||||||
from flask.helpers import send_file
|
|
||||||
from werkzeug.exceptions import NotFound, BadRequest, Forbidden
|
from werkzeug.exceptions import NotFound, BadRequest, Forbidden
|
||||||
|
|
||||||
from .. import logger
|
from flaschengeist import logger
|
||||||
from ..config import config
|
from flaschengeist.config import config
|
||||||
from ..database import db
|
from flaschengeist.database import db
|
||||||
from ..models import Notification, User, Role
|
from flaschengeist.models.notification import Notification
|
||||||
from ..models.user import _PasswordReset
|
from flaschengeist.utils.hook import Hook
|
||||||
from ..utils.hook import Hook
|
from flaschengeist.utils.datetime import from_iso_format
|
||||||
from ..utils.datetime import from_iso_format
|
from flaschengeist.models.user import User, Role, _PasswordReset
|
||||||
from ..controller import (
|
from flaschengeist.controller import messageController, sessionController
|
||||||
imageController,
|
|
||||||
messageController,
|
|
||||||
pluginController,
|
|
||||||
sessionController,
|
|
||||||
)
|
|
||||||
from ..plugins import AuthPlugin
|
|
||||||
|
|
||||||
|
|
||||||
def __active_users():
|
|
||||||
"""Return query for not deleted users"""
|
|
||||||
return User.query.filter(User.deleted == False)
|
|
||||||
|
|
||||||
|
|
||||||
def _generate_password_reset(user):
|
def _generate_password_reset(user):
|
||||||
|
@ -50,35 +30,17 @@ def _generate_password_reset(user):
|
||||||
return reset
|
return reset
|
||||||
|
|
||||||
|
|
||||||
def get_provider(userid: str) -> AuthPlugin:
|
|
||||||
return [p for p in pluginController.get_authentication_provider() if p.user_exists(userid)][0]
|
|
||||||
|
|
||||||
|
|
||||||
@Hook
|
|
||||||
def update_user(user: User, backend: Optional[AuthPlugin] = None):
|
|
||||||
"""Update user data from backend
|
|
||||||
|
|
||||||
This is seperate function to provide a hook"""
|
|
||||||
if not backend:
|
|
||||||
backend = get_provider(user.userid)
|
|
||||||
backend.update_user(user)
|
|
||||||
if not user.display_name:
|
|
||||||
user.display_name = "{} {}.".format(user.firstname, user.lastname[0])
|
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def login_user(username, password):
|
def login_user(username, password):
|
||||||
logger.info("login user {{ {} }}".format(username))
|
logger.info("login user {{ {} }}".format(username))
|
||||||
for provider in pluginController.get_authentication_provider():
|
|
||||||
uid = provider.login(username, password)
|
user = find_user(username)
|
||||||
if isinstance(uid, str):
|
if not user:
|
||||||
user = get_user(uid)
|
logger.debug("User not found in Database.")
|
||||||
if not user:
|
user = User(userid=username)
|
||||||
logger.debug("User not found in Database.")
|
db.session.add(user)
|
||||||
user = User(userid=uid)
|
if current_app.config["FG_AUTH_BACKEND"].login(user, password):
|
||||||
db.session.add(user)
|
update_user(user)
|
||||||
update_user(user, provider)
|
return user
|
||||||
return user
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
@ -111,30 +73,26 @@ def reset_password(token: str, password: str):
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
def set_roles(user: User, roles: list[str], create=False):
|
@Hook
|
||||||
"""Set roles of user
|
def update_user(user):
|
||||||
|
current_app.config["FG_AUTH_BACKEND"].update_user(user)
|
||||||
Args:
|
if not user.display_name:
|
||||||
user: User to set roles of
|
user.display_name = "{} {}.".format(user.firstname, user.lastname[0])
|
||||||
roles: List of role names
|
db.session.commit()
|
||||||
create: If set to true, create not existing roles
|
|
||||||
Raises:
|
|
||||||
BadRequest if invalid arguments given or not all roles found while *create* is set to false
|
|
||||||
"""
|
|
||||||
from .roleController import create_role
|
|
||||||
|
|
||||||
if not isinstance(roles, list) and any([not isinstance(r, str) for r in roles]):
|
|
||||||
raise BadRequest("Invalid role name")
|
|
||||||
fetched = Role.query.filter(Role.name.in_(roles)).all()
|
|
||||||
if len(fetched) < len(roles):
|
|
||||||
if not create:
|
|
||||||
raise BadRequest("Invalid role name, role not found")
|
|
||||||
# Create all new roles
|
|
||||||
fetched += [create_role(role_name) for role_name in roles if not any([role_name == r.name for r in fetched])]
|
|
||||||
user.roles_ = fetched
|
|
||||||
|
|
||||||
|
|
||||||
def modify_user(user: User, password: str, new_password: str = None):
|
def set_roles(user: User, roles: [str], create=False):
|
||||||
|
user.roles_.clear()
|
||||||
|
for role_name in roles:
|
||||||
|
role = Role.query.filter(Role.name == role_name).one_or_none()
|
||||||
|
if not role:
|
||||||
|
if not create:
|
||||||
|
raise BadRequest("Role not found >{}<".format(role_name))
|
||||||
|
role = Role(name=role_name)
|
||||||
|
user.roles_.append(role)
|
||||||
|
|
||||||
|
|
||||||
|
def modify_user(user, password, new_password=None):
|
||||||
"""Modify given user on the backend
|
"""Modify given user on the backend
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
@ -146,8 +104,7 @@ def modify_user(user: User, password: str, new_password: str = None):
|
||||||
NotImplemented: If backend is not capable of this operation
|
NotImplemented: If backend is not capable of this operation
|
||||||
BadRequest: Password is wrong or other logic issues
|
BadRequest: Password is wrong or other logic issues
|
||||||
"""
|
"""
|
||||||
provider = get_provider(user.userid)
|
current_app.config["FG_AUTH_BACKEND"].modify_user(user, password, new_password)
|
||||||
provider.modify_user(user, password, new_password)
|
|
||||||
|
|
||||||
if new_password:
|
if new_password:
|
||||||
logger.debug(f"Password changed for user {user.userid}")
|
logger.debug(f"Password changed for user {user.userid}")
|
||||||
|
@ -161,151 +118,99 @@ def modify_user(user: User, password: str, new_password: str = None):
|
||||||
messageController.send_message(messageController.Message(user, text, subject))
|
messageController.send_message(messageController.Message(user, text, subject))
|
||||||
|
|
||||||
|
|
||||||
def get_users(deleted=False):
|
def get_users():
|
||||||
query = __active_users() if not deleted else User.query
|
return User.query.all()
|
||||||
return query.all()
|
|
||||||
|
|
||||||
|
|
||||||
def get_user_by_role(role: Role):
|
def get_user_by_role(role: Role):
|
||||||
return User.query.join(User.roles_).filter_by(role_id=role.id).all()
|
return User.query.join(User.roles_).filter_by(role_id=role.id).all()
|
||||||
|
|
||||||
|
|
||||||
def get_user(uid, deleted=False) -> User:
|
def get_user(uid):
|
||||||
"""Get an user by userid from database
|
"""Get an user by userid from database
|
||||||
Args:
|
Args:
|
||||||
uid: Userid to search for
|
uid: Userid to search for
|
||||||
deleted: Set to true to also search deleted users
|
|
||||||
Returns:
|
Returns:
|
||||||
User fround
|
User fround
|
||||||
Raises:
|
Raises:
|
||||||
NotFound if not found"""
|
NotFound if not found"""
|
||||||
user = (__active_users() if not deleted else User.query).filter(User.userid == uid).one_or_none()
|
user = User.query.filter(User.userid == uid).one_or_none()
|
||||||
if not user:
|
if not user:
|
||||||
raise NotFound
|
raise NotFound
|
||||||
return user
|
return user
|
||||||
|
|
||||||
|
|
||||||
@Hook
|
def find_user(uid_mail):
|
||||||
def delete_user(user: User):
|
"""Finding an user by userid or mail in database or auth-backend
|
||||||
"""Delete given user"""
|
|
||||||
# First let the backend delete the user, as this might fail
|
|
||||||
provider = get_provider(user.userid)
|
|
||||||
provider.delete_user(user)
|
|
||||||
|
|
||||||
# Clear all easy relationships
|
|
||||||
user.avatar_ = None
|
|
||||||
user._attributes.clear()
|
|
||||||
user.roles_.clear()
|
|
||||||
user.sessions_.clear()
|
|
||||||
user.reset_requests_.clear()
|
|
||||||
# Now move all other references to the DELETED_USER
|
|
||||||
try:
|
|
||||||
deleted_user = get_user("__deleted_user__", True)
|
|
||||||
except NotFound:
|
|
||||||
deleted_user = User(
|
|
||||||
userid="__deleted_user__",
|
|
||||||
firstname="USER",
|
|
||||||
lastname="DELETED",
|
|
||||||
display_name="DELETED USER",
|
|
||||||
deleted=True,
|
|
||||||
)
|
|
||||||
db.session.add(user)
|
|
||||||
db.session.flush()
|
|
||||||
merge_references(user, deleted_user)
|
|
||||||
db.session.commit()
|
|
||||||
# Now try to delete the user for real
|
|
||||||
try:
|
|
||||||
db.session.delete(user)
|
|
||||||
db.session.commit()
|
|
||||||
except exc.IntegrityError:
|
|
||||||
logger.error(
|
|
||||||
"Delete of user failed, there might be ForeignKey contraits from disabled plugins",
|
|
||||||
exec_info=True,
|
|
||||||
)
|
|
||||||
# Remove at least all personal data
|
|
||||||
user.userid = f"__deleted_user__{user.id_}"
|
|
||||||
user.display_name = "DELETED USER"
|
|
||||||
user.firstname = ""
|
|
||||||
user.lastname = ""
|
|
||||||
user.deleted = True
|
|
||||||
user.birthday = None
|
|
||||||
user.mail = None
|
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
|
|
||||||
def register(data, passwd=None):
|
|
||||||
"""Register a new user
|
|
||||||
Args:
|
Args:
|
||||||
data: dictionary containing valid user properties
|
uid_mail: userid and or mail to search for
|
||||||
passwd: optional a password, default: 16byte random
|
Returns:
|
||||||
|
User if found or None
|
||||||
"""
|
"""
|
||||||
|
mail = uid_mail.split("@")
|
||||||
|
mail = len(mail) == 2 and len(mail[0]) > 0 and len(mail[1]) > 0
|
||||||
|
|
||||||
|
query = User.userid == uid_mail
|
||||||
|
if mail:
|
||||||
|
query |= User.mail == uid_mail
|
||||||
|
user = User.query.filter(query).one_or_none()
|
||||||
|
if user:
|
||||||
|
update_user(user)
|
||||||
|
else:
|
||||||
|
user = current_app.config["FG_AUTH_BACKEND"].find_user(uid_mail, uid_mail if mail else None)
|
||||||
|
if user:
|
||||||
|
db.session.add(user)
|
||||||
|
db.session.commit()
|
||||||
|
return user
|
||||||
|
|
||||||
|
|
||||||
|
def delete(user):
|
||||||
|
"""Delete given user"""
|
||||||
|
current_app.config["FG_AUTH_BACKEND"].delete_user(user)
|
||||||
|
db.session.delete(user)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def register(data):
|
||||||
allowed_keys = User().serialize().keys()
|
allowed_keys = User().serialize().keys()
|
||||||
values = {key: value for key, value in data.items() if key in allowed_keys}
|
values = {key: value for key, value in data.items() if key in allowed_keys}
|
||||||
roles = values.pop("roles", [])
|
roles = values.pop("roles", [])
|
||||||
if "birthday" in data:
|
if "birthday" in data:
|
||||||
if isinstance(data["birthday"], date):
|
values["birthday"] = from_iso_format(data["birthday"]).date()
|
||||||
values["birthday"] = data["birthday"]
|
|
||||||
else:
|
|
||||||
values["birthday"] = from_iso_format(data["birthday"]).date()
|
|
||||||
if "mail" in data and not re.match(r"[^@]+@[^@]+\.[^@]+", data["mail"]):
|
|
||||||
raise BadRequest("Invalid mail given")
|
|
||||||
user = User(**values)
|
user = User(**values)
|
||||||
set_roles(user, roles)
|
set_roles(user, roles)
|
||||||
|
|
||||||
password = passwd if passwd else secrets.token_urlsafe(16)
|
password = secrets.token_urlsafe(16)
|
||||||
try:
|
current_app.config["FG_AUTH_BACKEND"].create_user(user, password)
|
||||||
provider = [p for p in pluginController.get_authentication_provider() if p.can_register()][0]
|
db.session.add(user)
|
||||||
provider.create_user(user, password)
|
db.session.commit()
|
||||||
db.session.add(user)
|
|
||||||
db.session.commit()
|
|
||||||
except IndexError as e:
|
|
||||||
logger.error("No authentication backend, allowing registering new users, found.")
|
|
||||||
raise e
|
|
||||||
except exc.IntegrityError:
|
|
||||||
raise BadRequest("userid already in use")
|
|
||||||
|
|
||||||
if user.mail:
|
reset = _generate_password_reset(user)
|
||||||
reset = _generate_password_reset(user)
|
|
||||||
|
|
||||||
subject = str(config["MESSAGES"]["welcome_subject"]).format(name=user.display_name, username=user.userid)
|
subject = str(config["MESSAGES"]["welcome_subject"]).format(name=user.display_name, username=user.userid)
|
||||||
text = str(config["MESSAGES"]["welcome_text"]).format(
|
text = str(config["MESSAGES"]["welcome_text"]).format(
|
||||||
name=user.display_name,
|
name=user.display_name,
|
||||||
username=user.userid,
|
username=user.userid,
|
||||||
password_link=f'https://{config["FLASCHENGEIST"]["domain"]}/reset?token={reset.token}',
|
password_link=f'https://{config["FLASCHENGEIST"]["domain"]}/reset?token={reset.token}',
|
||||||
)
|
)
|
||||||
messageController.send_message(messageController.Message(user, text, subject))
|
messageController.send_message(messageController.Message(user, text, subject))
|
||||||
|
|
||||||
provider.update_user(user)
|
find_user(user.userid)
|
||||||
|
|
||||||
return user
|
return user
|
||||||
|
|
||||||
|
|
||||||
def get_last_modified(user: User):
|
def load_avatar(user: User):
|
||||||
"""Get the last modification date of the user"""
|
return current_app.config["FG_AUTH_BACKEND"].get_avatar(user)
|
||||||
return get_provider(user.userid).get_last_modified(user)
|
|
||||||
|
|
||||||
|
|
||||||
def load_avatar(user: User, etag: Union[str, None] = None):
|
def save_avatar(user, avatar):
|
||||||
if user.avatar_ is not None:
|
current_app.config["FG_AUTH_BACKEND"].set_avatar(user, avatar)
|
||||||
return imageController.send_image(image=user.avatar_)
|
|
||||||
else:
|
|
||||||
provider = get_provider(user.userid)
|
|
||||||
avatar = provider.get_avatar(user)
|
|
||||||
new_etag = hashlib.md5(avatar.binary).hexdigest()
|
|
||||||
if new_etag == etag:
|
|
||||||
return make_response("", 304)
|
|
||||||
if len(avatar.binary) > 0:
|
|
||||||
return send_file(BytesIO(avatar.binary), avatar.mimetype, etag=new_etag)
|
|
||||||
raise NotFound
|
|
||||||
|
|
||||||
|
|
||||||
def save_avatar(user, file):
|
|
||||||
get_provider(user.userid).set_avatar(user, file)
|
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
def delete_avatar(user):
|
def delete_avatar(user):
|
||||||
get_provider(user.userid).delete_avatar(user)
|
current_app.config["FG_AUTH_BACKEND"].delete_avatar(user)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,11 @@
|
||||||
|
from flask_sqlalchemy import SQLAlchemy
|
||||||
|
|
||||||
|
db = SQLAlchemy()
|
||||||
|
|
||||||
|
|
||||||
|
def case_sensitive(s):
|
||||||
|
if db.session.bind.dialect.name == "mysql":
|
||||||
|
from sqlalchemy import func
|
||||||
|
|
||||||
|
return func.binary(s)
|
||||||
|
return s
|
|
@ -1,75 +0,0 @@
|
||||||
import os
|
|
||||||
from flask_migrate import Migrate, Config
|
|
||||||
from flask_sqlalchemy import SQLAlchemy
|
|
||||||
from importlib.metadata import EntryPoint, entry_points, distribution
|
|
||||||
from sqlalchemy import MetaData
|
|
||||||
|
|
||||||
from flaschengeist.alembic import alembic_script_path
|
|
||||||
from flaschengeist import logger
|
|
||||||
|
|
||||||
# from flaschengeist.controller import pluginController
|
|
||||||
|
|
||||||
# https://alembic.sqlalchemy.org/en/latest/naming.html
|
|
||||||
metadata = MetaData(
|
|
||||||
naming_convention={
|
|
||||||
"pk": "pk_%(table_name)s",
|
|
||||||
"ix": "ix_%(table_name)s_%(column_0_name)s",
|
|
||||||
"uq": "uq_%(table_name)s_%(column_0_name)s",
|
|
||||||
"ck": "ck_%(table_name)s_%(constraint_name)s",
|
|
||||||
"fk": "fk_%(table_name)s_%(column_0_name)s_%(referred_table_name)s",
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
db = SQLAlchemy(metadata=metadata, session_options={"expire_on_commit": False})
|
|
||||||
migrate = Migrate()
|
|
||||||
|
|
||||||
|
|
||||||
@migrate.configure
|
|
||||||
def configure_alembic(config: Config):
|
|
||||||
"""Alembic configuration hook
|
|
||||||
|
|
||||||
Inject all migrations paths into the ``version_locations`` config option.
|
|
||||||
This includes even disabled plugins, as simply disabling a plugin without
|
|
||||||
uninstall can break the alembic version management.
|
|
||||||
"""
|
|
||||||
# Set main script location
|
|
||||||
config.set_main_option("script_location", alembic_script_path)
|
|
||||||
|
|
||||||
# Set Flaschengeist's migrations
|
|
||||||
migrations = [config.get_main_option("script_location") + "/migrations"]
|
|
||||||
|
|
||||||
# Gather all migration paths
|
|
||||||
for entry_point in entry_points(group="flaschengeist.plugins"):
|
|
||||||
try:
|
|
||||||
directory = entry_point.dist.locate_file("")
|
|
||||||
for loc in entry_point.module.split(".") + ["migrations"]:
|
|
||||||
directory /= loc
|
|
||||||
if directory.exists():
|
|
||||||
logger.debug(f"Adding migration version path {directory}")
|
|
||||||
migrations.append(str(directory.resolve()))
|
|
||||||
except:
|
|
||||||
logger.warning(f"Could not load migrations of plugin {entry_point.name} for database migration.")
|
|
||||||
logger.debug("Plugin loading failed", exc_info=True)
|
|
||||||
|
|
||||||
# write back seperator (we changed it if neither seperator nor locations were specified)
|
|
||||||
config.set_main_option("version_path_separator", os.pathsep)
|
|
||||||
config.set_main_option("version_locations", os.pathsep.join(set(migrations)))
|
|
||||||
return config
|
|
||||||
|
|
||||||
|
|
||||||
def case_sensitive(s):
|
|
||||||
"""
|
|
||||||
Compare string as case sensitive on the database
|
|
||||||
|
|
||||||
Args:
|
|
||||||
s: string to compare
|
|
||||||
|
|
||||||
Example:
|
|
||||||
User.query.filter(User.name == case_sensitive(some_string))
|
|
||||||
"""
|
|
||||||
if db.session.bind.dialect.name == "mysql":
|
|
||||||
from sqlalchemy import func
|
|
||||||
|
|
||||||
return func.binary(s)
|
|
||||||
return s
|
|
|
@ -1,97 +0,0 @@
|
||||||
from importlib import import_module
|
|
||||||
import datetime
|
|
||||||
|
|
||||||
from sqlalchemy import BigInteger, util
|
|
||||||
from sqlalchemy.dialects import mysql, sqlite
|
|
||||||
from sqlalchemy.types import DateTime, TypeDecorator
|
|
||||||
|
|
||||||
|
|
||||||
class ModelSerializeMixin:
|
|
||||||
"""Mixin class used for models to serialize them automatically
|
|
||||||
Ignores private and protected members as well as members marked as not to publish (name ends with _)
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __is_optional(self, param):
|
|
||||||
import typing
|
|
||||||
|
|
||||||
module = import_module("flaschengeist.models").__dict__
|
|
||||||
|
|
||||||
try:
|
|
||||||
hint = typing.get_type_hints(self.__class__, globalns=module, locals=locals())[param]
|
|
||||||
if (
|
|
||||||
typing.get_origin(hint) is typing.Union
|
|
||||||
and len(typing.get_args(hint)) == 2
|
|
||||||
and typing.get_args(hint)[1] is type(None)
|
|
||||||
):
|
|
||||||
return getattr(self, param) is None
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def serialize(self):
|
|
||||||
"""Serialize class to dict
|
|
||||||
Returns:
|
|
||||||
Dict of all not private or protected annotated member variables.
|
|
||||||
"""
|
|
||||||
d = {
|
|
||||||
param: getattr(self, param)
|
|
||||||
for param in self.__class__.__annotations__
|
|
||||||
if not param.startswith("_") and not param.endswith("_") and not self.__is_optional(param)
|
|
||||||
}
|
|
||||||
if len(d) == 1:
|
|
||||||
_, value = d.popitem()
|
|
||||||
return value
|
|
||||||
return d
|
|
||||||
|
|
||||||
def __str__(self) -> str:
|
|
||||||
return self.serialize().__str__()
|
|
||||||
|
|
||||||
|
|
||||||
class Serial(TypeDecorator):
|
|
||||||
"""Same as MariaDB Serial used for IDs"""
|
|
||||||
|
|
||||||
cache_ok = True
|
|
||||||
impl = BigInteger().with_variant(mysql.BIGINT(unsigned=True), "mysql").with_variant(sqlite.INTEGER(), "sqlite")
|
|
||||||
|
|
||||||
# https://alembic.sqlalchemy.org/en/latest/autogenerate.html?highlight=custom%20column#affecting-the-rendering-of-types-themselves
|
|
||||||
def __repr__(self) -> str:
|
|
||||||
return util.generic_repr(self)
|
|
||||||
|
|
||||||
|
|
||||||
class UtcDateTime(TypeDecorator):
|
|
||||||
"""Almost equivalent to `sqlalchemy.types.DateTime` with
|
|
||||||
``timezone=True`` option, but it differs from that by:
|
|
||||||
|
|
||||||
- Never silently take naive :class:`datetime.datetime`, instead it
|
|
||||||
always raise :exc:`ValueError` unless time zone aware value.
|
|
||||||
- :class:`datetime.datetime` value's :attr:`datetime.datetime.tzinfo`
|
|
||||||
is always converted to UTC.
|
|
||||||
- Unlike SQLAlchemy's built-in :class:`sqlalchemy.types.DateTime`,
|
|
||||||
it never return naive :class:`datetime.datetime`, but time zone
|
|
||||||
aware value, even with SQLite or MySQL.
|
|
||||||
"""
|
|
||||||
|
|
||||||
cache_ok = True
|
|
||||||
impl = DateTime(timezone=True)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def current_utc():
|
|
||||||
return datetime.datetime.now(tz=datetime.timezone.utc)
|
|
||||||
|
|
||||||
def process_bind_param(self, value, dialect):
|
|
||||||
if value is not None:
|
|
||||||
if not isinstance(value, datetime.datetime):
|
|
||||||
raise TypeError("expected datetime.datetime, not " + repr(value))
|
|
||||||
elif value.tzinfo is None:
|
|
||||||
raise ValueError("naive datetime is disallowed")
|
|
||||||
return value.astimezone(datetime.timezone.utc)
|
|
||||||
|
|
||||||
def process_result_value(self, value, dialect):
|
|
||||||
if value is not None:
|
|
||||||
if value.tzinfo is not None:
|
|
||||||
value = value.astimezone(datetime.timezone.utc)
|
|
||||||
value = value.replace(tzinfo=datetime.timezone.utc)
|
|
||||||
return value
|
|
||||||
|
|
||||||
# https://alembic.sqlalchemy.org/en/latest/autogenerate.html?highlight=custom%20column#affecting-the-rendering-of-types-themselves
|
|
||||||
def __repr__(self) -> str:
|
|
||||||
return util.generic_repr(self)
|
|
|
@ -11,29 +11,21 @@ root = "/api"
|
||||||
# Set secret key
|
# Set secret key
|
||||||
secret_key = "V3ryS3cr3t"
|
secret_key = "V3ryS3cr3t"
|
||||||
# Domain used by frontend
|
# Domain used by frontend
|
||||||
|
#domain = "flaschengeist.local"
|
||||||
[DATABASE]
|
|
||||||
# engine = "mysql" (default)
|
|
||||||
host = "localhost"
|
|
||||||
user = "flaschengeist"
|
|
||||||
password = "flaschengeist"
|
|
||||||
database = "flaschengeist"
|
|
||||||
|
|
||||||
[LOGGING]
|
[LOGGING]
|
||||||
# You can override all settings from the logging.toml here
|
# You can override all settings from the logging.toml here
|
||||||
# Default: Logging to WSGI stream (commonly stderr)
|
# E.g. override the formatters etc
|
||||||
|
#
|
||||||
# Logging level, possible: TRACE DEBUG INFO WARNING ERROR CRITICAL
|
# Logging level, possible: DEBUG INFO WARNING ERROR
|
||||||
# On TRACE level additionally every request will get logged
|
|
||||||
level = "DEBUG"
|
level = "DEBUG"
|
||||||
|
# Uncomment to enable logging to a file
|
||||||
|
# file = "/tmp/flaschengeist-debug.log"
|
||||||
|
# Uncomment to disable console logging
|
||||||
|
# console = False
|
||||||
|
|
||||||
# If you want the logger to log to a file, you could use:
|
[DATABASE]
|
||||||
#[LOGGING.handlers.file]
|
# engine = "mysql" (default)
|
||||||
# class = "logging.handlers.WatchedFileHandler"
|
|
||||||
# level = "WARNING"
|
|
||||||
# formatter = "extended"
|
|
||||||
# encoding = "utf8"
|
|
||||||
# filename = "flaschengeist.log"
|
|
||||||
|
|
||||||
[FILES]
|
[FILES]
|
||||||
# Path for file / image uploads
|
# Path for file / image uploads
|
||||||
|
@ -48,13 +40,12 @@ allowed_mimetypes = [
|
||||||
"image/webp"
|
"image/webp"
|
||||||
]
|
]
|
||||||
|
|
||||||
[scheduler]
|
[auth_plain]
|
||||||
# Possible values are: "passive_web" (default), "active_web" and "system"
|
enabled = true
|
||||||
# See documentation
|
|
||||||
# cron = "passive_web"
|
|
||||||
|
|
||||||
[auth_ldap]
|
[auth_ldap]
|
||||||
# Full documentation https://flaschengeist.dev/Flaschengeist/flaschengeist/wiki/plugins_auth_ldap
|
# Full documentation https://flaschengeist.dev/Flaschengeist/flaschengeist/wiki/plugins_auth_ldap
|
||||||
|
enabled = false
|
||||||
# host = "localhost"
|
# host = "localhost"
|
||||||
# port = 389
|
# port = 389
|
||||||
# base_dn = "dc=example,dc=com"
|
# base_dn = "dc=example,dc=com"
|
||||||
|
@ -123,5 +114,3 @@ If this was not you, please contact the support.
|
||||||
# enabled = true
|
# enabled = true
|
||||||
# Enable a default limit, will be set if no other limit is set
|
# Enable a default limit, will be set if no other limit is set
|
||||||
# limit = -10.00
|
# limit = -10.00
|
||||||
# Uncomment to allow active services to debit other users (requires events plugin)
|
|
||||||
# allow_service_debit = true
|
|
||||||
|
|
|
@ -6,16 +6,22 @@ disable_existing_loggers = false
|
||||||
|
|
||||||
[formatters]
|
[formatters]
|
||||||
[formatters.simple]
|
[formatters.simple]
|
||||||
format = "[%(asctime)s] %(levelname)s - %(message)s"
|
format = "%(asctime)s - %(levelname)s - %(message)s"
|
||||||
[formatters.extended]
|
[formatters.extended]
|
||||||
format = "[%(asctime)s] %(levelname)s %(filename)s - %(funcName)s - %(lineno)d - %(threadName)s - %(name)s — %(message)s"
|
format = "%(asctime)s — %(filename)s - %(funcName)s - %(lineno)d - %(threadName)s - %(name)s — %(levelname)s — %(message)s"
|
||||||
|
|
||||||
[handlers]
|
[handlers]
|
||||||
[handlers.wsgi]
|
[handlers.console]
|
||||||
stream = "ext://flask.logging.wsgi_errors_stream"
|
|
||||||
class = "logging.StreamHandler"
|
class = "logging.StreamHandler"
|
||||||
formatter = "simple"
|
|
||||||
level = "DEBUG"
|
level = "DEBUG"
|
||||||
|
formatter = "simple"
|
||||||
|
stream = "ext://sys.stdout"
|
||||||
|
[handlers.file]
|
||||||
|
class = "logging.handlers.WatchedFileHandler"
|
||||||
|
level = "WARNING"
|
||||||
|
formatter = "extended"
|
||||||
|
encoding = "utf8"
|
||||||
|
filename = "flaschengeist.log"
|
||||||
|
|
||||||
[loggers]
|
[loggers]
|
||||||
[loggers.werkzeug]
|
[loggers.werkzeug]
|
||||||
|
@ -23,4 +29,4 @@ disable_existing_loggers = false
|
||||||
|
|
||||||
[root]
|
[root]
|
||||||
level = "WARNING"
|
level = "WARNING"
|
||||||
handlers = ["wsgi"]
|
handlers = ["console"]
|
|
@ -1,5 +1,84 @@
|
||||||
from .session import *
|
import sys
|
||||||
from .user import *
|
import datetime
|
||||||
from .plugin import *
|
|
||||||
from .notification import *
|
from sqlalchemy import BigInteger
|
||||||
from .image import *
|
from sqlalchemy.dialects import mysql, sqlite
|
||||||
|
from sqlalchemy.types import DateTime, TypeDecorator
|
||||||
|
|
||||||
|
|
||||||
|
class ModelSerializeMixin:
|
||||||
|
"""Mixin class used for models to serialize them automatically
|
||||||
|
Ignores private and protected members as well as members marked as not to publish (name ends with _)
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __is_optional(self, param):
|
||||||
|
if sys.version_info < (3, 8):
|
||||||
|
return False
|
||||||
|
|
||||||
|
import typing
|
||||||
|
|
||||||
|
hint = typing.get_type_hints(self.__class__)[param]
|
||||||
|
if (
|
||||||
|
typing.get_origin(hint) is typing.Union
|
||||||
|
and len(typing.get_args(hint)) == 2
|
||||||
|
and typing.get_args(hint)[1] is type(None)
|
||||||
|
):
|
||||||
|
return getattr(self, param) is None
|
||||||
|
|
||||||
|
def serialize(self):
|
||||||
|
"""Serialize class to dict
|
||||||
|
Returns:
|
||||||
|
Dict of all not private or protected annotated member variables.
|
||||||
|
"""
|
||||||
|
d = {
|
||||||
|
param: getattr(self, param)
|
||||||
|
for param in self.__class__.__annotations__
|
||||||
|
if not param.startswith("_") and not param.endswith("_") and not self.__is_optional(param)
|
||||||
|
}
|
||||||
|
if len(d) == 1:
|
||||||
|
key, value = d.popitem()
|
||||||
|
return value
|
||||||
|
return d
|
||||||
|
|
||||||
|
|
||||||
|
class Serial(TypeDecorator):
|
||||||
|
"""Same as MariaDB Serial used for IDs"""
|
||||||
|
|
||||||
|
cache_ok = True
|
||||||
|
impl = BigInteger().with_variant(mysql.BIGINT(unsigned=True), "mysql").with_variant(sqlite.INTEGER, "sqlite")
|
||||||
|
|
||||||
|
|
||||||
|
class UtcDateTime(TypeDecorator):
|
||||||
|
"""Almost equivalent to `sqlalchemy.types.DateTime` with
|
||||||
|
``timezone=True`` option, but it differs from that by:
|
||||||
|
|
||||||
|
- Never silently take naive :class:`datetime.datetime`, instead it
|
||||||
|
always raise :exc:`ValueError` unless time zone aware value.
|
||||||
|
- :class:`datetime.datetime` value's :attr:`datetime.datetime.tzinfo`
|
||||||
|
is always converted to UTC.
|
||||||
|
- Unlike SQLAlchemy's built-in :class:`sqlalchemy.types.DateTime`,
|
||||||
|
it never return naive :class:`datetime.datetime`, but time zone
|
||||||
|
aware value, even with SQLite or MySQL.
|
||||||
|
"""
|
||||||
|
|
||||||
|
cache_ok = True
|
||||||
|
impl = DateTime(timezone=True)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def current_utc():
|
||||||
|
return datetime.datetime.now(tz=datetime.timezone.utc)
|
||||||
|
|
||||||
|
def process_bind_param(self, value, dialect):
|
||||||
|
if value is not None:
|
||||||
|
if not isinstance(value, datetime.datetime):
|
||||||
|
raise TypeError("expected datetime.datetime, not " + repr(value))
|
||||||
|
elif value.tzinfo is None:
|
||||||
|
raise ValueError("naive datetime is disallowed")
|
||||||
|
return value.astimezone(datetime.timezone.utc)
|
||||||
|
|
||||||
|
def process_result_value(self, value, dialect):
|
||||||
|
if value is not None:
|
||||||
|
if value.tzinfo is not None:
|
||||||
|
value = value.astimezone(datetime.timezone.utc)
|
||||||
|
value = value.replace(tzinfo=datetime.timezone.utc)
|
||||||
|
return value
|
||||||
|
|
|
@ -1,20 +1,19 @@
|
||||||
from __future__ import annotations # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
from sqlalchemy import event
|
from sqlalchemy import event
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
from . import ModelSerializeMixin, Serial
|
||||||
from ..database import db
|
from ..database import db
|
||||||
from ..database.types import ModelSerializeMixin, Serial
|
|
||||||
|
|
||||||
|
|
||||||
class Image(db.Model, ModelSerializeMixin):
|
class Image(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "image"
|
__tablename__ = "image"
|
||||||
id: int = db.Column(Serial, primary_key=True)
|
id: int = db.Column("id", Serial, primary_key=True)
|
||||||
filename_: str = db.Column("filename", db.String(255), nullable=False)
|
filename_: str = db.Column(db.String(127), nullable=False)
|
||||||
mimetype_: str = db.Column("mimetype", db.String(127), nullable=False)
|
mimetype_: str = db.Column(db.String(30), nullable=False)
|
||||||
thumbnail_: str = db.Column("thumbnail", db.String(255))
|
thumbnail_: str = db.Column(db.String(127))
|
||||||
path_: str = db.Column("path", db.String(255))
|
path_: str = db.Column(db.String(127))
|
||||||
|
|
||||||
def open(self):
|
def open(self):
|
||||||
return open(self.path_, "rb")
|
return open(self.path_, "rb")
|
||||||
|
|
|
@ -1,28 +1,19 @@
|
||||||
from __future__ import annotations # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
|
from . import Serial, UtcDateTime, ModelSerializeMixin
|
||||||
from ..database import db
|
from ..database import db
|
||||||
from ..database.types import Serial, UtcDateTime, ModelSerializeMixin
|
from .user import User
|
||||||
|
|
||||||
|
|
||||||
class Notification(db.Model, ModelSerializeMixin):
|
class Notification(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "notification"
|
__tablename__ = "notification"
|
||||||
id: int = db.Column("id", Serial, primary_key=True)
|
id: int = db.Column("id", Serial, primary_key=True)
|
||||||
|
plugin: str = db.Column(db.String(30), nullable=False)
|
||||||
text: str = db.Column(db.Text)
|
text: str = db.Column(db.Text)
|
||||||
data: Any = db.Column(db.PickleType(protocol=4))
|
data: Any = db.Column(db.PickleType(protocol=4))
|
||||||
time: datetime = db.Column(UtcDateTime, nullable=False, default=UtcDateTime.current_utc)
|
time: datetime = db.Column(UtcDateTime, nullable=False, default=UtcDateTime.current_utc)
|
||||||
|
|
||||||
user_id_: int = db.Column("user", Serial, db.ForeignKey("user.id"), nullable=False)
|
user_id_: int = db.Column("user_id", Serial, db.ForeignKey("user.id"), nullable=False)
|
||||||
plugin_id_: int = db.Column("plugin", Serial, db.ForeignKey("plugin.id"), nullable=False)
|
|
||||||
user_: User = db.relationship("User")
|
user_: User = db.relationship("User")
|
||||||
plugin_: Plugin = db.relationship(
|
|
||||||
"Plugin", backref=db.backref("notifications_", cascade="all, delete, delete-orphan")
|
|
||||||
)
|
|
||||||
plugin: str
|
|
||||||
|
|
||||||
@property
|
|
||||||
def plugin(self) -> str:
|
|
||||||
return self.plugin_.name
|
|
||||||
|
|
|
@ -1,74 +0,0 @@
|
||||||
from __future__ import annotations # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
|
||||||
|
|
||||||
from typing import Any, List, Dict
|
|
||||||
from sqlalchemy.orm.collections import attribute_mapped_collection
|
|
||||||
|
|
||||||
from ..database import db
|
|
||||||
from ..database.types import Serial
|
|
||||||
|
|
||||||
|
|
||||||
class PluginSetting(db.Model):
|
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "plugin_setting"
|
|
||||||
id = db.Column("id", Serial, primary_key=True)
|
|
||||||
plugin_id: int = db.Column("plugin", Serial, db.ForeignKey("plugin.id"))
|
|
||||||
name: str = db.Column(db.String(127), nullable=False)
|
|
||||||
value: Any = db.Column(db.PickleType(protocol=4))
|
|
||||||
|
|
||||||
|
|
||||||
class BasePlugin(db.Model):
|
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "plugin"
|
|
||||||
id: int = db.Column("id", Serial, primary_key=True)
|
|
||||||
name: str = db.Column(db.String(127), nullable=False)
|
|
||||||
"""Name of the plugin, loaded from distribution"""
|
|
||||||
installed_version: str = db.Column("version", db.String(30), nullable=False)
|
|
||||||
"""The latest installed version"""
|
|
||||||
enabled: bool = db.Column(db.Boolean, default=False)
|
|
||||||
"""Enabled state of the plugin"""
|
|
||||||
permissions: List["Permission"] = db.relationship(
|
|
||||||
"Permission", cascade="all, delete, delete-orphan", back_populates="plugin_", lazy="select"
|
|
||||||
)
|
|
||||||
"""Optional list of custom permissions used by the plugin
|
|
||||||
|
|
||||||
A good style is to name the permissions with a prefix related to the plugin name,
|
|
||||||
to prevent clashes with other plugins. E. g. instead of *delete* use *plugin_delete*.
|
|
||||||
"""
|
|
||||||
|
|
||||||
__settings: Dict[str, "PluginSetting"] = db.relationship(
|
|
||||||
"PluginSetting",
|
|
||||||
collection_class=attribute_mapped_collection("name"),
|
|
||||||
cascade="all, delete, delete-orphan",
|
|
||||||
lazy="subquery",
|
|
||||||
)
|
|
||||||
|
|
||||||
def get_setting(self, name: str, **kwargs):
|
|
||||||
"""Get plugin setting
|
|
||||||
|
|
||||||
Args:
|
|
||||||
name: string identifying the setting
|
|
||||||
default: Default value
|
|
||||||
Returns:
|
|
||||||
Value stored in database (native python)
|
|
||||||
Raises:
|
|
||||||
`KeyError` if no such setting exists in the database
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
return self.__settings[name].value
|
|
||||||
except KeyError as e:
|
|
||||||
if "default" in kwargs:
|
|
||||||
return kwargs["default"]
|
|
||||||
raise e
|
|
||||||
|
|
||||||
def set_setting(self, name: str, value):
|
|
||||||
"""Save setting in database
|
|
||||||
|
|
||||||
Args:
|
|
||||||
name: String identifying the setting
|
|
||||||
value: Value to be stored
|
|
||||||
"""
|
|
||||||
if value is None and name in self.__settings.keys():
|
|
||||||
del self.__settings[name]
|
|
||||||
else:
|
|
||||||
setting = self.__settings.setdefault(name, PluginSetting(plugin_id=self.id, name=name, value=None))
|
|
||||||
setting.value = value
|
|
|
@ -1,11 +1,12 @@
|
||||||
from __future__ import annotations # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
from datetime import datetime, timedelta, timezone
|
from datetime import datetime, timedelta, timezone
|
||||||
from secrets import compare_digest
|
|
||||||
|
|
||||||
from .. import logger
|
from . import ModelSerializeMixin, UtcDateTime, Serial
|
||||||
from ..database import db
|
from .user import User
|
||||||
from ..database.types import ModelSerializeMixin, UtcDateTime, Serial
|
from flaschengeist.database import db
|
||||||
|
from secrets import compare_digest
|
||||||
|
from flaschengeist import logger
|
||||||
|
|
||||||
|
|
||||||
class Session(db.Model, ModelSerializeMixin):
|
class Session(db.Model, ModelSerializeMixin):
|
||||||
|
@ -17,13 +18,12 @@ class Session(db.Model, ModelSerializeMixin):
|
||||||
token: String to verify access later.
|
token: String to verify access later.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "session"
|
__tablename__ = "session"
|
||||||
expires: datetime = db.Column(UtcDateTime)
|
expires: datetime = db.Column(UtcDateTime)
|
||||||
token: str = db.Column(db.String(32), unique=True)
|
token: str = db.Column(db.String(32), unique=True)
|
||||||
lifetime: int = db.Column(db.Integer)
|
lifetime: int = db.Column(db.Integer)
|
||||||
browser: str = db.Column(db.String(127))
|
browser: str = db.Column(db.String(30))
|
||||||
platform: str = db.Column(db.String(64))
|
platform: str = db.Column(db.String(30))
|
||||||
userid: str = ""
|
userid: str = ""
|
||||||
|
|
||||||
_id = db.Column("id", Serial, primary_key=True)
|
_id = db.Column("id", Serial, primary_key=True)
|
||||||
|
|
|
@ -0,0 +1,13 @@
|
||||||
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from . import Serial
|
||||||
|
from ..database import db
|
||||||
|
|
||||||
|
|
||||||
|
class _PluginSetting(db.Model):
|
||||||
|
__tablename__ = "plugin_setting"
|
||||||
|
id = db.Column("id", Serial, primary_key=True)
|
||||||
|
plugin: str = db.Column(db.String(30))
|
||||||
|
name: str = db.Column(db.String(30), nullable=False)
|
||||||
|
value: Any = db.Column(db.PickleType(protocol=4))
|
|
@ -1,13 +1,13 @@
|
||||||
from __future__ import (
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
annotations,
|
|
||||||
) # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
|
||||||
|
|
||||||
from typing import Optional, Union, List
|
from flask import url_for
|
||||||
|
from typing import Optional
|
||||||
from datetime import date, datetime
|
from datetime import date, datetime
|
||||||
from sqlalchemy.orm.collections import attribute_mapped_collection
|
from sqlalchemy.orm.collections import attribute_mapped_collection
|
||||||
|
|
||||||
from ..database import db
|
from ..database import db
|
||||||
from ..database.types import ModelSerializeMixin, UtcDateTime, Serial
|
from . import ModelSerializeMixin, UtcDateTime, Serial
|
||||||
|
|
||||||
|
|
||||||
association_table = db.Table(
|
association_table = db.Table(
|
||||||
"user_x_role",
|
"user_x_role",
|
||||||
|
@ -23,21 +23,17 @@ role_permission_association_table = db.Table(
|
||||||
|
|
||||||
|
|
||||||
class Permission(db.Model, ModelSerializeMixin):
|
class Permission(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "permission"
|
__tablename__ = "permission"
|
||||||
name: str = db.Column(db.String(30), unique=True)
|
name: str = db.Column(db.String(30), unique=True)
|
||||||
|
|
||||||
id_ = db.Column("id", Serial, primary_key=True)
|
_id = db.Column("id", Serial, primary_key=True)
|
||||||
plugin_id_: int = db.Column("plugin", Serial, db.ForeignKey("plugin.id"))
|
|
||||||
plugin_ = db.relationship("Plugin", lazy="subquery", back_populates="permissions", enable_typechecks=False)
|
|
||||||
|
|
||||||
|
|
||||||
class Role(db.Model, ModelSerializeMixin):
|
class Role(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "role"
|
__tablename__ = "role"
|
||||||
id: int = db.Column(Serial, primary_key=True)
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
name: str = db.Column(db.String(30), unique=True)
|
name: str = db.Column(db.String(30), unique=True)
|
||||||
permissions: List[Permission] = db.relationship("Permission", secondary=role_permission_association_table)
|
permissions: list[Permission] = db.relationship("Permission", secondary=role_permission_association_table)
|
||||||
|
|
||||||
|
|
||||||
class User(db.Model, ModelSerializeMixin):
|
class User(db.Model, ModelSerializeMixin):
|
||||||
|
@ -47,7 +43,7 @@ class User(db.Model, ModelSerializeMixin):
|
||||||
|
|
||||||
Attributes:
|
Attributes:
|
||||||
id: Id in Database as Primary Key.
|
id: Id in Database as Primary Key.
|
||||||
userid: User ID used by authentication provider
|
uid: User ID used by authentication provider
|
||||||
display_name: Name to show
|
display_name: Name to show
|
||||||
firstname: Firstname of the User
|
firstname: Firstname of the User
|
||||||
lastname: Lastname of the User
|
lastname: Lastname of the User
|
||||||
|
@ -55,35 +51,33 @@ class User(db.Model, ModelSerializeMixin):
|
||||||
birthday: Birthday of the user
|
birthday: Birthday of the user
|
||||||
"""
|
"""
|
||||||
|
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "user"
|
__tablename__ = "user"
|
||||||
userid: str = db.Column(db.String(30), unique=True, nullable=False)
|
userid: str = db.Column(db.String(30), nullable=False)
|
||||||
display_name: str = db.Column(db.String(30))
|
display_name: str = db.Column(db.String(30))
|
||||||
firstname: str = db.Column(db.String(50), nullable=False)
|
firstname: str = db.Column(db.String(50), nullable=False)
|
||||||
lastname: str = db.Column(db.String(50), nullable=False)
|
lastname: str = db.Column(db.String(50), nullable=False)
|
||||||
deleted: bool = db.Column(db.Boolean(), default=False)
|
|
||||||
birthday: Optional[date] = db.Column(db.Date)
|
|
||||||
mail: str = db.Column(db.String(60))
|
mail: str = db.Column(db.String(60))
|
||||||
roles: List[str] = []
|
birthday: Optional[date] = db.Column(db.Date)
|
||||||
permissions: Optional[list[str]] = []
|
roles: list[str] = []
|
||||||
|
permissions: Optional[list[str]] = None
|
||||||
|
avatar_url: Optional[str] = ""
|
||||||
|
|
||||||
# Protected stuff for backend use only
|
|
||||||
id_ = db.Column("id", Serial, primary_key=True)
|
id_ = db.Column("id", Serial, primary_key=True)
|
||||||
roles_: List[Role] = db.relationship("Role", secondary=association_table, cascade="save-update, merge")
|
roles_: list[Role] = db.relationship("Role", secondary=association_table, cascade="save-update, merge")
|
||||||
sessions_: List[Session] = db.relationship("Session", back_populates="user_", cascade="all, delete, delete-orphan")
|
sessions_ = db.relationship("Session", back_populates="user_")
|
||||||
avatar_: Optional[Image] = db.relationship("Image", cascade="all, delete, delete-orphan", single_parent=True)
|
|
||||||
reset_requests_: List["_PasswordReset"] = db.relationship("_PasswordReset", cascade="all, delete, delete-orphan")
|
|
||||||
|
|
||||||
# Private stuff for internal use
|
|
||||||
_avatar_id = db.Column("avatar", Serial, db.ForeignKey("image.id"))
|
|
||||||
_attributes = db.relationship(
|
_attributes = db.relationship(
|
||||||
"_UserAttribute",
|
"_UserAttribute",
|
||||||
collection_class=attribute_mapped_collection("name"),
|
collection_class=attribute_mapped_collection("name"),
|
||||||
cascade="all, delete, delete-orphan",
|
cascade="all, delete",
|
||||||
)
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def roles(self) -> List[str]:
|
def avatar_url(self):
|
||||||
|
return url_for("users.get_avatar", userid=self.userid) if self.userid else None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def roles(self):
|
||||||
return [role.name for role in self.roles_]
|
return [role.name for role in self.roles_]
|
||||||
|
|
||||||
def set_attribute(self, name, value):
|
def set_attribute(self, name, value):
|
||||||
|
@ -112,7 +106,6 @@ class User(db.Model, ModelSerializeMixin):
|
||||||
|
|
||||||
|
|
||||||
class _UserAttribute(db.Model, ModelSerializeMixin):
|
class _UserAttribute(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "user_attribute"
|
__tablename__ = "user_attribute"
|
||||||
id = db.Column("id", Serial, primary_key=True)
|
id = db.Column("id", Serial, primary_key=True)
|
||||||
user: User = db.Column("user", Serial, db.ForeignKey("user.id"), nullable=False)
|
user: User = db.Column("user", Serial, db.ForeignKey("user.id"), nullable=False)
|
||||||
|
@ -123,10 +116,9 @@ class _UserAttribute(db.Model, ModelSerializeMixin):
|
||||||
class _PasswordReset(db.Model):
|
class _PasswordReset(db.Model):
|
||||||
"""Table containing password reset requests"""
|
"""Table containing password reset requests"""
|
||||||
|
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "password_reset"
|
__tablename__ = "password_reset"
|
||||||
_user_id: User = db.Column("user", Serial, db.ForeignKey("user.id"), primary_key=True)
|
_user_id: User = db.Column("user", Serial, db.ForeignKey("user.id"), primary_key=True)
|
||||||
user: User = db.relationship("User", back_populates="reset_requests_", foreign_keys=[_user_id])
|
user: User = db.relationship("User", foreign_keys=[_user_id])
|
||||||
token: str = db.Column(db.String(32))
|
token: str = db.Column(db.String(32))
|
||||||
expires: datetime = db.Column(UtcDateTime)
|
expires: datetime = db.Column(UtcDateTime)
|
||||||
|
|
||||||
|
|
|
@ -1,159 +1,103 @@
|
||||||
"""Flaschengeist Plugins
|
import sqlalchemy
|
||||||
|
import pkg_resources
|
||||||
|
from werkzeug.exceptions import MethodNotAllowed, NotFound
|
||||||
|
|
||||||
.. include:: docs/plugin_development.md
|
from flaschengeist.database import db
|
||||||
|
from flaschengeist.models.notification import Notification
|
||||||
"""
|
from flaschengeist.models.user import _Avatar
|
||||||
|
from flaschengeist.models.setting import _PluginSetting
|
||||||
from typing import Union, List
|
|
||||||
from importlib.metadata import entry_points
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
from werkzeug.datastructures import FileStorage
|
|
||||||
|
|
||||||
from flaschengeist.models.plugin import BasePlugin
|
|
||||||
from flaschengeist.models.user import _Avatar, Permission
|
|
||||||
from flaschengeist.utils.hook import HookBefore, HookAfter
|
from flaschengeist.utils.hook import HookBefore, HookAfter
|
||||||
|
|
||||||
__all__ = [
|
|
||||||
"plugins_installed",
|
|
||||||
"plugins_loaded",
|
|
||||||
"before_delete_user",
|
|
||||||
"before_role_updated",
|
|
||||||
"before_update_user",
|
|
||||||
"after_role_updated",
|
|
||||||
"Plugin",
|
|
||||||
"AuthPlugin",
|
|
||||||
]
|
|
||||||
|
|
||||||
# Documentation hacks, see https://github.com/mitmproxy/pdoc/issues/320
|
|
||||||
plugins_installed = HookAfter("plugins.installed")
|
|
||||||
plugins_installed.__doc__ = """Hook decorator for when all plugins are installed
|
|
||||||
|
|
||||||
Possible use case would be to populate the database with some presets.
|
|
||||||
"""
|
|
||||||
|
|
||||||
plugins_loaded = HookAfter("plugins.loaded")
|
|
||||||
plugins_loaded.__doc__ = """Hook decorator for when all plugins are loaded
|
|
||||||
|
|
||||||
Possible use case would be to check if a specific other plugin is loaded and change own behavior
|
|
||||||
|
|
||||||
Passed args:
|
|
||||||
- *app:* Current flask app instance (args)
|
|
||||||
"""
|
|
||||||
|
|
||||||
before_role_updated = HookBefore("update_role")
|
before_role_updated = HookBefore("update_role")
|
||||||
before_role_updated.__doc__ = """Hook decorator for when roles are modified
|
"""Hook decorator for when roles are modified
|
||||||
|
Args:
|
||||||
Passed args:
|
role: Role object to modify
|
||||||
- *role:* `flaschengeist.models.user.Role` to modify
|
new_name: New name if the name was changed (None if delete)
|
||||||
- *new_name:* New name if the name was changed (*None* if delete)
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
after_role_updated = HookAfter("update_role")
|
after_role_updated = HookAfter("update_role")
|
||||||
after_role_updated.__doc__ = """Hook decorator for when roles are modified
|
"""Hook decorator for when roles are modified
|
||||||
|
Args:
|
||||||
Passed args:
|
role: Role object containing the modified role
|
||||||
- *role:* modified `flaschengeist.models.user.Role`
|
new_name: New name if the name was changed (None if deleted)
|
||||||
- *new_name:* New name if the name was changed (*None* if deleted)
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
before_update_user = HookBefore("update_user")
|
before_update_user = HookBefore("update_user")
|
||||||
before_update_user.__doc__ = """Hook decorator, when ever an user update is done, this is called before.
|
"""Hook decorator, when ever an user update is done, this is called before.
|
||||||
|
Args:
|
||||||
Passed args:
|
user: User object
|
||||||
- *user:* `flaschengeist.models.user.User` object
|
|
||||||
"""
|
|
||||||
|
|
||||||
before_delete_user = HookBefore("delete_user")
|
|
||||||
before_delete_user.__doc__ = """Hook decorator,this is called before an user gets deleted.
|
|
||||||
|
|
||||||
Passed args:
|
|
||||||
- *user:* `flaschengeist.models.user.User` object
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
class Plugin(BasePlugin):
|
class Plugin:
|
||||||
"""Base class for all Plugins
|
"""Base class for all Plugins
|
||||||
|
If your class uses custom models add a static property called ``models``"""
|
||||||
|
|
||||||
All plugins must derived from this class.
|
blueprint = None # You have to override
|
||||||
|
permissions = [] # You have to override
|
||||||
|
id = "dev.flaschengeist.plugin" # You have to override
|
||||||
|
name = "plugin" # You have to override
|
||||||
|
models = None # You have to override
|
||||||
|
|
||||||
Optional:
|
def __init__(self, config=None):
|
||||||
- *blueprint*: `flask.Blueprint` providing your routes
|
"""Constructor called by create_app
|
||||||
- *permissions*: List of your custom permissions
|
Args:
|
||||||
- *models*: Your models, used for API export
|
config: Dict configuration containing the plugin section
|
||||||
"""
|
"""
|
||||||
|
self.version = pkg_resources.get_distribution(self.__module__.split(".")[0]).version
|
||||||
blueprint = None
|
|
||||||
"""Optional `flask.blueprint` if the plugin uses custom routes"""
|
|
||||||
|
|
||||||
models = None
|
|
||||||
"""Optional module containing the SQLAlchemy models used by the plugin"""
|
|
||||||
|
|
||||||
@property
|
|
||||||
def version(self) -> str:
|
|
||||||
"""Version of the plugin, loaded from Distribution"""
|
|
||||||
return self.dist.version
|
|
||||||
|
|
||||||
@property
|
|
||||||
def dist(self):
|
|
||||||
"""Distribution of this plugin"""
|
|
||||||
return self.entry_point.dist
|
|
||||||
|
|
||||||
@property
|
|
||||||
def entry_point(self):
|
|
||||||
ep = tuple(entry_points(group="flaschengeist.plugins", name=self.name))
|
|
||||||
return ep[0]
|
|
||||||
|
|
||||||
def load(self):
|
|
||||||
"""__init__ like function that is called when the plugin is initially loaded"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
def install(self):
|
def install(self):
|
||||||
"""Installation routine
|
"""Installation routine
|
||||||
|
Is always called with Flask application context
|
||||||
Also called when updating the plugin, compare `version` and `installed_version`.
|
|
||||||
|
|
||||||
Is always called with Flask application context,
|
|
||||||
it is called after the plugin permissions are installed.
|
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def uninstall(self):
|
def post_install(self):
|
||||||
"""Uninstall routine
|
"""Fill database or do other stuff
|
||||||
|
Called after all plugins are installed
|
||||||
If the plugin has custom database tables, make sure to remove them.
|
|
||||||
This can be either done by downgrading the plugin *head* to the *base*.
|
|
||||||
Or use custom migrations for the uninstall and *stamp* some version.
|
|
||||||
|
|
||||||
Is always called with Flask application context.
|
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_setting(self, name: str, **kwargs):
|
||||||
|
"""Get plugin setting from database
|
||||||
|
Args:
|
||||||
|
name: string identifying the setting
|
||||||
|
default: Default value
|
||||||
|
Returns:
|
||||||
|
Value stored in database (native python)
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
setting = (
|
||||||
|
_PluginSetting.query.filter(_PluginSetting.plugin == self.name)
|
||||||
|
.filter(_PluginSetting.name == name)
|
||||||
|
.one()
|
||||||
|
)
|
||||||
|
return setting.value
|
||||||
|
except sqlalchemy.orm.exc.NoResultFound:
|
||||||
|
if "default" in kwargs:
|
||||||
|
return kwargs["default"]
|
||||||
|
else:
|
||||||
|
raise KeyError
|
||||||
|
|
||||||
|
def set_setting(self, name: str, value):
|
||||||
|
"""Save setting in database
|
||||||
|
Args:
|
||||||
|
name: String identifying the setting
|
||||||
|
value: Value to be stored
|
||||||
|
"""
|
||||||
|
setting = (
|
||||||
|
_PluginSetting.query.filter(_PluginSetting.plugin == self.name)
|
||||||
|
.filter(_PluginSetting.name == name)
|
||||||
|
.one_or_none()
|
||||||
|
)
|
||||||
|
if setting is not None:
|
||||||
|
setting.value = value
|
||||||
|
else:
|
||||||
|
db.session.add(_PluginSetting(plugin=self.name, name=name, value=value))
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
def notify(self, user, text: str, data=None):
|
def notify(self, user, text: str, data=None):
|
||||||
"""Create a new notification for an user
|
n = Notification(text=text, data=data, plugin=self.id, user_=user)
|
||||||
|
db.session.add(n)
|
||||||
Args:
|
db.session.commit()
|
||||||
user: `flaschengeist.models.user.User` to notify
|
|
||||||
text: Visibile notification text
|
|
||||||
data: Optional data passed to the notificaton
|
|
||||||
Returns:
|
|
||||||
ID of the created `flaschengeist.models.notification.Notification`
|
|
||||||
|
|
||||||
Hint: use the data for frontend actions.
|
|
||||||
"""
|
|
||||||
from ..controller import pluginController
|
|
||||||
|
|
||||||
return pluginController.notify(self.id, user, text, data)
|
|
||||||
|
|
||||||
@property
|
|
||||||
def notifications(self) -> List["Notification"]:
|
|
||||||
"""Get all notifications for this plugin
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
List of `flaschengeist.models.notification.Notification`
|
|
||||||
"""
|
|
||||||
from ..controller import pluginController
|
|
||||||
|
|
||||||
return pluginController.get_notifications(self.id)
|
|
||||||
|
|
||||||
def serialize(self):
|
def serialize(self):
|
||||||
"""Serialize a plugin into a dict
|
"""Serialize a plugin into a dict
|
||||||
|
@ -163,53 +107,35 @@ class Plugin(BasePlugin):
|
||||||
"""
|
"""
|
||||||
return {"version": self.version, "permissions": self.permissions}
|
return {"version": self.version, "permissions": self.permissions}
|
||||||
|
|
||||||
def install_permissions(self, permissions: list[str]):
|
|
||||||
"""Helper for installing a list of strings as permissions
|
|
||||||
|
|
||||||
Args:
|
|
||||||
permissions: List of permissions to install
|
|
||||||
"""
|
|
||||||
cur_perm = set(x for x in self.permissions or [])
|
|
||||||
all_perm = set(permissions)
|
|
||||||
|
|
||||||
new_perms = all_perm - cur_perm
|
|
||||||
_perms = [Permission(name=x, plugin_=self) for x in new_perms]
|
|
||||||
# self.permissions = list(filter(lambda x: x.name in permissions, self.permissions and isinstance(self.permissions, list) or []))
|
|
||||||
self.permissions.extend(_perms)
|
|
||||||
|
|
||||||
|
|
||||||
class AuthPlugin(Plugin):
|
class AuthPlugin(Plugin):
|
||||||
"""Base class for all authentification plugins
|
def login(self, user, pw):
|
||||||
|
|
||||||
See also `Plugin`
|
|
||||||
"""
|
|
||||||
|
|
||||||
def login(self, login_name, password) -> Union[bool, str]:
|
|
||||||
"""Login routine, MUST BE IMPLEMENTED!
|
"""Login routine, MUST BE IMPLEMENTED!
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
login_name: The name the user entered
|
user: User class containing at least the uid
|
||||||
password: The password the user used to log in
|
pw: given password
|
||||||
Returns:
|
Returns:
|
||||||
Must return False if not found or invalid credentials, otherwise the UID is returned
|
Must return False if not found or invalid credentials, True if success
|
||||||
"""
|
"""
|
||||||
raise NotImplemented
|
raise NotImplemented
|
||||||
|
|
||||||
def update_user(self, user: "User"):
|
def update_user(self, user):
|
||||||
"""If backend is using external data, then update this user instance with external data
|
"""If backend is using external data, then update this user instance with external data
|
||||||
Args:
|
Args:
|
||||||
user: User object
|
user: User object
|
||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def user_exists(self, userid) -> bool:
|
def find_user(self, userid, mail=None):
|
||||||
"""Check if user exists on this backend
|
"""Find an user by userid or mail
|
||||||
Args:
|
Args:
|
||||||
userid: Userid to search
|
userid: Userid to search
|
||||||
|
mail: If set, mail to search
|
||||||
Returns:
|
Returns:
|
||||||
True or False
|
None or User
|
||||||
"""
|
"""
|
||||||
raise NotImplemented
|
return None
|
||||||
|
|
||||||
def modify_user(self, user, password, new_password=None):
|
def modify_user(self, user, password, new_password=None):
|
||||||
"""If backend is using (writeable) external data, then update the external database with the user provided.
|
"""If backend is using (writeable) external data, then update the external database with the user provided.
|
||||||
|
@ -220,14 +146,11 @@ class AuthPlugin(Plugin):
|
||||||
password: Password (some backends need the current password for changes) if None force edit (admin)
|
password: Password (some backends need the current password for changes) if None force edit (admin)
|
||||||
new_password: If set a password change is requested
|
new_password: If set a password change is requested
|
||||||
Raises:
|
Raises:
|
||||||
|
NotImplemented: If backend does not support this feature (or no password change)
|
||||||
BadRequest: Logic error, e.g. password is wrong.
|
BadRequest: Logic error, e.g. password is wrong.
|
||||||
Error: Other errors if backend went mad (are not handled and will result in a 500 error)
|
Error: Other errors if backend went mad (are not handled and will result in a 500 error)
|
||||||
"""
|
"""
|
||||||
pass
|
raise NotImplemented
|
||||||
|
|
||||||
def can_register(self):
|
|
||||||
"""Check if this backend allows to register new users"""
|
|
||||||
return False
|
|
||||||
|
|
||||||
def create_user(self, user, password):
|
def create_user(self, user, password):
|
||||||
"""If backend is using (writeable) external data, then create a new user on the external database.
|
"""If backend is using (writeable) external data, then create a new user on the external database.
|
||||||
|
@ -237,7 +160,7 @@ class AuthPlugin(Plugin):
|
||||||
password: string
|
password: string
|
||||||
|
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
raise MethodNotAllowed
|
||||||
|
|
||||||
def delete_user(self, user):
|
def delete_user(self, user):
|
||||||
"""If backend is using (writeable) external data, then delete the user from external database.
|
"""If backend is using (writeable) external data, then delete the user from external database.
|
||||||
|
@ -246,25 +169,11 @@ class AuthPlugin(Plugin):
|
||||||
user: User object
|
user: User object
|
||||||
|
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
raise MethodNotAllowed
|
||||||
|
|
||||||
def get_modified_time(self, user):
|
|
||||||
"""If backend is using external data, then return the timestamp of the last modification
|
|
||||||
|
|
||||||
Args:
|
|
||||||
user: User object
|
|
||||||
Returns:
|
|
||||||
Timestamp of last modification
|
|
||||||
"""
|
|
||||||
pass
|
|
||||||
|
|
||||||
def get_avatar(self, user) -> _Avatar:
|
def get_avatar(self, user) -> _Avatar:
|
||||||
"""Retrieve avatar for given user (if supported by auth backend)
|
"""Retrieve avatar for given user (if supported by auth backend)
|
||||||
|
|
||||||
Default behavior is to use native Image objects,
|
|
||||||
so by default this function is never called, as the userController checks
|
|
||||||
native Image objects first.
|
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
user: User to retrieve the avatar for
|
user: User to retrieve the avatar for
|
||||||
Raises:
|
Raises:
|
||||||
|
@ -272,33 +181,24 @@ class AuthPlugin(Plugin):
|
||||||
"""
|
"""
|
||||||
raise NotFound
|
raise NotFound
|
||||||
|
|
||||||
def set_avatar(self, user, file: FileStorage):
|
def set_avatar(self, user, avatar: _Avatar):
|
||||||
"""Set the avatar for given user (if supported by auth backend)
|
"""Set the avatar for given user (if supported by auth backend)
|
||||||
|
|
||||||
Default behavior is to use native Image objects stored on the Flaschengeist server
|
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
user: User to set the avatar for
|
user: User to set the avatar for
|
||||||
file: `werkzeug.datastructures.FileStorage` uploaded by the user
|
avatar: Avatar to set
|
||||||
Raises:
|
Raises:
|
||||||
MethodNotAllowed: If not supported by Backend
|
MethodNotAllowed: If not supported by Backend
|
||||||
Any valid HTTP exception
|
|
||||||
"""
|
"""
|
||||||
# By default save the image to the avatar,
|
raise MethodNotAllowed
|
||||||
# deleting would happen by unsetting it
|
|
||||||
from ..controller import imageController
|
|
||||||
|
|
||||||
user.avatar_ = imageController.upload_image(file)
|
|
||||||
|
|
||||||
def delete_avatar(self, user):
|
def delete_avatar(self, user):
|
||||||
"""Delete the avatar for given user (if supported by auth backend)
|
"""Delete the avatar for given user (if supported by auth backend)
|
||||||
|
|
||||||
Default behavior is to use the imageController and native Image objects.
|
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
user: Uset to delete the avatar for
|
user: Uset to delete the avatar for
|
||||||
|
|
||||||
Raises:
|
Raises:
|
||||||
MethodNotAllowed: If not supported by Backend
|
MethodNotAllowed: If not supported by Backend
|
||||||
"""
|
"""
|
||||||
user.avatar_ = None
|
raise MethodNotAllowed
|
||||||
|
|
|
@ -13,7 +13,8 @@ from flaschengeist.controller import sessionController, userController
|
||||||
|
|
||||||
|
|
||||||
class AuthRoutePlugin(Plugin):
|
class AuthRoutePlugin(Plugin):
|
||||||
blueprint = Blueprint("auth", __name__)
|
name = "auth"
|
||||||
|
blueprint = Blueprint(name, __name__)
|
||||||
|
|
||||||
|
|
||||||
@AuthRoutePlugin.blueprint.route("/auth", methods=["POST"])
|
@AuthRoutePlugin.blueprint.route("/auth", methods=["POST"])
|
||||||
|
@ -36,13 +37,13 @@ def login():
|
||||||
except (KeyError, ValueError, TypeError):
|
except (KeyError, ValueError, TypeError):
|
||||||
raise BadRequest("Missing parameter(s)")
|
raise BadRequest("Missing parameter(s)")
|
||||||
|
|
||||||
logger.debug(f"search user {userid} in database")
|
logger.debug("search user {{ {} }} in database".format(userid))
|
||||||
user = userController.login_user(userid, password)
|
user = userController.login_user(userid, password)
|
||||||
if not user:
|
if not user:
|
||||||
raise Unauthorized
|
raise Unauthorized
|
||||||
session = sessionController.create(user, request_headers=request.headers)
|
session = sessionController.create(user, user_agent=request.user_agent)
|
||||||
logger.debug(f"token is {session.token}")
|
logger.debug("token is {{ {} }}".format(session.token))
|
||||||
logger.info(f"User {userid} logged in.")
|
logger.info("User {{ {} }} success login.".format(userid))
|
||||||
|
|
||||||
# Lets cleanup the DB
|
# Lets cleanup the DB
|
||||||
sessionController.clear_expired()
|
sessionController.clear_expired()
|
||||||
|
@ -165,7 +166,7 @@ def get_assocd_user(token, current_session, **kwargs):
|
||||||
def reset_password():
|
def reset_password():
|
||||||
data = request.get_json()
|
data = request.get_json()
|
||||||
if "userid" in data:
|
if "userid" in data:
|
||||||
user = userController.get_user(data["userid"])
|
user = userController.find_user(data["userid"])
|
||||||
if user:
|
if user:
|
||||||
userController.request_reset(user)
|
userController.request_reset(user)
|
||||||
elif "password" in data and "token" in data:
|
elif "password" in data and "token" in data:
|
||||||
|
|
|
@ -1,82 +1,78 @@
|
||||||
"""LDAP Authentication Provider Plugin"""
|
"""LDAP Authentication Provider Plugin"""
|
||||||
|
import io
|
||||||
import os
|
import os
|
||||||
import ssl
|
import ssl
|
||||||
from PIL import Image
|
from typing import Optional
|
||||||
from io import BytesIO
|
|
||||||
from flask_ldapconn import LDAPConn
|
from flask_ldapconn import LDAPConn
|
||||||
from flask import current_app as app
|
from flask import current_app as app
|
||||||
from ldap3 import SUBTREE, MODIFY_REPLACE, MODIFY_ADD, MODIFY_DELETE
|
|
||||||
from ldap3.core.exceptions import LDAPPasswordIsMandatoryError, LDAPBindError
|
from ldap3.core.exceptions import LDAPPasswordIsMandatoryError, LDAPBindError
|
||||||
|
from ldap3 import SUBTREE, MODIFY_REPLACE, MODIFY_ADD, MODIFY_DELETE
|
||||||
from werkzeug.exceptions import BadRequest, InternalServerError, NotFound
|
from werkzeug.exceptions import BadRequest, InternalServerError, NotFound
|
||||||
from werkzeug.datastructures import FileStorage
|
|
||||||
|
|
||||||
from datetime import datetime
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
from flaschengeist import logger
|
||||||
from flaschengeist.config import config
|
|
||||||
from flaschengeist.controller import userController
|
|
||||||
from flaschengeist.models import User, Role
|
|
||||||
from flaschengeist.models.user import _Avatar
|
|
||||||
from flaschengeist.plugins import AuthPlugin, before_role_updated
|
from flaschengeist.plugins import AuthPlugin, before_role_updated
|
||||||
|
from flaschengeist.models.user import User, Role, _Avatar
|
||||||
|
import flaschengeist.controller.userController as userController
|
||||||
|
|
||||||
|
|
||||||
class AuthLDAP(AuthPlugin):
|
class AuthLDAP(AuthPlugin):
|
||||||
def load(self):
|
def __init__(self, config):
|
||||||
self.config = config.get("auth_ldap", None)
|
super().__init__()
|
||||||
if self.config is None:
|
|
||||||
logger.error("auth_ldap was not configured in flaschengeist.toml", exc_info=True)
|
|
||||||
raise InternalServerError
|
|
||||||
app.config.update(
|
app.config.update(
|
||||||
LDAP_SERVER=self.config.get("host", "localhost"),
|
LDAP_SERVER=config.get("host", "localhost"),
|
||||||
LDAP_PORT=self.config.get("port", 389),
|
LDAP_PORT=config.get("port", 389),
|
||||||
LDAP_BINDDN=self.config.get("bind_dn", None),
|
LDAP_BINDDN=config.get("bind_dn", None),
|
||||||
LDAP_SECRET=self.config.get("secret", None),
|
LDAP_SECRET=config.get("secret", None),
|
||||||
LDAP_USE_SSL=self.config.get("use_ssl", False),
|
LDAP_USE_SSL=config.get("use_ssl", False),
|
||||||
# That's not TLS, its dirty StartTLS on unencrypted LDAP
|
# That's not TLS, its dirty StartTLS on unencrypted LDAP
|
||||||
LDAP_USE_TLS=False,
|
LDAP_USE_TLS=False,
|
||||||
LDAP_TLS_VERSION=ssl.PROTOCOL_TLS,
|
LDAP_TLS_VERSION=ssl.PROTOCOL_TLS,
|
||||||
FORCE_ATTRIBUTE_VALUE_AS_LIST=True,
|
FORCE_ATTRIBUTE_VALUE_AS_LIST=True,
|
||||||
)
|
)
|
||||||
if "ca_cert" in config:
|
if "ca_cert" in config:
|
||||||
app.config["LDAP_CA_CERTS_FILE"] = self.config["ca_cert"]
|
app.config["LDAP_CA_CERTS_FILE"] = config["ca_cert"]
|
||||||
else:
|
else:
|
||||||
# Default is CERT_REQUIRED
|
# Default is CERT_REQUIRED
|
||||||
app.config["LDAP_REQUIRE_CERT"] = ssl.CERT_OPTIONAL
|
app.config["LDAP_REQUIRE_CERT"] = ssl.CERT_OPTIONAL
|
||||||
self.ldap = LDAPConn(app)
|
self.ldap = LDAPConn(app)
|
||||||
self.base_dn = self.config["base_dn"]
|
self.base_dn = config["base_dn"]
|
||||||
self.search_dn = self.config.get("search_dn", "ou=people,{base_dn}").format(base_dn=self.base_dn)
|
self.search_dn = config.get("search_dn", "ou=people,{base_dn}").format(base_dn=self.base_dn)
|
||||||
self.group_dn = self.config.get("group_dn", "ou=group,{base_dn}").format(base_dn=self.base_dn)
|
self.group_dn = config.get("group_dn", "ou=group,{base_dn}").format(base_dn=self.base_dn)
|
||||||
self.password_hash = self.config.get("password_hash", "SSHA").upper()
|
self.password_hash = config.get("password_hash", "SSHA").upper()
|
||||||
self.object_classes = self.config.get("object_classes", ["inetOrgPerson"])
|
self.object_classes = config.get("object_classes", ["inetOrgPerson"])
|
||||||
self.user_attributes: dict = self.config.get("user_attributes", {})
|
self.user_attributes: dict = config.get("user_attributes", {})
|
||||||
self.dn_template = self.config.get("dn_template")
|
self.dn_template = config.get("dn_template")
|
||||||
|
|
||||||
# TODO: might not be set if modify is called
|
# TODO: might not be set if modify is called
|
||||||
self.root_dn = self.config.get("root_dn", None)
|
self.root_dn = config.get("root_dn", None)
|
||||||
self.root_secret = self.config.get("root_secret", None)
|
self.root_secret = config.get("root_secret", None)
|
||||||
|
|
||||||
@before_role_updated
|
@before_role_updated
|
||||||
def _role_updated(role, new_name):
|
def _role_updated(role, new_name):
|
||||||
logger.debug(f"LDAP: before_role_updated called with ({role}, {new_name})")
|
logger.debug(f"LDAP: before_role_updated called with ({role}, {new_name})")
|
||||||
self.__modify_role(role, new_name)
|
self.__modify_role(role, new_name)
|
||||||
|
|
||||||
def login(self, login_name, password):
|
def login(self, user, password):
|
||||||
if not login_name:
|
if not user:
|
||||||
return False
|
return False
|
||||||
return login_name if self.ldap.authenticate(login_name, password, "uid", self.base_dn) else False
|
return self.ldap.authenticate(user.userid, password, "uid", self.base_dn)
|
||||||
|
|
||||||
def user_exists(self, userid) -> bool:
|
def find_user(self, userid, mail=None):
|
||||||
attr = self.__find(userid, None)
|
attr = self.__find(userid, mail)
|
||||||
return attr is not None
|
if attr is not None:
|
||||||
|
user = User(userid=attr["uid"][0])
|
||||||
|
self.__update(user, attr)
|
||||||
|
return user
|
||||||
|
|
||||||
def update_user(self, user):
|
def update_user(self, user):
|
||||||
attr = self.__find(user.userid)
|
attr = self.__find(user.userid)
|
||||||
self.__update(user, attr)
|
self.__update(user, attr)
|
||||||
|
|
||||||
def can_register(self):
|
|
||||||
return self.root_dn is not None
|
|
||||||
|
|
||||||
def create_user(self, user, password):
|
def create_user(self, user, password):
|
||||||
|
if self.root_dn is None:
|
||||||
|
logger.error("root_dn missing in ldap config!")
|
||||||
|
raise InternalServerError
|
||||||
|
|
||||||
try:
|
try:
|
||||||
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
||||||
attributes = self.user_attributes.copy()
|
attributes = self.user_attributes.copy()
|
||||||
|
@ -118,7 +114,7 @@ class AuthLDAP(AuthPlugin):
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
if user.display_name:
|
if user.display_name:
|
||||||
attributes.update({"displayName": user.display_name})
|
attributes.update( {"displayName": user.display_name})
|
||||||
ldap_conn.add(dn, self.object_classes, attributes)
|
ldap_conn.add(dn, self.object_classes, attributes)
|
||||||
self._set_roles(user)
|
self._set_roles(user)
|
||||||
self.update_user(user)
|
self.update_user(user)
|
||||||
|
@ -128,12 +124,9 @@ class AuthLDAP(AuthPlugin):
|
||||||
def modify_user(self, user: User, password=None, new_password=None):
|
def modify_user(self, user: User, password=None, new_password=None):
|
||||||
try:
|
try:
|
||||||
dn = user.get_attribute("DN")
|
dn = user.get_attribute("DN")
|
||||||
logger.debug(f"LDAP: modify_user for user {user.userid} with dn {dn}")
|
|
||||||
if password:
|
if password:
|
||||||
logger.debug(f"LDAP: modify_user for user {user.userid} with password")
|
|
||||||
ldap_conn = self.ldap.connect(dn, password)
|
ldap_conn = self.ldap.connect(dn, password)
|
||||||
else:
|
else:
|
||||||
logger.debug(f"LDAP: modify_user for user {user.userid} with root_dn")
|
|
||||||
if self.root_dn is None:
|
if self.root_dn is None:
|
||||||
logger.error("root_dn missing in ldap config!")
|
logger.error("root_dn missing in ldap config!")
|
||||||
raise InternalServerError
|
raise InternalServerError
|
||||||
|
@ -146,31 +139,14 @@ class AuthLDAP(AuthPlugin):
|
||||||
("display_name", "displayName"),
|
("display_name", "displayName"),
|
||||||
]:
|
]:
|
||||||
if hasattr(user, name):
|
if hasattr(user, name):
|
||||||
attribute = getattr(user, name)
|
modifier[ldap_name] = [(MODIFY_REPLACE, [getattr(user, name)])]
|
||||||
if attribute:
|
|
||||||
modifier[ldap_name] = [(MODIFY_REPLACE, [getattr(user, name)])]
|
|
||||||
if new_password:
|
if new_password:
|
||||||
modifier["userPassword"] = [(MODIFY_REPLACE, [self.__hash(new_password)])]
|
modifier["userPassword"] = [(MODIFY_REPLACE, [self.__hash(new_password)])]
|
||||||
if "userPassword" in modifier:
|
|
||||||
logger.debug(f"LDAP: modify_user for user {user.userid} with password change (can't show >modifier<)")
|
|
||||||
else:
|
|
||||||
logger.debug(f"LDAP: modify_user for user {user.userid} with modifier {modifier}")
|
|
||||||
ldap_conn.modify(dn, modifier)
|
ldap_conn.modify(dn, modifier)
|
||||||
self._set_roles(user)
|
self._set_roles(user)
|
||||||
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
||||||
raise BadRequest
|
raise BadRequest
|
||||||
|
|
||||||
def get_modified_time(self, user):
|
|
||||||
self.ldap.connection.search(
|
|
||||||
self.search_dn,
|
|
||||||
"(uid={})".format(user.userid),
|
|
||||||
SUBTREE,
|
|
||||||
attributes=["modifyTimestamp"],
|
|
||||||
)
|
|
||||||
r = self.ldap.connection.response[0]["attributes"]
|
|
||||||
modified_time = r["modifyTimestamp"][0]
|
|
||||||
return datetime.strptime(modified_time, "%Y%m%d%H%M%SZ")
|
|
||||||
|
|
||||||
def get_avatar(self, user):
|
def get_avatar(self, user):
|
||||||
self.ldap.connection.search(
|
self.ldap.connection.search(
|
||||||
self.search_dn,
|
self.search_dn,
|
||||||
|
@ -188,23 +164,31 @@ class AuthLDAP(AuthPlugin):
|
||||||
else:
|
else:
|
||||||
raise NotFound
|
raise NotFound
|
||||||
|
|
||||||
def set_avatar(self, user: User, file: FileStorage):
|
def set_avatar(self, user, avatar: _Avatar):
|
||||||
if self.root_dn is None:
|
if self.root_dn is None:
|
||||||
logger.error("root_dn missing in ldap config!")
|
logger.error("root_dn missing in ldap config!")
|
||||||
raise InternalServerError
|
raise InternalServerError
|
||||||
|
|
||||||
image_bytes = BytesIO()
|
if avatar.mimetype != "image/jpeg":
|
||||||
try:
|
# Try converting using Pillow (if installed)
|
||||||
# Make sure converted to RGB, e.g. png support RGBA but jpeg does not
|
try:
|
||||||
image = Image.open(file).convert("RGB")
|
from PIL import Image
|
||||||
image.save(image_bytes, format="JPEG")
|
|
||||||
except IOError:
|
image = Image.open(io.BytesIO(avatar.binary))
|
||||||
logger.debug(f"Could not convert avatar from '{file.mimetype}' to JPEG")
|
image_bytes = io.BytesIO()
|
||||||
raise BadRequest("Unsupported image format")
|
image.save(image_bytes, format="JPEG")
|
||||||
|
avatar.binary = image_bytes.getvalue()
|
||||||
|
avatar.mimetype = "image/jpeg"
|
||||||
|
except ImportError:
|
||||||
|
logger.debug("Pillow not installed for image conversion")
|
||||||
|
raise BadRequest("Unsupported image format")
|
||||||
|
except IOError:
|
||||||
|
logger.debug(f"Could not convert avatar from '{avatar.mimetype}' to JPEG")
|
||||||
|
raise BadRequest("Unsupported image format")
|
||||||
|
|
||||||
dn = user.get_attribute("DN")
|
dn = user.get_attribute("DN")
|
||||||
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
ldap_conn = self.ldap.connect(self.root_dn, self.root_secret)
|
||||||
ldap_conn.modify(dn, {"jpegPhoto": [(MODIFY_REPLACE, [image_bytes.getvalue()])]})
|
ldap_conn.modify(dn, {"jpegPhoto": [(MODIFY_REPLACE, [avatar.binary])]})
|
||||||
|
|
||||||
def delete_avatar(self, user):
|
def delete_avatar(self, user):
|
||||||
if self.root_dn is None:
|
if self.root_dn is None:
|
||||||
|
@ -241,7 +225,7 @@ class AuthLDAP(AuthPlugin):
|
||||||
def __modify_role(
|
def __modify_role(
|
||||||
self,
|
self,
|
||||||
role: Role,
|
role: Role,
|
||||||
new_name,
|
new_name: Optional[str],
|
||||||
):
|
):
|
||||||
if self.root_dn is None:
|
if self.root_dn is None:
|
||||||
logger.error("root_dn missing in ldap config!")
|
logger.error("root_dn missing in ldap config!")
|
||||||
|
@ -328,5 +312,3 @@ class AuthLDAP(AuthPlugin):
|
||||||
|
|
||||||
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
except (LDAPPasswordIsMandatoryError, LDAPBindError):
|
||||||
raise BadRequest
|
raise BadRequest
|
||||||
except IndexError as e:
|
|
||||||
logger.error("Roles in LDAP", exc_info=True)
|
|
||||||
|
|
|
@ -1,47 +0,0 @@
|
||||||
import click
|
|
||||||
from flask import current_app
|
|
||||||
from flask.cli import with_appcontext
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
|
|
||||||
|
|
||||||
@click.command(no_args_is_help=True)
|
|
||||||
@click.option("--sync", is_flag=True, default=False, help="Synchronize users from LDAP -> database")
|
|
||||||
@click.option("--sync-ldap", is_flag=True, default=False, help="Synchronize users from database -> LDAP")
|
|
||||||
@with_appcontext
|
|
||||||
@click.pass_context
|
|
||||||
def ldap(ctx, sync, sync_ldap):
|
|
||||||
"""Tools for the LDAP authentification"""
|
|
||||||
from flaschengeist.controller import userController
|
|
||||||
from flaschengeist.plugins.auth_ldap import AuthLDAP
|
|
||||||
if sync:
|
|
||||||
click.echo("Synchronizing users from LDAP -> database")
|
|
||||||
from ldap3 import SUBTREE
|
|
||||||
from flaschengeist.models import User
|
|
||||||
from flaschengeist.database import db
|
|
||||||
|
|
||||||
auth_ldap: AuthLDAP = current_app.config.get("FG_PLUGINS").get("auth_ldap")
|
|
||||||
if auth_ldap is None or not isinstance(auth_ldap, AuthLDAP):
|
|
||||||
ctx.fail("auth_ldap plugin not found or not enabled!")
|
|
||||||
conn = auth_ldap.ldap.connection
|
|
||||||
if not conn:
|
|
||||||
conn = auth_ldap.ldap.connect(auth_ldap.root_dn, auth_ldap.root_secret)
|
|
||||||
conn.search(auth_ldap.search_dn, "(uid=*)", SUBTREE, attributes=["uid", "givenName", "sn", "mail"])
|
|
||||||
ldap_users_response = conn.response
|
|
||||||
for ldap_user in ldap_users_response:
|
|
||||||
uid = ldap_user["attributes"]["uid"][0]
|
|
||||||
try:
|
|
||||||
user = userController.get_user(uid)
|
|
||||||
except NotFound:
|
|
||||||
user = User(userid=uid)
|
|
||||||
db.session.add(user)
|
|
||||||
userController.update_user(user, auth_ldap)
|
|
||||||
if sync_ldap:
|
|
||||||
click.echo("Synchronizing users from database -> LDAP")
|
|
||||||
|
|
||||||
auth_ldap: AuthLDAP = current_app.config.get("FG_PLUGINS").get("auth_ldap")
|
|
||||||
if auth_ldap is None or not isinstance(auth_ldap, AuthLDAP):
|
|
||||||
ctx.fail("auth_ldap plugin not found or not enabled!")
|
|
||||||
users = userController.get_users()
|
|
||||||
for user in users:
|
|
||||||
userController.update_user(user, auth_ldap)
|
|
||||||
|
|
|
@ -6,26 +6,39 @@ Flaschengeist database (as User attribute)
|
||||||
import os
|
import os
|
||||||
import hashlib
|
import hashlib
|
||||||
import binascii
|
import binascii
|
||||||
from werkzeug.exceptions import BadRequest
|
from werkzeug.exceptions import BadRequest, NotFound
|
||||||
|
|
||||||
from flaschengeist.plugins import AuthPlugin
|
from flaschengeist.plugins import AuthPlugin
|
||||||
from flaschengeist.models import User, Role, Permission
|
from flaschengeist.models.user import User, Role, Permission
|
||||||
from flaschengeist.database import db
|
from flaschengeist.database import db
|
||||||
from flaschengeist import logger
|
from flaschengeist import logger
|
||||||
|
|
||||||
|
|
||||||
class AuthPlain(AuthPlugin):
|
class AuthPlain(AuthPlugin):
|
||||||
def can_register(self):
|
def post_install(self):
|
||||||
return True
|
if User.query.first() is None:
|
||||||
|
logger.info("Installing admin user")
|
||||||
|
role = Role(name="Superuser", permissions=Permission.query.all())
|
||||||
|
admin = User(
|
||||||
|
userid="admin",
|
||||||
|
firstname="Admin",
|
||||||
|
lastname="Admin",
|
||||||
|
mail="",
|
||||||
|
roles_=[role],
|
||||||
|
)
|
||||||
|
self.modify_user(admin, None, "admin")
|
||||||
|
db.session.add(admin)
|
||||||
|
db.session.commit()
|
||||||
|
logger.warning(
|
||||||
|
"New administrator user was added, please change the password or remove it before going into"
|
||||||
|
"production mode. Initial credentials:\n"
|
||||||
|
"name: admin\n"
|
||||||
|
"password: admin"
|
||||||
|
)
|
||||||
|
|
||||||
def login(self, login_name, password):
|
def login(self, user: User, password: str):
|
||||||
users: list[User] = (
|
if user.has_attribute("password"):
|
||||||
User.query.filter((User.userid == login_name) | (User.mail == login_name))
|
return AuthPlain._verify_password(user.get_attribute("password"), password)
|
||||||
.filter(User._attributes.any(name="password"))
|
|
||||||
.all()
|
|
||||||
)
|
|
||||||
for user in users:
|
|
||||||
if AuthPlain._verify_password(user.get_attribute("password"), password):
|
|
||||||
return user.userid
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def modify_user(self, user, password, new_password=None):
|
def modify_user(self, user, password, new_password=None):
|
||||||
|
@ -34,12 +47,6 @@ class AuthPlain(AuthPlugin):
|
||||||
if new_password:
|
if new_password:
|
||||||
user.set_attribute("password", AuthPlain._hash_password(new_password))
|
user.set_attribute("password", AuthPlain._hash_password(new_password))
|
||||||
|
|
||||||
def user_exists(self, userid) -> bool:
|
|
||||||
return (
|
|
||||||
db.session.query(User.id_).filter(User.userid == userid, User._attributes.any(name="password")).first()
|
|
||||||
is not None
|
|
||||||
)
|
|
||||||
|
|
||||||
def create_user(self, user, password):
|
def create_user(self, user, password):
|
||||||
if not user.userid:
|
if not user.userid:
|
||||||
raise BadRequest("userid is missing for new user")
|
raise BadRequest("userid is missing for new user")
|
||||||
|
@ -49,6 +56,17 @@ class AuthPlain(AuthPlugin):
|
||||||
def delete_user(self, user):
|
def delete_user(self, user):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_avatar(self, user):
|
||||||
|
if not user.has_attribute("avatar"):
|
||||||
|
raise NotFound
|
||||||
|
return user.get_attribute("avatar")
|
||||||
|
|
||||||
|
def set_avatar(self, user, avatar):
|
||||||
|
user.set_attribute("avatar", avatar)
|
||||||
|
|
||||||
|
def delete_avatar(self, user):
|
||||||
|
user.delete_attribute("avatar")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _hash_password(password):
|
def _hash_password(password):
|
||||||
salt = hashlib.sha256(os.urandom(60)).hexdigest().encode("ascii")
|
salt = hashlib.sha256(os.urandom(60)).hexdigest().encode("ascii")
|
||||||
|
@ -57,7 +75,7 @@ class AuthPlain(AuthPlugin):
|
||||||
return (salt + pass_hash).decode("ascii")
|
return (salt + pass_hash).decode("ascii")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def _verify_password(stored_password: str, provided_password: str):
|
def _verify_password(stored_password, provided_password):
|
||||||
salt = stored_password[:64]
|
salt = stored_password[:64]
|
||||||
stored_password = stored_password[64:]
|
stored_password = stored_password[64:]
|
||||||
pass_hash = hashlib.pbkdf2_hmac("sha3-512", provided_password.encode("utf-8"), salt.encode("ascii"), 100000)
|
pass_hash = hashlib.pbkdf2_hmac("sha3-512", provided_password.encode("utf-8"), salt.encode("ascii"), 100000)
|
||||||
|
|
|
@ -3,87 +3,31 @@
|
||||||
Extends users plugin with balance functions
|
Extends users plugin with balance functions
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from flask import current_app
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
from werkzeug.local import LocalProxy
|
from werkzeug.local import LocalProxy
|
||||||
|
from flask import Blueprint, current_app
|
||||||
|
|
||||||
from flaschengeist import logger
|
from flaschengeist import logger
|
||||||
from flaschengeist.config import config
|
from flaschengeist.plugins import Plugin, before_update_user
|
||||||
from flaschengeist.plugins import Plugin, plugins_loaded, before_update_user
|
|
||||||
from flaschengeist.plugins.scheduler import add_scheduled
|
|
||||||
|
|
||||||
from . import permissions, models
|
from . import permissions, models
|
||||||
|
|
||||||
|
|
||||||
def service_debit():
|
|
||||||
from flaschengeist.database import db
|
|
||||||
from flaschengeist.models import UtcDateTime
|
|
||||||
from flaschengeist.models.user import User
|
|
||||||
from flaschengeist.controller import roleController
|
|
||||||
from flaschengeist_events.models import Service, Job
|
|
||||||
|
|
||||||
role = BalancePlugin.plugin.get_setting("service_role", default=None)
|
|
||||||
if role is None:
|
|
||||||
try:
|
|
||||||
role = roleController.get("__has_service")
|
|
||||||
except NotFound:
|
|
||||||
role = roleController.create_role("__has_service", [permissions.DEBIT])
|
|
||||||
BalancePlugin.plugin.set_setting("service_role", role.id)
|
|
||||||
else:
|
|
||||||
role = roleController.get(role)
|
|
||||||
|
|
||||||
active_services = (
|
|
||||||
User.query.join(Service, User.id_ == Service._user_id)
|
|
||||||
.join(Job, Service.job_)
|
|
||||||
.filter(Job.start <= UtcDateTime.current_utc(), Job.end.is_(None) | (Job.end >= UtcDateTime.current_utc()))
|
|
||||||
.distinct(User.id_)
|
|
||||||
.all()
|
|
||||||
)
|
|
||||||
previous_services = BalancePlugin.plugin.get_setting("service_debit", default=[])
|
|
||||||
|
|
||||||
logger.debug(f"Found {len(active_services)} users doing their service.")
|
|
||||||
|
|
||||||
if len(previous_services) > 0:
|
|
||||||
previous_services = User.query.filter(User.userid.in_(previous_services)).all()
|
|
||||||
# Remove not active users
|
|
||||||
for user in [u for u in previous_services if u not in active_services]:
|
|
||||||
user.roles_ = [r for r in user.roles_ if r.id != role.id]
|
|
||||||
# Add active
|
|
||||||
for user in [u for u in active_services if u not in previous_services]:
|
|
||||||
if role not in user.roles_:
|
|
||||||
user.roles_.append(role)
|
|
||||||
db.session.commit()
|
|
||||||
BalancePlugin.plugin.set_setting("service_debit", [u.userid for u in active_services])
|
|
||||||
|
|
||||||
|
|
||||||
class BalancePlugin(Plugin):
|
class BalancePlugin(Plugin):
|
||||||
# id = "dev.flaschengeist.balance"
|
name = "balance"
|
||||||
|
blueprint = Blueprint(name, __name__)
|
||||||
|
permissions = permissions.permissions
|
||||||
|
plugin = LocalProxy(lambda: current_app.config["FG_PLUGINS"][BalancePlugin.name])
|
||||||
models = models
|
models = models
|
||||||
|
|
||||||
def install(self):
|
def __init__(self, config):
|
||||||
self.install_permissions(permissions.permissions)
|
super(BalancePlugin, self).__init__(config)
|
||||||
|
from . import routes, balance_controller
|
||||||
def load(self):
|
|
||||||
from .routes import blueprint
|
|
||||||
|
|
||||||
self.blueprint = blueprint
|
|
||||||
|
|
||||||
@plugins_loaded
|
|
||||||
def post_loaded(*args, **kwargs):
|
|
||||||
if config.get("allow_service_debit", False) and "events" in current_app.config["FG_PLUGINS"]:
|
|
||||||
add_scheduled(f"{id}.service_debit", service_debit, minutes=1)
|
|
||||||
|
|
||||||
@before_update_user
|
@before_update_user
|
||||||
def set_default_limit(user, *args):
|
def set_default_limit(user):
|
||||||
from . import balance_controller
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
limit = self.get_setting("limit")
|
limit = self.get_setting("limit")
|
||||||
logger.debug("Setting default limit of {} to user {}".format(limit, user.userid))
|
logger.debug("Setting default limit of {} to user {}".format(limit, user.userid))
|
||||||
balance_controller.set_limit(user, limit, override=False)
|
balance_controller.set_limit(user, limit, override=False)
|
||||||
except KeyError:
|
except KeyError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def getPlugin() -> LocalProxy["BalancePlugin"]:
|
|
||||||
return LocalProxy(lambda: current_app.config["FG_PLUGINS"]["balance"])
|
|
||||||
|
|
|
@ -2,15 +2,14 @@
|
||||||
# Haben -> Zugang aufs Konto
|
# Haben -> Zugang aufs Konto
|
||||||
# English: Debit -> from account
|
# English: Debit -> from account
|
||||||
# Credit -> to account
|
# Credit -> to account
|
||||||
from enum import IntEnum
|
|
||||||
from sqlalchemy import func, case, and_, or_
|
from sqlalchemy import func, case, and_
|
||||||
from sqlalchemy.ext.hybrid import hybrid_property
|
from sqlalchemy.ext.hybrid import hybrid_property
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from werkzeug.exceptions import BadRequest, NotFound, Conflict
|
from werkzeug.exceptions import BadRequest, NotFound, Conflict
|
||||||
|
|
||||||
from flaschengeist.database import db
|
from flaschengeist.database import db
|
||||||
from flaschengeist.models.user import User, _UserAttribute
|
from flaschengeist.models.user import User, _UserAttribute
|
||||||
from flaschengeist.app import logger
|
|
||||||
|
|
||||||
from .models import Transaction
|
from .models import Transaction
|
||||||
from . import permissions, BalancePlugin
|
from . import permissions, BalancePlugin
|
||||||
|
@ -18,13 +17,6 @@ from . import permissions, BalancePlugin
|
||||||
__attribute_limit = "balance_limit"
|
__attribute_limit = "balance_limit"
|
||||||
|
|
||||||
|
|
||||||
class NotifyType(IntEnum):
|
|
||||||
SEND_TO = 0x01
|
|
||||||
SEND_FROM = 0x02
|
|
||||||
ADD_FROM = 0x03
|
|
||||||
SUB_FROM = 0x04
|
|
||||||
|
|
||||||
|
|
||||||
def set_limit(user: User, limit: float, override=True):
|
def set_limit(user: User, limit: float, override=True):
|
||||||
if override or not user.has_attribute(__attribute_limit):
|
if override or not user.has_attribute(__attribute_limit):
|
||||||
user.set_attribute(__attribute_limit, limit)
|
user.set_attribute(__attribute_limit, limit)
|
||||||
|
@ -36,7 +28,7 @@ def get_limit(user: User) -> float:
|
||||||
|
|
||||||
|
|
||||||
def get_balance(user, start: datetime = None, end: datetime = None):
|
def get_balance(user, start: datetime = None, end: datetime = None):
|
||||||
query = db.session.query(func.sum(Transaction._amount))
|
query = db.session.query(func.sum(Transaction.amount))
|
||||||
if start:
|
if start:
|
||||||
query = query.filter(start <= Transaction.time)
|
query = query.filter(start <= Transaction.time)
|
||||||
if end:
|
if end:
|
||||||
|
@ -47,26 +39,10 @@ def get_balance(user, start: datetime = None, end: datetime = None):
|
||||||
return credit, debit, credit - debit
|
return credit, debit, credit - debit
|
||||||
|
|
||||||
|
|
||||||
def get_balances(
|
def get_balances(start: datetime = None, end: datetime = None, limit=None, offset=None, descending=None, sortBy=None):
|
||||||
start: datetime = None,
|
|
||||||
end: datetime = None,
|
|
||||||
limit=None,
|
|
||||||
offset=None,
|
|
||||||
descending=None,
|
|
||||||
sortBy=None,
|
|
||||||
_filter=None,
|
|
||||||
):
|
|
||||||
logger.debug(
|
|
||||||
f"get_balances(start={start}, end={end}, limit={limit}, offset={offset}, descending={descending}, sortBy={sortBy}, _filter={_filter})"
|
|
||||||
)
|
|
||||||
|
|
||||||
class _User(User):
|
class _User(User):
|
||||||
_debit = db.relationship(Transaction, back_populates="sender_", foreign_keys=[Transaction._sender_id])
|
_debit = db.relationship(Transaction, back_populates="sender_", foreign_keys=[Transaction._sender_id])
|
||||||
_credit = db.relationship(
|
_credit = db.relationship(Transaction, back_populates="receiver_", foreign_keys=[Transaction._receiver_id])
|
||||||
Transaction,
|
|
||||||
back_populates="receiver_",
|
|
||||||
foreign_keys=[Transaction._receiver_id],
|
|
||||||
)
|
|
||||||
|
|
||||||
@hybrid_property
|
@hybrid_property
|
||||||
def debit(self):
|
def debit(self):
|
||||||
|
@ -75,8 +51,8 @@ def get_balances(
|
||||||
@debit.expression
|
@debit.expression
|
||||||
def debit(cls):
|
def debit(cls):
|
||||||
a = (
|
a = (
|
||||||
db.select(func.sum(Transaction._amount))
|
db.select(func.sum(Transaction.amount))
|
||||||
.where(cls.id_ == Transaction._sender_id, Transaction._amount)
|
.where(cls.id_ == Transaction._sender_id, Transaction.amount)
|
||||||
.scalar_subquery()
|
.scalar_subquery()
|
||||||
)
|
)
|
||||||
return case([(a, a)], else_=0)
|
return case([(a, a)], else_=0)
|
||||||
|
@ -88,8 +64,8 @@ def get_balances(
|
||||||
@credit.expression
|
@credit.expression
|
||||||
def credit(cls):
|
def credit(cls):
|
||||||
b = (
|
b = (
|
||||||
db.select(func.sum(Transaction._amount))
|
db.select(func.sum(Transaction.amount))
|
||||||
.where(cls.id_ == Transaction._receiver_id, Transaction._amount)
|
.where(cls.id_ == Transaction._receiver_id, Transaction.amount)
|
||||||
.scalar_subquery()
|
.scalar_subquery()
|
||||||
)
|
)
|
||||||
return case([(b, b)], else_=0)
|
return case([(b, b)], else_=0)
|
||||||
|
@ -102,12 +78,7 @@ def get_balances(
|
||||||
def limit(cls):
|
def limit(cls):
|
||||||
return (
|
return (
|
||||||
db.select(_UserAttribute.value)
|
db.select(_UserAttribute.value)
|
||||||
.where(
|
.where(and_(cls.id_ == _UserAttribute.user, _UserAttribute.name == "balance_limit"))
|
||||||
and_(
|
|
||||||
cls.id_ == _UserAttribute.user,
|
|
||||||
_UserAttribute.name == "balance_limit",
|
|
||||||
)
|
|
||||||
)
|
|
||||||
.scalar_subquery()
|
.scalar_subquery()
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -140,27 +111,11 @@ def get_balances(
|
||||||
q2 = query.join(_User._debit).filter(Transaction.time <= end)
|
q2 = query.join(_User._debit).filter(Transaction.time <= end)
|
||||||
query = q1.union(q2)
|
query = q1.union(q2)
|
||||||
|
|
||||||
if _filter:
|
|
||||||
query = query.filter(
|
|
||||||
or_(
|
|
||||||
_User.firstname.ilike(f"%{_filter.lower()}%"),
|
|
||||||
_User.lastname.ilike(f"%{_filter.lower()}%"),
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
if sortBy == "balance":
|
if sortBy == "balance":
|
||||||
if descending:
|
if descending:
|
||||||
query = query.order_by(
|
query = query.order_by((_User.credit - _User.debit).desc(), _User.lastname.asc(), _User.firstname.asc())
|
||||||
(_User.credit - _User.debit).desc(),
|
|
||||||
_User.lastname.asc(),
|
|
||||||
_User.firstname.asc(),
|
|
||||||
)
|
|
||||||
else:
|
else:
|
||||||
query = query.order_by(
|
query = query.order_by((_User.credit - _User.debit).asc(), _User.lastname.asc(), _User.firstname.asc())
|
||||||
(_User.credit - _User.debit).asc(),
|
|
||||||
_User.lastname.asc(),
|
|
||||||
_User.firstname.asc(),
|
|
||||||
)
|
|
||||||
elif sortBy == "limit":
|
elif sortBy == "limit":
|
||||||
if descending:
|
if descending:
|
||||||
query = query.order_by(_User.limit.desc(), User.lastname.asc(), User.firstname.asc())
|
query = query.order_by(_User.limit.desc(), User.lastname.asc(), User.firstname.asc())
|
||||||
|
@ -187,6 +142,7 @@ def get_balances(
|
||||||
all = {}
|
all = {}
|
||||||
|
|
||||||
for user in users:
|
for user in users:
|
||||||
|
|
||||||
all[user.userid] = [user.get_credit(start, end), 0]
|
all[user.userid] = [user.get_credit(start, end), 0]
|
||||||
all[user.userid][1] = user.get_debit(start, end)
|
all[user.userid][1] = user.get_debit(start, end)
|
||||||
|
|
||||||
|
@ -206,7 +162,6 @@ def send(sender: User, receiver, amount: float, author: User):
|
||||||
Raises:
|
Raises:
|
||||||
BadRequest if amount <= 0
|
BadRequest if amount <= 0
|
||||||
"""
|
"""
|
||||||
logger.debug(f"send(sender={sender}, receiver={receiver}, amount={amount}, author={author})")
|
|
||||||
if amount <= 0:
|
if amount <= 0:
|
||||||
raise BadRequest
|
raise BadRequest
|
||||||
|
|
||||||
|
@ -220,48 +175,9 @@ def send(sender: User, receiver, amount: float, author: User):
|
||||||
db.session.add(transaction)
|
db.session.add(transaction)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
if sender is not None and sender.id_ != author.id_:
|
if sender is not None and sender.id_ != author.id_:
|
||||||
if receiver is not None:
|
BalancePlugin.plugin.notify(sender, "Neue Transaktion")
|
||||||
BalancePlugin.getPlugin().notify(
|
|
||||||
sender,
|
|
||||||
"Neue Transaktion",
|
|
||||||
{
|
|
||||||
"type": NotifyType.SEND_FROM,
|
|
||||||
"receiver_id": receiver.userid,
|
|
||||||
"author_id": author.userid,
|
|
||||||
"amount": amount,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
BalancePlugin.getPlugin().notify(
|
|
||||||
sender,
|
|
||||||
"Neue Transaktion",
|
|
||||||
{
|
|
||||||
"type": NotifyType.SUB_FROM,
|
|
||||||
"author_id": author.userid,
|
|
||||||
"amount": amount,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if receiver is not None and receiver.id_ != author.id_:
|
if receiver is not None and receiver.id_ != author.id_:
|
||||||
if sender is not None:
|
BalancePlugin.plugin.notify(receiver, "Neue Transaktion")
|
||||||
BalancePlugin.getPlugin().notify(
|
|
||||||
receiver,
|
|
||||||
"Neue Transaktion",
|
|
||||||
{
|
|
||||||
"type": NotifyType.SEND_TO,
|
|
||||||
"sender_id": sender.userid,
|
|
||||||
"amount": amount,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
BalancePlugin.getPlugin().notify(
|
|
||||||
receiver,
|
|
||||||
"Neue Transaktion",
|
|
||||||
{
|
|
||||||
"type": NotifyType.ADD_FROM,
|
|
||||||
"author_id": author.userid,
|
|
||||||
"amount": amount,
|
|
||||||
},
|
|
||||||
)
|
|
||||||
return transaction
|
return transaction
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,47 +0,0 @@
|
||||||
"""balance: initial
|
|
||||||
|
|
||||||
Revision ID: 98f2733bbe45
|
|
||||||
Revises:
|
|
||||||
Create Date: 2022-02-23 14:41:03.089145
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
import flaschengeist
|
|
||||||
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = "98f2733bbe45"
|
|
||||||
down_revision = None
|
|
||||||
branch_labels = ("balance",)
|
|
||||||
depends_on = "flaschengeist"
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.create_table(
|
|
||||||
"balance_transaction",
|
|
||||||
sa.Column("receiver_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("sender_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("author_id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("time", flaschengeist.models.UtcDateTime(), nullable=False),
|
|
||||||
sa.Column("amount", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=False),
|
|
||||||
sa.Column("reversal_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["author_id"], ["user.id"], name=op.f("fk_balance_transaction_author_id_user")),
|
|
||||||
sa.ForeignKeyConstraint(["receiver_id"], ["user.id"], name=op.f("fk_balance_transaction_receiver_id_user")),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["reversal_id"],
|
|
||||||
["balance_transaction.id"],
|
|
||||||
name=op.f("fk_balance_transaction_reversal_id_balance_transaction"),
|
|
||||||
),
|
|
||||||
sa.ForeignKeyConstraint(["sender_id"], ["user.id"], name=op.f("fk_balance_transaction_sender_id_user")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_balance_transaction")),
|
|
||||||
)
|
|
||||||
# ### end Alembic commands ###
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.drop_table("balance_transaction")
|
|
||||||
# ### end Alembic commands ###
|
|
|
@ -1,16 +1,15 @@
|
||||||
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
from sqlalchemy.ext.hybrid import hybrid_property
|
from sqlalchemy.ext.hybrid import hybrid_property
|
||||||
from math import floor
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.database import db
|
from flaschengeist.database import db
|
||||||
from flaschengeist.models.user import User
|
from flaschengeist.models.user import User
|
||||||
from flaschengeist.models import ModelSerializeMixin, UtcDateTime, Serial
|
from flaschengeist.models import ModelSerializeMixin, UtcDateTime, Serial
|
||||||
|
|
||||||
|
|
||||||
class Transaction(db.Model, ModelSerializeMixin):
|
class Transaction(db.Model, ModelSerializeMixin):
|
||||||
__allow_unmapped__ = True
|
|
||||||
__tablename__ = "balance_transaction"
|
__tablename__ = "balance_transaction"
|
||||||
# Protected foreign key properties
|
# Protected foreign key properties
|
||||||
_receiver_id = db.Column("receiver_id", Serial, db.ForeignKey("user.id"))
|
_receiver_id = db.Column("receiver_id", Serial, db.ForeignKey("user.id"))
|
||||||
|
@ -20,9 +19,8 @@ class Transaction(db.Model, ModelSerializeMixin):
|
||||||
# Public and exported member
|
# Public and exported member
|
||||||
id: int = db.Column("id", Serial, primary_key=True)
|
id: int = db.Column("id", Serial, primary_key=True)
|
||||||
time: datetime = db.Column(UtcDateTime, nullable=False, default=UtcDateTime.current_utc)
|
time: datetime = db.Column(UtcDateTime, nullable=False, default=UtcDateTime.current_utc)
|
||||||
_amount: float = db.Column("amount", db.Numeric(precision=5, scale=2, asdecimal=False), nullable=False)
|
amount: float = db.Column(db.Numeric(precision=5, scale=2, asdecimal=False), nullable=False)
|
||||||
reversal_id: Optional[int] = db.Column(Serial, db.ForeignKey("balance_transaction.id"))
|
reversal_id: Optional[int] = db.Column(Serial, db.ForeignKey("balance_transaction.id"))
|
||||||
amount: float
|
|
||||||
|
|
||||||
# Dummy properties used for JSON serialization (userid instead of full user)
|
# Dummy properties used for JSON serialization (userid instead of full user)
|
||||||
author_id: Optional[str] = None
|
author_id: Optional[str] = None
|
||||||
|
@ -42,7 +40,7 @@ class Transaction(db.Model, ModelSerializeMixin):
|
||||||
|
|
||||||
@sender_id.expression
|
@sender_id.expression
|
||||||
def sender_id(cls):
|
def sender_id(cls):
|
||||||
return db.select([User.userid]).where(cls._sender_id == User.id_).scalar_subquery()
|
return db.select([User.userid]).where(cls._sender_id == User.id_).as_scalar()
|
||||||
|
|
||||||
@hybrid_property
|
@hybrid_property
|
||||||
def receiver_id(self):
|
def receiver_id(self):
|
||||||
|
@ -50,7 +48,7 @@ class Transaction(db.Model, ModelSerializeMixin):
|
||||||
|
|
||||||
@receiver_id.expression
|
@receiver_id.expression
|
||||||
def receiver_id(cls):
|
def receiver_id(cls):
|
||||||
return db.select([User.userid]).where(cls._receiver_id == User.id_).scalar_subquery()
|
return db.select([User.userid]).where(cls._receiver_id == User.id_).as_scalar()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def author_id(self):
|
def author_id(self):
|
||||||
|
@ -59,14 +57,3 @@ class Transaction(db.Model, ModelSerializeMixin):
|
||||||
@property
|
@property
|
||||||
def original_id(self):
|
def original_id(self):
|
||||||
return self.original_.id if self.original_ else None
|
return self.original_.id if self.original_ else None
|
||||||
|
|
||||||
@property
|
|
||||||
def amount(self):
|
|
||||||
return self._amount
|
|
||||||
|
|
||||||
@amount.setter
|
|
||||||
def amount(self, value):
|
|
||||||
self._amount = floor(value * 100) / 100
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
return f"<Transaction {self.id} {self.amount} {self.time} {self.sender_id} {self.receiver_id} {self.author_id}>"
|
|
||||||
|
|
|
@ -1,14 +1,12 @@
|
||||||
from datetime import datetime, timezone
|
from datetime import datetime, timezone
|
||||||
from logging import log
|
|
||||||
from werkzeug.exceptions import Forbidden, BadRequest
|
from werkzeug.exceptions import Forbidden, BadRequest
|
||||||
from flask import Blueprint, request, jsonify
|
from flask import request, jsonify
|
||||||
|
|
||||||
from flaschengeist.utils import HTTP
|
from flaschengeist.utils import HTTP
|
||||||
from flaschengeist.models.session import Session
|
from flaschengeist.models.session import Session
|
||||||
from flaschengeist.utils.datetime import from_iso_format
|
from flaschengeist.utils.datetime import from_iso_format
|
||||||
from flaschengeist.utils.decorators import login_required
|
from flaschengeist.utils.decorators import login_required
|
||||||
from flaschengeist.controller import userController
|
from flaschengeist.controller import userController
|
||||||
from flaschengeist.app import logger
|
|
||||||
from . import BalancePlugin, balance_controller, permissions
|
from . import BalancePlugin, balance_controller, permissions
|
||||||
|
|
||||||
|
|
||||||
|
@ -20,10 +18,7 @@ def str2bool(string: str):
|
||||||
raise ValueError
|
raise ValueError
|
||||||
|
|
||||||
|
|
||||||
blueprint = Blueprint("balance", __package__)
|
@BalancePlugin.blueprint.route("/users/<userid>/balance/shortcuts", methods=["GET", "PUT"])
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance/shortcuts", methods=["GET", "PUT"])
|
|
||||||
@login_required()
|
@login_required()
|
||||||
def get_shortcuts(userid, current_session: Session):
|
def get_shortcuts(userid, current_session: Session):
|
||||||
"""Get balance shortcuts of an user
|
"""Get balance shortcuts of an user
|
||||||
|
@ -55,7 +50,7 @@ def get_shortcuts(userid, current_session: Session):
|
||||||
return HTTP.no_content()
|
return HTTP.no_content()
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance/limit", methods=["GET"])
|
@BalancePlugin.blueprint.route("/users/<userid>/balance/limit", methods=["GET"])
|
||||||
@login_required()
|
@login_required()
|
||||||
def get_limit(userid, current_session: Session):
|
def get_limit(userid, current_session: Session):
|
||||||
"""Get limit of an user
|
"""Get limit of an user
|
||||||
|
@ -78,7 +73,7 @@ def get_limit(userid, current_session: Session):
|
||||||
return {"limit": balance_controller.get_limit(user)}
|
return {"limit": balance_controller.get_limit(user)}
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance/limit", methods=["PUT"])
|
@BalancePlugin.blueprint.route("/users/<userid>/balance/limit", methods=["PUT"])
|
||||||
@login_required(permissions.SET_LIMIT)
|
@login_required(permissions.SET_LIMIT)
|
||||||
def set_limit(userid, current_session: Session):
|
def set_limit(userid, current_session: Session):
|
||||||
"""Set the limit of an user
|
"""Set the limit of an user
|
||||||
|
@ -104,7 +99,7 @@ def set_limit(userid, current_session: Session):
|
||||||
return HTTP.no_content()
|
return HTTP.no_content()
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/balance/limit", methods=["GET", "PUT"])
|
@BalancePlugin.blueprint.route("/users/balance/limit", methods=["GET", "PUT"])
|
||||||
@login_required(permission=permissions.SET_LIMIT)
|
@login_required(permission=permissions.SET_LIMIT)
|
||||||
def limits(current_session: Session):
|
def limits(current_session: Session):
|
||||||
"""Get, Modify limit of all users
|
"""Get, Modify limit of all users
|
||||||
|
@ -129,14 +124,14 @@ def limits(current_session: Session):
|
||||||
return HTTP.no_content()
|
return HTTP.no_content()
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance", methods=["GET"])
|
@BalancePlugin.blueprint.route("/users/<userid>/balance", methods=["GET"])
|
||||||
@login_required(permission=permissions.SHOW)
|
@login_required(permission=permissions.SHOW)
|
||||||
def get_balance(userid, current_session: Session):
|
def get_balance(userid, current_session: Session):
|
||||||
"""Get balance of user, optionally filtered
|
"""Get balance of user, optionally filtered
|
||||||
|
|
||||||
Route: ``/users/<userid>/balance`` | Method: ``GET``
|
Route: ``/users/<userid>/balance`` | Method: ``GET``
|
||||||
|
|
||||||
GET-parameters: ``{from?: string, to?: string}``
|
GET-parameters: ```{from?: string, to?: string}```
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
userid: Userid of user to get balance from
|
userid: Userid of user to get balance from
|
||||||
|
@ -164,11 +159,10 @@ def get_balance(userid, current_session: Session):
|
||||||
end = datetime.now(tz=timezone.utc)
|
end = datetime.now(tz=timezone.utc)
|
||||||
|
|
||||||
balance = balance_controller.get_balance(user, start, end)
|
balance = balance_controller.get_balance(user, start, end)
|
||||||
logger.debug(f"Balance of {user.userid} from {start} to {end}: {balance}")
|
|
||||||
return {"credit": balance[0], "debit": balance[1], "balance": balance[2]}
|
return {"credit": balance[0], "debit": balance[1], "balance": balance[2]}
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance/transactions", methods=["GET"])
|
@BalancePlugin.blueprint.route("/users/<userid>/balance/transactions", methods=["GET"])
|
||||||
@login_required(permission=permissions.SHOW)
|
@login_required(permission=permissions.SHOW)
|
||||||
def get_transactions(userid, current_session: Session):
|
def get_transactions(userid, current_session: Session):
|
||||||
"""Get transactions of user, optionally filtered
|
"""Get transactions of user, optionally filtered
|
||||||
|
@ -176,7 +170,7 @@ def get_transactions(userid, current_session: Session):
|
||||||
|
|
||||||
Route: ``/users/<userid>/balance/transactions`` | Method: ``GET``
|
Route: ``/users/<userid>/balance/transactions`` | Method: ``GET``
|
||||||
|
|
||||||
GET-parameters: ``{from?: string, to?: string, limit?: int, offset?: int}``
|
GET-parameters: ```{from?: string, to?: string, limit?: int, offset?: int}```
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
userid: Userid of user to get transactions from
|
userid: Userid of user to get transactions from
|
||||||
|
@ -226,11 +220,10 @@ def get_transactions(userid, current_session: Session):
|
||||||
show_cancelled=show_cancelled,
|
show_cancelled=show_cancelled,
|
||||||
descending=descending,
|
descending=descending,
|
||||||
)
|
)
|
||||||
logger.debug(f"transactions: {transactions}")
|
|
||||||
return {"transactions": transactions, "count": count}
|
return {"transactions": transactions, "count": count}
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/users/<userid>/balance", methods=["PUT"])
|
@BalancePlugin.blueprint.route("/users/<userid>/balance", methods=["PUT"])
|
||||||
@login_required()
|
@login_required()
|
||||||
def change_balance(userid, current_session: Session):
|
def change_balance(userid, current_session: Session):
|
||||||
"""Change balance of an user
|
"""Change balance of an user
|
||||||
|
@ -279,7 +272,7 @@ def change_balance(userid, current_session: Session):
|
||||||
raise Forbidden
|
raise Forbidden
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/balance/<int:transaction_id>", methods=["DELETE"])
|
@BalancePlugin.blueprint.route("/balance/<int:transaction_id>", methods=["DELETE"])
|
||||||
@login_required()
|
@login_required()
|
||||||
def reverse_transaction(transaction_id, current_session: Session):
|
def reverse_transaction(transaction_id, current_session: Session):
|
||||||
"""Reverse a transaction
|
"""Reverse a transaction
|
||||||
|
@ -304,7 +297,7 @@ def reverse_transaction(transaction_id, current_session: Session):
|
||||||
raise Forbidden
|
raise Forbidden
|
||||||
|
|
||||||
|
|
||||||
@blueprint.route("/balance", methods=["GET"])
|
@BalancePlugin.blueprint.route("/balance", methods=["GET"])
|
||||||
@login_required(permission=permissions.SHOW_OTHER)
|
@login_required(permission=permissions.SHOW_OTHER)
|
||||||
def get_balances(current_session: Session):
|
def get_balances(current_session: Session):
|
||||||
"""Get all balances
|
"""Get all balances
|
||||||
|
@ -321,15 +314,7 @@ def get_balances(current_session: Session):
|
||||||
offset = request.args.get("offset", type=int)
|
offset = request.args.get("offset", type=int)
|
||||||
descending = request.args.get("descending", False, type=bool)
|
descending = request.args.get("descending", False, type=bool)
|
||||||
sortBy = request.args.get("sortBy", type=str)
|
sortBy = request.args.get("sortBy", type=str)
|
||||||
_filter = request.args.get("filter", None, type=str)
|
balances, count = balance_controller.get_balances(limit=limit, offset=offset, descending=descending, sortBy=sortBy)
|
||||||
logger.debug(f"request.args: {request.args}")
|
|
||||||
balances, count = balance_controller.get_balances(
|
|
||||||
limit=limit,
|
|
||||||
offset=offset,
|
|
||||||
descending=descending,
|
|
||||||
sortBy=sortBy,
|
|
||||||
_filter=_filter,
|
|
||||||
)
|
|
||||||
return jsonify(
|
return jsonify(
|
||||||
{
|
{
|
||||||
"balances": [{"userid": u, "credit": v[0], "debit": v[1]} for u, v in balances.items()],
|
"balances": [{"userid": u, "credit": v[0], "debit": v[1]} for u, v in balances.items()],
|
||||||
|
|
|
@ -0,0 +1,22 @@
|
||||||
|
"""Events plugin
|
||||||
|
|
||||||
|
Provides duty schedule / duty roster functions
|
||||||
|
"""
|
||||||
|
from flask import Blueprint, current_app
|
||||||
|
from werkzeug.local import LocalProxy
|
||||||
|
|
||||||
|
from flaschengeist.plugins import Plugin
|
||||||
|
from . import permissions, models
|
||||||
|
|
||||||
|
|
||||||
|
class EventPlugin(Plugin):
|
||||||
|
name = "events"
|
||||||
|
id = "dev.flaschengeist.events"
|
||||||
|
plugin = LocalProxy(lambda: current_app.config["FG_PLUGINS"][EventPlugin.name])
|
||||||
|
permissions = permissions.permissions
|
||||||
|
blueprint = Blueprint(name, __name__)
|
||||||
|
models = models
|
||||||
|
|
||||||
|
def __init__(self, cfg):
|
||||||
|
super(EventPlugin, self).__init__(cfg)
|
||||||
|
from . import routes
|
|
@ -0,0 +1,354 @@
|
||||||
|
from datetime import datetime, timedelta, timezone
|
||||||
|
from enum import IntEnum
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
from werkzeug.exceptions import BadRequest, Conflict, NotFound
|
||||||
|
from sqlalchemy.exc import IntegrityError
|
||||||
|
from sqlalchemy.orm.util import was_deleted
|
||||||
|
|
||||||
|
from flaschengeist import logger
|
||||||
|
from flaschengeist.database import db
|
||||||
|
from flaschengeist.plugins.events import EventPlugin
|
||||||
|
from flaschengeist.plugins.events.models import EventType, Event, Invitation, Job, JobType, Service
|
||||||
|
from flaschengeist.utils.scheduler import scheduled
|
||||||
|
|
||||||
|
|
||||||
|
# STUB
|
||||||
|
def _(x):
|
||||||
|
return x
|
||||||
|
|
||||||
|
|
||||||
|
class NotifyType(IntEnum):
|
||||||
|
# Invitations 0x00..0x0F
|
||||||
|
INVITE = 0x01
|
||||||
|
TRANSFER = 0x02
|
||||||
|
# Invitation responsed 0x10..0x1F
|
||||||
|
INVITATION_ACCEPTED = 0x10
|
||||||
|
INVITATION_REJECTED = 0x11
|
||||||
|
|
||||||
|
|
||||||
|
def update():
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def get_event_types():
|
||||||
|
return EventType.query.all()
|
||||||
|
|
||||||
|
|
||||||
|
def get_event_type(identifier):
|
||||||
|
"""Get EventType by ID (int) or name (string)"""
|
||||||
|
|
||||||
|
if isinstance(identifier, int):
|
||||||
|
et = EventType.query.get(identifier)
|
||||||
|
elif isinstance(identifier, str):
|
||||||
|
et = EventType.query.filter(EventType.name == identifier).one_or_none()
|
||||||
|
else:
|
||||||
|
logger.debug("Invalid identifier type for EventType")
|
||||||
|
raise BadRequest
|
||||||
|
if not et:
|
||||||
|
raise NotFound
|
||||||
|
return et
|
||||||
|
|
||||||
|
|
||||||
|
def create_event_type(name):
|
||||||
|
try:
|
||||||
|
event = EventType(name=name)
|
||||||
|
db.session.add(event)
|
||||||
|
db.session.commit()
|
||||||
|
return event
|
||||||
|
except IntegrityError:
|
||||||
|
raise Conflict("Name already exists")
|
||||||
|
|
||||||
|
|
||||||
|
def rename_event_type(identifier, new_name):
|
||||||
|
event_type = get_event_type(identifier)
|
||||||
|
event_type.name = new_name
|
||||||
|
try:
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
raise Conflict("Name already exists")
|
||||||
|
|
||||||
|
|
||||||
|
def delete_event_type(name):
|
||||||
|
event_type = get_event_type(name)
|
||||||
|
db.session.delete(event_type)
|
||||||
|
try:
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
raise BadRequest("Type still in use")
|
||||||
|
|
||||||
|
|
||||||
|
def get_job_types():
|
||||||
|
return JobType.query.all()
|
||||||
|
|
||||||
|
|
||||||
|
def get_job_type(type_id):
|
||||||
|
job_type = JobType.query.get(type_id)
|
||||||
|
print(job_type)
|
||||||
|
if not job_type:
|
||||||
|
raise NotFound
|
||||||
|
return job_type
|
||||||
|
|
||||||
|
|
||||||
|
def create_job_type(name):
|
||||||
|
try:
|
||||||
|
job_type = JobType(name=name)
|
||||||
|
db.session.add(job_type)
|
||||||
|
db.session.commit()
|
||||||
|
return job_type
|
||||||
|
except IntegrityError:
|
||||||
|
raise BadRequest("Name already exists")
|
||||||
|
|
||||||
|
|
||||||
|
def rename_job_type(name, new_name):
|
||||||
|
job_type = get_job_type(name)
|
||||||
|
job_type.name = new_name
|
||||||
|
try:
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
raise BadRequest("Name already exists")
|
||||||
|
|
||||||
|
|
||||||
|
def delete_job_type(name):
|
||||||
|
job_type = get_job_type(name)
|
||||||
|
db.session.delete(job_type)
|
||||||
|
try:
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
raise BadRequest("Type still in use")
|
||||||
|
|
||||||
|
|
||||||
|
def clear_backup(event: Event):
|
||||||
|
for job in event.jobs:
|
||||||
|
services = []
|
||||||
|
for service in job.services:
|
||||||
|
if not service.is_backup:
|
||||||
|
services.append(service)
|
||||||
|
job.services = services
|
||||||
|
|
||||||
|
|
||||||
|
def get_event(event_id, with_backup=False) -> Event:
|
||||||
|
event = Event.query.get(event_id)
|
||||||
|
if event is None:
|
||||||
|
raise NotFound
|
||||||
|
if not with_backup:
|
||||||
|
clear_backup(event)
|
||||||
|
return event
|
||||||
|
|
||||||
|
|
||||||
|
def get_templates():
|
||||||
|
return Event.query.filter(Event.is_template == True).all()
|
||||||
|
|
||||||
|
|
||||||
|
def get_events(
|
||||||
|
start: Optional[datetime] = None,
|
||||||
|
end: Optional[datetime] = None,
|
||||||
|
limit: Optional[int] = None,
|
||||||
|
offset: Optional[int] = None,
|
||||||
|
descending: Optional[bool] = False,
|
||||||
|
with_backup=False,
|
||||||
|
):
|
||||||
|
"""Query events which start from begin until end
|
||||||
|
Args:
|
||||||
|
start (datetime): Earliest start
|
||||||
|
end (datetime): Latest start
|
||||||
|
with_backup (bool): Export also backup services
|
||||||
|
|
||||||
|
Returns: collection of Event objects
|
||||||
|
"""
|
||||||
|
query = Event.query.filter(Event.is_template.__eq__(False))
|
||||||
|
if start is not None:
|
||||||
|
query = query.filter(start <= Event.start)
|
||||||
|
if end is not None:
|
||||||
|
query = query.filter(Event.start < end)
|
||||||
|
if descending:
|
||||||
|
query = query.order_by(Event.start.desc())
|
||||||
|
else:
|
||||||
|
query = query.order_by(Event.start)
|
||||||
|
if limit is not None:
|
||||||
|
query = query.limit(limit)
|
||||||
|
if offset is not None and offset > 0:
|
||||||
|
query = query.offset(offset)
|
||||||
|
events = query.all()
|
||||||
|
if not with_backup:
|
||||||
|
for event in events:
|
||||||
|
clear_backup(event)
|
||||||
|
return events
|
||||||
|
|
||||||
|
|
||||||
|
def delete_event(event_id):
|
||||||
|
"""Delete event with given ID
|
||||||
|
Args:
|
||||||
|
event_id: id of Event to delete
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
NotFound if not found
|
||||||
|
"""
|
||||||
|
event = get_event(event_id, True)
|
||||||
|
for job in event.jobs:
|
||||||
|
delete_job(job)
|
||||||
|
db.session.delete(event)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def create_event(event_type, start, end=None, jobs=[], is_template=None, name=None, description=None):
|
||||||
|
try:
|
||||||
|
logger.debug(event_type)
|
||||||
|
event = Event(
|
||||||
|
start=start,
|
||||||
|
end=end,
|
||||||
|
name=name,
|
||||||
|
description=description,
|
||||||
|
type=event_type,
|
||||||
|
is_template=is_template,
|
||||||
|
jobs=jobs,
|
||||||
|
)
|
||||||
|
db.session.add(event)
|
||||||
|
db.session.commit()
|
||||||
|
return event
|
||||||
|
except IntegrityError:
|
||||||
|
logger.debug("Database error when creating new event", exc_info=True)
|
||||||
|
raise BadRequest
|
||||||
|
|
||||||
|
|
||||||
|
def get_job(job_id, event_id=None) -> Job:
|
||||||
|
query = Job.query.filter(Job.id == job_id)
|
||||||
|
if event_id is not None:
|
||||||
|
query = query.filter(Job.event_id_ == event_id)
|
||||||
|
job = query.one_or_none()
|
||||||
|
if job is None:
|
||||||
|
raise NotFound
|
||||||
|
return job
|
||||||
|
|
||||||
|
|
||||||
|
def add_job(event, job_type, required_services, start, end=None, comment=None):
|
||||||
|
job = Job(
|
||||||
|
required_services=required_services,
|
||||||
|
type=job_type,
|
||||||
|
start=start,
|
||||||
|
end=end,
|
||||||
|
comment=comment,
|
||||||
|
)
|
||||||
|
event.jobs.append(job)
|
||||||
|
update()
|
||||||
|
return job
|
||||||
|
|
||||||
|
|
||||||
|
def update():
|
||||||
|
try:
|
||||||
|
db.session.commit()
|
||||||
|
except IntegrityError:
|
||||||
|
logger.debug(
|
||||||
|
"Error, looks like a Job with that type already exists on an event",
|
||||||
|
exc_info=True,
|
||||||
|
)
|
||||||
|
raise BadRequest()
|
||||||
|
|
||||||
|
|
||||||
|
def delete_job(job: Job):
|
||||||
|
for service in job.services:
|
||||||
|
unassign_job(service=service, notify=True)
|
||||||
|
for invitation in job.invitations_:
|
||||||
|
respond_invitation(invitation, False)
|
||||||
|
db.session.delete(job)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def assign_job(job: Job, user, value, is_backup=False):
|
||||||
|
assert value > 0
|
||||||
|
service = Service.query.get((job.id, user.id_))
|
||||||
|
if service:
|
||||||
|
service.value = value
|
||||||
|
else:
|
||||||
|
job.services.append(Service(user_=user, value=value, is_backup=is_backup, job_=job))
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def unassign_job(job: Job = None, user=None, service=None, notify=False):
|
||||||
|
if service is None:
|
||||||
|
assert job is not None and user is not None
|
||||||
|
service = Service.query.get((job.id, user.id_))
|
||||||
|
else:
|
||||||
|
user = service.user_
|
||||||
|
if not service:
|
||||||
|
raise BadRequest
|
||||||
|
|
||||||
|
event_id = service.job_.event_id_
|
||||||
|
|
||||||
|
db.session.delete(service)
|
||||||
|
db.session.commit()
|
||||||
|
if notify:
|
||||||
|
EventPlugin.plugin.notify(user, "Your assignmet was cancelled", {"event_id": event_id})
|
||||||
|
|
||||||
|
|
||||||
|
def invite(job: Job, invitee, inviter, transferee=None):
|
||||||
|
inv = Invitation(job_=job, inviter_=inviter, invitee_=invitee, transferee_=transferee)
|
||||||
|
db.session.add(inv)
|
||||||
|
update()
|
||||||
|
if transferee is None:
|
||||||
|
EventPlugin.plugin.notify(invitee, _("Job invitation"), {"type": NotifyType.INVITE, "invitation": inv.id})
|
||||||
|
else:
|
||||||
|
EventPlugin.plugin.notify(invitee, _("Job transfer"), {"type": NotifyType.TRANSFER, "invitation": inv.id})
|
||||||
|
return inv
|
||||||
|
|
||||||
|
|
||||||
|
def get_invitation(id: int):
|
||||||
|
inv: Invitation = Invitation.query.get(id)
|
||||||
|
if inv is None:
|
||||||
|
raise NotFound
|
||||||
|
return inv
|
||||||
|
|
||||||
|
|
||||||
|
def cancel_invitation(inv: Invitation):
|
||||||
|
db.session.delete(inv)
|
||||||
|
db.session.commit()
|
||||||
|
|
||||||
|
|
||||||
|
def respond_invitation(invite: Invitation, accepted=True):
|
||||||
|
inviter = invite.inviter_
|
||||||
|
job = invite.job_
|
||||||
|
|
||||||
|
db.session.delete(invite)
|
||||||
|
db.session.commit()
|
||||||
|
if not was_deleted(invite):
|
||||||
|
raise Conflict
|
||||||
|
|
||||||
|
if not accepted:
|
||||||
|
EventPlugin.plugin.notify(inviter, _("Invitation rejected"), {"type": NotifyType.INVITATION_REJECTED, "event": job.event_id_, "job": invite.job_id, "invitee": invite.invitee_id})
|
||||||
|
else:
|
||||||
|
if invite.transferee_id is None:
|
||||||
|
assign_job(job, invite.invitee_, 1)
|
||||||
|
else:
|
||||||
|
service = filter(lambda s: s.userid == invite.transferee_id, job.services)
|
||||||
|
if not service:
|
||||||
|
raise Conflict
|
||||||
|
unassign_job(job, invite.transferee_, service[0], True)
|
||||||
|
assign_job(job, invite.invitee_, service[0].value)
|
||||||
|
EventPlugin.plugin.notify(inviter, _("Invitation accepted"), {"type": NotifyType.INVITATION_ACCEPTED, "event": job.event_id_, "job": invite.job_id, "invitee": invite.invitee_id})
|
||||||
|
|
||||||
|
|
||||||
|
@scheduled
|
||||||
|
def assign_backups():
|
||||||
|
logger.debug("Notifications")
|
||||||
|
now = datetime.now(tz=timezone.utc)
|
||||||
|
# now + backup_time + next cron tick
|
||||||
|
start = now + timedelta(hours=16) + timedelta(minutes=30)
|
||||||
|
services = Service.query.filter(Service.is_backup == True).join(Job).filter(Job.start <= start).all()
|
||||||
|
for service in services:
|
||||||
|
if service.job_.start <= now or service.job_.is_full():
|
||||||
|
EventPlugin.plugin.notify(
|
||||||
|
service.user_,
|
||||||
|
"Your backup assignment was cancelled.",
|
||||||
|
{"event_id": service.job_.event_id_},
|
||||||
|
)
|
||||||
|
logger.debug(f"Service is outdated or full, removing. {service.serialize()}")
|
||||||
|
db.session.delete(service)
|
||||||
|
else:
|
||||||
|
service.is_backup = False
|
||||||
|
logger.debug(f"Service not full, assigning backup. {service.serialize()}")
|
||||||
|
EventPlugin.plugin.notify(
|
||||||
|
service.user_,
|
||||||
|
"Your backup assignment was accepted.",
|
||||||
|
{"event_id": service.job_.event_id_},
|
||||||
|
)
|
||||||
|
db.session.commit()
|
|
@ -0,0 +1,139 @@
|
||||||
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Optional, Union
|
||||||
|
|
||||||
|
from sqlalchemy import UniqueConstraint
|
||||||
|
|
||||||
|
from flaschengeist.models import ModelSerializeMixin, UtcDateTime, Serial
|
||||||
|
from flaschengeist.models.user import User
|
||||||
|
from flaschengeist.database import db
|
||||||
|
|
||||||
|
#########
|
||||||
|
# Types #
|
||||||
|
#########
|
||||||
|
|
||||||
|
_table_prefix_ = "events_"
|
||||||
|
|
||||||
|
|
||||||
|
class EventType(db.Model, ModelSerializeMixin):
|
||||||
|
__tablename__ = _table_prefix_ + "event_type"
|
||||||
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
|
name: str = db.Column(db.String(30), nullable=False, unique=True)
|
||||||
|
|
||||||
|
|
||||||
|
class JobType(db.Model, ModelSerializeMixin):
|
||||||
|
__tablename__ = _table_prefix_ + "job_type"
|
||||||
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
|
name: str = db.Column(db.String(30), nullable=False, unique=True)
|
||||||
|
|
||||||
|
|
||||||
|
########
|
||||||
|
# Jobs #
|
||||||
|
########
|
||||||
|
|
||||||
|
|
||||||
|
class Service(db.Model, ModelSerializeMixin):
|
||||||
|
__tablename__ = _table_prefix_ + "service"
|
||||||
|
userid: str = ""
|
||||||
|
is_backup: bool = db.Column(db.Boolean, default=False)
|
||||||
|
value: float = db.Column(db.Numeric(precision=3, scale=2, asdecimal=False), nullable=False)
|
||||||
|
|
||||||
|
_job_id = db.Column(
|
||||||
|
"job_id",
|
||||||
|
Serial,
|
||||||
|
db.ForeignKey(f"{_table_prefix_}job.id"),
|
||||||
|
nullable=False,
|
||||||
|
primary_key=True,
|
||||||
|
)
|
||||||
|
_user_id = db.Column("user_id", Serial, db.ForeignKey("user.id"), nullable=False, primary_key=True)
|
||||||
|
|
||||||
|
user_: User = db.relationship("User")
|
||||||
|
job_: Job = db.relationship("Job")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def userid(self):
|
||||||
|
return self.user_.userid
|
||||||
|
|
||||||
|
|
||||||
|
class Job(db.Model, ModelSerializeMixin):
|
||||||
|
__tablename__ = _table_prefix_ + "job"
|
||||||
|
_type_id = db.Column("type_id", Serial, db.ForeignKey(f"{_table_prefix_}job_type.id"), nullable=False)
|
||||||
|
|
||||||
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
|
start: datetime = db.Column(UtcDateTime, nullable=False)
|
||||||
|
end: Optional[datetime] = db.Column(UtcDateTime)
|
||||||
|
type: Union[JobType, int] = db.relationship("JobType")
|
||||||
|
comment: Optional[str] = db.Column(db.String(256))
|
||||||
|
locked: bool = db.Column(db.Boolean(), default=False, nullable=False)
|
||||||
|
services: list[Service] = db.relationship(
|
||||||
|
"Service", back_populates="job_", cascade="save-update, merge, delete, delete-orphan"
|
||||||
|
)
|
||||||
|
required_services: float = db.Column(db.Numeric(precision=4, scale=2, asdecimal=False), nullable=False)
|
||||||
|
|
||||||
|
event_ = db.relationship("Event", back_populates="jobs")
|
||||||
|
event_id_ = db.Column("event_id", Serial, db.ForeignKey(f"{_table_prefix_}event.id"), nullable=False)
|
||||||
|
invitations_ = db.relationship("Invitation", cascade="all,delete,delete-orphan", back_populates="job_")
|
||||||
|
|
||||||
|
__table_args__ = (UniqueConstraint("type_id", "start", "event_id", name="_type_start_uc"),)
|
||||||
|
|
||||||
|
|
||||||
|
##########
|
||||||
|
# Events #
|
||||||
|
##########
|
||||||
|
class Event(db.Model, ModelSerializeMixin):
|
||||||
|
"""Model for an Event"""
|
||||||
|
|
||||||
|
__tablename__ = _table_prefix_ + "event"
|
||||||
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
|
start: datetime = db.Column(UtcDateTime, nullable=False)
|
||||||
|
end: Optional[datetime] = db.Column(UtcDateTime)
|
||||||
|
name: Optional[str] = db.Column(db.String(255))
|
||||||
|
description: Optional[str] = db.Column(db.String(512))
|
||||||
|
type: Union[EventType, int] = db.relationship("EventType")
|
||||||
|
is_template: bool = db.Column(db.Boolean, default=False)
|
||||||
|
jobs: list[Job] = db.relationship(
|
||||||
|
"Job",
|
||||||
|
back_populates="event_",
|
||||||
|
cascade="all,delete,delete-orphan",
|
||||||
|
order_by="[Job.start, Job.end]",
|
||||||
|
)
|
||||||
|
# Protected for internal use
|
||||||
|
_type_id = db.Column(
|
||||||
|
"type_id",
|
||||||
|
Serial,
|
||||||
|
db.ForeignKey(f"{_table_prefix_}event_type.id", ondelete="CASCADE"),
|
||||||
|
nullable=False,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class Invitation(db.Model, ModelSerializeMixin):
|
||||||
|
__tablename__ = _table_prefix_ + "invitation"
|
||||||
|
|
||||||
|
id: int = db.Column(Serial, primary_key=True)
|
||||||
|
job_id: int = db.Column(Serial, db.ForeignKey(_table_prefix_ + "job.id"), nullable=False)
|
||||||
|
# Dummy properties for API export
|
||||||
|
invitee_id: str = None # User who was invited to take over
|
||||||
|
inviter_id: str = None # User who invited the invitee
|
||||||
|
transferee_id: Optional[str] = None # In case of a transfer: The user who is transfered out of the job
|
||||||
|
# Not exported properties for backend use
|
||||||
|
job_: Job = db.relationship(Job, foreign_keys="Invitation.job_id")
|
||||||
|
invitee_: User = db.relationship("User", foreign_keys="Invitation._invitee_id")
|
||||||
|
inviter_: User = db.relationship("User", foreign_keys="Invitation._inviter_id")
|
||||||
|
transferee_: User = db.relationship("User", foreign_keys="Invitation._transferee_id")
|
||||||
|
# Protected properties needed for internal use
|
||||||
|
_invitee_id = db.Column("invitee_id", Serial, db.ForeignKey("user.id"), nullable=False)
|
||||||
|
_inviter_id = db.Column("inviter_id", Serial, db.ForeignKey("user.id"), nullable=False)
|
||||||
|
_transferee_id = db.Column("transferee_id", Serial, db.ForeignKey("user.id"))
|
||||||
|
|
||||||
|
@property
|
||||||
|
def invitee_id(self):
|
||||||
|
return self.invitee_.userid
|
||||||
|
|
||||||
|
@property
|
||||||
|
def inviter_id(self):
|
||||||
|
return self.inviter_.userid
|
||||||
|
|
||||||
|
@property
|
||||||
|
def transferee_id(self):
|
||||||
|
return self.transferee_.userid if self.transferee_ else None
|
|
@ -0,0 +1,28 @@
|
||||||
|
CREATE = "events_create"
|
||||||
|
"""Can create events"""
|
||||||
|
|
||||||
|
EDIT = "events_edit"
|
||||||
|
"""Can edit events"""
|
||||||
|
|
||||||
|
DELETE = "events_delete"
|
||||||
|
"""Can delete events"""
|
||||||
|
|
||||||
|
EVENT_TYPE = "events_event_type"
|
||||||
|
"""Can create and edit EventTypes"""
|
||||||
|
|
||||||
|
JOB_TYPE = "events_job_type"
|
||||||
|
"""Can create and edit JobTypes"""
|
||||||
|
|
||||||
|
ASSIGN = "events_assign"
|
||||||
|
"""Can self assign to jobs"""
|
||||||
|
|
||||||
|
ASSIGN_OTHER = "events_assign_other"
|
||||||
|
"""Can assign other users to jobs"""
|
||||||
|
|
||||||
|
SEE_BACKUP = "events_see_backup"
|
||||||
|
"""Can see users assigned as backup"""
|
||||||
|
|
||||||
|
LOCK_JOBS = "events_lock_jobs"
|
||||||
|
"""Can lock jobs, no further services can be assigned or unassigned"""
|
||||||
|
|
||||||
|
permissions = [value for key, value in globals().items() if not key.startswith("_")]
|
|
@ -0,0 +1,576 @@
|
||||||
|
from datetime import datetime, timedelta, timezone
|
||||||
|
from http.client import NO_CONTENT
|
||||||
|
from re import template
|
||||||
|
from flask import request, jsonify
|
||||||
|
from sqlalchemy import exc
|
||||||
|
from werkzeug.exceptions import BadRequest, NotFound, Forbidden
|
||||||
|
|
||||||
|
from flaschengeist.models.session import Session
|
||||||
|
from flaschengeist.plugins.events.models import Job
|
||||||
|
from flaschengeist.utils.decorators import login_required
|
||||||
|
from flaschengeist.utils.datetime import from_iso_format
|
||||||
|
from flaschengeist.controller import userController
|
||||||
|
|
||||||
|
from . import event_controller, permissions, EventPlugin
|
||||||
|
from ...utils.HTTP import no_content
|
||||||
|
|
||||||
|
|
||||||
|
def dict_get(self, key, default=None, type=None):
|
||||||
|
"""Same as .get from MultiDict"""
|
||||||
|
try:
|
||||||
|
rv = self[key]
|
||||||
|
except KeyError:
|
||||||
|
return default
|
||||||
|
if type is not None:
|
||||||
|
try:
|
||||||
|
rv = type(rv)
|
||||||
|
except ValueError:
|
||||||
|
rv = default
|
||||||
|
return rv
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/templates", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_templates(current_session):
|
||||||
|
return jsonify(event_controller.get_templates())
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/event-types", methods=["GET"])
|
||||||
|
@EventPlugin.blueprint.route("/events/event-types/<int:identifier>", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_event_types(current_session, identifier=None):
|
||||||
|
"""Get EventType(s)
|
||||||
|
|
||||||
|
Route: ``/events/event-types`` | Method: ``GET``
|
||||||
|
Route: ``/events/event-types/<identifier>`` | Method: ``GET``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
identifier: If querying a specific EventType
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded (list of) EventType(s) or HTTP-error
|
||||||
|
"""
|
||||||
|
if identifier:
|
||||||
|
result = event_controller.get_event_type(identifier)
|
||||||
|
else:
|
||||||
|
result = event_controller.get_event_types()
|
||||||
|
return jsonify(result)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/event-types", methods=["POST"])
|
||||||
|
@login_required(permission=permissions.EVENT_TYPE)
|
||||||
|
def new_event_type(current_session):
|
||||||
|
"""Create a new EventType
|
||||||
|
|
||||||
|
Route: ``/events/event-types`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: ``{name: string}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-Created or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
if "name" not in data:
|
||||||
|
raise BadRequest
|
||||||
|
event_type = event_controller.create_event_type(data["name"])
|
||||||
|
return jsonify(event_type)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/event-types/<int:identifier>", methods=["PUT", "DELETE"])
|
||||||
|
@login_required(permission=permissions.EVENT_TYPE)
|
||||||
|
def modify_event_type(identifier, current_session):
|
||||||
|
"""Rename or delete an event type
|
||||||
|
|
||||||
|
Route: ``/events/event-types/<id>`` | Method: ``PUT`` or ``DELETE``
|
||||||
|
|
||||||
|
POST-data: (if renaming) ``{name: string}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
identifier: Identifier of the EventType
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-NoContent or HTTP-error
|
||||||
|
"""
|
||||||
|
if request.method == "DELETE":
|
||||||
|
event_controller.delete_event_type(identifier)
|
||||||
|
else:
|
||||||
|
data = request.get_json()
|
||||||
|
if "name" not in data:
|
||||||
|
raise BadRequest("Parameter missing in data")
|
||||||
|
event_controller.rename_event_type(identifier, data["name"])
|
||||||
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/job-types", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_job_types(current_session):
|
||||||
|
"""Get all JobTypes
|
||||||
|
|
||||||
|
Route: ``/events/job-types`` | Method: ``GET``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded list of JobType HTTP-error
|
||||||
|
"""
|
||||||
|
types = event_controller.get_job_types()
|
||||||
|
return jsonify(types)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/job-types", methods=["POST"])
|
||||||
|
@login_required(permission=permissions.JOB_TYPE)
|
||||||
|
def new_job_type(current_session):
|
||||||
|
"""Create a new JobType
|
||||||
|
|
||||||
|
Route: ``/events/job-types`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: ``{name: string}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded JobType or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
if "name" not in data:
|
||||||
|
raise BadRequest
|
||||||
|
jt = event_controller.create_job_type(data["name"])
|
||||||
|
return jsonify(jt)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/job-types/<int:type_id>", methods=["PUT", "DELETE"])
|
||||||
|
@login_required(permission=permissions.JOB_TYPE)
|
||||||
|
def modify_job_type(type_id, current_session):
|
||||||
|
"""Rename or delete a JobType
|
||||||
|
|
||||||
|
Route: ``/events/job-types/<name>`` | Method: ``PUT`` or ``DELETE``
|
||||||
|
|
||||||
|
POST-data: (if renaming) ``{name: string}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
type_id: Identifier of the JobType
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-NoContent or HTTP-error
|
||||||
|
"""
|
||||||
|
if request.method == "DELETE":
|
||||||
|
event_controller.delete_job_type(type_id)
|
||||||
|
else:
|
||||||
|
data = request.get_json()
|
||||||
|
if "name" not in data:
|
||||||
|
raise BadRequest("Parameter missing in data")
|
||||||
|
event_controller.rename_job_type(type_id, data["name"])
|
||||||
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_event(event_id, current_session):
|
||||||
|
"""Get event by id
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>`` | Method: ``GET``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: ID identifying the event
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded event object
|
||||||
|
"""
|
||||||
|
event = event_controller.get_event(
|
||||||
|
event_id,
|
||||||
|
with_backup=current_session.user_.has_permission(permissions.SEE_BACKUP),
|
||||||
|
)
|
||||||
|
return jsonify(event)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_filtered_events(current_session):
|
||||||
|
begin = request.args.get("from", type=from_iso_format)
|
||||||
|
end = request.args.get("to", type=from_iso_format)
|
||||||
|
limit = request.args.get("limit", type=int)
|
||||||
|
offset = request.args.get("offset", type=int)
|
||||||
|
descending = "descending" in request.args
|
||||||
|
if begin is None and end is None:
|
||||||
|
begin = datetime.now()
|
||||||
|
return jsonify(
|
||||||
|
event_controller.get_events(
|
||||||
|
start=begin,
|
||||||
|
end=end,
|
||||||
|
limit=limit,
|
||||||
|
offset=offset,
|
||||||
|
descending=descending,
|
||||||
|
with_backup=current_session.user_.has_permission(permissions.SEE_BACKUP),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:year>/<int:month>", methods=["GET"])
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:year>/<int:month>/<int:day>", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_events(current_session, year=datetime.now().year, month=datetime.now().month, day=None):
|
||||||
|
"""Get Event objects for specified date (or month or year),
|
||||||
|
if nothing set then events for current month are returned
|
||||||
|
|
||||||
|
Route: ``/events[/<year>/<month>[/<int:day>]]`` | Method: ``GET``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
year (int, optional): year to query, defaults to current year
|
||||||
|
month (int, optional): month to query (if set), defaults to current month
|
||||||
|
day (int, optional): day to query events for (if set)
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded list containing events found or HTTP-error
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
begin = datetime(year=year, month=month, day=1, tzinfo=timezone.utc)
|
||||||
|
if day:
|
||||||
|
begin += timedelta(days=day - 1)
|
||||||
|
end = begin + timedelta(days=1)
|
||||||
|
else:
|
||||||
|
if month == 12:
|
||||||
|
end = datetime(year=year + 1, month=1, day=1, tzinfo=timezone.utc)
|
||||||
|
else:
|
||||||
|
end = datetime(year=year, month=month + 1, day=1, tzinfo=timezone.utc)
|
||||||
|
|
||||||
|
events = event_controller.get_events(
|
||||||
|
begin,
|
||||||
|
end,
|
||||||
|
with_backup=current_session.user_.has_permission(permissions.SEE_BACKUP),
|
||||||
|
)
|
||||||
|
return jsonify(events)
|
||||||
|
except ValueError:
|
||||||
|
raise BadRequest("Invalid date given")
|
||||||
|
|
||||||
|
|
||||||
|
def _add_job(event, data):
|
||||||
|
try:
|
||||||
|
start = from_iso_format(data["start"])
|
||||||
|
end = dict_get(data, "end", None, type=from_iso_format)
|
||||||
|
required_services = data["required_services"]
|
||||||
|
job_type = data["type"]
|
||||||
|
if isinstance(job_type, dict):
|
||||||
|
job_type = data["type"]["id"]
|
||||||
|
except (KeyError, ValueError):
|
||||||
|
raise BadRequest("Missing or invalid POST parameter")
|
||||||
|
|
||||||
|
job_type = event_controller.get_job_type(job_type)
|
||||||
|
event_controller.add_job(
|
||||||
|
event,
|
||||||
|
job_type,
|
||||||
|
required_services,
|
||||||
|
start,
|
||||||
|
end,
|
||||||
|
comment=dict_get(data, "comment", None, str),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events", methods=["POST"])
|
||||||
|
@login_required(permission=permissions.CREATE)
|
||||||
|
def create_event(current_session):
|
||||||
|
"""Create an new event
|
||||||
|
|
||||||
|
Route: ``/events`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: See interfaces for Event, can already contain jobs
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded Event object or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
try:
|
||||||
|
start = from_iso_format(data["start"])
|
||||||
|
end = dict_get(data, "end", None, type=from_iso_format)
|
||||||
|
data_type = data["type"]
|
||||||
|
if isinstance(data_type, dict):
|
||||||
|
data_type = data["type"]["id"]
|
||||||
|
event_type = event_controller.get_event_type(data_type)
|
||||||
|
except KeyError:
|
||||||
|
raise BadRequest("Missing POST parameter")
|
||||||
|
except (NotFound, ValueError):
|
||||||
|
raise BadRequest("Invalid parameter")
|
||||||
|
|
||||||
|
event = event_controller.create_event(
|
||||||
|
start=start,
|
||||||
|
end=end,
|
||||||
|
name=dict_get(data, "name", None),
|
||||||
|
is_template=dict_get(data, "is_template", None),
|
||||||
|
event_type=event_type,
|
||||||
|
description=dict_get(data, "description", None),
|
||||||
|
)
|
||||||
|
if "jobs" in data:
|
||||||
|
for job in data["jobs"]:
|
||||||
|
_add_job(event, job)
|
||||||
|
|
||||||
|
return jsonify(event)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>", methods=["PUT"])
|
||||||
|
@login_required(permission=permissions.EDIT)
|
||||||
|
def modify_event(event_id, current_session):
|
||||||
|
"""Modify an event
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>`` | Method: ``PUT``
|
||||||
|
|
||||||
|
POST-data: See interfaces for Event, can already contain slots
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: Identifier of the event
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded Event object or HTTP-error
|
||||||
|
"""
|
||||||
|
event = event_controller.get_event(event_id)
|
||||||
|
data = request.get_json()
|
||||||
|
event.start = dict_get(data, "start", event.start, type=from_iso_format)
|
||||||
|
event.end = dict_get(data, "end", event.end, type=from_iso_format)
|
||||||
|
event.name = dict_get(data, "name", event.name, type=str)
|
||||||
|
event.description = dict_get(data, "description", event.description, type=str)
|
||||||
|
if "type" in data:
|
||||||
|
event_type = event_controller.get_event_type(data["type"])
|
||||||
|
event.type = event_type
|
||||||
|
|
||||||
|
event_controller.update()
|
||||||
|
return jsonify(event)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>", methods=["DELETE"])
|
||||||
|
@login_required(permission=permissions.DELETE)
|
||||||
|
def delete_event(event_id, current_session):
|
||||||
|
"""Delete an event
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>`` | Method: ``DELETE``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: Identifier of the event
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-NoContent or HTTP-error
|
||||||
|
"""
|
||||||
|
event_controller.delete_event(event_id)
|
||||||
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>/jobs", methods=["POST"])
|
||||||
|
@login_required(permission=permissions.EDIT)
|
||||||
|
def add_job(event_id, current_session):
|
||||||
|
"""Add an new Job to an Event / EventSlot
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>/jobs`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: See Job
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: Identifier of the event
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded Event object or HTTP-error
|
||||||
|
"""
|
||||||
|
event = event_controller.get_event(event_id)
|
||||||
|
_add_job(event, request.get_json())
|
||||||
|
return jsonify(event)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>/jobs/<int:job_id>", methods=["DELETE"])
|
||||||
|
@login_required(permission=permissions.DELETE)
|
||||||
|
def delete_job(event_id, job_id, current_session):
|
||||||
|
"""Delete a Job
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>/jobs/<job_id>`` | Method: ``DELETE``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: Identifier of the event
|
||||||
|
job_id: Identifier of the Job
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-no-content or HTTP error
|
||||||
|
"""
|
||||||
|
job = event_controller.get_job(job_id, event_id)
|
||||||
|
event_controller.delete_job(job)
|
||||||
|
return no_content()
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/<int:event_id>/jobs/<int:job_id>", methods=["PUT"])
|
||||||
|
@login_required()
|
||||||
|
def update_job(event_id, job_id, current_session: Session):
|
||||||
|
"""Edit Job
|
||||||
|
|
||||||
|
Route: ``/events/<event_id>/jobs/<job_id>`` | Method: ``PUT``
|
||||||
|
|
||||||
|
POST-data: See TS interface for Job
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: Identifier of the event
|
||||||
|
job_id: Identifier of the Job
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded Job object or HTTP-error
|
||||||
|
"""
|
||||||
|
if not current_session.user_.has_permission(permissions.EDIT):
|
||||||
|
raise Forbidden
|
||||||
|
|
||||||
|
data = request.get_json()
|
||||||
|
if not data:
|
||||||
|
raise BadRequest
|
||||||
|
|
||||||
|
job = event_controller.get_job(job_id, event_id)
|
||||||
|
try:
|
||||||
|
if "type" in data:
|
||||||
|
job.type = event_controller.get_job_type(data["type"])
|
||||||
|
job.start = from_iso_format(data.get("start", job.start))
|
||||||
|
job.end = from_iso_format(data.get("end", job.end))
|
||||||
|
job.comment = str(data.get("comment", job.comment))
|
||||||
|
job.locked = bool(data.get("locked", job.locked))
|
||||||
|
job.required_services = float(data.get("required_services", job.required_services))
|
||||||
|
event_controller.update()
|
||||||
|
except NotFound:
|
||||||
|
raise BadRequest("Invalid JobType")
|
||||||
|
except ValueError:
|
||||||
|
raise BadRequest("Invalid POST data")
|
||||||
|
|
||||||
|
return jsonify(job)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/jobs/<int:job_id>/assign", methods=["POST"])
|
||||||
|
@login_required()
|
||||||
|
def assign_job(job_id, current_session: Session):
|
||||||
|
"""Assign / unassign user to the Job
|
||||||
|
|
||||||
|
Route: ``/events/jobs/<job_id>/assign`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: a Service object, see TS interface for Service
|
||||||
|
|
||||||
|
Args:
|
||||||
|
job_id: Identifier of the Job
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
JSON encoded Job or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
job = event_controller.get_job(job_id)
|
||||||
|
try:
|
||||||
|
user = userController.get_user(data["userid"])
|
||||||
|
value = data["value"]
|
||||||
|
if (user == current_session.user_ and not user.has_permission(permissions.ASSIGN)) or (
|
||||||
|
user != current_session.user_ and not current_session.user_.has_permission(permissions.ASSIGN_OTHER)
|
||||||
|
):
|
||||||
|
raise Forbidden
|
||||||
|
if value > 0:
|
||||||
|
event_controller.assign_job(job, user, value, data.get("is_backup", False))
|
||||||
|
else:
|
||||||
|
event_controller.unassign_job(job, user, notify=user != current_session.user_)
|
||||||
|
except (TypeError, KeyError, ValueError):
|
||||||
|
raise BadRequest
|
||||||
|
return jsonify(job)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/jobs/<int:job_id>/lock", methods=["POST"])
|
||||||
|
@login_required(permissions.LOCK_JOBS)
|
||||||
|
def lock_job(job_id, current_session: Session):
|
||||||
|
"""Lock / unlock the Job
|
||||||
|
|
||||||
|
Route: ``/events/jobs/<job_id>/lock`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: ``{locked: boolean}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
job_id: Identifier of the Job
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
HTTP-No-Content or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
job = event_controller.get_job(job_id)
|
||||||
|
try:
|
||||||
|
locked = bool(userController.get_user(data["locked"]))
|
||||||
|
job.locked = locked
|
||||||
|
event_controller.update()
|
||||||
|
except (TypeError, KeyError, ValueError):
|
||||||
|
raise BadRequest
|
||||||
|
return no_content()
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/invitations", methods=["POST"])
|
||||||
|
@login_required()
|
||||||
|
def invite(current_session: Session):
|
||||||
|
"""Invite an user to a job or transfer job
|
||||||
|
|
||||||
|
Route: ``/events/invites`` | Method: ``POST``
|
||||||
|
|
||||||
|
POST-data: ``{job: number, invitees: string[], is_transfer?: boolean}``
|
||||||
|
|
||||||
|
Args:
|
||||||
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
List of Invitation objects or HTTP-error
|
||||||
|
"""
|
||||||
|
data = request.get_json()
|
||||||
|
transferee = data.get("transferee", None)
|
||||||
|
if (
|
||||||
|
transferee is not None
|
||||||
|
and transferee != current_session.userid
|
||||||
|
and not current_session.user_.has_permission(permissions.ASSIGN_OTHER)
|
||||||
|
):
|
||||||
|
raise Forbidden
|
||||||
|
|
||||||
|
try:
|
||||||
|
job = event_controller.get_job(data["job"])
|
||||||
|
if not isinstance(data["invitees"], list):
|
||||||
|
raise BadRequest
|
||||||
|
return jsonify(
|
||||||
|
[
|
||||||
|
event_controller.invite(job, invitee, current_session.user_, transferee)
|
||||||
|
for invitee in [userController.get_user(uid) for uid in data["invitees"]]
|
||||||
|
]
|
||||||
|
)
|
||||||
|
except (TypeError, KeyError, ValueError):
|
||||||
|
raise BadRequest
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/invitations/<int:invitation_id>", methods=["GET"])
|
||||||
|
@login_required()
|
||||||
|
def get_invitation(invitation_id: int, current_session: Session):
|
||||||
|
inv = event_controller.get_invitation(invitation_id)
|
||||||
|
if current_session.userid not in [inv.invitee_id, inv.inviter_id, inv.transferee_id]:
|
||||||
|
raise Forbidden
|
||||||
|
return jsonify(inv)
|
||||||
|
|
||||||
|
|
||||||
|
@EventPlugin.blueprint.route("/events/invitations/<int:invitation_id>", methods=["DELETE", "PUT"])
|
||||||
|
@login_required()
|
||||||
|
def respond_invitation(invitation_id: int, current_session: Session):
|
||||||
|
inv = event_controller.get_invitation(invitation_id)
|
||||||
|
if request.method == "DELETE":
|
||||||
|
if current_session.userid == inv.invitee_id:
|
||||||
|
event_controller.respond_invitation(inv, False)
|
||||||
|
elif current_session.userid == inv.inviter_id:
|
||||||
|
event_controller.cancel_invitation(inv)
|
||||||
|
else:
|
||||||
|
raise Forbidden
|
||||||
|
else:
|
||||||
|
# maybe validate data is something like ({accepted: true})
|
||||||
|
if current_session.userid != inv.invitee_id:
|
||||||
|
raise Forbidden
|
||||||
|
event_controller.respond_invitation(inv)
|
||||||
|
return no_content()
|
|
@ -1,37 +1,31 @@
|
||||||
import smtplib
|
import smtplib
|
||||||
from email.mime.text import MIMEText
|
from email.mime.text import MIMEText
|
||||||
from email.mime.multipart import MIMEMultipart
|
from email.mime.multipart import MIMEMultipart
|
||||||
from werkzeug.exceptions import InternalServerError
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
from flaschengeist import logger
|
||||||
from flaschengeist.models import User
|
from flaschengeist.models.user import User
|
||||||
from flaschengeist.plugins import Plugin
|
|
||||||
from flaschengeist.utils.hook import HookAfter
|
from flaschengeist.utils.hook import HookAfter
|
||||||
from flaschengeist.controller import userController
|
from flaschengeist.controller import userController
|
||||||
from flaschengeist.controller.messageController import Message
|
from flaschengeist.controller.messageController import Message
|
||||||
from flaschengeist.config import config
|
|
||||||
|
from . import Plugin
|
||||||
|
|
||||||
|
|
||||||
class MailMessagePlugin(Plugin):
|
class MailMessagePlugin(Plugin):
|
||||||
def load(self):
|
def __init__(self, config):
|
||||||
self.config = config.get("mail", None)
|
super().__init__()
|
||||||
if self.config is None:
|
self.server = config["SERVER"]
|
||||||
logger.error("mail was not configured in flaschengeist.toml")
|
self.port = config["PORT"]
|
||||||
raise InternalServerError
|
self.user = config["USER"]
|
||||||
self.server = self.config["SERVER"]
|
self.password = config["PASSWORD"]
|
||||||
self.port = self.config["PORT"]
|
self.crypt = config["CRYPT"]
|
||||||
self.user = self.config["USER"]
|
self.mail = config["MAIL"]
|
||||||
self.password = self.config["PASSWORD"]
|
|
||||||
self.crypt = self.config["CRYPT"]
|
|
||||||
self.mail = self.config["MAIL"]
|
|
||||||
|
|
||||||
@HookAfter("send_message")
|
@HookAfter("send_message")
|
||||||
def dummy_send(msg, *args, **kwargs):
|
def dummy_send(msg):
|
||||||
logger.info(f"(dummy_send) Sending message to {msg.receiver}")
|
|
||||||
self.send_mail(msg)
|
self.send_mail(msg)
|
||||||
|
|
||||||
def send_mail(self, msg: Message):
|
def send_mail(self, msg: Message):
|
||||||
logger.debug(f"Sending mail to {msg.receiver} with subject {msg.subject}")
|
|
||||||
if isinstance(msg.receiver, User):
|
if isinstance(msg.receiver, User):
|
||||||
if not msg.receiver.mail:
|
if not msg.receiver.mail:
|
||||||
logger.warning("Could not send Mail, mail missing: {}".format(msg.receiver))
|
logger.warning("Could not send Mail, mail missing: {}".format(msg.receiver))
|
||||||
|
@ -45,8 +39,9 @@ class MailMessagePlugin(Plugin):
|
||||||
mail["To"] = ", ".join(recipients)
|
mail["To"] = ", ".join(recipients)
|
||||||
mail["Subject"] = msg.subject
|
mail["Subject"] = msg.subject
|
||||||
mail.attach(MIMEText(msg.message))
|
mail.attach(MIMEText(msg.message))
|
||||||
with self.__connect() as smtp:
|
if not hasattr(self, "smtp"):
|
||||||
smtp.sendmail(self.mail, recipients, mail.as_string())
|
self.__connect()
|
||||||
|
self.smtp.sendmail(self.mail, recipients, mail.as_string())
|
||||||
|
|
||||||
def __connect(self):
|
def __connect(self):
|
||||||
if self.crypt == "SSL":
|
if self.crypt == "SSL":
|
||||||
|
@ -57,4 +52,3 @@ class MailMessagePlugin(Plugin):
|
||||||
else:
|
else:
|
||||||
raise ValueError("Invalid CRYPT given")
|
raise ValueError("Invalid CRYPT given")
|
||||||
self.smtp.login(self.user, self.password)
|
self.smtp.login(self.user, self.password)
|
||||||
return self.smtp
|
|
||||||
|
|
|
@ -1,12 +1,14 @@
|
||||||
"""Pricelist plugin"""
|
"""Pricelist plugin"""
|
||||||
from flask import Blueprint, jsonify, request
|
|
||||||
|
from flask import Blueprint, jsonify, request, current_app
|
||||||
|
from werkzeug.local import LocalProxy
|
||||||
from werkzeug.exceptions import BadRequest, Forbidden, NotFound, Unauthorized
|
from werkzeug.exceptions import BadRequest, Forbidden, NotFound, Unauthorized
|
||||||
|
|
||||||
from flaschengeist import logger
|
from flaschengeist import logger
|
||||||
from flaschengeist.controller import userController
|
from flaschengeist.controller import userController
|
||||||
from flaschengeist.controller.imageController import send_image, send_thumbnail
|
from flaschengeist.controller.imageController import send_image, send_thumbnail
|
||||||
from flaschengeist.plugins import Plugin
|
from flaschengeist.plugins import Plugin
|
||||||
from flaschengeist.utils.decorators import login_required, extract_session
|
from flaschengeist.utils.decorators import login_required, extract_session, headers
|
||||||
from flaschengeist.utils.HTTP import no_content
|
from flaschengeist.utils.HTTP import no_content
|
||||||
|
|
||||||
from . import models
|
from . import models
|
||||||
|
@ -14,15 +16,16 @@ from . import pricelist_controller, permissions
|
||||||
|
|
||||||
|
|
||||||
class PriceListPlugin(Plugin):
|
class PriceListPlugin(Plugin):
|
||||||
|
name = "pricelist"
|
||||||
|
permissions = permissions.permissions
|
||||||
|
blueprint = Blueprint(name, __name__, url_prefix="/pricelist")
|
||||||
|
plugin = LocalProxy(lambda: current_app.config["FG_PLUGINS"][PriceListPlugin.name])
|
||||||
models = models
|
models = models
|
||||||
blueprint = Blueprint("pricelist", __name__, url_prefix="/pricelist")
|
|
||||||
|
|
||||||
def install(self):
|
def __init__(self, cfg):
|
||||||
self.install_permissions(permissions.permissions)
|
super().__init__(cfg)
|
||||||
|
|
||||||
def load(self):
|
|
||||||
config = {"discount": 0}
|
config = {"discount": 0}
|
||||||
config.update(config)
|
config.update(cfg)
|
||||||
|
|
||||||
|
|
||||||
@PriceListPlugin.blueprint.route("/drink-types", methods=["GET"])
|
@PriceListPlugin.blueprint.route("/drink-types", methods=["GET"])
|
||||||
|
|
|
@ -1,141 +0,0 @@
|
||||||
"""pricelist: initial
|
|
||||||
|
|
||||||
Revision ID: 58ab9b6a8839
|
|
||||||
Revises:
|
|
||||||
Create Date: 2022-02-23 14:45:30.563647
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
import flaschengeist
|
|
||||||
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = "58ab9b6a8839"
|
|
||||||
down_revision = None
|
|
||||||
branch_labels = ("pricelist",)
|
|
||||||
depends_on = "flaschengeist"
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.create_table(
|
|
||||||
"drink_extra_ingredient",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=False),
|
|
||||||
sa.Column("price", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=True),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_extra_ingredient")),
|
|
||||||
sa.UniqueConstraint("name", name=op.f("uq_drink_extra_ingredient_name")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_tag",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=False),
|
|
||||||
sa.Column("color", sa.String(length=7), nullable=False),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_tag")),
|
|
||||||
sa.UniqueConstraint("name", name=op.f("uq_drink_tag_name")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_type",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("name", sa.String(length=30), nullable=False),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_type")),
|
|
||||||
sa.UniqueConstraint("name", name=op.f("uq_drink_type_name")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("article_id", sa.String(length=64), nullable=True),
|
|
||||||
sa.Column("package_size", sa.Integer(), nullable=True),
|
|
||||||
sa.Column("name", sa.String(length=60), nullable=False),
|
|
||||||
sa.Column("volume", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=True),
|
|
||||||
sa.Column("cost_per_volume", sa.Numeric(precision=5, scale=3, asdecimal=False), nullable=True),
|
|
||||||
sa.Column("cost_per_package", sa.Numeric(precision=5, scale=3, asdecimal=False), nullable=True),
|
|
||||||
sa.Column("receipt", sa.PickleType(), nullable=True),
|
|
||||||
sa.Column("type_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("image_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["image_id"], ["image.id"], name=op.f("fk_drink_image_id_image")),
|
|
||||||
sa.ForeignKeyConstraint(["type_id"], ["drink_type.id"], name=op.f("fk_drink_type_id_drink_type")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_ingredient",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("volume", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=False),
|
|
||||||
sa.Column("ingredient_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["ingredient_id"], ["drink.id"], name=op.f("fk_drink_ingredient_ingredient_id_drink")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_ingredient")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_price_volume",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("drink_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("volume", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["drink_id"], ["drink.id"], name=op.f("fk_drink_price_volume_drink_id_drink")),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_price_volume")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_x_tag",
|
|
||||||
sa.Column("drink_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("tag_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["drink_id"], ["drink.id"], name=op.f("fk_drink_x_tag_drink_id_drink")),
|
|
||||||
sa.ForeignKeyConstraint(["tag_id"], ["drink_tag.id"], name=op.f("fk_drink_x_tag_tag_id_drink_tag")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_x_type",
|
|
||||||
sa.Column("drink_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("type_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(["drink_id"], ["drink.id"], name=op.f("fk_drink_x_type_drink_id_drink")),
|
|
||||||
sa.ForeignKeyConstraint(["type_id"], ["drink_type.id"], name=op.f("fk_drink_x_type_type_id_drink_type")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_ingredient_association",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("volume_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("_drink_ingredient_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("_extra_ingredient_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["_drink_ingredient_id"],
|
|
||||||
["drink_ingredient.id"],
|
|
||||||
name=op.f("fk_drink_ingredient_association__drink_ingredient_id_drink_ingredient"),
|
|
||||||
),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["_extra_ingredient_id"],
|
|
||||||
["drink_extra_ingredient.id"],
|
|
||||||
name=op.f("fk_drink_ingredient_association__extra_ingredient_id_drink_extra_ingredient"),
|
|
||||||
),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["volume_id"],
|
|
||||||
["drink_price_volume.id"],
|
|
||||||
name=op.f("fk_drink_ingredient_association_volume_id_drink_price_volume"),
|
|
||||||
),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_ingredient_association")),
|
|
||||||
)
|
|
||||||
op.create_table(
|
|
||||||
"drink_price",
|
|
||||||
sa.Column("id", flaschengeist.models.Serial(), nullable=False),
|
|
||||||
sa.Column("price", sa.Numeric(precision=5, scale=2, asdecimal=False), nullable=True),
|
|
||||||
sa.Column("volume_id", flaschengeist.models.Serial(), nullable=True),
|
|
||||||
sa.Column("public", sa.Boolean(), nullable=True),
|
|
||||||
sa.Column("description", sa.String(length=30), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(
|
|
||||||
["volume_id"], ["drink_price_volume.id"], name=op.f("fk_drink_price_volume_id_drink_price_volume")
|
|
||||||
),
|
|
||||||
sa.PrimaryKeyConstraint("id", name=op.f("pk_drink_price")),
|
|
||||||
)
|
|
||||||
# ### end Alembic commands ###
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.drop_table("drink_price")
|
|
||||||
op.drop_table("drink_ingredient_association")
|
|
||||||
op.drop_table("drink_x_type")
|
|
||||||
op.drop_table("drink_x_tag")
|
|
||||||
op.drop_table("drink_price_volume")
|
|
||||||
op.drop_table("drink_ingredient")
|
|
||||||
op.drop_table("drink")
|
|
||||||
op.drop_table("drink_type")
|
|
||||||
op.drop_table("drink_tag")
|
|
||||||
op.drop_table("drink_extra_ingredient")
|
|
||||||
# ### end Alembic commands ###
|
|
|
@ -1,11 +1,10 @@
|
||||||
from __future__ import annotations # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered)
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
|
||||||
from typing import Optional
|
|
||||||
|
|
||||||
from flaschengeist.database import db
|
from flaschengeist.database import db
|
||||||
from flaschengeist.database.types import ModelSerializeMixin, Serial
|
from flaschengeist.models import ModelSerializeMixin, Serial
|
||||||
from flaschengeist.models import Image
|
from flaschengeist.models.image import Image
|
||||||
|
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
drink_tag_association = db.Table(
|
drink_tag_association = db.Table(
|
||||||
"drink_x_tag",
|
"drink_x_tag",
|
||||||
|
|
|
@ -5,20 +5,21 @@ Provides routes used to configure roles and permissions of users / roles.
|
||||||
|
|
||||||
from werkzeug.exceptions import BadRequest
|
from werkzeug.exceptions import BadRequest
|
||||||
from flask import Blueprint, request, jsonify
|
from flask import Blueprint, request, jsonify
|
||||||
|
from http.client import NO_CONTENT
|
||||||
|
|
||||||
from flaschengeist.plugins import Plugin
|
from flaschengeist.plugins import Plugin
|
||||||
from flaschengeist.controller import roleController
|
|
||||||
from flaschengeist.utils.HTTP import created, no_content
|
|
||||||
from flaschengeist.utils.decorators import login_required
|
from flaschengeist.utils.decorators import login_required
|
||||||
|
from flaschengeist.controller import roleController
|
||||||
|
from flaschengeist.utils.HTTP import created
|
||||||
|
|
||||||
from . import permissions
|
_permission_edit = "roles_edit"
|
||||||
|
_permission_delete = "roles_delete"
|
||||||
|
|
||||||
|
|
||||||
class RolesPlugin(Plugin):
|
class RolesPlugin(Plugin):
|
||||||
blueprint = Blueprint("roles", __name__)
|
name = "roles"
|
||||||
|
blueprint = Blueprint(name, __name__)
|
||||||
def install(self):
|
permissions = [_permission_edit, _permission_delete]
|
||||||
self.install_permissions(permissions.permissions)
|
|
||||||
|
|
||||||
|
|
||||||
@RolesPlugin.blueprint.route("/roles", methods=["GET"])
|
@RolesPlugin.blueprint.route("/roles", methods=["GET"])
|
||||||
|
@ -39,7 +40,7 @@ def list_roles(current_session):
|
||||||
|
|
||||||
|
|
||||||
@RolesPlugin.blueprint.route("/roles", methods=["POST"])
|
@RolesPlugin.blueprint.route("/roles", methods=["POST"])
|
||||||
@login_required(permission=permissions.EDIT)
|
@login_required(permission=_permission_edit)
|
||||||
def create_role(current_session):
|
def create_role(current_session):
|
||||||
"""Create new role
|
"""Create new role
|
||||||
|
|
||||||
|
@ -97,7 +98,7 @@ def get_role(role_name, current_session):
|
||||||
|
|
||||||
|
|
||||||
@RolesPlugin.blueprint.route("/roles/<int:role_id>", methods=["PUT"])
|
@RolesPlugin.blueprint.route("/roles/<int:role_id>", methods=["PUT"])
|
||||||
@login_required(permission=permissions.EDIT)
|
@login_required(permission=_permission_edit)
|
||||||
def edit_role(role_id, current_session):
|
def edit_role(role_id, current_session):
|
||||||
"""Edit role, rename and / or set permissions
|
"""Edit role, rename and / or set permissions
|
||||||
|
|
||||||
|
@ -117,13 +118,13 @@ def edit_role(role_id, current_session):
|
||||||
data = request.get_json()
|
data = request.get_json()
|
||||||
if "permissions" in data:
|
if "permissions" in data:
|
||||||
roleController.set_permissions(role, data["permissions"])
|
roleController.set_permissions(role, data["permissions"])
|
||||||
if "name" in data and data["name"] != role.name:
|
if "name" in data:
|
||||||
roleController.update_role(role, data["name"])
|
roleController.update_role(role, data["name"])
|
||||||
return no_content()
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
@RolesPlugin.blueprint.route("/roles/<int:role_id>", methods=["DELETE"])
|
@RolesPlugin.blueprint.route("/roles/<int:role_id>", methods=["DELETE"])
|
||||||
@login_required(permission=permissions.DELETE)
|
@login_required(permission=_permission_delete)
|
||||||
def delete_role(role_id, current_session):
|
def delete_role(role_id, current_session):
|
||||||
"""Delete role
|
"""Delete role
|
||||||
|
|
||||||
|
@ -134,8 +135,8 @@ def delete_role(role_id, current_session):
|
||||||
current_session: Session sent with Authorization Header
|
current_session: Session sent with Authorization Header
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
HTTP-204 or HTTP error (HTTP-409 Conflict if role still in use)
|
HTTP-204 or HTTP error
|
||||||
"""
|
"""
|
||||||
role = roleController.get(role_id)
|
role = roleController.get(role_id)
|
||||||
roleController.delete(role)
|
roleController.delete(role)
|
||||||
return no_content()
|
return "", NO_CONTENT
|
||||||
|
|
|
@ -1,7 +0,0 @@
|
||||||
EDIT = "roles_edit"
|
|
||||||
"""Can edit roles, assign permissions to roles and change names"""
|
|
||||||
|
|
||||||
DELETE = "roles_delete"
|
|
||||||
"""Can delete roles"""
|
|
||||||
|
|
||||||
permissions = [value for key, value in globals().items() if not key.startswith("_")]
|
|
|
@ -1,86 +0,0 @@
|
||||||
from flask import Blueprint
|
|
||||||
from datetime import datetime, timedelta
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.config import config
|
|
||||||
from flaschengeist.plugins import Plugin
|
|
||||||
from flaschengeist.utils.HTTP import no_content
|
|
||||||
|
|
||||||
|
|
||||||
class __Task:
|
|
||||||
def __init__(self, function, **kwags):
|
|
||||||
self.function = function
|
|
||||||
self.interval = timedelta(**kwags)
|
|
||||||
|
|
||||||
|
|
||||||
_scheduled_tasks: dict[__Task] = dict()
|
|
||||||
|
|
||||||
|
|
||||||
def add_scheduled(id: str, function, replace=False, **kwargs):
|
|
||||||
if id not in _scheduled_tasks or replace:
|
|
||||||
_scheduled_tasks[id] = __Task(function, **kwargs)
|
|
||||||
logger.info(f"Registered task: {id}")
|
|
||||||
else:
|
|
||||||
logger.debug(f"Skipping already registered task: {id}")
|
|
||||||
|
|
||||||
|
|
||||||
def scheduled(id: str, replace=False, **kwargs):
|
|
||||||
"""
|
|
||||||
kwargs: days, hours, minutes
|
|
||||||
"""
|
|
||||||
|
|
||||||
def real_decorator(function):
|
|
||||||
add_scheduled(id, function, replace, **kwargs)
|
|
||||||
return function
|
|
||||||
|
|
||||||
if not isinstance(id, str):
|
|
||||||
raise TypeError
|
|
||||||
return real_decorator
|
|
||||||
|
|
||||||
|
|
||||||
class SchedulerPlugin(Plugin):
|
|
||||||
blueprint = Blueprint("scheduler", __name__)
|
|
||||||
|
|
||||||
def load(self):
|
|
||||||
def __view_func():
|
|
||||||
self.run_tasks()
|
|
||||||
return no_content()
|
|
||||||
|
|
||||||
def __passiv_func(v):
|
|
||||||
try:
|
|
||||||
self.run_tasks()
|
|
||||||
except:
|
|
||||||
logger.error("Error while executing scheduled tasks!", exc_info=True)
|
|
||||||
|
|
||||||
cron = config.get("scheduler", {}).get("cron", "passive_web").lower()
|
|
||||||
|
|
||||||
if cron == "passive_web":
|
|
||||||
self.blueprint.teardown_app_request(__passiv_func)
|
|
||||||
elif cron == "active_web":
|
|
||||||
self.blueprint.add_url_rule("/cron", view_func=__view_func)
|
|
||||||
|
|
||||||
def run_tasks(self):
|
|
||||||
from ..database import db
|
|
||||||
|
|
||||||
self = db.session.merge(self)
|
|
||||||
|
|
||||||
changed = False
|
|
||||||
now = datetime.now()
|
|
||||||
status = self.get_setting("status", default=dict())
|
|
||||||
|
|
||||||
for id, task in _scheduled_tasks.items():
|
|
||||||
last_run = status.setdefault(id, now)
|
|
||||||
if last_run + task.interval <= now:
|
|
||||||
logger.debug(
|
|
||||||
f"Run task {id}, was scheduled for {last_run + task.interval}, next iteration: {now + task.interval}"
|
|
||||||
)
|
|
||||||
task.function()
|
|
||||||
changed = True
|
|
||||||
else:
|
|
||||||
logger.debug(f"Skip task {id}, is scheduled for {last_run + task.interval}")
|
|
||||||
if changed:
|
|
||||||
# Remove not registered tasks
|
|
||||||
for id in status.keys():
|
|
||||||
if id not in _scheduled_tasks.keys():
|
|
||||||
del status[id]
|
|
||||||
self.set_setting("status", status)
|
|
|
@ -2,30 +2,25 @@
|
||||||
|
|
||||||
Provides routes used to manage users
|
Provides routes used to manage users
|
||||||
"""
|
"""
|
||||||
|
from http.client import NO_CONTENT, CREATED
|
||||||
from datetime import datetime
|
from flask import Blueprint, request, jsonify, make_response, Response
|
||||||
from http.client import CREATED
|
from werkzeug.exceptions import BadRequest, Forbidden, MethodNotAllowed, NotFound
|
||||||
|
|
||||||
from flask import Blueprint, Response, after_this_request, jsonify, make_response, request
|
|
||||||
from werkzeug.exceptions import BadRequest, Forbidden, MethodNotAllowed
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.config import config
|
|
||||||
from flaschengeist.controller import userController
|
|
||||||
from flaschengeist.models import User
|
|
||||||
from flaschengeist.plugins import Plugin
|
|
||||||
from flaschengeist.utils.datetime import from_iso_format
|
|
||||||
from flaschengeist.utils.decorators import extract_session, headers, login_required
|
|
||||||
from flaschengeist.utils.HTTP import created, no_content
|
|
||||||
|
|
||||||
from . import permissions
|
from . import permissions
|
||||||
|
from flaschengeist import logger
|
||||||
|
from flaschengeist.config import config
|
||||||
|
from flaschengeist.plugins import Plugin
|
||||||
|
from flaschengeist.models.user import User, _Avatar
|
||||||
|
from flaschengeist.utils.decorators import login_required, extract_session, headers
|
||||||
|
from flaschengeist.controller import userController
|
||||||
|
from flaschengeist.utils.HTTP import created, no_content
|
||||||
|
from flaschengeist.utils.datetime import from_iso_format
|
||||||
|
|
||||||
|
|
||||||
class UsersPlugin(Plugin):
|
class UsersPlugin(Plugin):
|
||||||
blueprint = Blueprint("users", __name__)
|
name = "users"
|
||||||
|
blueprint = Blueprint(name, __name__)
|
||||||
def install(self):
|
permissions = permissions.permissions
|
||||||
self.install_permissions(permissions.permissions)
|
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users", methods=["POST"])
|
@UsersPlugin.blueprint.route("/users", methods=["POST"])
|
||||||
|
@ -61,7 +56,7 @@ def register():
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users", methods=["GET"])
|
@UsersPlugin.blueprint.route("/users", methods=["GET"])
|
||||||
@login_required()
|
@login_required()
|
||||||
# @headers({"Cache-Control": "private, must-revalidate, max-age=3600"})
|
@headers({"Cache-Control": "private, must-revalidate, max-age=3600"})
|
||||||
def list_users(current_session):
|
def list_users(current_session):
|
||||||
"""List all existing users
|
"""List all existing users
|
||||||
|
|
||||||
|
@ -94,9 +89,7 @@ def get_user(userid, current_session):
|
||||||
JSON encoded `flaschengeist.models.user.User` or if userid is current user also containing permissions or HTTP error
|
JSON encoded `flaschengeist.models.user.User` or if userid is current user also containing permissions or HTTP error
|
||||||
"""
|
"""
|
||||||
logger.debug("Get information of user {{ {} }}".format(userid))
|
logger.debug("Get information of user {{ {} }}".format(userid))
|
||||||
user: User = userController.get_user(
|
user: User = userController.get_user(userid)
|
||||||
userid, True
|
|
||||||
) # This is the only API point that should return data for deleted users
|
|
||||||
serial = user.serialize()
|
serial = user.serialize()
|
||||||
if userid == current_session.user_.userid:
|
if userid == current_session.user_.userid:
|
||||||
serial["permissions"] = user.get_permissions()
|
serial["permissions"] = user.get_permissions()
|
||||||
|
@ -110,7 +103,7 @@ def frontend(userid, current_session):
|
||||||
raise Forbidden
|
raise Forbidden
|
||||||
|
|
||||||
if request.method == "POST":
|
if request.method == "POST":
|
||||||
if request.content_length > 1024**2:
|
if request.content_length > 1024 ** 2:
|
||||||
raise BadRequest
|
raise BadRequest
|
||||||
current_session.user_.set_attribute("frontend", request.get_json())
|
current_session.user_.set_attribute("frontend", request.get_json())
|
||||||
return no_content()
|
return no_content()
|
||||||
|
@ -122,13 +115,15 @@ def frontend(userid, current_session):
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users/<userid>/avatar", methods=["GET"])
|
@UsersPlugin.blueprint.route("/users/<userid>/avatar", methods=["GET"])
|
||||||
@headers({"Cache-Control": "public, must-revalidate, max-age=10"})
|
@headers({"Cache-Control": "public, max-age=604800"})
|
||||||
def get_avatar(userid):
|
def get_avatar(userid):
|
||||||
etag = None
|
|
||||||
if "If-None-Match" in request.headers:
|
|
||||||
etag = request.headers["If-None-Match"]
|
|
||||||
user = userController.get_user(userid)
|
user = userController.get_user(userid)
|
||||||
return userController.load_avatar(user, etag)
|
avatar = userController.load_avatar(user)
|
||||||
|
if len(avatar.binary) > 0:
|
||||||
|
response = Response(avatar.binary, mimetype=avatar.mimetype)
|
||||||
|
response.add_etag()
|
||||||
|
return response.make_conditional(request)
|
||||||
|
raise NotFound
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users/<userid>/avatar", methods=["POST"])
|
@UsersPlugin.blueprint.route("/users/<userid>/avatar", methods=["POST"])
|
||||||
|
@ -140,7 +135,10 @@ def set_avatar(userid, current_session):
|
||||||
|
|
||||||
file = request.files.get("file")
|
file = request.files.get("file")
|
||||||
if file:
|
if file:
|
||||||
userController.save_avatar(user, file)
|
avatar = _Avatar()
|
||||||
|
avatar.mimetype = file.content_type
|
||||||
|
avatar.binary = bytearray(file.stream.read())
|
||||||
|
userController.save_avatar(user, avatar)
|
||||||
return created()
|
return created()
|
||||||
else:
|
else:
|
||||||
raise BadRequest
|
raise BadRequest
|
||||||
|
@ -153,7 +151,7 @@ def delete_avatar(userid, current_session):
|
||||||
if userid != current_session.user_.userid and not current_session.user_.has_permission(permissions.EDIT):
|
if userid != current_session.user_.userid and not current_session.user_.has_permission(permissions.EDIT):
|
||||||
raise Forbidden
|
raise Forbidden
|
||||||
userController.delete_avatar(user)
|
userController.delete_avatar(user)
|
||||||
return no_content()
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users/<userid>", methods=["DELETE"])
|
@UsersPlugin.blueprint.route("/users/<userid>", methods=["DELETE"])
|
||||||
|
@ -172,8 +170,8 @@ def delete_user(userid, current_session):
|
||||||
"""
|
"""
|
||||||
logger.debug("Delete user {{ {} }}".format(userid))
|
logger.debug("Delete user {{ {} }}".format(userid))
|
||||||
user = userController.get_user(userid)
|
user = userController.get_user(userid)
|
||||||
userController.delete_user(user)
|
userController.delete(user)
|
||||||
return no_content()
|
return "", NO_CONTENT
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users/<userid>", methods=["PUT"])
|
@UsersPlugin.blueprint.route("/users/<userid>", methods=["PUT"])
|
||||||
|
@ -225,10 +223,8 @@ def edit_user(userid, current_session):
|
||||||
userController.set_roles(user, roles)
|
userController.set_roles(user, roles)
|
||||||
|
|
||||||
userController.modify_user(user, password, new_password)
|
userController.modify_user(user, password, new_password)
|
||||||
userController.update_user(
|
userController.update_user(user)
|
||||||
user,
|
return "", NO_CONTENT
|
||||||
)
|
|
||||||
return no_content()
|
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/notifications", methods=["GET"])
|
@UsersPlugin.blueprint.route("/notifications", methods=["GET"])
|
||||||
|
@ -242,7 +238,7 @@ def notifications(current_session):
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/notifications/<nid>", methods=["DELETE"])
|
@UsersPlugin.blueprint.route("/notifications/<nid>", methods=["DELETE"])
|
||||||
@login_required()
|
@login_required()
|
||||||
def remove_notification(nid, current_session):
|
def remove_notifications(nid, current_session):
|
||||||
userController.delete_notification(nid, current_session.user_)
|
userController.delete_notification(nid, current_session.user_)
|
||||||
return no_content()
|
return no_content()
|
||||||
|
|
||||||
|
@ -263,21 +259,3 @@ def shortcuts(userid, current_session):
|
||||||
user.set_attribute("users_link_shortcuts", data)
|
user.set_attribute("users_link_shortcuts", data)
|
||||||
userController.persist()
|
userController.persist()
|
||||||
return no_content()
|
return no_content()
|
||||||
|
|
||||||
|
|
||||||
@UsersPlugin.blueprint.route("/users/<userid>/setting/<setting>", methods=["GET", "PUT"])
|
|
||||||
@login_required()
|
|
||||||
def settings(userid, setting, current_session):
|
|
||||||
if userid != current_session.user_.userid:
|
|
||||||
raise Forbidden
|
|
||||||
user = userController.get_user(userid)
|
|
||||||
if request.method == "GET":
|
|
||||||
retVal = user.get_attribute(setting, None)
|
|
||||||
logger.debug(f"Get setting >>{setting}<< for user >>{user.userid}<< with >>{retVal}<<")
|
|
||||||
return jsonify(retVal)
|
|
||||||
else:
|
|
||||||
data = request.get_json()
|
|
||||||
logger.debug(f"Set setting >>{setting}<< for user >>{user.userid}<< to >>{data}<<")
|
|
||||||
user.set_attribute(setting, data)
|
|
||||||
userController.persist()
|
|
||||||
return no_content()
|
|
||||||
|
|
|
@ -1,91 +0,0 @@
|
||||||
import click
|
|
||||||
import sqlalchemy.exc
|
|
||||||
from flask.cli import with_appcontext
|
|
||||||
from werkzeug.exceptions import NotFound
|
|
||||||
|
|
||||||
from flaschengeist import logger
|
|
||||||
from flaschengeist.database import db
|
|
||||||
from flaschengeist.controller import roleController, userController
|
|
||||||
|
|
||||||
|
|
||||||
USER_KEY = f"{__name__}.user"
|
|
||||||
|
|
||||||
|
|
||||||
def user(ctx, param, value):
|
|
||||||
if not value or ctx.resilient_parsing:
|
|
||||||
return
|
|
||||||
|
|
||||||
click.echo("Adding new user")
|
|
||||||
ctx.meta[USER_KEY] = {}
|
|
||||||
try:
|
|
||||||
ctx.meta[USER_KEY]["userid"] = click.prompt("userid", type=str)
|
|
||||||
ctx.meta[USER_KEY]["firstname"] = click.prompt("firstname", type=str)
|
|
||||||
ctx.meta[USER_KEY]["lastname"] = click.prompt("lastname", type=str)
|
|
||||||
ctx.meta[USER_KEY]["display_name"] = click.prompt("displayed name", type=str, default="")
|
|
||||||
ctx.meta[USER_KEY]["mail"] = click.prompt("mail", type=str, default="")
|
|
||||||
ctx.meta[USER_KEY]["password"] = click.prompt("password", type=str, confirmation_prompt=True, hide_input=True)
|
|
||||||
ctx.meta[USER_KEY] = {k: v for k, v in ctx.meta[USER_KEY].items() if v != ""}
|
|
||||||
|
|
||||||
except click.Abort:
|
|
||||||
click.echo("\n!!! User was not added, aborted.")
|
|
||||||
del ctx.meta[USER_KEY]
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@click.option("--create", help="Add new role", is_flag=True)
|
|
||||||
@click.option("--delete", help="Delete role", is_flag=True)
|
|
||||||
@click.option("--set-admin", is_flag=True, help="Make a role an admin role, adding all permissions", type=str)
|
|
||||||
@click.argument("role", nargs=-1, required=True, type=str)
|
|
||||||
def role(create, delete, set_admin, role):
|
|
||||||
"""Manage roles"""
|
|
||||||
ctx = click.get_current_context()
|
|
||||||
|
|
||||||
if (create and delete) or (set_admin and delete):
|
|
||||||
ctx.fail("Do not mix --delete with --create or --set-admin")
|
|
||||||
|
|
||||||
for role_name in role:
|
|
||||||
if create:
|
|
||||||
r = roleController.create_role(role_name)
|
|
||||||
else:
|
|
||||||
r = roleController.get(role_name)
|
|
||||||
if delete:
|
|
||||||
roleController.delete(r)
|
|
||||||
if set_admin:
|
|
||||||
r.permissions = roleController.get_permissions()
|
|
||||||
db.session.commit()
|
|
||||||
|
|
||||||
|
|
||||||
@click.command()
|
|
||||||
@click.option("--add-role", help="Add a role to an user", type=str)
|
|
||||||
@click.option("--create", help="Create new user interactivly", callback=user, is_flag=True, expose_value=False)
|
|
||||||
@click.option("--delete", help="Delete a user", is_flag=True)
|
|
||||||
@click.argument("user", nargs=-1, type=str)
|
|
||||||
@with_appcontext
|
|
||||||
def user(add_role, delete, user):
|
|
||||||
"""Manage users"""
|
|
||||||
from flaschengeist.database import db
|
|
||||||
|
|
||||||
ctx = click.get_current_context()
|
|
||||||
|
|
||||||
try:
|
|
||||||
if USER_KEY in ctx.meta:
|
|
||||||
userController.register(ctx.meta[USER_KEY], ctx.meta[USER_KEY]["password"])
|
|
||||||
else:
|
|
||||||
if not isinstance(user, list) or not isinstance(user, tuple):
|
|
||||||
user = [user]
|
|
||||||
for uid in user:
|
|
||||||
logger.debug(f"Userid: {uid}")
|
|
||||||
user = userController.get_user(uid)
|
|
||||||
logger.debug(f"User: {user}")
|
|
||||||
if delete:
|
|
||||||
logger.debug(f"Deleting user {user}")
|
|
||||||
userController.delete_user(user)
|
|
||||||
elif add_role:
|
|
||||||
logger.debug(f"Adding role {add_role} to user {user}")
|
|
||||||
role = roleController.get(add_role)
|
|
||||||
logger.debug(f"Role: {role}")
|
|
||||||
user.roles_.append(role)
|
|
||||||
userController.modify_user(user, None)
|
|
||||||
db.session.commit()
|
|
||||||
except NotFound:
|
|
||||||
ctx.fail(f"User not found {uid}")
|
|
|
@ -2,24 +2,6 @@ from http.client import NO_CONTENT, CREATED
|
||||||
|
|
||||||
from flask import make_response, jsonify
|
from flask import make_response, jsonify
|
||||||
|
|
||||||
from flaschengeist.utils.datetime import from_iso_format
|
|
||||||
|
|
||||||
|
|
||||||
def get_filter_args():
|
|
||||||
"""
|
|
||||||
Get filter parameter from request
|
|
||||||
returns: FROM, TO, LIMIT, OFFSET, DESCENDING
|
|
||||||
"""
|
|
||||||
from flask import request
|
|
||||||
|
|
||||||
return (
|
|
||||||
request.args.get("from", type=from_iso_format),
|
|
||||||
request.args.get("to", type=from_iso_format),
|
|
||||||
request.args.get("limit", type=int),
|
|
||||||
request.args.get("offset", type=int),
|
|
||||||
"descending" in request.args,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def no_content():
|
def no_content():
|
||||||
return make_response(jsonify(""), NO_CONTENT)
|
return make_response(jsonify(""), NO_CONTENT)
|
||||||
|
|
|
@ -1 +0,0 @@
|
||||||
"""Common utilities"""
|
|
|
@ -1,4 +1,10 @@
|
||||||
import datetime
|
import datetime
|
||||||
|
import sys
|
||||||
|
|
||||||
|
if sys.version_info < (3, 7):
|
||||||
|
from backports.datetime_fromisoformat import MonkeyPatch
|
||||||
|
|
||||||
|
MonkeyPatch.patch_fromisoformat()
|
||||||
|
|
||||||
|
|
||||||
def from_iso_format(date_str):
|
def from_iso_format(date_str):
|
||||||
|
|
|
@ -14,7 +14,7 @@ def extract_session(permission=None):
|
||||||
logger.debug("Missing Authorization header or ill-formed")
|
logger.debug("Missing Authorization header or ill-formed")
|
||||||
raise Unauthorized
|
raise Unauthorized
|
||||||
|
|
||||||
session = sessionController.validate_token(token, request.headers, permission)
|
session = sessionController.validate_token(token, request.user_agent, permission)
|
||||||
return session
|
return session
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,81 +1,43 @@
|
||||||
from functools import wraps
|
_hook_dict = ({}, {})
|
||||||
|
|
||||||
|
|
||||||
_hooks_before = {}
|
class Hook(object):
|
||||||
_hooks_after = {}
|
"""Decorator for Hooks
|
||||||
|
Use to decorate system hooks where plugins should be able to hook in
|
||||||
|
|
||||||
def Hook(function=None, id=None):
|
|
||||||
"""Hook decorator
|
|
||||||
|
|
||||||
Use to decorate functions as hooks, so plugins can hook up their custom functions.
|
|
||||||
"""
|
"""
|
||||||
# `id` passed as `arg` not `kwarg`
|
|
||||||
if isinstance(function, str):
|
|
||||||
return Hook(id=function)
|
|
||||||
|
|
||||||
def decorator(function):
|
def __init__(self, function):
|
||||||
@wraps(function)
|
self.function = function
|
||||||
def wrapped(*args, **kwargs):
|
|
||||||
_id = id if id is not None else function.__qualname__
|
|
||||||
# Hooks before
|
|
||||||
for f in _hooks_before.get(_id, []):
|
|
||||||
f(*args, **kwargs)
|
|
||||||
# Main function
|
|
||||||
result = function(*args, **kwargs)
|
|
||||||
# Hooks after
|
|
||||||
for f in _hooks_after.get(_id, []):
|
|
||||||
f(*args, hook_result=result, **kwargs)
|
|
||||||
return result
|
|
||||||
|
|
||||||
return wrapped
|
def __call__(self, *args, **kwargs):
|
||||||
|
# Hooks before
|
||||||
# Called @Hook or we are in the second step
|
for function in _hook_dict[0].get(self.function.__name__, []):
|
||||||
if callable(function):
|
function(*args, **kwargs)
|
||||||
return decorator(function)
|
# Main function
|
||||||
else:
|
ret = self.function(*args, **kwargs)
|
||||||
return decorator
|
# Hooks after
|
||||||
|
for function in _hook_dict[1].get(self.function.__name__, []):
|
||||||
|
function(*args, **kwargs)
|
||||||
|
return ret
|
||||||
|
|
||||||
|
|
||||||
def HookBefore(id: str):
|
class HookBefore(object):
|
||||||
"""Decorator for functions to be called before a Hook-Function is called
|
"""Decorator for functions to be called before a Hook-Function is called"""
|
||||||
|
|
||||||
The hooked up function must accept the same arguments as the function hooked onto,
|
def __init__(self, name):
|
||||||
as the functions are called with the same arguments.
|
self.name = name
|
||||||
|
|
||||||
Hint: This enables you to modify the arguments!
|
def __call__(self, function):
|
||||||
"""
|
_hook_dict[0].setdefault(self.name, []).append(function)
|
||||||
if not id or not isinstance(id, str):
|
|
||||||
raise TypeError("HookBefore requires the ID of the function to hook up")
|
|
||||||
|
|
||||||
def wrapped(function):
|
|
||||||
_hooks_before.setdefault(id, []).append(function)
|
|
||||||
return function
|
return function
|
||||||
|
|
||||||
return wrapped
|
|
||||||
|
|
||||||
|
class HookAfter(object):
|
||||||
|
"""Decorator for functions to be called after a Hook-Function is called"""
|
||||||
|
|
||||||
def HookAfter(id: str):
|
def __init__(self, name):
|
||||||
"""Decorator for functions to be called after a Hook-Function is called
|
self.name = name
|
||||||
|
|
||||||
As with the HookBefore, the hooked up function must accept the same
|
def __call__(self, function):
|
||||||
arguments as the function hooked onto, but also receives a
|
_hook_dict[1].setdefault(self.name, []).append(function)
|
||||||
`hook_result` kwarg containing the result of the function.
|
|
||||||
|
|
||||||
Example:
|
|
||||||
```py
|
|
||||||
@HookAfter("some.id")
|
|
||||||
def my_func(hook_result):
|
|
||||||
# This function is executed after the function registered with "some.id"
|
|
||||||
print(hook_result) # This is the result of the function
|
|
||||||
```
|
|
||||||
"""
|
|
||||||
|
|
||||||
if not id or not isinstance(id, str):
|
|
||||||
raise TypeError("HookAfter requires the ID of the function to hook up")
|
|
||||||
|
|
||||||
def wrapped(function):
|
|
||||||
_hooks_after.setdefault(id, []).append(function)
|
|
||||||
return function
|
return function
|
||||||
|
|
||||||
return wrapped
|
|
||||||
|
|
|
@ -0,0 +1,17 @@
|
||||||
|
from flask import current_app
|
||||||
|
|
||||||
|
from flaschengeist.utils.HTTP import no_content
|
||||||
|
|
||||||
|
_scheduled = set()
|
||||||
|
|
||||||
|
|
||||||
|
def scheduled(func):
|
||||||
|
_scheduled.add(func)
|
||||||
|
return func
|
||||||
|
|
||||||
|
|
||||||
|
@current_app.route("/cron")
|
||||||
|
def __run_scheduled():
|
||||||
|
for function in _scheduled:
|
||||||
|
function()
|
||||||
|
return no_content()
|
|
@ -1,6 +0,0 @@
|
||||||
[build-system]
|
|
||||||
requires = ["setuptools", "wheel"]
|
|
||||||
build-backend = "setuptools.build_meta"
|
|
||||||
|
|
||||||
[tool.black]
|
|
||||||
line-length = 120
|
|
|
@ -0,0 +1,77 @@
|
||||||
|
# Flaschengeist
|
||||||
|
This is the backend of the Flaschengeist.
|
||||||
|
|
||||||
|
## Installation
|
||||||
|
### Requirements
|
||||||
|
- `mysql` or `mariadb`
|
||||||
|
- maybe `libmariadb` development files[1]
|
||||||
|
- python 3.7+
|
||||||
|
|
||||||
|
[1] By default Flaschengeist uses mysql as database backend, if you are on Windows Flaschengeist uses `PyMySQL`, but on
|
||||||
|
Linux / Mac the faster `mysqlclient` is used, if it is not already installed installing from pypi requires the
|
||||||
|
development files for `libmariadb` to be present on your system.
|
||||||
|
|
||||||
|
### Install python files
|
||||||
|
pip3 install --user .
|
||||||
|
or with ldap support
|
||||||
|
|
||||||
|
pip3 install --user ".[ldap]"
|
||||||
|
or if you want to also run the tests:
|
||||||
|
|
||||||
|
pip3 install --user ".[ldap,test]"
|
||||||
|
|
||||||
|
You will also need a MySQL driver, recommended drivers are
|
||||||
|
- `mysqlclient`
|
||||||
|
- `PyMySQL`
|
||||||
|
|
||||||
|
`setup.py` will try to install a matching driver.
|
||||||
|
|
||||||
|
#### Windows
|
||||||
|
Same as above, but if you want to use `mysqlclient` instead of `PyMySQL` (performance?) you have to follow this guide:
|
||||||
|
|
||||||
|
https://www.radishlogic.com/coding/python-3/installing-mysqldb-for-python-3-in-windows/
|
||||||
|
|
||||||
|
### Configuration
|
||||||
|
Configuration is done within the a `flaschengeist.toml`file, you can copy the one located inside the module path
|
||||||
|
(where flaschegeist is installed) or create an empty one and place it inside either:
|
||||||
|
1. `~/.config/`
|
||||||
|
2. A custom path and set environment variable `FLASCHENGEIST_CONF`
|
||||||
|
|
||||||
|
Uncomment and change at least all the database parameters!
|
||||||
|
|
||||||
|
### Database installation
|
||||||
|
The user needs to have full permissions to the database.
|
||||||
|
If not you need to create user and database manually do (or similar on Windows):
|
||||||
|
|
||||||
|
(
|
||||||
|
echo "CREATE DATABASE flaschengeist;"
|
||||||
|
echo "CREATE USER 'flaschengeist'@'localhost' IDENTIFIED BY 'flaschengeist';"
|
||||||
|
echo "GRANT ALL PRIVILEGES ON flaschengeist.* TO 'flaschengeist'@'localhost';"
|
||||||
|
echo "FLUSH PRIVILEGES;"
|
||||||
|
) | sudo mysql
|
||||||
|
|
||||||
|
Then you can install the database tables and initial entries:
|
||||||
|
|
||||||
|
run_flaschengeist install
|
||||||
|
|
||||||
|
### Run
|
||||||
|
run_flaschengeist run
|
||||||
|
or with debug messages:
|
||||||
|
|
||||||
|
run_flaschengeist run --debug
|
||||||
|
|
||||||
|
This will run the backend on http://localhost:5000
|
||||||
|
|
||||||
|
## Tests
|
||||||
|
$ pip install '.[test]'
|
||||||
|
$ pytest
|
||||||
|
Run with coverage report:
|
||||||
|
|
||||||
|
$ coverage run -m pytest
|
||||||
|
$ coverage report
|
||||||
|
Or with html output (open `htmlcov/index.html` in a browser):
|
||||||
|
|
||||||
|
$ coverage html
|
||||||
|
|
||||||
|
## Development
|
||||||
|
Please refer to our [development wiki](https://flaschengeist.dev/Flaschengeist/flaschengeist/wiki/Development).
|
|
@ -0,0 +1,221 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
from __future__ import annotations # TODO: Remove if python requirement is >= 3.10
|
||||||
|
import inspect
|
||||||
|
import argparse
|
||||||
|
import sys
|
||||||
|
|
||||||
|
import pkg_resources
|
||||||
|
|
||||||
|
from flaschengeist.config import config
|
||||||
|
|
||||||
|
|
||||||
|
class PrefixMiddleware(object):
|
||||||
|
def __init__(self, app, prefix=""):
|
||||||
|
self.app = app
|
||||||
|
self.prefix = prefix
|
||||||
|
|
||||||
|
def __call__(self, environ, start_response):
|
||||||
|
|
||||||
|
if environ["PATH_INFO"].startswith(self.prefix):
|
||||||
|
environ["PATH_INFO"] = environ["PATH_INFO"][len(self.prefix) :]
|
||||||
|
environ["SCRIPT_NAME"] = self.prefix
|
||||||
|
return self.app(environ, start_response)
|
||||||
|
else:
|
||||||
|
start_response("404", [("Content-Type", "text/plain")])
|
||||||
|
return ["This url does not belong to the app.".encode()]
|
||||||
|
|
||||||
|
|
||||||
|
class InterfaceGenerator:
|
||||||
|
known = []
|
||||||
|
classes = {}
|
||||||
|
mapper = {
|
||||||
|
"str": "string",
|
||||||
|
"int": "number",
|
||||||
|
"float": "number",
|
||||||
|
"date": "Date",
|
||||||
|
"datetime": "Date",
|
||||||
|
"NoneType": "null",
|
||||||
|
"bool": "boolean",
|
||||||
|
}
|
||||||
|
|
||||||
|
def __init__(self, namespace, filename):
|
||||||
|
self.basename = ""
|
||||||
|
self.namespace = namespace
|
||||||
|
self.filename = filename
|
||||||
|
self.this_type = None
|
||||||
|
|
||||||
|
def pytype(self, cls):
|
||||||
|
a = self._pytype(cls)
|
||||||
|
print(f"{cls} -> {a}")
|
||||||
|
return a
|
||||||
|
|
||||||
|
def _pytype(self, cls):
|
||||||
|
import typing
|
||||||
|
|
||||||
|
origin = typing.get_origin(cls)
|
||||||
|
arguments = typing.get_args(cls)
|
||||||
|
|
||||||
|
if origin is typing.ForwardRef: # isinstance(cls, typing.ForwardRef):
|
||||||
|
return "", "this" if cls.__forward_arg__ == self.this_type else cls.__forward_arg__
|
||||||
|
if origin is typing.Union:
|
||||||
|
print(f"A1: {arguments[1]}")
|
||||||
|
if len(arguments) == 2 and arguments[1] is type(None):
|
||||||
|
return "?", self.pytype(arguments[0])[1]
|
||||||
|
else:
|
||||||
|
return "", "|".join([self.pytype(pt)[1] for pt in arguments])
|
||||||
|
if origin is list:
|
||||||
|
return "", "Array<{}>".format("|".join([self.pytype(a_type)[1] for a_type in arguments]))
|
||||||
|
|
||||||
|
name = cls.__name__ if hasattr(cls, "__name__") else cls if isinstance(cls, str) else None
|
||||||
|
if name is not None:
|
||||||
|
if name in self.mapper:
|
||||||
|
return "", self.mapper[name]
|
||||||
|
else:
|
||||||
|
return "", name
|
||||||
|
print(
|
||||||
|
"WARNING: This python version might not detect all types (try >= 3.9). Could not identify >{}<".format(cls)
|
||||||
|
)
|
||||||
|
return "?", "any"
|
||||||
|
|
||||||
|
def walker(self, module):
|
||||||
|
if sys.version_info < (3, 9):
|
||||||
|
raise RuntimeError("Python >= 3.9 is required to export API")
|
||||||
|
import typing
|
||||||
|
|
||||||
|
if (
|
||||||
|
inspect.ismodule(module[1])
|
||||||
|
and module[1].__name__.startswith(self.basename)
|
||||||
|
and module[1].__name__ not in self.known
|
||||||
|
):
|
||||||
|
self.known.append(module[1].__name__)
|
||||||
|
for cls in inspect.getmembers(module[1], lambda x: inspect.isclass(x) or inspect.ismodule(x)):
|
||||||
|
self.walker(cls)
|
||||||
|
elif (
|
||||||
|
inspect.isclass(module[1])
|
||||||
|
and module[1].__module__.startswith(self.basename)
|
||||||
|
and module[0] not in self.classes
|
||||||
|
and not module[0].startswith("_")
|
||||||
|
and hasattr(module[1], "__annotations__")
|
||||||
|
):
|
||||||
|
self.this_type = module[0]
|
||||||
|
print("\n\n" + module[0] + "\n")
|
||||||
|
d = {}
|
||||||
|
for param, ptype in typing.get_type_hints(module[1], globalns=None, localns=None).items():
|
||||||
|
if not param.startswith("_") and not param.endswith("_"):
|
||||||
|
print(f"{param} ::: {ptype}")
|
||||||
|
d[param] = self.pytype(ptype)
|
||||||
|
|
||||||
|
if len(d) == 1:
|
||||||
|
key, value = d.popitem()
|
||||||
|
self.classes[module[0]] = value[1]
|
||||||
|
else:
|
||||||
|
self.classes[module[0]] = d
|
||||||
|
|
||||||
|
def run(self, models):
|
||||||
|
self.basename = models.__name__
|
||||||
|
self.walker(("models", models))
|
||||||
|
|
||||||
|
def write(self):
|
||||||
|
with open(self.filename, "w") as file:
|
||||||
|
file.write("declare namespace {} {{\n".format(self.namespace))
|
||||||
|
for cls, params in self.classes.items():
|
||||||
|
if isinstance(params, str):
|
||||||
|
file.write("\ttype {} = {};\n".format(cls, params))
|
||||||
|
else:
|
||||||
|
file.write("\tinterface {} {{\n".format(cls))
|
||||||
|
for name in params:
|
||||||
|
file.write("\t\t{}{}: {};\n".format(name, *params[name]))
|
||||||
|
file.write("\t}\n")
|
||||||
|
file.write("}\n")
|
||||||
|
|
||||||
|
|
||||||
|
def install(arguments):
|
||||||
|
from flaschengeist.app import create_app, install_all
|
||||||
|
|
||||||
|
app = create_app()
|
||||||
|
with app.app_context():
|
||||||
|
install_all()
|
||||||
|
|
||||||
|
|
||||||
|
def run(arguments):
|
||||||
|
from flaschengeist.app import create_app
|
||||||
|
|
||||||
|
app = create_app()
|
||||||
|
with app.app_context():
|
||||||
|
app.wsgi_app = PrefixMiddleware(app.wsgi_app, prefix=config["FLASCHENGEIST"].get("root", ""))
|
||||||
|
if arguments.debug:
|
||||||
|
app.run(arguments.host, arguments.port, debug=True)
|
||||||
|
else:
|
||||||
|
app.run(arguments.host, arguments.port, debug=False)
|
||||||
|
|
||||||
|
|
||||||
|
def export(arguments):
|
||||||
|
import flaschengeist.models as models
|
||||||
|
from flaschengeist.app import create_app
|
||||||
|
|
||||||
|
app = create_app()
|
||||||
|
with app.app_context():
|
||||||
|
gen = InterfaceGenerator(arguments.namespace, arguments.file)
|
||||||
|
if not arguments.no_core:
|
||||||
|
gen.run(models)
|
||||||
|
if arguments.plugins is not None:
|
||||||
|
for entry_point in pkg_resources.iter_entry_points("flaschengeist.plugin"):
|
||||||
|
if len(arguments.plugins) == 0 or entry_point.name in arguments.plugins:
|
||||||
|
plg = entry_point.load()
|
||||||
|
if hasattr(plg, "models") and plg.models is not None:
|
||||||
|
gen.run(plg.models)
|
||||||
|
gen.write()
|
||||||
|
|
||||||
|
|
||||||
|
def ldap_sync(arguments):
|
||||||
|
from flaschengeist.app import create_app
|
||||||
|
from flaschengeist.controller import userController
|
||||||
|
from flaschengeist.plugins.auth_ldap import AuthLDAP
|
||||||
|
from ldap3 import SUBTREE
|
||||||
|
|
||||||
|
app = create_app()
|
||||||
|
with app.app_context():
|
||||||
|
auth_ldap: AuthLDAP = app.config.get("FG_PLUGINS").get("auth_ldap")
|
||||||
|
if auth_ldap:
|
||||||
|
conn = auth_ldap.ldap.connection
|
||||||
|
if not conn:
|
||||||
|
conn = auth_ldap.ldap.connect(auth_ldap.root_dn, auth_ldap.root_secret)
|
||||||
|
conn.search(auth_ldap.search_dn, "(uid=*)", SUBTREE, attributes=["uid", "givenName", "sn", "mail"])
|
||||||
|
ldap_users_response = conn.response
|
||||||
|
for ldap_user in ldap_users_response:
|
||||||
|
uid = ldap_user["attributes"]["uid"][0]
|
||||||
|
userController.find_user(uid)
|
||||||
|
exit()
|
||||||
|
raise Exception("auth_ldap not found")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
# create the top-level parser
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
subparsers = parser.add_subparsers(help="sub-command help", dest="sub_command")
|
||||||
|
subparsers.required = True
|
||||||
|
parser_run = subparsers.add_parser("run", help="run flaschengeist")
|
||||||
|
parser_run.set_defaults(func=run)
|
||||||
|
parser_run.add_argument("--host", help="set hostname to listen on", default="127.0.0.1")
|
||||||
|
parser_run.add_argument("--port", help="set port to listen on", type=int, default=5000)
|
||||||
|
parser_run.add_argument("--debug", help="run in debug mode", action="store_true")
|
||||||
|
parser_install = subparsers.add_parser(
|
||||||
|
"install", help="run database setup for flaschengeist and all installed plugins"
|
||||||
|
)
|
||||||
|
parser_install.set_defaults(func=install)
|
||||||
|
parser_export = subparsers.add_parser("export", help="export models to typescript interfaces")
|
||||||
|
parser_export.set_defaults(func=export)
|
||||||
|
parser_export.add_argument("--file", help="Filename where to save", default="flaschengeist.d.ts")
|
||||||
|
parser_export.add_argument("--namespace", help="Namespace of declarations", default="FG")
|
||||||
|
parser_export.add_argument(
|
||||||
|
"--no-core",
|
||||||
|
help="Do not export core declarations (only useful in conjunction with --plugins)",
|
||||||
|
action="store_true",
|
||||||
|
)
|
||||||
|
parser_export.add_argument("--plugins", help="Also export plugins (none means all)", nargs="*")
|
||||||
|
|
||||||
|
parser_ldap_sync = subparsers.add_parser("ldap_sync", help="synch ldap-users with database")
|
||||||
|
parser_ldap_sync.set_defaults(func=ldap_sync)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
args.func(args)
|
55
setup.cfg
55
setup.cfg
|
@ -1,6 +1,6 @@
|
||||||
[metadata]
|
[metadata]
|
||||||
license = MIT
|
license = MIT
|
||||||
version = 2.1.0
|
version = 2.0.0.dev0
|
||||||
name = flaschengeist
|
name = flaschengeist
|
||||||
author = Tim Gröger
|
author = Tim Gröger
|
||||||
author_email = flaschengeist@wu5.de
|
author_email = flaschengeist@wu5.de
|
||||||
|
@ -17,57 +17,6 @@ classifiers =
|
||||||
License :: OSI Approved :: MIT License
|
License :: OSI Approved :: MIT License
|
||||||
Operating System :: OS Independent
|
Operating System :: OS Independent
|
||||||
|
|
||||||
[options]
|
|
||||||
include_package_data = True
|
|
||||||
python_requires = >=3.10
|
|
||||||
packages = find:
|
|
||||||
install_requires =
|
|
||||||
#Flask>=2.2.2, <2.3
|
|
||||||
Flask>=2.2.2, <2.9
|
|
||||||
Pillow>=9.2
|
|
||||||
flask_cors
|
|
||||||
flask_migrate>=3.1.0
|
|
||||||
flask_sqlalchemy>=2.5.1
|
|
||||||
sqlalchemy_utils>=0.38.3
|
|
||||||
# Importlib requirement can be dropped when python requirement is >= 3.10
|
|
||||||
importlib_metadata>=4.3
|
|
||||||
#sqlalchemy>=1.4.40, <2.0
|
|
||||||
sqlalchemy >= 2.0
|
|
||||||
toml
|
|
||||||
werkzeug>=2.2.2
|
|
||||||
ua-parser>=0.16.1
|
|
||||||
|
|
||||||
[options.extras_require]
|
|
||||||
argon = argon2-cffi
|
|
||||||
ldap = flask_ldapconn @ git+https://github.com/rroemhild/flask-ldapconn.git; ldap3
|
|
||||||
tests = pytest; pytest-depends; coverage
|
|
||||||
mysql =
|
|
||||||
PyMySQL;platform_system=='Windows'
|
|
||||||
mysqlclient;platform_system!='Windows'
|
|
||||||
|
|
||||||
[options.package_data]
|
|
||||||
* = *.toml, script.py.mako, *.ini, */migrations/*, migrations/versions/*
|
|
||||||
|
|
||||||
[options.entry_points]
|
|
||||||
console_scripts =
|
|
||||||
flaschengeist = flaschengeist.cli:main
|
|
||||||
flask.commands =
|
|
||||||
ldap = flaschengeist.plugins.auth_ldap.cli:ldap
|
|
||||||
user = flaschengeist.plugins.users.cli:user
|
|
||||||
role = flaschengeist.plugins.users.cli:role
|
|
||||||
flaschengeist.plugins =
|
|
||||||
# Authentication providers
|
|
||||||
auth_plain = flaschengeist.plugins.auth_plain:AuthPlain
|
|
||||||
auth_ldap = flaschengeist.plugins.auth_ldap:AuthLDAP [ldap]
|
|
||||||
# Route providers (and misc)
|
|
||||||
auth = flaschengeist.plugins.auth:AuthRoutePlugin
|
|
||||||
users = flaschengeist.plugins.users:UsersPlugin
|
|
||||||
roles = flaschengeist.plugins.roles:RolesPlugin
|
|
||||||
balance = flaschengeist.plugins.balance:BalancePlugin
|
|
||||||
mail = flaschengeist.plugins.message_mail:MailMessagePlugin
|
|
||||||
pricelist = flaschengeist.plugins.pricelist:PriceListPlugin
|
|
||||||
scheduler = flaschengeist.plugins.scheduler:SchedulerPlugin
|
|
||||||
|
|
||||||
[bdist_wheel]
|
[bdist_wheel]
|
||||||
universal = True
|
universal = True
|
||||||
|
|
||||||
|
@ -76,4 +25,4 @@ testpaths = tests
|
||||||
|
|
||||||
[coverage:run]
|
[coverage:run]
|
||||||
branch = True
|
branch = True
|
||||||
source = flaschengeist
|
source = flaschengeist
|
|
@ -0,0 +1,77 @@
|
||||||
|
from setuptools import setup, find_packages, Command
|
||||||
|
import subprocess
|
||||||
|
import os
|
||||||
|
|
||||||
|
mysql_driver = "PyMySQL" if os.name == "nt" else "mysqlclient"
|
||||||
|
|
||||||
|
|
||||||
|
class DocsCommand(Command):
|
||||||
|
description = "Generate and export API documentation using pdoc3"
|
||||||
|
user_options = [
|
||||||
|
# The format is (long option, short option, description).
|
||||||
|
("output=", "o", "Documentation output path"),
|
||||||
|
]
|
||||||
|
|
||||||
|
def initialize_options(self):
|
||||||
|
self.output = "./docs"
|
||||||
|
|
||||||
|
def finalize_options(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
"""Run command."""
|
||||||
|
command = [
|
||||||
|
"python",
|
||||||
|
"-m",
|
||||||
|
"pdoc",
|
||||||
|
"--skip-errors",
|
||||||
|
"--html",
|
||||||
|
"--output-dir",
|
||||||
|
self.output,
|
||||||
|
"flaschengeist",
|
||||||
|
]
|
||||||
|
self.announce(
|
||||||
|
"Running command: %s" % str(command),
|
||||||
|
)
|
||||||
|
subprocess.check_call(command)
|
||||||
|
|
||||||
|
|
||||||
|
setup(
|
||||||
|
packages=find_packages(),
|
||||||
|
package_data={"": ["*.toml"]},
|
||||||
|
scripts=["run_flaschengeist"],
|
||||||
|
python_requires=">=3.7",
|
||||||
|
install_requires=[
|
||||||
|
"Flask >= 2.0",
|
||||||
|
"toml",
|
||||||
|
"sqlalchemy>=1.4.26",
|
||||||
|
"flask_sqlalchemy>=2.5",
|
||||||
|
"flask_cors",
|
||||||
|
"Pillow>=8.4.0",
|
||||||
|
"werkzeug",
|
||||||
|
mysql_driver,
|
||||||
|
],
|
||||||
|
extras_require={
|
||||||
|
"ldap": ["flask_ldapconn", "ldap3"],
|
||||||
|
"argon": ["argon2-cffi"],
|
||||||
|
"test": ["pytest", "coverage"],
|
||||||
|
},
|
||||||
|
entry_points={
|
||||||
|
"flaschengeist.plugin": [
|
||||||
|
# Authentication providers
|
||||||
|
"auth_plain = flaschengeist.plugins.auth_plain:AuthPlain",
|
||||||
|
"auth_ldap = flaschengeist.plugins.auth_ldap:AuthLDAP [ldap]",
|
||||||
|
# Route providers (and misc)
|
||||||
|
"auth = flaschengeist.plugins.auth:AuthRoutePlugin",
|
||||||
|
"users = flaschengeist.plugins.users:UsersPlugin",
|
||||||
|
"roles = flaschengeist.plugins.roles:RolesPlugin",
|
||||||
|
"balance = flaschengeist.plugins.balance:BalancePlugin",
|
||||||
|
"events = flaschengeist.plugins.events:EventPlugin",
|
||||||
|
"mail = flaschengeist.plugins.message_mail:MailMessagePlugin",
|
||||||
|
"pricelist = flaschengeist.plugins.pricelist:PriceListPlugin",
|
||||||
|
],
|
||||||
|
},
|
||||||
|
cmdclass={
|
||||||
|
"docs": DocsCommand,
|
||||||
|
},
|
||||||
|
)
|
|
@ -3,7 +3,8 @@ import tempfile
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
from flaschengeist import database
|
from flaschengeist import database
|
||||||
from flaschengeist.app import create_app
|
from flaschengeist.app import create_app, install_all
|
||||||
|
|
||||||
|
|
||||||
# read in SQL for populating test data
|
# read in SQL for populating test data
|
||||||
with open(os.path.join(os.path.dirname(__file__), "data.sql"), "r") as f:
|
with open(os.path.join(os.path.dirname(__file__), "data.sql"), "r") as f:
|
||||||
|
@ -24,14 +25,12 @@ def app():
|
||||||
app = create_app(
|
app = create_app(
|
||||||
{
|
{
|
||||||
"TESTING": True,
|
"TESTING": True,
|
||||||
"DATABASE": {"engine": "sqlite", "database": f"/{db_path}"},
|
"DATABASE": {"file_path": f"/{db_path}"},
|
||||||
"LOGGING": {"level": "DEBUG"},
|
"LOGGING": {"level": "DEBUG"},
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
with app.app_context():
|
with app.app_context():
|
||||||
database.db.create_all()
|
install_all()
|
||||||
database.db.session.commit()
|
|
||||||
|
|
||||||
engine = database.db.engine
|
engine = database.db.engine
|
||||||
with engine.connect() as connection:
|
with engine.connect() as connection:
|
||||||
for statement in _data_sql:
|
for statement in _data_sql:
|
||||||
|
|
|
@ -1,8 +1,4 @@
|
||||||
INSERT INTO "user" ('userid', 'firstname', 'lastname', 'mail', 'deleted', 'id') VALUES ('user', 'Max', 'Mustermann', 'abc@def.gh', 0, 1);
|
INSERT INTO user ('userid', 'firstname', 'lastname', 'mail', 'id') VALUES ('user', 'Max', 'Mustermann', 'abc@def.gh', 1);
|
||||||
INSERT INTO "user" ('userid', 'firstname', 'lastname', 'mail', 'deleted', 'id') VALUES ('deleted_user', 'John', 'Doe', 'doe@example.com', 1, 2);
|
|
||||||
-- Password = 1234
|
-- Password = 1234
|
||||||
INSERT INTO user_attribute VALUES(1,1,'password',X'800495c4000000000000008cc0373731346161336536623932613830366664353038656631323932623134393936393561386463353536623037363761323037623238346264623833313265323333373066376233663462643332666332653766303537333564366335393133366463366234356539633865613835643661643435343931376636626663343163653333643635646530386634396231323061316236386162613164373663663333306564306463303737303733336136353363393538396536343266393865942e');
|
INSERT INTO user_attribute VALUES(1,1,'password',X'800495c4000000000000008cc0373731346161336536623932613830366664353038656631323932623134393936393561386463353536623037363761323037623238346264623833313265323333373066376233663462643332666332653766303537333564366335393133366463366234356539633865613835643661643435343931376636626663343163653333643635646530386634396231323061316236386162613164373663663333306564306463303737303733336136353363393538396536343266393865942e');
|
||||||
INSERT INTO session ('expires', 'token', 'lifetime', 'id', 'user_id') VALUES ('2999-01-01 00:00:00', 'f4ecbe14be3527ca998143a49200e294', 600, 1, 1);
|
INSERT INTO session ('expires', 'token', 'lifetime', 'id', 'user_id') VALUES ('2999-01-01 00:00:00', 'f4ecbe14be3527ca998143a49200e294', 600, 1, 1);
|
||||||
-- ROLES
|
|
||||||
INSERT INTO role ('name', 'id') VALUES ('role_1', 1);
|
|
||||||
INSERT INTO permission ('name', 'id') VALUES ('permission_1', 1);
|
|
|
@ -15,9 +15,9 @@ def test_login(client):
|
||||||
# Login successful
|
# Login successful
|
||||||
assert result.status_code == 201
|
assert result.status_code == 201
|
||||||
# User set correctly
|
# User set correctly
|
||||||
assert json["userid"] == USERID
|
assert json["user"]["userid"] == USERID
|
||||||
# Token works
|
# Token works
|
||||||
assert client.get("/auth", headers={"Authorization": f"Bearer {json['token']}"}).status_code == 200
|
assert client.get("/auth", headers={"Authorization": f"Bearer {json['session']['token']}"}).status_code == 200
|
||||||
|
|
||||||
|
|
||||||
def test_login_decorator(client):
|
def test_login_decorator(client):
|
||||||
|
|
|
@ -0,0 +1,17 @@
|
||||||
|
import pytest
|
||||||
|
from werkzeug.exceptions import BadRequest
|
||||||
|
|
||||||
|
import flaschengeist.plugins.events.event_controller as event_controller
|
||||||
|
from flaschengeist.plugins.events.models import EventType
|
||||||
|
|
||||||
|
VALID_TOKEN = "f4ecbe14be3527ca998143a49200e294"
|
||||||
|
EVENT_TYPE_NAME = "Test Type"
|
||||||
|
|
||||||
|
|
||||||
|
def test_create_event_type(app):
|
||||||
|
with app.app_context():
|
||||||
|
type = event_controller.create_event_type(EVENT_TYPE_NAME)
|
||||||
|
assert isinstance(type, EventType)
|
||||||
|
|
||||||
|
with pytest.raises(BadRequest):
|
||||||
|
event_controller.create_event_type(EVENT_TYPE_NAME)
|
|
@ -1,52 +0,0 @@
|
||||||
import pytest
|
|
||||||
from werkzeug.exceptions import BadRequest, NotFound
|
|
||||||
from flaschengeist.controller import roleController, userController
|
|
||||||
from flaschengeist.models.user import User
|
|
||||||
|
|
||||||
VALID_TOKEN = "f4ecbe14be3527ca998143a49200e294"
|
|
||||||
|
|
||||||
|
|
||||||
def test_get_user(app):
|
|
||||||
with app.app_context():
|
|
||||||
user = userController.get_user("user")
|
|
||||||
assert user is not None and isinstance(user, User)
|
|
||||||
assert user.userid == "user"
|
|
||||||
|
|
||||||
user = userController.get_user("deleted_user", deleted=True)
|
|
||||||
assert user is not None and isinstance(user, User)
|
|
||||||
assert user.userid == "deleted_user"
|
|
||||||
|
|
||||||
with pytest.raises(NotFound):
|
|
||||||
user = userController.get_user("__does_not_exist__")
|
|
||||||
with pytest.raises(NotFound):
|
|
||||||
user = userController.get_user("__does_not_exist__", deleted=True)
|
|
||||||
with pytest.raises(NotFound):
|
|
||||||
user = userController.get_user("deleted_user")
|
|
||||||
|
|
||||||
|
|
||||||
def test_set_roles(app):
|
|
||||||
with app.app_context():
|
|
||||||
user = userController.get_user("user")
|
|
||||||
userController.set_roles(user, [])
|
|
||||||
assert user.roles_ == []
|
|
||||||
|
|
||||||
userController.set_roles(user, ["role_1"])
|
|
||||||
assert len(user.roles_) == 1 and user.roles_[0].id == 1
|
|
||||||
|
|
||||||
# Test unknown role + no create flag -> raise no changes
|
|
||||||
with pytest.raises(BadRequest):
|
|
||||||
userController.set_roles(user, ["__custom__"])
|
|
||||||
assert len(user.roles_) == 1
|
|
||||||
|
|
||||||
userController.set_roles(user, ["__custom__"], create=True)
|
|
||||||
assert len(user.roles_) == 1 and user.roles_[0].name == "__custom__"
|
|
||||||
assert roleController.get("__custom__").id == user.roles_[0].id
|
|
||||||
|
|
||||||
userController.set_roles(user, ["__custom__"], create=True)
|
|
||||||
assert len(user.roles_) == 1
|
|
||||||
|
|
||||||
userController.set_roles(user, ["__custom__", "role_1"])
|
|
||||||
assert len(user.roles_) == 2
|
|
||||||
|
|
||||||
userController.set_roles(user, [])
|
|
||||||
assert len(user.roles_) == 0
|
|
Loading…
Reference in New Issue