Merge pull request 'feature/migrations, closes #19' (#20) from feature/migrations into develop
Reviewed-on: #20
This commit is contained in:
		
						commit
						cfbb557539
					
				
							
								
								
									
										55
									
								
								README.md
								
								
								
								
							
							
						
						
									
										55
									
								
								README.md
								
								
								
								
							|  | @ -31,18 +31,41 @@ or if you want to also run the tests: | |||
| 
 | ||||
|     pip3 install --user ".[ldap,tests]" | ||||
| 
 | ||||
| You will also need a MySQL driver, recommended drivers are | ||||
| - `mysqlclient` | ||||
| - `PyMySQL` | ||||
| You will also need a MySQL driver, by default one of this is installed: | ||||
| - `mysqlclient` (non Windows) | ||||
| - `PyMySQL` (on Windows) | ||||
| 
 | ||||
| `setup.py` will try to install a matching driver. | ||||
| #### 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/) | ||||
| 
 | ||||
| #### Windows | ||||
| Same as above, but if you want to use `mysqlclient` instead of `PyMySQL` (performance?) you have to follow this guide: | ||||
| ### 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): | ||||
| 
 | ||||
| https://www.radishlogic.com/coding/python-3/installing-mysqldb-for-python-3-in-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 | ||||
| 
 | ||||
| ### Configuration | ||||
| 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/` | ||||
|  | @ -63,21 +86,6 @@ So you have to configure one of the following options to call flaschengeists CRO | |||
|   - 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 | ||||
| 
 | ||||
| ### 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: | ||||
| 
 | ||||
|     $ flaschengeist install | ||||
| 
 | ||||
| ### Run | ||||
| Flaschengeist provides a CLI, based on the flask CLI, respectivly called `flaschengeist`. | ||||
| 
 | ||||
|  | @ -86,6 +94,7 @@ with the difference of the main logger will be forced to output to `stderr` and | |||
| of the CLI will override the logging level you have configured for the main logger. | ||||
| 
 | ||||
|     $ flaschengeist run | ||||
| 
 | ||||
| or with debug messages: | ||||
| 
 | ||||
|     $ flaschengeist run --debug | ||||
|  |  | |||
|  | @ -5,9 +5,53 @@ | |||
|     - 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` | ||||
|  |  | |||
|  | @ -0,0 +1,5 @@ | |||
| from pathlib import Path | ||||
| 
 | ||||
| 
 | ||||
| alembic_migrations_path = str(Path(__file__).resolve().parent / "migrations") | ||||
| alembic_script_path = str(Path(__file__).resolve().parent) | ||||
|  | @ -0,0 +1,53 @@ | |||
| # 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 | ||||
|  | @ -0,0 +1,74 @@ | |||
| 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() | ||||
|  | @ -0,0 +1,154 @@ | |||
| """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 ### | ||||
|  | @ -0,0 +1,25 @@ | |||
| """${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,18 +1,16 @@ | |||
| import enum | ||||
| 
 | ||||
| from flask import Flask, current_app | ||||
| from flask import Flask | ||||
| from flask_cors import CORS | ||||
| from datetime import datetime, date | ||||
| from flask.json import JSONEncoder, jsonify | ||||
| from importlib_metadata import entry_points | ||||
| from sqlalchemy.exc import OperationalError | ||||
| from werkzeug.exceptions import HTTPException | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.controller import pluginController | ||||
| from flaschengeist.utils.hook import Hook | ||||
| from flaschengeist.plugins import AuthPlugin | ||||
| from flaschengeist.controller import roleController | ||||
| from flaschengeist.config import config, configure_app | ||||
| from flaschengeist.config import configure_app | ||||
| 
 | ||||
| 
 | ||||
| class CustomJSONEncoder(JSONEncoder): | ||||
|  | @ -37,78 +35,47 @@ class CustomJSONEncoder(JSONEncoder): | |||
| 
 | ||||
| 
 | ||||
| @Hook("plugins.loaded") | ||||
| def __load_plugins(app): | ||||
|     logger.debug("Search for plugins") | ||||
| 
 | ||||
| def load_plugins(app: Flask): | ||||
|     app.config["FG_PLUGINS"] = {} | ||||
|     for entry_point in entry_points(group="flaschengeist.plugins"): | ||||
|         logger.debug(f"Found plugin: {entry_point.name} ({entry_point.dist.version})") | ||||
| 
 | ||||
|         if entry_point.name == config["FLASCHENGEIST"]["auth"] or ( | ||||
|             entry_point.name in config and config[entry_point.name].get("enabled", False) | ||||
|         ): | ||||
|             logger.debug(f"Load plugin {entry_point.name}") | ||||
|             try: | ||||
|                 plugin = entry_point.load()(entry_point, config=config.get(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, | ||||
|                 ) | ||||
|                 continue | ||||
|             if isinstance(plugin, AuthPlugin): | ||||
|                 if entry_point.name != config["FLASCHENGEIST"]["auth"]: | ||||
|                     logger.debug(f"Unload not configured AuthPlugin {entry_point.name}") | ||||
|                     del plugin | ||||
|                     continue | ||||
|                 else: | ||||
|                     logger.info(f"Using authentication plugin: {entry_point.name}") | ||||
|                     app.config["FG_AUTH_BACKEND"] = plugin | ||||
|             else: | ||||
|                 logger.info(f"Using plugin: {entry_point.name}") | ||||
|             app.config["FG_PLUGINS"][entry_point.name] = plugin | ||||
|         else: | ||||
|             logger.debug(f"Skip disabled plugin {entry_point.name}") | ||||
|     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") | ||||
| 
 | ||||
| 
 | ||||
| @Hook("plugins.installed") | ||||
| def install_all(): | ||||
|     from flaschengeist.database import db | ||||
| 
 | ||||
|     db.create_all() | ||||
|     db.session.commit() | ||||
|     for name, plugin in current_app.config["FG_PLUGINS"].items(): | ||||
|         if not plugin: | ||||
|             logger.debug(f"Skip disabled plugin: {name}") | ||||
|     for plugin in pluginController.get_enabled_plugins(): | ||||
|         logger.debug(f"Searching for enabled plugin {plugin.name}") | ||||
|         try: | ||||
|             # Load class | ||||
|             cls = plugin.entry_point.load() | ||||
|             plugin = cls.query.get(plugin.id) if plugin.id is not None else plugin | ||||
|             # Custom loading tasks | ||||
|             plugin.load() | ||||
|             # 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 | ||||
|         logger.info(f"Install plugin {name}") | ||||
|         plugin.install() | ||||
|         if plugin.permissions: | ||||
|             roleController.create_permissions(plugin.permissions) | ||||
|         logger.info(f"Loaded plugin: {plugin.name}") | ||||
|         app.config["FG_PLUGINS"][plugin.name] = plugin | ||||
| 
 | ||||
| 
 | ||||
| def create_app(test_config=None): | ||||
| def create_app(test_config=None, cli=False): | ||||
|     app = Flask("flaschengeist") | ||||
|     app.json_encoder = CustomJSONEncoder | ||||
|     CORS(app) | ||||
| 
 | ||||
|     with app.app_context(): | ||||
|         from flaschengeist.database import db | ||||
|         from flaschengeist.database import db, migrate | ||||
| 
 | ||||
|         configure_app(app, test_config) | ||||
|         db.init_app(app) | ||||
|         __load_plugins(app) | ||||
|         migrate.init_app(app, db, compare_type=True) | ||||
|         load_plugins(app) | ||||
| 
 | ||||
|     @app.route("/", methods=["GET"]) | ||||
|     def __get_state(): | ||||
|         from . import __version__ as version | ||||
| 
 | ||||
|         return jsonify({"plugins": app.config["FG_PLUGINS"], "version": version}) | ||||
|         return jsonify({"plugins": pluginController.get_loaded_plugins(), "version": version}) | ||||
| 
 | ||||
|     @app.errorhandler(Exception) | ||||
|     def handle_exception(e): | ||||
|  |  | |||
|  | @ -84,16 +84,18 @@ def cli(): | |||
| 
 | ||||
| 
 | ||||
| def main(*args, **kwargs): | ||||
|     # from .plugin_cmd import plugin | ||||
|     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 | ||||
| 
 | ||||
|     # Override logging level | ||||
|     environ.setdefault("FG_LOGGING", logging.getLevelName(LOGGING_MAX)) | ||||
| 
 | ||||
|     # cli.add_command(plugin) | ||||
|     cli.add_command(export) | ||||
|     cli.add_command(docs) | ||||
|     cli.add_command(install) | ||||
|     cli.add_command(plugin) | ||||
|     cli.add_command(run) | ||||
|     cli(*args, **kwargs) | ||||
|  |  | |||
|  | @ -1,4 +1,5 @@ | |||
| import click | ||||
| from importlib.metadata import entry_points | ||||
| 
 | ||||
| 
 | ||||
| @click.command() | ||||
|  | @ -8,14 +9,21 @@ import click | |||
| @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.app import get_plugins | ||||
|     from .InterfaceGenerator import InterfaceGenerator | ||||
| 
 | ||||
|     gen = InterfaceGenerator(namespace, output, logger) | ||||
|     if not no_core: | ||||
|         gen.run(models) | ||||
|     if plugin: | ||||
|         for plugin_class in get_plugins(): | ||||
|             if (len(plugin) == 0 or plugin_class.id in plugin) and plugin_class.models is not None: | ||||
|                 gen.run(plugin_class.models) | ||||
|         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() | ||||
|  |  | |||
|  | @ -0,0 +1,23 @@ | |||
| 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) | ||||
|  | @ -0,0 +1,145 @@ | |||
| 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')}") | ||||
|  | @ -77,17 +77,6 @@ def configure_app(app, test_config=None): | |||
| 
 | ||||
|     configure_logger() | ||||
| 
 | ||||
|     # Always enable this builtin plugins! | ||||
|     update_dict( | ||||
|         config, | ||||
|         { | ||||
|             "auth": {"enabled": True}, | ||||
|             "roles": {"enabled": True}, | ||||
|             "users": {"enabled": True}, | ||||
|             "scheduler": {"enabled": True}, | ||||
|         }, | ||||
|     ) | ||||
| 
 | ||||
|     if "secret_key" not in config["FLASCHENGEIST"]: | ||||
|         logger.critical("No secret key was configured, please configure one for production systems!") | ||||
|         raise RuntimeError("No secret key was configured") | ||||
|  |  | |||
|  | @ -1,15 +1,14 @@ | |||
| from datetime import date | ||||
| from flask import send_file | ||||
| from pathlib import Path | ||||
| from flask import send_file | ||||
| from PIL import Image as PImage | ||||
| 
 | ||||
| from werkzeug.exceptions import NotFound, UnprocessableEntity | ||||
| from werkzeug.datastructures import FileStorage | ||||
| from werkzeug.utils import secure_filename | ||||
| from werkzeug.datastructures import FileStorage | ||||
| from werkzeug.exceptions import NotFound, UnprocessableEntity | ||||
| 
 | ||||
| from flaschengeist.models.image import Image | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist.config import config | ||||
| from ..models import Image | ||||
| from ..database import db | ||||
| from ..config import config | ||||
| 
 | ||||
| 
 | ||||
| def check_mimetype(mime: str): | ||||
|  |  | |||
|  | @ -1,5 +1,5 @@ | |||
| from flaschengeist.utils.hook import Hook | ||||
| from flaschengeist.models.user import User, Role | ||||
| from ..utils.hook import Hook | ||||
| from ..models import User, Role | ||||
| 
 | ||||
| 
 | ||||
| class Message: | ||||
|  |  | |||
|  | @ -3,57 +3,57 @@ | |||
| Used by plugins for setting and notification functionality. | ||||
| """ | ||||
| 
 | ||||
| import sqlalchemy | ||||
| from typing import Union | ||||
| 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 ..models.setting import _PluginSetting | ||||
| from ..models.notification import Notification | ||||
| from ..utils.hook import Hook | ||||
| from ..plugins import Plugin, AuthPlugin | ||||
| from ..models import Notification | ||||
| 
 | ||||
| 
 | ||||
| def get_setting(plugin_id: str, name: str, **kwargs): | ||||
|     """Get plugin setting from database | ||||
| __required_plugins = ["users", "roles", "scheduler", "auth"] | ||||
| 
 | ||||
|     Args: | ||||
|         plugin_id: ID of the plugin | ||||
|         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 | ||||
|     """ | ||||
| 
 | ||||
| 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: | ||||
|         setting = ( | ||||
|             _PluginSetting.query.filter(_PluginSetting.plugin == plugin_id).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(plugin_id: str, name: str, value): | ||||
|     """Save setting in database | ||||
| 
 | ||||
|     Args: | ||||
|         plugin_id: ID of the plugin | ||||
|         name: String identifying the setting | ||||
|         value: Value to be stored | ||||
|     """ | ||||
|     setting = ( | ||||
|         _PluginSetting.query.filter(_PluginSetting.plugin == plugin_id) | ||||
|         .filter(_PluginSetting.name == name) | ||||
|         .one_or_none() | ||||
|     ) | ||||
|     if setting is not None: | ||||
|         if value is None: | ||||
|             db.session.delete(setting) | ||||
|         else: | ||||
|             setting.value = value | ||||
|     else: | ||||
|         db.session.add(_PluginSetting(plugin=plugin_id, name=name, value=value)) | ||||
|     db.session.commit() | ||||
|         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: str, user, text: str, data=None): | ||||
|  | @ -74,3 +74,81 @@ def notify(plugin_id: str, user, text: str, data=None): | |||
|         db.session.add(n) | ||||
|         db.session.commit() | ||||
|         return n.id | ||||
| 
 | ||||
| 
 | ||||
| @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("") | ||||
|     for loc in entry_point[0].module.split(".") + ["migrations"]: | ||||
|         directory /= loc | ||||
|     if directory.exists(): | ||||
|         database_upgrade(revision=f"{plugin_name}@head") | ||||
| 
 | ||||
|     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 | ||||
|  |  | |||
|  | @ -2,10 +2,10 @@ from typing import Union | |||
| from sqlalchemy.exc import IntegrityError | ||||
| from werkzeug.exceptions import BadRequest, Conflict, NotFound | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.models.user import Role, Permission | ||||
| from flaschengeist.database import db, case_sensitive | ||||
| from flaschengeist.utils.hook import Hook | ||||
| from .. import logger | ||||
| from ..models import Role, Permission | ||||
| from ..database import db, case_sensitive | ||||
| from ..utils.hook import Hook | ||||
| 
 | ||||
| 
 | ||||
| def get_all(): | ||||
|  |  | |||
|  | @ -1,14 +1,46 @@ | |||
| import secrets | ||||
| from flaschengeist.models.session import Session | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist import logger | ||||
| from werkzeug.exceptions import Forbidden, Unauthorized | ||||
| 
 | ||||
| from datetime import datetime, timezone | ||||
| from werkzeug.exceptions import Forbidden, Unauthorized | ||||
| 
 | ||||
| from .. import logger | ||||
| from ..models import Session | ||||
| from ..database import db | ||||
| 
 | ||||
| 
 | ||||
| lifetime = 1800 | ||||
| 
 | ||||
| 
 | ||||
| def validate_token(token, user_agent, permission): | ||||
| def __get_user_agent_platform(ua: str): | ||||
|     if "Win" in ua: | ||||
|         return "windows" | ||||
|     if "Mac" in ua: | ||||
|         return "macintosh" | ||||
|     if "Linux" in ua: | ||||
|         return "linux" | ||||
|     if "Android" in ua: | ||||
|         return "android" | ||||
|     if "like Mac" in ua: | ||||
|         return "ios" | ||||
|     return "unknown" | ||||
| 
 | ||||
| 
 | ||||
| def __get_user_agent_browser(ua: str): | ||||
|     ua_str = ua.lower() | ||||
|     if "firefox" in ua_str or "fxios" in ua_str: | ||||
|         return "firefox" | ||||
|     if "safari" in ua_str: | ||||
|         return "safari" | ||||
|     if "opr/" in ua_str: | ||||
|         return "opera" | ||||
|     if "edg" in ua_str: | ||||
|         return "edge" | ||||
|     if "chrom" in ua_str or "crios" in ua_str: | ||||
|         return "chrome" | ||||
|     return "unknown" | ||||
| 
 | ||||
| 
 | ||||
| def validate_token(token, request_headers, permission): | ||||
|     """Verify session | ||||
| 
 | ||||
|     Verify a Session and Roles so if the User has permission or not. | ||||
|  | @ -16,7 +48,7 @@ def validate_token(token, user_agent, permission): | |||
| 
 | ||||
|     Args: | ||||
|         token: Token to verify. | ||||
|         user_agent: User agent of browser to check | ||||
|         request_headers: Headers to validate user agent of browser | ||||
|         permission: Permission needed to access restricted routes | ||||
|     Returns: | ||||
|         A Session for this given Token | ||||
|  | @ -28,8 +60,16 @@ def validate_token(token, user_agent, permission): | |||
|     session = Session.query.filter_by(token=token).one_or_none() | ||||
|     if session: | ||||
|         logger.debug("token found, check if expired or invalid user agent differs") | ||||
| 
 | ||||
|         platform = request_headers.get("Sec-CH-UA-Platform", None) or __get_user_agent_platform( | ||||
|             request_headers.get("User-Agent", "") | ||||
|         ) | ||||
|         browser = request_headers.get("Sec-CH-UA", None) or __get_user_agent_browser( | ||||
|             request_headers.get("User-Agent", "") | ||||
|         ) | ||||
| 
 | ||||
|         if session.expires >= datetime.now(timezone.utc) and ( | ||||
|             session.browser == user_agent.browser and session.platform == user_agent.platform | ||||
|             session.browser == browser and session.platform == platform | ||||
|         ): | ||||
|             if not permission or session.user_.has_permission(permission): | ||||
|                 session.refresh() | ||||
|  | @ -44,12 +84,12 @@ def validate_token(token, user_agent, permission): | |||
|     raise Unauthorized | ||||
| 
 | ||||
| 
 | ||||
| def create(user, user_agent=None) -> Session: | ||||
| def create(user, request_headers=None) -> Session: | ||||
|     """Create a Session | ||||
| 
 | ||||
|     Args: | ||||
|         user: For which User is to create a Session | ||||
|         user_agent: User agent to identify session | ||||
|         request_headers: Headers to validate user agent of browser | ||||
| 
 | ||||
|     Returns: | ||||
|             Session: A created Token for User | ||||
|  | @ -60,8 +100,10 @@ def create(user, user_agent=None) -> Session: | |||
|         token=token_str, | ||||
|         user_=user, | ||||
|         lifetime=lifetime, | ||||
|         browser=user_agent.browser, | ||||
|         platform=user_agent.platform, | ||||
|         platform=request_headers.get("Sec-CH-UA-Platform", None) | ||||
|         or __get_user_agent_platform(request_headers.get("User-Agent", "")), | ||||
|         browser=request_headers.get("Sec-CH-UA", None) | ||||
|         or __get_user_agent_browser(request_headers.get("User-Agent", "")), | ||||
|     ) | ||||
|     session.refresh() | ||||
|     db.session.add(session) | ||||
|  |  | |||
|  | @ -1,21 +1,22 @@ | |||
| import secrets | ||||
| import re | ||||
| import secrets | ||||
| 
 | ||||
| from io import BytesIO | ||||
| from sqlalchemy import exc | ||||
| from flask import current_app | ||||
| from sqlalchemy_utils import merge_references | ||||
| from datetime import datetime, timedelta, timezone | ||||
| from flask.helpers import send_file | ||||
| from werkzeug.exceptions import NotFound, BadRequest, Forbidden | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.config import config | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist.models.notification import Notification | ||||
| from flaschengeist.utils.hook import Hook | ||||
| from flaschengeist.utils.datetime import from_iso_format | ||||
| from flaschengeist.utils.foreign_keys import merge_references | ||||
| from flaschengeist.models.user import User, Role, _PasswordReset | ||||
| from flaschengeist.controller import imageController, messageController, sessionController | ||||
| from .. import logger | ||||
| from ..config import config | ||||
| from ..database import db | ||||
| from ..models import Notification, User, Role | ||||
| from ..models.user import _PasswordReset | ||||
| from ..utils.hook import Hook | ||||
| from ..utils.datetime import from_iso_format | ||||
| from ..controller import imageController, messageController, pluginController, sessionController | ||||
| from ..plugins import AuthPlugin | ||||
| 
 | ||||
| 
 | ||||
| def __active_users(): | ||||
|  | @ -40,17 +41,33 @@ def _generate_password_reset(user): | |||
|     return reset | ||||
| 
 | ||||
| 
 | ||||
| def get_provider(userid: str): | ||||
|     return [p for p in pluginController.get_authentication_provider() if p.user_exists(userid)][0] | ||||
| 
 | ||||
| 
 | ||||
| @Hook | ||||
| def update_user(user: User, backend: AuthPlugin): | ||||
|     """Update user data from backend | ||||
| 
 | ||||
|     This is seperate function to provide a hook""" | ||||
|     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): | ||||
|     logger.info("login user {{ {} }}".format(username)) | ||||
| 
 | ||||
|     user = find_user(username) | ||||
|     if not user: | ||||
|         logger.debug("User not found in Database.") | ||||
|         user = User(userid=username) | ||||
|         db.session.add(user) | ||||
|     if current_app.config["FG_AUTH_BACKEND"].login(user, password): | ||||
|         update_user(user) | ||||
|         return user | ||||
|     for provider in pluginController.get_authentication_provider(): | ||||
|         uid = provider.login(username, password) | ||||
|         if isinstance(uid, str): | ||||
|             user = get_user(uid) | ||||
|             if not user: | ||||
|                 logger.debug("User not found in Database.") | ||||
|                 user = User(userid=uid) | ||||
|                 db.session.add(user) | ||||
|             update_user(user, provider) | ||||
|             return user | ||||
|     return None | ||||
| 
 | ||||
| 
 | ||||
|  | @ -83,14 +100,6 @@ def reset_password(token: str, password: str): | |||
|     db.session.commit() | ||||
| 
 | ||||
| 
 | ||||
| @Hook | ||||
| def update_user(user): | ||||
|     current_app.config["FG_AUTH_BACKEND"].update_user(user) | ||||
|     if not user.display_name: | ||||
|         user.display_name = "{} {}.".format(user.firstname, user.lastname[0]) | ||||
|     db.session.commit() | ||||
| 
 | ||||
| 
 | ||||
| def set_roles(user: User, roles: list[str], create=False): | ||||
|     """Set roles of user | ||||
| 
 | ||||
|  | @ -114,7 +123,7 @@ def set_roles(user: User, roles: list[str], create=False): | |||
|     user.roles_ = fetched | ||||
| 
 | ||||
| 
 | ||||
| def modify_user(user, password, new_password=None): | ||||
| def modify_user(user: User, password: str, new_password: str = None): | ||||
|     """Modify given user on the backend | ||||
| 
 | ||||
|     Args: | ||||
|  | @ -126,7 +135,8 @@ def modify_user(user, password, new_password=None): | |||
|         NotImplemented: If backend is not capable of this operation | ||||
|         BadRequest: Password is wrong or other logic issues | ||||
|     """ | ||||
|     current_app.config["FG_AUTH_BACKEND"].modify_user(user, password, new_password) | ||||
|     provider = get_provider(user.userid) | ||||
|     provider.modify_user(user, password, new_password) | ||||
| 
 | ||||
|     if new_password: | ||||
|         logger.debug(f"Password changed for user {user.userid}") | ||||
|  | @ -164,37 +174,13 @@ def get_user(uid, deleted=False) -> User: | |||
|     return user | ||||
| 
 | ||||
| 
 | ||||
| def find_user(uid_mail): | ||||
|     """Finding an user by userid or mail in database or auth-backend | ||||
|     Args: | ||||
|         uid_mail: userid and or mail to search for | ||||
|     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: | ||||
|             if not user.display_name: | ||||
|                 user.display_name = "{} {}.".format(user.firstname, user.lastname[0]) | ||||
|             db.session.add(user) | ||||
|             db.session.commit() | ||||
|     return user | ||||
| 
 | ||||
| 
 | ||||
| @Hook | ||||
| def delete_user(user: User): | ||||
|     """Delete given user""" | ||||
|     # First let the backend delete the user, as this might fail | ||||
|     current_app.config["FG_AUTH_BACKEND"].delete_user(user) | ||||
|     provider = get_provider(user.userid) | ||||
|     provider.delete_user(user) | ||||
| 
 | ||||
|     # Clear all easy relationships | ||||
|     user.avatar_ = None | ||||
|     user._attributes.clear() | ||||
|  | @ -246,10 +232,14 @@ def register(data, passwd=None): | |||
|     set_roles(user, roles) | ||||
| 
 | ||||
|     password = passwd if passwd else secrets.token_urlsafe(16) | ||||
|     current_app.config["FG_AUTH_BACKEND"].create_user(user, password) | ||||
|     try: | ||||
|         provider = [p for p in pluginController.get_authentication_provider() if p.can_register()][0] | ||||
|         provider.create_user(user, password) | ||||
|         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") | ||||
| 
 | ||||
|  | @ -264,7 +254,7 @@ def register(data, passwd=None): | |||
|         ) | ||||
|         messageController.send_message(messageController.Message(user, text, subject)) | ||||
| 
 | ||||
|     find_user(user.userid) | ||||
|     provider.update_user(user) | ||||
| 
 | ||||
|     return user | ||||
| 
 | ||||
|  | @ -273,19 +263,20 @@ def load_avatar(user: User): | |||
|     if user.avatar_ is not None: | ||||
|         return imageController.send_image(image=user.avatar_) | ||||
|     else: | ||||
|         avatar = current_app.config["FG_AUTH_BACKEND"].get_avatar(user) | ||||
|         provider = get_provider(user.userid) | ||||
|         avatar = provider.get_avatar(user) | ||||
|         if len(avatar.binary) > 0: | ||||
|             return send_file(BytesIO(avatar.binary), avatar.mimetype) | ||||
|     raise NotFound | ||||
| 
 | ||||
| 
 | ||||
| def save_avatar(user, file): | ||||
|     current_app.config["FG_AUTH_BACKEND"].set_avatar(user, file) | ||||
|     get_provider(user.userid).set_avatar(user, file) | ||||
|     db.session.commit() | ||||
| 
 | ||||
| 
 | ||||
| def delete_avatar(user): | ||||
|     current_app.config["FG_AUTH_BACKEND"].delete_avatar(user) | ||||
|     get_provider(user.userid).delete_avatar(user) | ||||
|     db.session.commit() | ||||
| 
 | ||||
| 
 | ||||
|  |  | |||
|  | @ -1,33 +0,0 @@ | |||
| from flask_sqlalchemy import SQLAlchemy | ||||
| from sqlalchemy import MetaData | ||||
| 
 | ||||
| # 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) | ||||
| 
 | ||||
| 
 | ||||
| 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 | ||||
|  | @ -0,0 +1,74 @@ | |||
| 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) | ||||
| 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 | ||||
|  | @ -0,0 +1,94 @@ | |||
| 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__ | ||||
| 
 | ||||
|         hint = typing.get_type_hints(self.__class__, globalns=module)[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 | ||||
| 
 | ||||
|     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) | ||||
|  | @ -1,87 +1,5 @@ | |||
| import sys | ||||
| import datetime | ||||
| 
 | ||||
| from sqlalchemy import BigInteger | ||||
| 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 | ||||
| 
 | ||||
|     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") | ||||
| 
 | ||||
| 
 | ||||
| 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 | ||||
| from .session import * | ||||
| from .user import * | ||||
| from .plugin import * | ||||
| from .notification import * | ||||
| from .image import * | ||||
|  |  | |||
|  | @ -1,19 +1,19 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from sqlalchemy import event | ||||
| from pathlib import Path | ||||
| 
 | ||||
| from . import ModelSerializeMixin, Serial | ||||
| from ..database import db | ||||
| from ..database.types import ModelSerializeMixin, Serial | ||||
| 
 | ||||
| 
 | ||||
| class Image(db.Model, ModelSerializeMixin): | ||||
|     __tablename__ = "image" | ||||
|     id: int = db.Column("id", Serial, primary_key=True) | ||||
|     filename_: str = db.Column(db.String(127), nullable=False) | ||||
|     mimetype_: str = db.Column(db.String(30), nullable=False) | ||||
|     thumbnail_: str = db.Column(db.String(127)) | ||||
|     path_: str = db.Column(db.String(127)) | ||||
|     id: int = db.Column(Serial, primary_key=True) | ||||
|     filename_: str = db.Column("filename", db.String(255), nullable=False) | ||||
|     mimetype_: str = db.Column("mimetype", db.String(127), nullable=False) | ||||
|     thumbnail_: str = db.Column("thumbnail", db.String(255)) | ||||
|     path_: str = db.Column("path", db.String(255)) | ||||
| 
 | ||||
|     def open(self): | ||||
|         return open(self.path_, "rb") | ||||
|  |  | |||
|  | @ -1,19 +1,26 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from datetime import datetime | ||||
| from typing import Any | ||||
| 
 | ||||
| from . import Serial, UtcDateTime, ModelSerializeMixin | ||||
| from ..database import db | ||||
| from .user import User | ||||
| from ..database.types import Serial, UtcDateTime, ModelSerializeMixin | ||||
| 
 | ||||
| 
 | ||||
| class Notification(db.Model, ModelSerializeMixin): | ||||
|     __tablename__ = "notification" | ||||
|     id: int = db.Column("id", Serial, primary_key=True) | ||||
|     plugin: str = db.Column(db.String(127), nullable=False) | ||||
|     text: str = db.Column(db.Text) | ||||
|     data: Any = db.Column(db.PickleType(protocol=4)) | ||||
|     time: datetime = db.Column(UtcDateTime, nullable=False, default=UtcDateTime.current_utc) | ||||
| 
 | ||||
|     user_id_: int = db.Column("user_id", Serial, db.ForeignKey("user.id"), nullable=False) | ||||
|     user_id_: int = db.Column("user", 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") | ||||
|     plugin_: Plugin = db.relationship( | ||||
|         "Plugin", backref=db.backref("notifications_", cascade="all, delete, delete-orphan") | ||||
|     ) | ||||
| 
 | ||||
|     @property | ||||
|     def plugin(self): | ||||
|         return self.plugin_.name | ||||
|  |  | |||
|  | @ -0,0 +1,72 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from typing import Any | ||||
| from sqlalchemy.orm.collections import attribute_mapped_collection | ||||
| 
 | ||||
| from ..database import db | ||||
| from ..database.types import Serial | ||||
| 
 | ||||
| 
 | ||||
| class PluginSetting(db.Model): | ||||
|     __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): | ||||
|     __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 = 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="select", | ||||
|     ) | ||||
| 
 | ||||
|     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,12 +1,11 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from datetime import datetime, timedelta, timezone | ||||
| 
 | ||||
| from . import ModelSerializeMixin, UtcDateTime, Serial | ||||
| from .user import User | ||||
| from flaschengeist.database import db | ||||
| from secrets import compare_digest | ||||
| from flaschengeist import logger | ||||
| 
 | ||||
| from .. import logger | ||||
| from ..database import db | ||||
| from ..database.types import ModelSerializeMixin, UtcDateTime, Serial | ||||
| 
 | ||||
| 
 | ||||
| class Session(db.Model, ModelSerializeMixin): | ||||
|  | @ -22,8 +21,8 @@ class Session(db.Model, ModelSerializeMixin): | |||
|     expires: datetime = db.Column(UtcDateTime) | ||||
|     token: str = db.Column(db.String(32), unique=True) | ||||
|     lifetime: int = db.Column(db.Integer) | ||||
|     browser: str = db.Column(db.String(30)) | ||||
|     platform: str = db.Column(db.String(30)) | ||||
|     browser: str = db.Column(db.String(127)) | ||||
|     platform: str = db.Column(db.String(64)) | ||||
|     userid: str = "" | ||||
| 
 | ||||
|     _id = db.Column("id", Serial, primary_key=True) | ||||
|  |  | |||
|  | @ -1,13 +0,0 @@ | |||
| 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,14 +1,11 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from flask import url_for | ||||
| from typing import Optional | ||||
| from datetime import date, datetime | ||||
| from sqlalchemy.orm.collections import attribute_mapped_collection | ||||
| 
 | ||||
| from ..database import db | ||||
| from . import ModelSerializeMixin, UtcDateTime, Serial | ||||
| from .image import Image | ||||
| 
 | ||||
| from ..database.types import ModelSerializeMixin, UtcDateTime, Serial | ||||
| 
 | ||||
| association_table = db.Table( | ||||
|     "user_x_role", | ||||
|  | @ -27,7 +24,9 @@ class Permission(db.Model, ModelSerializeMixin): | |||
|     __tablename__ = "permission" | ||||
|     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="select", back_populates="permissions", enable_typechecks=False) | ||||
| 
 | ||||
| 
 | ||||
| class Role(db.Model, ModelSerializeMixin): | ||||
|  | @ -66,9 +65,7 @@ class User(db.Model, ModelSerializeMixin): | |||
|     # Protected stuff for backend use only | ||||
|     id_ = db.Column("id", Serial, primary_key=True) | ||||
|     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_: list[Session] = db.relationship("Session", back_populates="user_", cascade="all, delete, delete-orphan") | ||||
|     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") | ||||
| 
 | ||||
|  |  | |||
|  | @ -3,124 +3,130 @@ | |||
| .. include:: docs/plugin_development.md | ||||
| 
 | ||||
| """ | ||||
| from importlib_metadata import Distribution, EntryPoint | ||||
| from werkzeug.datastructures import FileStorage | ||||
| from werkzeug.exceptions import MethodNotAllowed, NotFound | ||||
| 
 | ||||
| from flaschengeist.models.user import _Avatar, User | ||||
| from typing import Union | ||||
| 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 | ||||
| 
 | ||||
| __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") | ||||
| """Hook decorator for when all plugins are installed | ||||
|     Possible use case would be to populate the database with some presets. | ||||
| plugins_installed.__doc__ = """Hook decorator for when all plugins are installed | ||||
| 
 | ||||
|     Args: | ||||
|         hook_result: void (kwargs) | ||||
| Possible use case would be to populate the database with some presets. | ||||
| """ | ||||
| 
 | ||||
| plugins_loaded = HookAfter("plugins.loaded") | ||||
| """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 | ||||
| plugins_loaded.__doc__ = """Hook decorator for when all plugins are loaded | ||||
| 
 | ||||
|     Args: | ||||
|         app: Current flask app instance (args) | ||||
|         hook_result: void (kwargs) | ||||
| 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") | ||||
| """Hook decorator for when roles are modified | ||||
| Args: | ||||
|     role: Role object to modify | ||||
|     new_name: New name if the name was changed (None if delete) | ||||
| before_role_updated.__doc__ = """Hook decorator for when roles are modified | ||||
|      | ||||
| Passed args: | ||||
|     - *role:* `flaschengeist.models.user.Role` to modify | ||||
|     - *new_name:* New name if the name was changed (*None* if delete) | ||||
| """ | ||||
| 
 | ||||
| after_role_updated = HookAfter("update_role") | ||||
| """Hook decorator for when roles are modified | ||||
| Args: | ||||
|     role: Role object containing the modified role | ||||
|     new_name: New name if the name was changed (None if deleted) | ||||
| after_role_updated.__doc__ = """Hook decorator for when roles are modified | ||||
| 
 | ||||
| Passed args: | ||||
|     - *role:* modified `flaschengeist.models.user.Role` | ||||
|     - *new_name:* New name if the name was changed (*None* if deleted) | ||||
| """ | ||||
| 
 | ||||
| before_update_user = HookBefore("update_user") | ||||
| """Hook decorator, when ever an user update is done, this is called before. | ||||
| Args: | ||||
|     user: User object | ||||
| before_update_user.__doc__ = """Hook decorator, when ever an user update is done, this is called before. | ||||
| 
 | ||||
| Passed args: | ||||
|     - *user:* `flaschengeist.models.user.User` object | ||||
| """ | ||||
| 
 | ||||
| before_delete_user = HookBefore("delete_user") | ||||
| """Hook decorator,this is called before an user gets deleted. | ||||
| Args: | ||||
|     user: User object | ||||
| before_delete_user.__doc__ = """Hook decorator,this is called before an user gets deleted. | ||||
| 
 | ||||
| Passed args: | ||||
|     - *user:* `flaschengeist.models.user.User` object | ||||
| """ | ||||
| 
 | ||||
| 
 | ||||
| class Plugin: | ||||
| class Plugin(BasePlugin): | ||||
|     """Base class for all Plugins | ||||
| 
 | ||||
|     If your class uses custom models add a static property called ``models``. | ||||
|     All plugins must derived from this class. | ||||
| 
 | ||||
|     Optional: | ||||
|         - *blueprint*: `flask.Blueprint` providing your routes | ||||
|         - *permissions*: List of your custom permissions | ||||
|         - *models*: Your models, used for API export | ||||
|     """ | ||||
| 
 | ||||
|     name: str | ||||
|     """Name of the plugin, loaded from EntryPoint""" | ||||
| 
 | ||||
|     version: str | ||||
|     """Version of the plugin, loaded from Distribution""" | ||||
| 
 | ||||
|     dist: Distribution | ||||
|     """Distribution of this plugin""" | ||||
| 
 | ||||
|     blueprint = None | ||||
|     """Optional `flask.blueprint` if the plugin uses custom routes""" | ||||
| 
 | ||||
|     permissions: list[str] = [] | ||||
|     """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*. | ||||
|     """ | ||||
| 
 | ||||
|     models = None | ||||
|     """Optional module containing the SQLAlchemy models used by the plugin""" | ||||
| 
 | ||||
|     migrations_path = None | ||||
|     """Optional location of the path to migration files, required if custome db tables are used""" | ||||
|     @property | ||||
|     def version(self) -> str: | ||||
|         """Version of the plugin, loaded from Distribution""" | ||||
|         return self.dist.version | ||||
| 
 | ||||
|     def __init__(self, entry_point: EntryPoint, config=None): | ||||
|         """Constructor called by create_app | ||||
|         Args: | ||||
|             config: Dict configuration containing the plugin section | ||||
|         """ | ||||
|         self.version = entry_point.dist.version | ||||
|         self.name = entry_point.name | ||||
|         self.dist = entry_point.dist | ||||
|     @property | ||||
|     def dist(self): | ||||
|         """Distribution of this plugin""" | ||||
|         return self.entry_point.dist | ||||
| 
 | ||||
|     @property | ||||
|     def entry_point(self): | ||||
|         ep = 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): | ||||
|         """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 | ||||
| 
 | ||||
|     def get_setting(self, name: str, **kwargs): | ||||
|         """Get plugin setting from database | ||||
|     def uninstall(self): | ||||
|         """Uninstall routine | ||||
| 
 | ||||
|         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 | ||||
|         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. | ||||
|         """ | ||||
|         from ..controller import pluginController | ||||
| 
 | ||||
|         return pluginController.get_setting(self.name, name, **kwargs) | ||||
| 
 | ||||
|     def set_setting(self, name: str, value): | ||||
|         """Save setting in database | ||||
| 
 | ||||
|         Args: | ||||
|             name: String identifying the setting | ||||
|             value: Value to be stored | ||||
|         """ | ||||
|         from ..controller import pluginController | ||||
| 
 | ||||
|         return pluginController.set_setting(self.name, name, value) | ||||
|         pass | ||||
| 
 | ||||
|     def notify(self, user, text: str, data=None): | ||||
|         """Create a new notification for an user | ||||
|  | @ -146,35 +152,53 @@ class Plugin: | |||
|         """ | ||||
|         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.name for x in self.permissions) | ||||
|         all_perm = set(permissions) | ||||
| 
 | ||||
|         new_perms = all_perm - cur_perm | ||||
|         self.permissions = list(filter(lambda x: x.name in permissions, self.permissions)) + [ | ||||
|             Permission(name=x, plugin_=self) for x in new_perms | ||||
|         ] | ||||
| 
 | ||||
| 
 | ||||
| class AuthPlugin(Plugin): | ||||
|     def login(self, user, pw): | ||||
|     """Base class for all authentification plugins | ||||
| 
 | ||||
|     See also `Plugin` | ||||
|     """ | ||||
| 
 | ||||
|     def login(self, login_name, password) -> Union[bool, str]: | ||||
|         """Login routine, MUST BE IMPLEMENTED! | ||||
| 
 | ||||
|         Args: | ||||
|             user: User class containing at least the uid | ||||
|             pw: given password | ||||
|             login_name: The name the user entered | ||||
|             password: The password the user used to log in | ||||
|         Returns: | ||||
|             Must return False if not found or invalid credentials, True if success | ||||
|             Must return False if not found or invalid credentials, otherwise the UID is returned | ||||
|         """ | ||||
|         raise NotImplemented | ||||
| 
 | ||||
|     def update_user(self, user): | ||||
|     def update_user(self, user: "User"): | ||||
|         """If backend is using external data, then update this user instance with external data | ||||
|         Args: | ||||
|             user: User object | ||||
|         """ | ||||
|         pass | ||||
| 
 | ||||
|     def find_user(self, userid, mail=None): | ||||
|         """Find an user by userid or mail | ||||
|     def user_exists(self, userid) -> bool: | ||||
|         """Check if user exists on this backend | ||||
|         Args: | ||||
|             userid: Userid to search | ||||
|             mail: If set, mail to search | ||||
|         Returns: | ||||
|             None or User | ||||
|             True or False | ||||
|         """ | ||||
|         return None | ||||
|         raise NotImplemented | ||||
| 
 | ||||
|     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. | ||||
|  | @ -185,11 +209,14 @@ class AuthPlugin(Plugin): | |||
|             password: Password (some backends need the current password for changes) if None force edit (admin) | ||||
|             new_password: If set a password change is requested | ||||
|         Raises: | ||||
|             NotImplemented: If backend does not support this feature (or no password change) | ||||
|             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) | ||||
|         """ | ||||
|         raise NotImplemented | ||||
|         pass | ||||
| 
 | ||||
|     def can_register(self): | ||||
|         """Check if this backend allows to register new users""" | ||||
|         return False | ||||
| 
 | ||||
|     def create_user(self, user, password): | ||||
|         """If backend is using (writeable) external data, then create a new user on the external database. | ||||
|  | @ -199,7 +226,7 @@ class AuthPlugin(Plugin): | |||
|             password: string | ||||
| 
 | ||||
|         """ | ||||
|         raise MethodNotAllowed | ||||
|         raise NotImplementedError | ||||
| 
 | ||||
|     def delete_user(self, user): | ||||
|         """If backend is using (writeable) external data, then delete the user from external database. | ||||
|  | @ -208,9 +235,9 @@ class AuthPlugin(Plugin): | |||
|             user: User object | ||||
| 
 | ||||
|         """ | ||||
|         raise MethodNotAllowed | ||||
|         raise NotImplementedError | ||||
| 
 | ||||
|     def get_avatar(self, user: User) -> _Avatar: | ||||
|     def get_avatar(self, user) -> _Avatar: | ||||
|         """Retrieve avatar for given user (if supported by auth backend) | ||||
| 
 | ||||
|         Default behavior is to use native Image objects, | ||||
|  | @ -224,14 +251,14 @@ class AuthPlugin(Plugin): | |||
|         """ | ||||
|         raise NotFound | ||||
| 
 | ||||
|     def set_avatar(self, user: User, file: FileStorage): | ||||
|     def set_avatar(self, user, file: FileStorage): | ||||
|         """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: | ||||
|             user: User to set the avatar for | ||||
|             file: FileStorage object uploaded by the user | ||||
|             file: `werkzeug.datastructures.FileStorage` uploaded by the user | ||||
|         Raises: | ||||
|             MethodNotAllowed: If not supported by Backend | ||||
|             Any valid HTTP exception | ||||
|  | @ -242,7 +269,7 @@ class AuthPlugin(Plugin): | |||
| 
 | ||||
|         user.avatar_ = imageController.upload_image(file) | ||||
| 
 | ||||
|     def delete_avatar(self, user: User): | ||||
|     def delete_avatar(self, user): | ||||
|         """Delete the avatar for given user (if supported by auth backend) | ||||
| 
 | ||||
|         Default behavior is to use the imageController and native Image objects. | ||||
|  |  | |||
|  | @ -13,8 +13,7 @@ from flaschengeist.controller import sessionController, userController | |||
| 
 | ||||
| 
 | ||||
| class AuthRoutePlugin(Plugin): | ||||
|     name = "auth" | ||||
|     blueprint = Blueprint(name, __name__) | ||||
|     blueprint = Blueprint("auth", __name__) | ||||
| 
 | ||||
| 
 | ||||
| @AuthRoutePlugin.blueprint.route("/auth", methods=["POST"]) | ||||
|  | @ -41,7 +40,7 @@ def login(): | |||
|     user = userController.login_user(userid, password) | ||||
|     if not user: | ||||
|         raise Unauthorized | ||||
|     session = sessionController.create(user, user_agent=request.user_agent) | ||||
|     session = sessionController.create(user, request_headers=request.headers) | ||||
|     logger.debug(f"token is {session.token}") | ||||
|     logger.info(f"User {userid} logged in.") | ||||
| 
 | ||||
|  |  | |||
|  | @ -11,69 +11,70 @@ from werkzeug.exceptions import BadRequest, InternalServerError, NotFound | |||
| from werkzeug.datastructures import FileStorage | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.config import config | ||||
| from flaschengeist.controller import userController | ||||
| from flaschengeist.models.user import User, Role, _Avatar | ||||
| from flaschengeist.models import User, Role | ||||
| from flaschengeist.models.user import _Avatar | ||||
| from flaschengeist.plugins import AuthPlugin, before_role_updated | ||||
| 
 | ||||
| 
 | ||||
| class AuthLDAP(AuthPlugin): | ||||
|     def __init__(self, entry_point, config): | ||||
|         super().__init__(entry_point) | ||||
|     def load(self): | ||||
|         self.config = config.get("auth_ldap", None) | ||||
|         if self.config is None: | ||||
|             logger.error("auth_ldap was not configured in flaschengeist.toml", exc_info=True) | ||||
|             raise InternalServerError | ||||
|         app.config.update( | ||||
|             LDAP_SERVER=config.get("host", "localhost"), | ||||
|             LDAP_PORT=config.get("port", 389), | ||||
|             LDAP_BINDDN=config.get("bind_dn", None), | ||||
|             LDAP_SECRET=config.get("secret", None), | ||||
|             LDAP_USE_SSL=config.get("use_ssl", False), | ||||
|             LDAP_SERVER=self.config.get("host", "localhost"), | ||||
|             LDAP_PORT=self.config.get("port", 389), | ||||
|             LDAP_BINDDN=self.config.get("bind_dn", None), | ||||
|             LDAP_SECRET=self.config.get("secret", None), | ||||
|             LDAP_USE_SSL=self.config.get("use_ssl", False), | ||||
|             # That's not TLS, its dirty StartTLS on unencrypted LDAP | ||||
|             LDAP_USE_TLS=False, | ||||
|             LDAP_TLS_VERSION=ssl.PROTOCOL_TLS, | ||||
|             FORCE_ATTRIBUTE_VALUE_AS_LIST=True, | ||||
|         ) | ||||
|         if "ca_cert" in config: | ||||
|             app.config["LDAP_CA_CERTS_FILE"] = config["ca_cert"] | ||||
|             app.config["LDAP_CA_CERTS_FILE"] = self.config["ca_cert"] | ||||
|         else: | ||||
|             # Default is CERT_REQUIRED | ||||
|             app.config["LDAP_REQUIRE_CERT"] = ssl.CERT_OPTIONAL | ||||
|         self.ldap = LDAPConn(app) | ||||
|         self.base_dn = config["base_dn"] | ||||
|         self.search_dn = config.get("search_dn", "ou=people,{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 = config.get("password_hash", "SSHA").upper() | ||||
|         self.object_classes = config.get("object_classes", ["inetOrgPerson"]) | ||||
|         self.user_attributes: dict = config.get("user_attributes", {}) | ||||
|         self.dn_template = config.get("dn_template") | ||||
|         self.base_dn = self.config["base_dn"] | ||||
|         self.search_dn = self.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.password_hash = self.config.get("password_hash", "SSHA").upper() | ||||
|         self.object_classes = self.config.get("object_classes", ["inetOrgPerson"]) | ||||
|         self.user_attributes: dict = self.config.get("user_attributes", {}) | ||||
|         self.dn_template = self.config.get("dn_template") | ||||
| 
 | ||||
|         # TODO: might not be set if modify is called | ||||
|         self.root_dn = config.get("root_dn", None) | ||||
|         self.root_secret = config.get("root_secret", None) | ||||
|         self.root_dn = self.config.get("root_dn", None) | ||||
|         self.root_secret = self.config.get("root_secret", None) | ||||
| 
 | ||||
|         @before_role_updated | ||||
|         def _role_updated(role, new_name): | ||||
|             logger.debug(f"LDAP: before_role_updated called with ({role}, {new_name})") | ||||
|             self.__modify_role(role, new_name) | ||||
| 
 | ||||
|     def login(self, user, password): | ||||
|         if not user: | ||||
|     def login(self, login_name, password): | ||||
|         if not login_name: | ||||
|             return False | ||||
|         return self.ldap.authenticate(user.userid, password, "uid", self.base_dn) | ||||
|         return login_name if self.ldap.authenticate(login_name, password, "uid", self.base_dn) else False | ||||
| 
 | ||||
|     def find_user(self, userid, mail=None): | ||||
|         attr = self.__find(userid, mail) | ||||
|         if attr is not None: | ||||
|             user = User(userid=attr["uid"][0]) | ||||
|             self.__update(user, attr) | ||||
|             return user | ||||
|     def user_exists(self, userid) -> bool: | ||||
|         attr = self.__find(userid, None) | ||||
|         return attr is not None | ||||
| 
 | ||||
|     def update_user(self, user): | ||||
|         attr = self.__find(user.userid) | ||||
|         self.__update(user, attr) | ||||
| 
 | ||||
|     def create_user(self, user, password): | ||||
|         if self.root_dn is None: | ||||
|             logger.error("root_dn missing in ldap config!") | ||||
|             raise InternalServerError | ||||
|     def can_register(self): | ||||
|         return self.root_dn is not None | ||||
| 
 | ||||
|     def create_user(self, user, password): | ||||
|         try: | ||||
|             ldap_conn = self.ldap.connect(self.root_dn, self.root_secret) | ||||
|             attributes = self.user_attributes.copy() | ||||
|  | @ -305,3 +306,5 @@ class AuthLDAP(AuthPlugin): | |||
| 
 | ||||
|         except (LDAPPasswordIsMandatoryError, LDAPBindError): | ||||
|             raise BadRequest | ||||
|         except IndexError as e: | ||||
|             logger.error("Roles in LDAP", exc_info=True) | ||||
|  |  | |||
|  | @ -1,6 +1,7 @@ | |||
| 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) | ||||
|  | @ -13,8 +14,10 @@ def ldap(ctx, sync): | |||
|         from flaschengeist.controller import userController | ||||
|         from flaschengeist.plugins.auth_ldap import AuthLDAP | ||||
|         from ldap3 import SUBTREE | ||||
|         from flaschengeist.models import User | ||||
|         from flaschengeist.database import db | ||||
| 
 | ||||
|         auth_ldap: AuthLDAP = current_app.config.get("FG_AUTH_BACKEND") | ||||
|         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 | ||||
|  | @ -24,4 +27,9 @@ def ldap(ctx, sync): | |||
|         ldap_users_response = conn.response | ||||
|         for ldap_user in ldap_users_response: | ||||
|             uid = ldap_user["attributes"]["uid"][0] | ||||
|             userController.find_user(uid) | ||||
|             try: | ||||
|                 user = userController.get_user(uid) | ||||
|             except NotFound: | ||||
|                 user = User(userid=uid) | ||||
|                 db.session.add(user) | ||||
|             userController.update_user(user, auth_ldap) | ||||
|  |  | |||
|  | @ -7,42 +7,25 @@ import os | |||
| import hashlib | ||||
| import binascii | ||||
| from werkzeug.exceptions import BadRequest | ||||
| from flaschengeist.plugins import AuthPlugin, plugins_installed | ||||
| from flaschengeist.models.user import User, Role, Permission | ||||
| from flaschengeist.plugins import AuthPlugin | ||||
| from flaschengeist.models import User, Role, Permission | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist import logger | ||||
| 
 | ||||
| 
 | ||||
| class AuthPlain(AuthPlugin): | ||||
|     def install(self): | ||||
|         plugins_installed(self.post_install) | ||||
|     def can_register(self): | ||||
|         return True | ||||
| 
 | ||||
|     def post_install(self, **kwargs): | ||||
|         if User.query.filter(User.deleted == False).count() == 0: | ||||
|             logger.info("Installing admin user") | ||||
|             role = Role.query.filter(Role.name == "Superuser").first() | ||||
|             if role is None: | ||||
|                 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, user: User, password: str): | ||||
|         if user.has_attribute("password"): | ||||
|             return AuthPlain._verify_password(user.get_attribute("password"), password) | ||||
|     def login(self, login_name, password): | ||||
|         users: list[User] = ( | ||||
|             User.query.filter((User.userid == login_name) | (User.mail == login_name)) | ||||
|             .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 | ||||
| 
 | ||||
|     def modify_user(self, user, password, new_password=None): | ||||
|  | @ -51,6 +34,12 @@ class AuthPlain(AuthPlugin): | |||
|         if 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): | ||||
|         if not user.userid: | ||||
|             raise BadRequest("userid is missing for new user") | ||||
|  | @ -68,7 +57,7 @@ class AuthPlain(AuthPlugin): | |||
|         return (salt + pass_hash).decode("ascii") | ||||
| 
 | ||||
|     @staticmethod | ||||
|     def _verify_password(stored_password, provided_password): | ||||
|     def _verify_password(stored_password: str, provided_password: str): | ||||
|         salt = stored_password[:64] | ||||
|         stored_password = stored_password[64:] | ||||
|         pass_hash = hashlib.pbkdf2_hmac("sha3-512", provided_password.encode("utf-8"), salt.encode("ascii"), 100000) | ||||
|  |  | |||
|  | @ -3,11 +3,11 @@ | |||
| Extends users plugin with balance functions | ||||
| """ | ||||
| 
 | ||||
| from flask import Blueprint, current_app | ||||
| from werkzeug.local import LocalProxy | ||||
| from flask import current_app | ||||
| from werkzeug.exceptions import NotFound | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.config import config | ||||
| from flaschengeist.plugins import Plugin, plugins_loaded, before_update_user | ||||
| from flaschengeist.plugins.scheduler import add_scheduled | ||||
| 
 | ||||
|  | @ -56,14 +56,13 @@ def service_debit(): | |||
| 
 | ||||
| 
 | ||||
| class BalancePlugin(Plugin): | ||||
|     permissions = permissions.permissions | ||||
|     plugin: "BalancePlugin" = LocalProxy(lambda: current_app.config["FG_PLUGINS"][BalancePlugin.name]) | ||||
|     models = models | ||||
| 
 | ||||
|     def __init__(self, entry_point, config): | ||||
|         super(BalancePlugin, self).__init__(entry_point, config) | ||||
|         from .routes import blueprint | ||||
|     def install(self): | ||||
|         self.install_permissions(permissions.permissions) | ||||
| 
 | ||||
|     def load(self): | ||||
|         from .routes import blueprint | ||||
|         self.blueprint = blueprint | ||||
| 
 | ||||
|         @plugins_loaded | ||||
|  |  | |||
|  | @ -0,0 +1,47 @@ | |||
| """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,5 +1,3 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| 
 | ||||
| from datetime import datetime | ||||
| from typing import Optional | ||||
| from sqlalchemy.ext.hybrid import hybrid_property | ||||
|  |  | |||
|  | @ -134,7 +134,7 @@ def get_balance(userid, current_session: Session): | |||
| 
 | ||||
|     Route: ``/users/<userid>/balance`` | Method: ``GET`` | ||||
| 
 | ||||
|     GET-parameters: ```{from?: string, to?: string}``` | ||||
|     GET-parameters: ``{from?: string, to?: string}`` | ||||
| 
 | ||||
|     Args: | ||||
|         userid: Userid of user to get balance from | ||||
|  | @ -173,7 +173,7 @@ def get_transactions(userid, current_session: Session): | |||
| 
 | ||||
|     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: | ||||
|         userid: Userid of user to get transactions from | ||||
|  |  | |||
|  | @ -3,17 +3,16 @@ from email.mime.text import MIMEText | |||
| from email.mime.multipart import MIMEMultipart | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.models.user import User | ||||
| from flaschengeist.models import User | ||||
| from flaschengeist.plugins import Plugin | ||||
| from flaschengeist.utils.hook import HookAfter | ||||
| from flaschengeist.controller import userController | ||||
| from flaschengeist.controller.messageController import Message | ||||
| 
 | ||||
| from . import Plugin | ||||
| 
 | ||||
| 
 | ||||
| class MailMessagePlugin(Plugin): | ||||
|     def __init__(self, entry_point, config): | ||||
|         super().__init__(entry_point) | ||||
|         super().__init__(entry_point, config) | ||||
|         self.server = config["SERVER"] | ||||
|         self.port = config["PORT"] | ||||
|         self.user = config["USER"] | ||||
|  |  | |||
|  | @ -1,37 +1,32 @@ | |||
| """Pricelist plugin""" | ||||
| 
 | ||||
| from flask import Blueprint, jsonify, request, current_app | ||||
| from werkzeug.local import LocalProxy | ||||
| from flask import Blueprint, jsonify, request | ||||
| from werkzeug.exceptions import BadRequest, Forbidden, NotFound, Unauthorized | ||||
| 
 | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.controller import userController | ||||
| from flaschengeist.controller.imageController import send_image, send_thumbnail | ||||
| from flaschengeist.plugins import Plugin | ||||
| from flaschengeist.utils.decorators import login_required, extract_session, headers | ||||
| from flaschengeist.utils.decorators import login_required, extract_session | ||||
| from flaschengeist.utils.HTTP import no_content | ||||
| 
 | ||||
| from . import models | ||||
| from . import pricelist_controller, permissions | ||||
| 
 | ||||
| 
 | ||||
| blueprint = Blueprint("pricelist", __name__, url_prefix="/pricelist") | ||||
| 
 | ||||
| 
 | ||||
| class PriceListPlugin(Plugin): | ||||
|     permissions = permissions.permissions | ||||
|     plugin = LocalProxy(lambda: current_app.config["FG_PLUGINS"][PriceListPlugin.name]) | ||||
|     models = models | ||||
|     blueprint = Blueprint("pricelist", __name__, url_prefix="/pricelist") | ||||
| 
 | ||||
|     def __init__(self, entry_point, config=None): | ||||
|         super().__init__(entry_point, config) | ||||
|         self.blueprint = blueprint | ||||
|     def install(self): | ||||
|         self.install_permissions(permissions.permissions) | ||||
| 
 | ||||
|     def load(self): | ||||
|         config = {"discount": 0} | ||||
|         config.update(config) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drink-types", methods=["GET"]) | ||||
| @blueprint.route("/drink-types/<int:identifier>", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drink-types", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drink-types/<int:identifier>", methods=["GET"]) | ||||
| def get_drink_types(identifier=None): | ||||
|     """Get DrinkType(s) | ||||
| 
 | ||||
|  | @ -51,7 +46,7 @@ def get_drink_types(identifier=None): | |||
|     return jsonify(result) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drink-types", methods=["POST"]) | ||||
| @PriceListPlugin.blueprint.route("/drink-types", methods=["POST"]) | ||||
| @login_required(permission=permissions.CREATE_TYPE) | ||||
| def new_drink_type(current_session): | ||||
|     """Create new DrinkType | ||||
|  | @ -73,7 +68,7 @@ def new_drink_type(current_session): | |||
|     return jsonify(drink_type) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drink-types/<int:identifier>", methods=["PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/drink-types/<int:identifier>", methods=["PUT"]) | ||||
| @login_required(permission=permissions.EDIT_TYPE) | ||||
| def update_drink_type(identifier, current_session): | ||||
|     """Modify DrinkType | ||||
|  | @ -96,7 +91,7 @@ def update_drink_type(identifier, current_session): | |||
|     return jsonify(drink_type) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drink-types/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/drink-types/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_TYPE) | ||||
| def delete_drink_type(identifier, current_session): | ||||
|     """Delete DrinkType | ||||
|  | @ -114,8 +109,8 @@ def delete_drink_type(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/tags", methods=["GET"]) | ||||
| @blueprint.route("/tags/<int:identifier>", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/tags", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/tags/<int:identifier>", methods=["GET"]) | ||||
| def get_tags(identifier=None): | ||||
|     """Get Tag(s) | ||||
| 
 | ||||
