flask-exp 0.1.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- flask_exp-0.1.0/LICENSE +21 -0
- flask_exp-0.1.0/PKG-INFO +139 -0
- flask_exp-0.1.0/README.md +108 -0
- flask_exp-0.1.0/pyproject.toml +47 -0
- flask_exp-0.1.0/src/flask_exp/__init__.py +5 -0
- flask_exp-0.1.0/src/flask_exp/exports.py +119 -0
- flask_exp-0.1.0/src/flask_exp/extension.py +155 -0
- flask_exp-0.1.0/src/flask_exp/py.typed +0 -0
flask_exp-0.1.0/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 Imran
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
flask_exp-0.1.0/PKG-INFO
ADDED
|
@@ -0,0 +1,139 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: flask-exp
|
|
3
|
+
Version: 0.1.0
|
|
4
|
+
Summary: Библиотека, расширяющая возможности Flask для быстрого подключения полезных функций в приложениях.
|
|
5
|
+
License: MIT
|
|
6
|
+
License-File: LICENSE
|
|
7
|
+
Keywords: flask,extension,web,python
|
|
8
|
+
Author: Podval
|
|
9
|
+
Requires-Python: >=3.9,<4.0
|
|
10
|
+
Classifier: Development Status :: 3 - Alpha
|
|
11
|
+
Classifier: Framework :: Flask
|
|
12
|
+
Classifier: Intended Audience :: Developers
|
|
13
|
+
Classifier: License :: OSI Approved :: MIT License
|
|
14
|
+
Classifier: Operating System :: OS Independent
|
|
15
|
+
Classifier: Programming Language :: Python :: 3
|
|
16
|
+
Classifier: Programming Language :: Python :: 3.9
|
|
17
|
+
Classifier: Programming Language :: Python :: 3.10
|
|
18
|
+
Classifier: Programming Language :: Python :: 3.11
|
|
19
|
+
Classifier: Programming Language :: Python :: 3.12
|
|
20
|
+
Classifier: Programming Language :: Python :: 3.13
|
|
21
|
+
Classifier: Programming Language :: Python :: 3.14
|
|
22
|
+
Classifier: Topic :: Software Development :: Libraries
|
|
23
|
+
Requires-Dist: flask (>=2.3)
|
|
24
|
+
Requires-Dist: openpyxl (>=3.1)
|
|
25
|
+
Requires-Dist: sqlalchemy (>=2.0)
|
|
26
|
+
Project-URL: Homepage, https://pypi.org/project/flask-exp/
|
|
27
|
+
Project-URL: Issues, https://github.com/example/flask-exp/issues
|
|
28
|
+
Project-URL: Repository, https://github.com/example/flask-exp
|
|
29
|
+
Description-Content-Type: text/markdown
|
|
30
|
+
|
|
31
|
+
# flask-exp
|
|
32
|
+
|
|
33
|
+
`flask-exp` — это библиотека, расширяющая возможности Flask.
|
|
34
|
+
|
|
35
|
+
Она добавляет полезные функции, которые ускоряют разработку:
|
|
36
|
+
- автоматический health-check endpoint;
|
|
37
|
+
- удобные JSON-ответы;
|
|
38
|
+
- декоратор для маршрутов, автоматически конвертирующий `dict` в JSON.
|
|
39
|
+
- импорт данных из `CSV/XLSX` в SQLAlchemy-модели по явному сопоставлению заголовков.
|
|
40
|
+
- автогенерацию CRUD API-роутов по SQLAlchemy-модели.
|
|
41
|
+
|
|
42
|
+
## Установка
|
|
43
|
+
|
|
44
|
+
```bash
|
|
45
|
+
pip install flask-exp
|
|
46
|
+
```
|
|
47
|
+
|
|
48
|
+
Подробная документация по использованию: [docs/USAGE.md](docs/USAGE.md)
|
|
49
|
+
|
|
50
|
+
## Быстрый старт
|
|
51
|
+
|
|
52
|
+
```python
|
|
53
|
+
from flask import Flask
|
|
54
|
+
from flask_exp import FlaskExp
|
|
55
|
+
|
|
56
|
+
app = Flask(__name__)
|
|
57
|
+
exp = FlaskExp(app)
|
|
58
|
+
|
|
59
|
+
|
|
60
|
+
@exp.route_with_json(app, "/info")
|
|
61
|
+
def info():
|
|
62
|
+
return {"service": "demo", "status": "running"}
|
|
63
|
+
|
|
64
|
+
|
|
65
|
+
if __name__ == "__main__":
|
|
66
|
+
app.run(debug=True)
|
|
67
|
+
```
|
|
68
|
+
|
|
69
|
+
После запуска:
|
|
70
|
+
- `GET /health` вернёт служебный статус;
|
|
71
|
+
- `GET /info` вернёт JSON.
|
|
72
|
+
|
|
73
|
+
## Импорт CSV/XLSX в SQLAlchemy
|
|
74
|
+
|
|
75
|
+
Пользователь задаёт явное сопоставление `заголовок -> поле модели`.
|
|
76
|
+
|
|
77
|
+
```python
|
|
78
|
+
from flask import Flask
|
|
79
|
+
from flask_exp import FlaskExp
|
|
80
|
+
|
|
81
|
+
app = Flask(__name__)
|
|
82
|
+
exp = FlaskExp(app)
|
|
83
|
+
|
|
84
|
+
report = exp.exports.import_to_model(
|
|
85
|
+
file_path="users.csv", # или users.xlsx
|
|
86
|
+
model=User,
|
|
87
|
+
mapping={
|
|
88
|
+
"Email": "email",
|
|
89
|
+
"Full Name": "name",
|
|
90
|
+
},
|
|
91
|
+
session=db_session,
|
|
92
|
+
strict=False, # True: остановка на первой ошибке
|
|
93
|
+
)
|
|
94
|
+
|
|
95
|
+
print(report.created, report.failed, report.errors)
|
|
96
|
+
```
|
|
97
|
+
|
|
98
|
+
## Авто-CRUD API по модели
|
|
99
|
+
|
|
100
|
+
Можно сгенерировать полный CRUD по SQLAlchemy-модели одной командой.
|
|
101
|
+
|
|
102
|
+
```python
|
|
103
|
+
routes = exp.create_crud_routes(app, User, db_session)
|
|
104
|
+
```
|
|
105
|
+
|
|
106
|
+
По умолчанию для модели `User` с `__tablename__ = "users"` создаются маршруты:
|
|
107
|
+
- `POST /users` — create
|
|
108
|
+
- `GET /users` — list
|
|
109
|
+
- `GET /users/<item_id>` — get by id
|
|
110
|
+
- `PUT /users/<item_id>` и `PATCH /users/<item_id>` — update
|
|
111
|
+
- `DELETE /users/<item_id>` — delete
|
|
112
|
+
|
|
113
|
+
Можно переопределить базовый путь:
|
|
114
|
+
|
|
115
|
+
```python
|
|
116
|
+
exp.create_crud_routes(app, User, db_session, base_route="/api/users")
|
|
117
|
+
```
|
|
118
|
+
|
|
119
|
+
## Разработка
|
|
120
|
+
|
|
121
|
+
```bash
|
|
122
|
+
poetry install
|
|
123
|
+
poetry run pytest
|
|
124
|
+
```
|
|
125
|
+
|
|
126
|
+
## Публикация в PyPI
|
|
127
|
+
|
|
128
|
+
```bash
|
|
129
|
+
poetry build
|
|
130
|
+
poetry publish
|
|
131
|
+
```
|
|
132
|
+
|
|
133
|
+
Для публикации через токен:
|
|
134
|
+
|
|
135
|
+
```bash
|
|
136
|
+
poetry config pypi-token.pypi <YOUR_PYPI_TOKEN>
|
|
137
|
+
poetry publish --build
|
|
138
|
+
```
|
|
139
|
+
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
# flask-exp
|
|
2
|
+
|
|
3
|
+
`flask-exp` — это библиотека, расширяющая возможности Flask.
|
|
4
|
+
|
|
5
|
+
Она добавляет полезные функции, которые ускоряют разработку:
|
|
6
|
+
- автоматический health-check endpoint;
|
|
7
|
+
- удобные JSON-ответы;
|
|
8
|
+
- декоратор для маршрутов, автоматически конвертирующий `dict` в JSON.
|
|
9
|
+
- импорт данных из `CSV/XLSX` в SQLAlchemy-модели по явному сопоставлению заголовков.
|
|
10
|
+
- автогенерацию CRUD API-роутов по SQLAlchemy-модели.
|
|
11
|
+
|
|
12
|
+
## Установка
|
|
13
|
+
|
|
14
|
+
```bash
|
|
15
|
+
pip install flask-exp
|
|
16
|
+
```
|
|
17
|
+
|
|
18
|
+
Подробная документация по использованию: [docs/USAGE.md](docs/USAGE.md)
|
|
19
|
+
|
|
20
|
+
## Быстрый старт
|
|
21
|
+
|
|
22
|
+
```python
|
|
23
|
+
from flask import Flask
|
|
24
|
+
from flask_exp import FlaskExp
|
|
25
|
+
|
|
26
|
+
app = Flask(__name__)
|
|
27
|
+
exp = FlaskExp(app)
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
@exp.route_with_json(app, "/info")
|
|
31
|
+
def info():
|
|
32
|
+
return {"service": "demo", "status": "running"}
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
if __name__ == "__main__":
|
|
36
|
+
app.run(debug=True)
|
|
37
|
+
```
|
|
38
|
+
|
|
39
|
+
После запуска:
|
|
40
|
+
- `GET /health` вернёт служебный статус;
|
|
41
|
+
- `GET /info` вернёт JSON.
|
|
42
|
+
|
|
43
|
+
## Импорт CSV/XLSX в SQLAlchemy
|
|
44
|
+
|
|
45
|
+
Пользователь задаёт явное сопоставление `заголовок -> поле модели`.
|
|
46
|
+
|
|
47
|
+
```python
|
|
48
|
+
from flask import Flask
|
|
49
|
+
from flask_exp import FlaskExp
|
|
50
|
+
|
|
51
|
+
app = Flask(__name__)
|
|
52
|
+
exp = FlaskExp(app)
|
|
53
|
+
|
|
54
|
+
report = exp.exports.import_to_model(
|
|
55
|
+
file_path="users.csv", # или users.xlsx
|
|
56
|
+
model=User,
|
|
57
|
+
mapping={
|
|
58
|
+
"Email": "email",
|
|
59
|
+
"Full Name": "name",
|
|
60
|
+
},
|
|
61
|
+
session=db_session,
|
|
62
|
+
strict=False, # True: остановка на первой ошибке
|
|
63
|
+
)
|
|
64
|
+
|
|
65
|
+
print(report.created, report.failed, report.errors)
|
|
66
|
+
```
|
|
67
|
+
|
|
68
|
+
## Авто-CRUD API по модели
|
|
69
|
+
|
|
70
|
+
Можно сгенерировать полный CRUD по SQLAlchemy-модели одной командой.
|
|
71
|
+
|
|
72
|
+
```python
|
|
73
|
+
routes = exp.create_crud_routes(app, User, db_session)
|
|
74
|
+
```
|
|
75
|
+
|
|
76
|
+
По умолчанию для модели `User` с `__tablename__ = "users"` создаются маршруты:
|
|
77
|
+
- `POST /users` — create
|
|
78
|
+
- `GET /users` — list
|
|
79
|
+
- `GET /users/<item_id>` — get by id
|
|
80
|
+
- `PUT /users/<item_id>` и `PATCH /users/<item_id>` — update
|
|
81
|
+
- `DELETE /users/<item_id>` — delete
|
|
82
|
+
|
|
83
|
+
Можно переопределить базовый путь:
|
|
84
|
+
|
|
85
|
+
```python
|
|
86
|
+
exp.create_crud_routes(app, User, db_session, base_route="/api/users")
|
|
87
|
+
```
|
|
88
|
+
|
|
89
|
+
## Разработка
|
|
90
|
+
|
|
91
|
+
```bash
|
|
92
|
+
poetry install
|
|
93
|
+
poetry run pytest
|
|
94
|
+
```
|
|
95
|
+
|
|
96
|
+
## Публикация в PyPI
|
|
97
|
+
|
|
98
|
+
```bash
|
|
99
|
+
poetry build
|
|
100
|
+
poetry publish
|
|
101
|
+
```
|
|
102
|
+
|
|
103
|
+
Для публикации через токен:
|
|
104
|
+
|
|
105
|
+
```bash
|
|
106
|
+
poetry config pypi-token.pypi <YOUR_PYPI_TOKEN>
|
|
107
|
+
poetry publish --build
|
|
108
|
+
```
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
[build-system]
|
|
2
|
+
requires = ["poetry-core>=1.9.0"]
|
|
3
|
+
build-backend = "poetry.core.masonry.api"
|
|
4
|
+
|
|
5
|
+
[tool.poetry]
|
|
6
|
+
name = "flask-exp"
|
|
7
|
+
version = "0.1.0"
|
|
8
|
+
description = "Библиотека, расширяющая возможности Flask для быстрого подключения полезных функций в приложениях."
|
|
9
|
+
readme = "README.md"
|
|
10
|
+
license = "MIT"
|
|
11
|
+
authors = [
|
|
12
|
+
"Podval"
|
|
13
|
+
]
|
|
14
|
+
keywords = ["flask", "extension", "web", "python"]
|
|
15
|
+
classifiers = [
|
|
16
|
+
"Development Status :: 3 - Alpha",
|
|
17
|
+
"Framework :: Flask",
|
|
18
|
+
"Intended Audience :: Developers",
|
|
19
|
+
"License :: OSI Approved :: MIT License",
|
|
20
|
+
"Operating System :: OS Independent",
|
|
21
|
+
"Programming Language :: Python :: 3",
|
|
22
|
+
"Programming Language :: Python :: 3.9",
|
|
23
|
+
"Programming Language :: Python :: 3.10",
|
|
24
|
+
"Programming Language :: Python :: 3.11",
|
|
25
|
+
"Programming Language :: Python :: 3.12",
|
|
26
|
+
"Topic :: Software Development :: Libraries",
|
|
27
|
+
]
|
|
28
|
+
packages = [{ include = "flask_exp", from = "src" }]
|
|
29
|
+
|
|
30
|
+
[tool.poetry.dependencies]
|
|
31
|
+
python = ">=3.9,<4.0"
|
|
32
|
+
flask = ">=2.3"
|
|
33
|
+
sqlalchemy = ">=2.0"
|
|
34
|
+
openpyxl = ">=3.1"
|
|
35
|
+
|
|
36
|
+
[tool.poetry.group.dev.dependencies]
|
|
37
|
+
pytest = ">=8.3"
|
|
38
|
+
ruff = ">=0.6"
|
|
39
|
+
|
|
40
|
+
[tool.poetry.urls]
|
|
41
|
+
Homepage = "https://pypi.org/project/flask-exp/"
|
|
42
|
+
Repository = "https://github.com/example/flask-exp"
|
|
43
|
+
Issues = "https://github.com/example/flask-exp/issues"
|
|
44
|
+
|
|
45
|
+
[tool.pytest.ini_options]
|
|
46
|
+
testpaths = ["tests"]
|
|
47
|
+
python_files = ["test_*.py"]
|
|
@@ -0,0 +1,119 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
import csv
|
|
4
|
+
from dataclasses import dataclass, field
|
|
5
|
+
from pathlib import Path
|
|
6
|
+
from typing import Any, Iterable
|
|
7
|
+
|
|
8
|
+
from openpyxl import load_workbook
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
@dataclass
|
|
12
|
+
class ImportReport:
|
|
13
|
+
created: int = 0
|
|
14
|
+
failed: int = 0
|
|
15
|
+
skipped: int = 0
|
|
16
|
+
dry_run: bool = False
|
|
17
|
+
errors: list[dict[str, Any]] = field(default_factory=list)
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
class Exports:
|
|
21
|
+
def import_to_model(
|
|
22
|
+
self,
|
|
23
|
+
*,
|
|
24
|
+
file_path: str | Path,
|
|
25
|
+
model: type,
|
|
26
|
+
mapping: dict[str, str],
|
|
27
|
+
session,
|
|
28
|
+
strict: bool = True,
|
|
29
|
+
dry_run: bool = False,
|
|
30
|
+
commit: bool = True,
|
|
31
|
+
encoding: str = "utf-8-sig",
|
|
32
|
+
) -> ImportReport:
|
|
33
|
+
path = Path(file_path)
|
|
34
|
+
headers, rows = self._read_rows(path, encoding=encoding)
|
|
35
|
+
self._validate_mapping(headers=headers, model=model, mapping=mapping)
|
|
36
|
+
|
|
37
|
+
report = ImportReport(dry_run=dry_run)
|
|
38
|
+
|
|
39
|
+
for row_number, row in rows:
|
|
40
|
+
payload = {model_field: row.get(source_header) for source_header, model_field in mapping.items()}
|
|
41
|
+
try:
|
|
42
|
+
if dry_run:
|
|
43
|
+
model(**payload)
|
|
44
|
+
else:
|
|
45
|
+
with session.begin_nested():
|
|
46
|
+
instance = model(**payload)
|
|
47
|
+
session.add(instance)
|
|
48
|
+
session.flush()
|
|
49
|
+
report.created += 1
|
|
50
|
+
except Exception as exc:
|
|
51
|
+
report.failed += 1
|
|
52
|
+
report.skipped += 1
|
|
53
|
+
report.errors.append(
|
|
54
|
+
{
|
|
55
|
+
"row_number": row_number,
|
|
56
|
+
"error": str(exc),
|
|
57
|
+
"row": row,
|
|
58
|
+
"payload": payload,
|
|
59
|
+
}
|
|
60
|
+
)
|
|
61
|
+
if strict:
|
|
62
|
+
if not dry_run:
|
|
63
|
+
session.rollback()
|
|
64
|
+
raise ValueError(f"Import failed on row {row_number}: {exc}") from exc
|
|
65
|
+
|
|
66
|
+
if not dry_run and commit:
|
|
67
|
+
session.commit()
|
|
68
|
+
|
|
69
|
+
return report
|
|
70
|
+
|
|
71
|
+
def _validate_mapping(self, *, headers: list[str], model: type, mapping: dict[str, str]) -> None:
|
|
72
|
+
missing_headers = [header for header in mapping.keys() if header not in headers]
|
|
73
|
+
if missing_headers:
|
|
74
|
+
missing = ", ".join(missing_headers)
|
|
75
|
+
raise ValueError(f"Headers not found in file: {missing}")
|
|
76
|
+
|
|
77
|
+
missing_model_fields = [field_name for field_name in mapping.values() if not hasattr(model, field_name)]
|
|
78
|
+
if missing_model_fields:
|
|
79
|
+
missing = ", ".join(missing_model_fields)
|
|
80
|
+
raise ValueError(f"Model fields not found: {missing}")
|
|
81
|
+
|
|
82
|
+
def _read_rows(self, path: Path, *, encoding: str) -> tuple[list[str], Iterable[tuple[int, dict[str, Any]]]]:
|
|
83
|
+
extension = path.suffix.lower()
|
|
84
|
+
if extension == ".csv":
|
|
85
|
+
return self._read_csv(path, encoding=encoding)
|
|
86
|
+
if extension == ".xlsx":
|
|
87
|
+
return self._read_xlsx(path)
|
|
88
|
+
raise ValueError("Only .csv and .xlsx files are supported")
|
|
89
|
+
|
|
90
|
+
def _read_csv(self, path: Path, *, encoding: str) -> tuple[list[str], Iterable[tuple[int, dict[str, Any]]]]:
|
|
91
|
+
with path.open("r", encoding=encoding, newline="") as file:
|
|
92
|
+
reader = csv.DictReader(file)
|
|
93
|
+
if reader.fieldnames is None:
|
|
94
|
+
raise ValueError("CSV file must contain a header row")
|
|
95
|
+
headers = [str(field) for field in reader.fieldnames]
|
|
96
|
+
rows = [(index, row) for index, row in enumerate(reader, start=2)]
|
|
97
|
+
return headers, rows
|
|
98
|
+
|
|
99
|
+
def _read_xlsx(self, path: Path) -> tuple[list[str], Iterable[tuple[int, dict[str, Any]]]]:
|
|
100
|
+
workbook = load_workbook(path, read_only=True, data_only=True)
|
|
101
|
+
worksheet = workbook.active
|
|
102
|
+
raw_rows = worksheet.iter_rows(values_only=True)
|
|
103
|
+
|
|
104
|
+
try:
|
|
105
|
+
header_row = next(raw_rows)
|
|
106
|
+
except StopIteration as exc:
|
|
107
|
+
raise ValueError("XLSX file is empty") from exc
|
|
108
|
+
|
|
109
|
+
if not header_row:
|
|
110
|
+
raise ValueError("XLSX file must contain a header row")
|
|
111
|
+
|
|
112
|
+
headers = [str(value).strip() if value is not None else "" for value in header_row]
|
|
113
|
+
rows = []
|
|
114
|
+
for row_number, values in enumerate(raw_rows, start=2):
|
|
115
|
+
row = {header: value for header, value in zip(headers, values)}
|
|
116
|
+
rows.append((row_number, row))
|
|
117
|
+
|
|
118
|
+
workbook.close()
|
|
119
|
+
return headers, rows
|
|
@@ -0,0 +1,155 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from functools import wraps
|
|
4
|
+
from typing import Any, Callable
|
|
5
|
+
|
|
6
|
+
from flask import Flask, jsonify, request
|
|
7
|
+
from sqlalchemy.inspection import inspect as sqlalchemy_inspect
|
|
8
|
+
|
|
9
|
+
from .exports import Exports
|
|
10
|
+
|
|
11
|
+
|
|
12
|
+
class FlaskExp:
|
|
13
|
+
def __init__(self, app: Flask | None = None, *, health_endpoint: str = "/health") -> None:
|
|
14
|
+
self.health_endpoint = health_endpoint
|
|
15
|
+
self._initialized = False
|
|
16
|
+
self.exports = Exports()
|
|
17
|
+
if app is not None:
|
|
18
|
+
self.init_app(app)
|
|
19
|
+
|
|
20
|
+
def init_app(self, app: Flask) -> None:
|
|
21
|
+
if self._initialized:
|
|
22
|
+
return
|
|
23
|
+
|
|
24
|
+
app.extensions["flask-exp"] = self
|
|
25
|
+
self._register_health_endpoint(app)
|
|
26
|
+
self._initialized = True
|
|
27
|
+
|
|
28
|
+
def _register_health_endpoint(self, app: Flask) -> None:
|
|
29
|
+
endpoint_name = "flask_exp_health"
|
|
30
|
+
if endpoint_name in app.view_functions:
|
|
31
|
+
return
|
|
32
|
+
|
|
33
|
+
@app.get(self.health_endpoint)
|
|
34
|
+
def flask_exp_health() -> tuple[dict[str, str], int]:
|
|
35
|
+
return {"status": "ok", "library": "flask-exp"}, 200
|
|
36
|
+
|
|
37
|
+
def json_response(self, payload: dict, status_code: int = 200):
|
|
38
|
+
response = jsonify(payload)
|
|
39
|
+
response.status_code = status_code
|
|
40
|
+
return response
|
|
41
|
+
|
|
42
|
+
def route_with_json(self, app: Flask, rule: str, **options):
|
|
43
|
+
def decorator(func: Callable):
|
|
44
|
+
@wraps(func)
|
|
45
|
+
def wrapper(*args, **kwargs):
|
|
46
|
+
result = func(*args, **kwargs)
|
|
47
|
+
if isinstance(result, dict):
|
|
48
|
+
return self.json_response(result)
|
|
49
|
+
return result
|
|
50
|
+
|
|
51
|
+
app.add_url_rule(rule, endpoint=options.pop("endpoint", None), view_func=wrapper, **options)
|
|
52
|
+
return wrapper
|
|
53
|
+
|
|
54
|
+
return decorator
|
|
55
|
+
|
|
56
|
+
def create_crud_routes(
|
|
57
|
+
self,
|
|
58
|
+
app: Flask,
|
|
59
|
+
model: type,
|
|
60
|
+
session,
|
|
61
|
+
*,
|
|
62
|
+
base_route: str | None = None,
|
|
63
|
+
endpoint_prefix: str | None = None,
|
|
64
|
+
) -> dict[str, str]:
|
|
65
|
+
mapper = sqlalchemy_inspect(model)
|
|
66
|
+
primary_keys = [column.key for column in mapper.primary_key]
|
|
67
|
+
if len(primary_keys) != 1:
|
|
68
|
+
raise ValueError("CRUD auto-router supports models with a single primary key")
|
|
69
|
+
|
|
70
|
+
primary_key = primary_keys[0]
|
|
71
|
+
all_columns = [column.key for column in mapper.columns]
|
|
72
|
+
writable_fields = [field for field in all_columns if field != primary_key]
|
|
73
|
+
|
|
74
|
+
resource = base_route or f"/{model.__tablename__}"
|
|
75
|
+
detail_resource = f"{resource}/<item_id>"
|
|
76
|
+
endpoint_base = endpoint_prefix or model.__name__.lower()
|
|
77
|
+
|
|
78
|
+
endpoints = {
|
|
79
|
+
"create": f"{endpoint_base}_create",
|
|
80
|
+
"list": f"{endpoint_base}_list",
|
|
81
|
+
"get": f"{endpoint_base}_get",
|
|
82
|
+
"update": f"{endpoint_base}_update",
|
|
83
|
+
"delete": f"{endpoint_base}_delete",
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
def serialize(instance) -> dict[str, Any]:
|
|
87
|
+
return {column: getattr(instance, column) for column in all_columns}
|
|
88
|
+
|
|
89
|
+
@app.post(resource, endpoint=endpoints["create"])
|
|
90
|
+
def create_item():
|
|
91
|
+
payload = request.get_json(silent=True)
|
|
92
|
+
if not isinstance(payload, dict):
|
|
93
|
+
return self.json_response({"error": "JSON body must be an object"}, 400)
|
|
94
|
+
|
|
95
|
+
data = {field: payload[field] for field in writable_fields if field in payload}
|
|
96
|
+
instance = model(**data)
|
|
97
|
+
try:
|
|
98
|
+
session.add(instance)
|
|
99
|
+
session.commit()
|
|
100
|
+
except Exception as exc:
|
|
101
|
+
session.rollback()
|
|
102
|
+
return self.json_response({"error": str(exc)}, 400)
|
|
103
|
+
|
|
104
|
+
return self.json_response(serialize(instance), 201)
|
|
105
|
+
|
|
106
|
+
@app.get(resource, endpoint=endpoints["list"])
|
|
107
|
+
def list_items():
|
|
108
|
+
items = session.query(model).all()
|
|
109
|
+
return self.json_response({"items": [serialize(item) for item in items]})
|
|
110
|
+
|
|
111
|
+
@app.get(detail_resource, endpoint=endpoints["get"])
|
|
112
|
+
def get_item(item_id):
|
|
113
|
+
instance = session.get(model, item_id)
|
|
114
|
+
if instance is None:
|
|
115
|
+
return self.json_response({"error": "Not found"}, 404)
|
|
116
|
+
return self.json_response(serialize(instance))
|
|
117
|
+
|
|
118
|
+
@app.put(detail_resource, endpoint=endpoints["update"])
|
|
119
|
+
@app.patch(detail_resource, endpoint=f"{endpoints['update']}_patch")
|
|
120
|
+
def update_item(item_id):
|
|
121
|
+
payload = request.get_json(silent=True)
|
|
122
|
+
if not isinstance(payload, dict):
|
|
123
|
+
return self.json_response({"error": "JSON body must be an object"}, 400)
|
|
124
|
+
|
|
125
|
+
instance = session.get(model, item_id)
|
|
126
|
+
if instance is None:
|
|
127
|
+
return self.json_response({"error": "Not found"}, 404)
|
|
128
|
+
|
|
129
|
+
for field in writable_fields:
|
|
130
|
+
if field in payload:
|
|
131
|
+
setattr(instance, field, payload[field])
|
|
132
|
+
|
|
133
|
+
try:
|
|
134
|
+
session.commit()
|
|
135
|
+
except Exception as exc:
|
|
136
|
+
session.rollback()
|
|
137
|
+
return self.json_response({"error": str(exc)}, 400)
|
|
138
|
+
|
|
139
|
+
return self.json_response(serialize(instance))
|
|
140
|
+
|
|
141
|
+
@app.delete(detail_resource, endpoint=endpoints["delete"])
|
|
142
|
+
def delete_item(item_id):
|
|
143
|
+
instance = session.get(model, item_id)
|
|
144
|
+
if instance is None:
|
|
145
|
+
return self.json_response({"error": "Not found"}, 404)
|
|
146
|
+
|
|
147
|
+
session.delete(instance)
|
|
148
|
+
session.commit()
|
|
149
|
+
return self.json_response({"status": "deleted"})
|
|
150
|
+
|
|
151
|
+
return {
|
|
152
|
+
"resource": resource,
|
|
153
|
+
"detail_resource": detail_resource,
|
|
154
|
+
"endpoints": endpoints,
|
|
155
|
+
}
|
|
File without changes
|