|  | @ -135,7 +130,7 @@ def get_tags(identifier=None): | |||
|     return jsonify(result) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/tags", methods=["POST"]) | ||||
| @PriceListPlugin.blueprint.route("/tags", methods=["POST"]) | ||||
| @login_required(permission=permissions.CREATE_TAG) | ||||
| def new_tag(current_session): | ||||
|     """Create Tag | ||||
|  | @ -155,7 +150,7 @@ def new_tag(current_session): | |||
|     return jsonify(drink_type) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/tags/<int:identifier>", methods=["PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/tags/<int:identifier>", methods=["PUT"]) | ||||
| @login_required(permission=permissions.EDIT_TAG) | ||||
| def update_tag(identifier, current_session): | ||||
|     """Modify Tag | ||||
|  | @ -176,7 +171,7 @@ def update_tag(identifier, current_session): | |||
|     return jsonify(tag) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/tags/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/tags/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_TAG) | ||||
| def delete_tag(identifier, current_session): | ||||
|     """Delete Tag | ||||
|  | @ -194,8 +189,8 @@ def delete_tag(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks", methods=["GET"]) | ||||
| @blueprint.route("/drinks/<int:identifier>", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/<int:identifier>", methods=["GET"]) | ||||
| def get_drinks(identifier=None): | ||||
|     """Get Drink(s) | ||||
| 
 | ||||
|  | @ -251,7 +246,7 @@ def get_drinks(identifier=None): | |||
|     return jsonify({"drinks": drinks, "count": count}) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/list", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/list", methods=["GET"]) | ||||
| def get_pricelist(): | ||||
|     """Get Priclist | ||||
|     Route: ``/pricelist/list`` | Method: ``GET`` | ||||
|  | @ -300,7 +295,7 @@ def get_pricelist(): | |||
|     return jsonify({"pricelist": pricelist, "count": count}) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks/search/<string:name>", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/search/<string:name>", methods=["GET"]) | ||||
| def search_drinks(name): | ||||
|     """Search Drink | ||||
| 
 | ||||
|  | @ -321,7 +316,7 @@ def search_drinks(name): | |||
|     return jsonify(pricelist_controller.get_drinks(name, public=public)) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks", methods=["POST"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks", methods=["POST"]) | ||||
| @login_required(permission=permissions.CREATE) | ||||
| def create_drink(current_session): | ||||
|     """Create Drink | ||||
|  | @ -373,7 +368,7 @@ def create_drink(current_session): | |||
|     return jsonify(pricelist_controller.set_drink(data)) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks/<int:identifier>", methods=["PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/<int:identifier>", methods=["PUT"]) | ||||
| @login_required(permission=permissions.EDIT) | ||||
| def update_drink(identifier, current_session): | ||||
|     """Modify Drink | ||||
|  | @ -427,7 +422,7 @@ def update_drink(identifier, current_session): | |||
|     return jsonify(pricelist_controller.update_drink(identifier, data)) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE) | ||||
| def delete_drink(identifier, current_session): | ||||
|     """Delete Drink | ||||
|  | @ -445,7 +440,7 @@ def delete_drink(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/prices/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/prices/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_PRICE) | ||||
| def delete_price(identifier, current_session): | ||||
|     """Delete Price | ||||
|  | @ -463,7 +458,7 @@ def delete_price(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/volumes/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/volumes/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_VOLUME) | ||||
| def delete_volume(identifier, current_session): | ||||
|     """Delete DrinkPriceVolume | ||||
|  | @ -481,7 +476,7 @@ def delete_volume(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/ingredients/extraIngredients", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/ingredients/extraIngredients", methods=["GET"]) | ||||
| @login_required() | ||||
| def get_extra_ingredients(current_session): | ||||
|     """Get ExtraIngredients | ||||
|  | @ -497,7 +492,7 @@ def get_extra_ingredients(current_session): | |||
|     return jsonify(pricelist_controller.get_extra_ingredients()) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/ingredients/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/ingredients/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_INGREDIENTS_DRINK) | ||||
| def delete_ingredient(identifier, current_session): | ||||
|     """Delete Ingredient | ||||
|  | @ -515,7 +510,7 @@ def delete_ingredient(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/ingredients/extraIngredients", methods=["POST"]) | ||||
| @PriceListPlugin.blueprint.route("/ingredients/extraIngredients", methods=["POST"]) | ||||
| @login_required(permission=permissions.EDIT_INGREDIENTS) | ||||
| def set_extra_ingredient(current_session): | ||||
|     """Create ExtraIngredient | ||||
|  | @ -534,7 +529,7 @@ def set_extra_ingredient(current_session): | |||
|     return jsonify(pricelist_controller.set_extra_ingredient(data)) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/ingredients/extraIngredients/<int:identifier>", methods=["PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/ingredients/extraIngredients/<int:identifier>", methods=["PUT"]) | ||||
| @login_required(permission=permissions.EDIT_INGREDIENTS) | ||||
| def update_extra_ingredient(identifier, current_session): | ||||
|     """Modify ExtraIngredient | ||||
|  | @ -554,7 +549,7 @@ def update_extra_ingredient(identifier, current_session): | |||
|     return jsonify(pricelist_controller.update_extra_ingredient(identifier, data)) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/ingredients/extraIngredients/<int:identifier>", methods=["DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/ingredients/extraIngredients/<int:identifier>", methods=["DELETE"]) | ||||
| @login_required(permission=permissions.DELETE_INGREDIENTS) | ||||
| def delete_extra_ingredient(identifier, current_session): | ||||
|     """Delete ExtraIngredient | ||||
|  | @ -572,7 +567,7 @@ def delete_extra_ingredient(identifier, current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/settings/min_prices", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/settings/min_prices", methods=["GET"]) | ||||
| @login_required() | ||||
| def get_pricelist_settings_min_prices(current_session): | ||||
|     """Get MinPrices | ||||
|  | @ -593,7 +588,7 @@ def get_pricelist_settings_min_prices(current_session): | |||
|     return jsonify(min_prices) | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/settings/min_prices", methods=["POST"]) | ||||
| @PriceListPlugin.blueprint.route("/settings/min_prices", methods=["POST"]) | ||||
| @login_required(permission=permissions.EDIT_MIN_PRICES) | ||||
| def post_pricelist_settings_min_prices(current_session): | ||||
|     """Create MinPrices | ||||
|  | @ -616,7 +611,7 @@ def post_pricelist_settings_min_prices(current_session): | |||
|     return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/users/<userid>/pricecalc_columns", methods=["GET", "PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/users/<userid>/pricecalc_columns", methods=["GET", "PUT"]) | ||||
| @login_required() | ||||
| def get_columns(userid, current_session): | ||||
|     """Get pricecalc_columns of an user | ||||
|  | @ -648,7 +643,7 @@ def get_columns(userid, current_session): | |||
|         return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/users/<userid>/pricecalc_columns_order", methods=["GET", "PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/users/<userid>/pricecalc_columns_order", methods=["GET", "PUT"]) | ||||
| @login_required() | ||||
| def get_columns_order(userid, current_session): | ||||
|     """Get pricecalc_columns_order of an user | ||||
|  | @ -679,7 +674,7 @@ def get_columns_order(userid, current_session): | |||
|         return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/users/<userid>/pricelist", methods=["GET", "PUT"]) | ||||
| @PriceListPlugin.blueprint.route("/users/<userid>/pricelist", methods=["GET", "PUT"]) | ||||
| @login_required() | ||||
| def get_priclist_setting(userid, current_session): | ||||
|     """Get pricelistsetting of an user | ||||
|  | @ -712,7 +707,7 @@ def get_priclist_setting(userid, current_session): | |||
|         return no_content() | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks/<int:identifier>/picture", methods=["POST", "DELETE"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/<int:identifier>/picture", methods=["POST", "DELETE"]) | ||||
| @login_required(permission=permissions.EDIT) | ||||
| def set_picture(identifier, current_session): | ||||
|     """Get, Create, Delete Drink Picture | ||||
|  | @ -739,7 +734,7 @@ def set_picture(identifier, current_session): | |||
|         raise BadRequest | ||||
| 
 | ||||
| 
 | ||||
| @blueprint.route("/drinks/<int:identifier>/picture", methods=["GET"]) | ||||
| @PriceListPlugin.blueprint.route("/drinks/<int:identifier>/picture", methods=["GET"]) | ||||
| # @headers({"Cache-Control": "private, must-revalidate"}) | ||||
| def _get_picture(identifier): | ||||
|     """Get Picture | ||||
|  |  | |||
|  | @ -0,0 +1,141 @@ | |||
| """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,12 @@ | |||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.10 | ||||
| 
 | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist.models import ModelSerializeMixin, Serial | ||||
| from flaschengeist.models.image import Image | ||||
| from __future__ import annotations  # TODO: Remove if python requirement is >= 3.12 (? PEP 563 is defered) | ||||
| 
 | ||||
| from typing import Optional | ||||
| 
 | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist.database.types import ModelSerializeMixin, Serial | ||||
| from flaschengeist.models import Image | ||||
| 
 | ||||
| 
 | ||||
| drink_tag_association = db.Table( | ||||
|     "drink_x_tag", | ||||
|     db.Column("drink_id", Serial, db.ForeignKey("drink.id")), | ||||
|  |  | |||
|  | @ -5,19 +5,20 @@ Provides routes used to configure roles and permissions of users / roles. | |||
| 
 | ||||
| from werkzeug.exceptions import BadRequest | ||||
| from flask import Blueprint, request, jsonify | ||||
| from http.client import NO_CONTENT | ||||
| 
 | ||||
| from flaschengeist.plugins import Plugin | ||||
| from flaschengeist.utils.decorators import login_required | ||||
| from flaschengeist.controller import roleController | ||||
| from flaschengeist.utils.HTTP import created, no_content | ||||
| from flaschengeist.utils.decorators import login_required | ||||
| 
 | ||||
| from . import permissions | ||||
| 
 | ||||
| 
 | ||||
| class RolesPlugin(Plugin): | ||||
|     blueprint = Blueprint("roles", __name__) | ||||
|     permissions = permissions.permissions | ||||
| 
 | ||||
|     def install(self): | ||||
|         self.install_permissions(permissions.permissions) | ||||
| 
 | ||||
| 
 | ||||
| @RolesPlugin.blueprint.route("/roles", methods=["GET"]) | ||||
|  |  | |||
|  | @ -2,10 +2,10 @@ 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 | ||||
| 
 | ||||
| from . import Plugin | ||||
| 
 | ||||
| 
 | ||||
| class __Task: | ||||
|     def __init__(self, function, **kwags): | ||||
|  | @ -39,10 +39,9 @@ def scheduled(id: str, replace=False, **kwargs): | |||
| 
 | ||||
| 
 | ||||
| class SchedulerPlugin(Plugin): | ||||
|     def __init__(self, entry_point, config=None): | ||||
|         super().__init__(entry_point, config) | ||||
|         self.blueprint = Blueprint(self.name, __name__) | ||||
|     blueprint = Blueprint("scheduler", __name__) | ||||
| 
 | ||||
|     def load(self): | ||||
|         def __view_func(): | ||||
|             self.run_tasks() | ||||
|             return no_content() | ||||
|  | @ -53,14 +52,17 @@ class SchedulerPlugin(Plugin): | |||
|             except: | ||||
|                 logger.error("Error while executing scheduled tasks!", exc_info=True) | ||||
| 
 | ||||
|         cron = None if config is None else config.get("cron", "passive_web").lower() | ||||
|         cron = config.get("scheduler", {}).get("cron", "passive_web").lower() | ||||
| 
 | ||||
|         if cron is None or cron == "passive_web": | ||||
|         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()) | ||||
|  |  | |||
|  | @ -2,15 +2,15 @@ | |||
| 
 | ||||
| Provides routes used to manage users | ||||
| """ | ||||
| from http.client import NO_CONTENT, CREATED | ||||
| from http.client import CREATED | ||||
| from flask import Blueprint, request, jsonify, make_response | ||||
| from werkzeug.exceptions import BadRequest, Forbidden, MethodNotAllowed, NotFound | ||||
| from werkzeug.exceptions import BadRequest, Forbidden, MethodNotAllowed | ||||
| 
 | ||||
| from . import permissions | ||||
| from flaschengeist import logger | ||||
| from flaschengeist.config import config | ||||
| from flaschengeist.plugins import Plugin | ||||
| from flaschengeist.models.user import User | ||||
| from flaschengeist.models import User | ||||
| from flaschengeist.utils.decorators import login_required, extract_session, headers | ||||
| from flaschengeist.controller import userController | ||||
| from flaschengeist.utils.HTTP import created, no_content | ||||
|  | @ -19,7 +19,9 @@ from flaschengeist.utils.datetime import from_iso_format | |||
| 
 | ||||
| class UsersPlugin(Plugin): | ||||
|     blueprint = Blueprint("users", __name__) | ||||
|     permissions = permissions.permissions | ||||
| 
 | ||||
|     def install(self): | ||||
|         self.install_permissions(permissions.permissions) | ||||
| 
 | ||||
| 
 | ||||
| @UsersPlugin.blueprint.route("/users", methods=["POST"]) | ||||
|  | @ -104,7 +106,7 @@ def frontend(userid, current_session): | |||
|         raise Forbidden | ||||
| 
 | ||||
|     if request.method == "POST": | ||||
|         if request.content_length > 1024**2: | ||||
|         if request.content_length > 1024 ** 2: | ||||
|             raise BadRequest | ||||
|         current_session.user_.set_attribute("frontend", request.get_json()) | ||||
|         return no_content() | ||||
|  |  | |||
|  | @ -1,6 +1,9 @@ | |||
| import click | ||||
| import sqlalchemy.exc | ||||
| from flask.cli import with_appcontext | ||||
| from werkzeug.exceptions import BadRequest, Conflict, NotFound | ||||
| from werkzeug.exceptions import NotFound | ||||
| 
 | ||||
| from flaschengeist.database import db | ||||
| from flaschengeist.controller import roleController, userController | ||||
| 
 | ||||
| 
 | ||||
|  | @ -28,23 +31,53 @@ def user(ctx, param, value): | |||
| 
 | ||||
| 
 | ||||
| @click.command() | ||||
| @click.option("--add-role", help="Add new role", type=str) | ||||
| @click.option("--set-admin", help="Make a role an admin role, adding all permissions", type=str) | ||||
| @click.option("--add-user", help="Add new user interactivly", callback=user, is_flag=True, expose_value=False) | ||||
| @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 users(add_role, set_admin): | ||||
| def user(add_role, delete, user): | ||||
|     """Manage users""" | ||||
|     from flaschengeist.database import db | ||||
| 
 | ||||
|     ctx = click.get_current_context() | ||||
| 
 | ||||
|     try: | ||||
|         if add_role: | ||||
|             roleController.create_role(add_role) | ||||
|         if set_admin: | ||||
|             role = roleController.get(set_admin) | ||||
|             role.permissions = roleController.get_permissions() | ||||
|             db.session.commit() | ||||
|         if USER_KEY in ctx.meta: | ||||
|             userController.register(ctx.meta[USER_KEY], ctx.meta[USER_KEY]["password"]) | ||||
|     except (BadRequest, NotFound) as e: | ||||
|         ctx.fail(e.description) | ||||
|         else: | ||||
|             for uid in user: | ||||
|                 user = userController.get_user(uid) | ||||
|                 if delete: | ||||
|                     userController.delete_user(user) | ||||
|                 elif add_role: | ||||
|                     role = roleController.get(add_role) | ||||
|                     user.roles_.append(role) | ||||
|                     userController.modify_user(user, None) | ||||
|                     db.session.commit() | ||||
|     except NotFound: | ||||
|         ctx.fail(f"User not found {uid}") | ||||
|  |  | |||
|  | @ -14,7 +14,7 @@ def extract_session(permission=None): | |||
|         logger.debug("Missing Authorization header or ill-formed") | ||||
|         raise Unauthorized | ||||
| 
 | ||||
|     session = sessionController.validate_token(token, request.user_agent, permission) | ||||
|     session = sessionController.validate_token(token, request.headers, permission) | ||||
|     return session | ||||
| 
 | ||||
| 
 | ||||
|  |  | |||
|  | @ -1,51 +0,0 @@ | |||
| # Borrowed from https://github.com/kvesteri/sqlalchemy-utils | ||||
| # Modifications see: https://github.com/kvesteri/sqlalchemy-utils/issues/561 | ||||
| # LICENSED under the BSD license, see upstream https://github.com/kvesteri/sqlalchemy-utils/blob/master/LICENSE | ||||
| 
 | ||||
| import sqlalchemy as sa | ||||
| from sqlalchemy.orm import object_session | ||||
| 
 | ||||
| 
 | ||||
| def get_foreign_key_values(fk, obj): | ||||
|     mapper = sa.inspect(obj.__class__) | ||||
|     return dict( | ||||
|         ( | ||||
|             fk.constraint.columns.values()[index], | ||||
|             getattr(obj, element.column.key) | ||||
|             if hasattr(obj, element.column.key) | ||||
|             else getattr(obj, mapper.get_property_by_column(element.column).key), | ||||
|         ) | ||||
|         for index, element in enumerate(fk.constraint.elements) | ||||
|     ) | ||||
| 
 | ||||
| 
 | ||||
| def get_referencing_foreign_keys(mixed): | ||||
|     tables = [mixed] | ||||
|     referencing_foreign_keys = set() | ||||
| 
 | ||||
|     for table in mixed.metadata.tables.values(): | ||||
|         if table not in tables: | ||||
|             for constraint in table.constraints: | ||||
|                 if isinstance(constraint, sa.sql.schema.ForeignKeyConstraint): | ||||
|                     for fk in constraint.elements: | ||||
|                         if any(fk.references(t) for t in tables): | ||||
|                             referencing_foreign_keys.add(fk) | ||||
|     return referencing_foreign_keys | ||||
| 
 | ||||
| 
 | ||||
| def merge_references(from_, to, foreign_keys=None): | ||||
|     """ | ||||
|     Merge the references of an entity into another entity. | ||||
|     """ | ||||
|     if from_.__tablename__ != to.__tablename__: | ||||
|         raise TypeError("The tables of given arguments do not match.") | ||||
| 
 | ||||
|     session = object_session(from_) | ||||
|     foreign_keys = get_referencing_foreign_keys(from_.__table__) | ||||
| 
 | ||||
|     for fk in foreign_keys: | ||||
|         old_values = get_foreign_key_values(fk, from_) | ||||
|         new_values = get_foreign_key_values(fk, to) | ||||
|         session.query(from_.__mapper__).filter(*[k == old_values[k] for k in old_values]).update( | ||||
|             new_values, synchronize_session=False | ||||
|         ) | ||||
|  | @ -7,6 +7,7 @@ _hooks_after = {} | |||
| 
 | ||||
| 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` | ||||
|  | @ -38,8 +39,10 @@ def Hook(function=None, id=None): | |||
| 
 | ||||
| def HookBefore(id: str): | ||||
|     """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, | ||||
|     as the functions are called with the same arguments. | ||||
| 
 | ||||
|     Hint: This enables you to modify the arguments! | ||||
|     """ | ||||
|     if not id or not isinstance(id, str): | ||||
|  | @ -54,9 +57,18 @@ def HookBefore(id: str): | |||
| 
 | ||||
| def HookAfter(id: str): | ||||
|     """Decorator for functions to be called after a Hook-Function is called | ||||
| 
 | ||||
|     As with the HookBefore, the hooked up function must accept the same | ||||
|     arguments as the function hooked onto, but also receives a | ||||
|     `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): | ||||
|  |  | |||
							
								
								
									
										20
									
								
								setup.cfg
								
								
								
								
							
							
						
						
									
										20
									
								
								setup.cfg
								
								
								
								
							|  | @ -19,19 +19,20 @@ classifiers = | |||
| 
 | ||||
| [options] | ||||
| include_package_data = True | ||||
| python_requires = >=3.9 | ||||
| python_requires = >=3.10 | ||||
| packages = find: | ||||
| install_requires = | ||||
|     Flask >= 2.0 | ||||
|     Pillow>=8.4.0 | ||||
|     Flask>=2.2.2 | ||||
|     Pillow>=9.2 | ||||
|     flask_cors | ||||
|     flask_sqlalchemy>=2.5 | ||||
|     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.26, <2.0 | ||||
|     sqlalchemy>=1.4.40, <2.0 | ||||
|     toml | ||||
|     werkzeug | ||||
|      | ||||
|     werkzeug>=2.2.2 | ||||
| 
 | ||||
| [options.extras_require] | ||||
| argon = argon2-cffi | ||||
|  | @ -42,14 +43,15 @@ mysql = | |||
|     mysqlclient;platform_system!='Windows' | ||||
| 
 | ||||
| [options.package_data] | ||||
| * = *.toml | ||||
| * = *.toml, script.py.mako | ||||
| 
 | ||||
| [options.entry_points] | ||||
| console_scripts = | ||||
|     flaschengeist = flaschengeist.cli:main | ||||
| flask.commands = | ||||
|     ldap = flaschengeist.plugins.auth_ldap.cli:ldap | ||||
|     users = flaschengeist.plugins.users.cli:users | ||||
|     user = flaschengeist.plugins.users.cli:user | ||||
|     role = flaschengeist.plugins.users.cli:role | ||||
| flaschengeist.plugins = | ||||
|     # Authentication providers | ||||
|     auth_plain = flaschengeist.plugins.auth_plain:AuthPlain | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue