mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 21:15:13 +00:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2099c33aad | ||
|
|
b809497aba | ||
|
|
a6fe8c0e8d | ||
|
|
629cf7ce45 | ||
|
|
fad6c05d41 | ||
|
|
77d59d3453 | ||
|
|
931d6b281a | ||
|
|
d929f0dbdc | ||
|
|
6d1b47903e | ||
|
|
8420ccf190 | ||
|
|
1e38e1918b | ||
|
|
52e78152a1 | ||
|
|
dcb07c60f2 | ||
|
|
33c9f78cf0 | ||
|
|
37b1033697 | ||
|
|
0aa4ebc635 | ||
|
|
30d4107670 | ||
|
|
dcf235db78 | ||
|
|
1dfcd7ae3c | ||
|
|
2b28244099 | ||
|
|
f71071f905 | ||
|
|
34c9c398ca | ||
|
|
f099777dd3 | ||
|
|
c95be82cb9 | ||
|
|
00ec039095 | ||
|
|
4d087355b4 | ||
|
|
d57fabcbb4 | ||
|
|
3a32522e2b | ||
|
|
22f8850bf7 | ||
|
|
eeb709c6aa | ||
|
|
a6bbf2971d | ||
|
|
cd48bb4210 | ||
|
|
4500dd0128 | ||
|
|
2b6d8222ad | ||
|
|
1b47ab7e83 | ||
|
|
bd804e0a38 | ||
|
|
8e8af2032a | ||
|
|
e09062eb39 | ||
|
|
8c00492e44 | ||
|
|
0237d4c7b1 | ||
|
|
fe3a3b4e86 | ||
|
|
65ec721a1c | ||
|
|
4eb76ad6ea | ||
|
|
1f8687155a | ||
|
|
8591d4e592 | ||
|
|
dedf4e7bae | ||
|
|
1e4aa88487 | ||
|
|
0dd745b7a7 | ||
|
|
6b2402f212 | ||
|
|
24a487eeea | ||
|
|
4f574189c7 | ||
|
|
b559a74cea | ||
|
|
83dcdbe59d | ||
|
|
d64f7c2911 | ||
|
|
53e64bce37 | ||
|
|
00a781357b | ||
|
|
d7f87ed0ad | ||
|
|
bda9b1d463 | ||
|
|
e0a106ec51 | ||
|
|
016361bd68 | ||
|
|
f2d3c39425 | ||
|
|
e45d2d6fe7 | ||
|
|
4ddf84fa22 | ||
|
|
326a1883e8 | ||
|
|
9ff36d4498 | ||
|
|
98ef07ad32 | ||
|
|
802b0d456a | ||
|
|
6767f0e2e6 | ||
|
|
d72a05c965 | ||
|
|
ce3be1db8e | ||
|
|
e84ef04f15 | ||
|
|
adee3a2177 | ||
|
|
526e0aab9a | ||
|
|
2bbbbe764a | ||
|
|
ef5e921c08 | ||
|
|
a2fa59285a | ||
|
|
494942e87e | ||
|
|
aa2b76673f | ||
|
|
099880dd1c | ||
|
|
d906a103af | ||
|
|
4ed49d2d13 | ||
|
|
7b5ae4696f | ||
|
|
54ea301152 | ||
|
|
69c27232ac | ||
|
|
f36dcb3f5c | ||
|
|
63d088d62c | ||
|
|
a461f78224 | ||
|
|
c306432708 | ||
|
|
97f778cc90 | ||
|
|
88695ac6ad | ||
|
|
3b04329b04 | ||
|
|
bd4acf8121 | ||
|
|
b9a9b1efe6 | ||
|
|
c050f58ee9 | ||
|
|
d47375ae41 | ||
|
|
0602c6144d | ||
|
|
c84f0658fb | ||
|
|
a2902bd23a | ||
|
|
69a9b90711 | ||
|
|
b96402e11d | ||
|
|
0bc6f6a719 | ||
|
|
ebd65c8246 | ||
|
|
234bf708b3 | ||
|
|
6d5e741d94 | ||
|
|
f9bfe8f510 | ||
|
|
408c866dda | ||
|
|
bac41a46f8 | ||
|
|
8638c26db1 | ||
|
|
319ba13df1 | ||
|
|
0d1f05d307 | ||
|
|
e6608d18ce | ||
|
|
7afd8415cd | ||
|
|
845f2f0ff8 | ||
|
|
11d7792c0f | ||
|
|
6883ec34c6 | ||
|
|
4c45e7ff27 | ||
|
|
5621daea98 | ||
|
|
71d168dfea | ||
|
|
2d4365189d | ||
|
|
0c28a46e60 | ||
|
|
a4c7dae99c | ||
|
|
bfda0a226c | ||
|
|
c9bbb105a3 | ||
|
|
b415dc8912 | ||
|
|
05b74d71a3 | ||
|
|
10c0c691a7 | ||
|
|
0bc3786fc6 | ||
|
|
8e041ccb1b | ||
|
|
a82e9b7b9d | ||
|
|
7bab8cc636 | ||
|
|
7659b63d88 | ||
|
|
33ebc5723f | ||
|
|
86d9efd785 | ||
|
|
57ae7f03a6 | ||
|
|
6ee3983846 | ||
|
|
f05ed5cdda | ||
|
|
187dfa5344 | ||
|
|
918c8a9d33 | ||
|
|
ccfe35ea85 | ||
|
|
4a118c166a | ||
|
|
5f3823cb6d | ||
|
|
19ef52da9b | ||
|
|
39c3339a8d | ||
|
|
9f52629dcb | ||
|
|
1cddd6e2d5 | ||
|
|
ed10e6ff47 | ||
|
|
48574fb50a | ||
|
|
ee7fe3087d | ||
|
|
270ac3d941 | ||
|
|
2f76988076 | ||
|
|
b275481742 | ||
|
|
f3f1ceb595 | ||
|
|
bcc764142b | ||
|
|
9b0d9f8296 | ||
|
|
63ba7646c1 | ||
|
|
1a7734185f | ||
|
|
8541059d23 | ||
|
|
61edc81333 | ||
|
|
8998687adb | ||
|
|
85b0294da9 | ||
|
|
a8ecb4ae3b | ||
|
|
a87ff74b56 | ||
|
|
3c5682988d | ||
|
|
1d32154498 | ||
|
|
f4290ddb8c | ||
|
|
1e54b0ba42 | ||
|
|
27b9cf4c64 | ||
|
|
87f29e52a1 |
@@ -1,14 +1,14 @@
|
|||||||
repos:
|
repos:
|
||||||
- repo: https://github.com/pre-commit/pre-commit-hooks
|
- repo: https://github.com/pre-commit/pre-commit-hooks
|
||||||
rev: v5.0.0
|
rev: v6.0.0
|
||||||
hooks:
|
hooks:
|
||||||
- id: check-added-large-files
|
# - id: check-added-large-files
|
||||||
- id: end-of-file-fixer
|
# - id: end-of-file-fixer
|
||||||
- id: check-yaml
|
- id: check-yaml
|
||||||
- id: check-toml
|
- id: check-toml
|
||||||
|
|
||||||
- repo: https://github.com/charliermarsh/ruff-pre-commit
|
- repo: https://github.com/charliermarsh/ruff-pre-commit
|
||||||
rev: v0.11.2
|
rev: v0.12.11
|
||||||
hooks:
|
hooks:
|
||||||
- id: ruff
|
- id: ruff
|
||||||
args:
|
args:
|
||||||
@@ -20,7 +20,7 @@ repos:
|
|||||||
- id: ruff-format
|
- id: ruff-format
|
||||||
|
|
||||||
- repo: https://github.com/astral-sh/uv-pre-commit
|
- repo: https://github.com/astral-sh/uv-pre-commit
|
||||||
rev: 0.6.10
|
rev: 0.8.14
|
||||||
hooks:
|
hooks:
|
||||||
- id: uv-lock
|
- id: uv-lock
|
||||||
- id: uv-export
|
- id: uv-export
|
||||||
|
|||||||
+293
@@ -1,3 +1,284 @@
|
|||||||
|
<a id="v1.8.0"></a>
|
||||||
|
# [v1.8.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.8.0) - 2025-08-15
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.7.0 by [@wu-clan](https://github.com/wu-clan) in [#729](https://github.com/fastapi-practices/fastapi_best_architecture/pull/729)
|
||||||
|
* Simplify task crontab expression validation by [@wu-clan](https://github.com/wu-clan) in [#733](https://github.com/fastapi-practices/fastapi_best_architecture/pull/733)
|
||||||
|
* Add distributed lock for scheduled task by [@wu-clan](https://github.com/wu-clan) in [#732](https://github.com/fastapi-practices/fastapi_best_architecture/pull/732)
|
||||||
|
* Update the default cache period for userinfo by [@wu-clan](https://github.com/wu-clan) in [#734](https://github.com/fastapi-practices/fastapi_best_architecture/pull/734)
|
||||||
|
* Fix celery CLI option to required by [@wu-clan](https://github.com/wu-clan) in [#737](https://github.com/fastapi-practices/fastapi_best_architecture/pull/737)
|
||||||
|
* Add auth whitelist regular expression config by [@wu-clan](https://github.com/wu-clan) in [#738](https://github.com/fastapi-practices/fastapi_best_architecture/pull/738)
|
||||||
|
* Fix the opera log field encryption by [@wu-clan](https://github.com/wu-clan) in [#739](https://github.com/fastapi-practices/fastapi_best_architecture/pull/739)
|
||||||
|
* Update the OAuth2 login password policy by [@wu-clan](https://github.com/wu-clan) in [#741](https://github.com/fastapi-practices/fastapi_best_architecture/pull/741)
|
||||||
|
* Add update support for user email and phone by [@wu-clan](https://github.com/wu-clan) in [#742](https://github.com/fastapi-practices/fastapi_best_architecture/pull/742)
|
||||||
|
* Fix the error trigger when model auto import by [@wu-clan](https://github.com/wu-clan) in [#743](https://github.com/fastapi-practices/fastapi_best_architecture/pull/743)
|
||||||
|
* Simplify the plugin status update logic by [@wu-clan](https://github.com/wu-clan) in [#744](https://github.com/fastapi-practices/fastapi_best_architecture/pull/744)
|
||||||
|
* Add some interfaces for user profiles by [@wu-clan](https://github.com/wu-clan) in [#745](https://github.com/fastapi-practices/fastapi_best_architecture/pull/745)
|
||||||
|
* Add schedule task demo that contains params by [@wu-clan](https://github.com/wu-clan) in [#746](https://github.com/fastapi-practices/fastapi_best_architecture/pull/746)
|
||||||
|
* Fix the kwargs params of schedule task by [@wu-clan](https://github.com/wu-clan) in [#747](https://github.com/fastapi-practices/fastapi_best_architecture/pull/747)
|
||||||
|
* Refactor code generation files and routes by [@wu-clan](https://github.com/wu-clan) in [#748](https://github.com/fastapi-practices/fastapi_best_architecture/pull/748)
|
||||||
|
* Refactor task routes and add control routes by [@wu-clan](https://github.com/wu-clan) in [#749](https://github.com/fastapi-practices/fastapi_best_architecture/pull/749)
|
||||||
|
* Fix message format in validation exception handler by [@wu-clan](https://github.com/wu-clan) in [#755](https://github.com/fastapi-practices/fastapi_best_architecture/pull/755)
|
||||||
|
* Update the opera log desensitization method by [@wu-clan](https://github.com/wu-clan) in [#756](https://github.com/fastapi-practices/fastapi_best_architecture/pull/756)
|
||||||
|
* Add business pagination in the code generator by [@wu-clan](https://github.com/wu-clan) in [#757](https://github.com/fastapi-practices/fastapi_best_architecture/pull/757)
|
||||||
|
* Optimize the data sort logic of tree nodes by [@wu-clan](https://github.com/wu-clan) in [#758](https://github.com/fastapi-practices/fastapi_best_architecture/pull/758)
|
||||||
|
* Update log output config and format by [@wu-clan](https://github.com/wu-clan) in [#759](https://github.com/fastapi-practices/fastapi_best_architecture/pull/759)
|
||||||
|
* Update the naming of table creation function by [@wu-clan](https://github.com/wu-clan) in [#760](https://github.com/fastapi-practices/fastapi_best_architecture/pull/760)
|
||||||
|
* Optimize the opera log storage logic through queue by [@IAseven](https://github.com/IAseven) in [#750](https://github.com/fastapi-practices/fastapi_best_architecture/pull/750)
|
||||||
|
* Optimize naming and preview in code generation by [@wu-clan](https://github.com/wu-clan) in [#764](https://github.com/fastapi-practices/fastapi_best_architecture/pull/764)
|
||||||
|
* Update the description for the run file by [@wu-clan](https://github.com/wu-clan) in [#766](https://github.com/fastapi-practices/fastapi_best_architecture/pull/766)
|
||||||
|
* Optimize the timezone datetime return encoder by [@wu-clan](https://github.com/wu-clan) in [#767](https://github.com/fastapi-practices/fastapi_best_architecture/pull/767)
|
||||||
|
* Update the content layout of the config file by [@wu-clan](https://github.com/wu-clan) in [#768](https://github.com/fastapi-practices/fastapi_best_architecture/pull/768)
|
||||||
|
* Add a standalone email sending plugin by [@wu-clan](https://github.com/wu-clan) in [#769](https://github.com/fastapi-practices/fastapi_best_architecture/pull/769)
|
||||||
|
* Add i18n support for response message by [@downdawn](https://github.com/downdawn) in [#753](https://github.com/fastapi-practices/fastapi_best_architecture/pull/753)
|
||||||
|
* Update the menu title in SQL scripts by [@wu-clan](https://github.com/wu-clan) in [#770](https://github.com/fastapi-practices/fastapi_best_architecture/pull/770)
|
||||||
|
* Update the version number to 1.8.0 by [@wu-clan](https://github.com/wu-clan) in [#771](https://github.com/fastapi-practices/fastapi_best_architecture/pull/771)
|
||||||
|
|
||||||
|
## New Contributors
|
||||||
|
* [@IAseven](https://github.com/IAseven) made their first contribution in [#750](https://github.com/fastapi-practices/fastapi_best_architecture/pull/750)
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.7.0...v1.8.0
|
||||||
|
|
||||||
|
[Changes][v1.8.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.7.0"></a>
|
||||||
|
# [v1.7.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.7.0) - 2025-07-16
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update the changelog for v1.6.0 by [@wu-clan](https://github.com/wu-clan) in [#703](https://github.com/fastapi-practices/fastapi_best_architecture/pull/703)
|
||||||
|
* Update the CLI to be executed async by [@wu-clan](https://github.com/wu-clan) in [#704](https://github.com/fastapi-practices/fastapi_best_architecture/pull/704)
|
||||||
|
* Fix the code generation schema template by [@wu-clan](https://github.com/wu-clan) in [#706](https://github.com/fastapi-practices/fastapi_best_architecture/pull/706)
|
||||||
|
* Replace gunicorn deployment to granian by [@wu-clan](https://github.com/wu-clan) in [#705](https://github.com/fastapi-practices/fastapi_best_architecture/pull/705)
|
||||||
|
* Fix the code generation delete schema template by [@wu-clan](https://github.com/wu-clan) in [#708](https://github.com/fastapi-practices/fastapi_best_architecture/pull/708)
|
||||||
|
* Update the refresh token verify mechanism by [@wu-clan](https://github.com/wu-clan) in [#710](https://github.com/fastapi-practices/fastapi_best_architecture/pull/710)
|
||||||
|
* Update the reload excludes for CLI run by [@wu-clan](https://github.com/wu-clan) in [#709](https://github.com/fastapi-practices/fastapi_best_architecture/pull/709)
|
||||||
|
* Add CLI support for execute sql scripts by [@wu-clan](https://github.com/wu-clan) in [#711](https://github.com/fastapi-practices/fastapi_best_architecture/pull/711)
|
||||||
|
* Update the granian env to command params by [@wu-clan](https://github.com/wu-clan) in [#712](https://github.com/fastapi-practices/fastapi_best_architecture/pull/712)
|
||||||
|
* Update the middleware logging accuracy by [@wu-clan](https://github.com/wu-clan) in [#713](https://github.com/fastapi-practices/fastapi_best_architecture/pull/713)
|
||||||
|
* Update the log output default style by [@wu-clan](https://github.com/wu-clan) in [#714](https://github.com/fastapi-practices/fastapi_best_architecture/pull/714)
|
||||||
|
* Optimize the analysis of get plugins by [@wu-clan](https://github.com/wu-clan) in [#716](https://github.com/fastapi-practices/fastapi_best_architecture/pull/716)
|
||||||
|
* Simplify user permission database queries by [@wu-clan](https://github.com/wu-clan) in [#717](https://github.com/fastapi-practices/fastapi_best_architecture/pull/717)
|
||||||
|
* Update the CLI startup service mode by [@wu-clan](https://github.com/wu-clan) in [#718](https://github.com/fastapi-practices/fastapi_best_architecture/pull/718)
|
||||||
|
* Add support for celery dynamic tasks by [@wu-clan](https://github.com/wu-clan) in [#715](https://github.com/fastapi-practices/fastapi_best_architecture/pull/715)
|
||||||
|
* Fix the celery task scheduler query by [@wu-clan](https://github.com/wu-clan) in [#719](https://github.com/fastapi-practices/fastapi_best_architecture/pull/719)
|
||||||
|
* Update the celery task comment and name by [@wu-clan](https://github.com/wu-clan) in [#720](https://github.com/fastapi-practices/fastapi_best_architecture/pull/720)
|
||||||
|
* Optimize celery integrations and events by [@wu-clan](https://github.com/wu-clan) in [#721](https://github.com/fastapi-practices/fastapi_best_architecture/pull/721)
|
||||||
|
* Simplify celery task crontab config by [@wu-clan](https://github.com/wu-clan) in [#722](https://github.com/fastapi-practices/fastapi_best_architecture/pull/722)
|
||||||
|
* Delete the default value of schema enum data by [@wu-clan](https://github.com/wu-clan) in [#723](https://github.com/fastapi-practices/fastapi_best_architecture/pull/723)
|
||||||
|
* Fix the parsing of execution task params by [@wu-clan](https://github.com/wu-clan) in [#725](https://github.com/fastapi-practices/fastapi_best_architecture/pull/725)
|
||||||
|
* Bump granian from 2.4.0 to 2.4.2 by [@wu-clan](https://github.com/wu-clan) in [#727](https://github.com/fastapi-practices/fastapi_best_architecture/pull/727)
|
||||||
|
* Add CLI support for startup celery services by [@wu-clan](https://github.com/wu-clan) in [#724](https://github.com/fastapi-practices/fastapi_best_architecture/pull/724)
|
||||||
|
* Fix login and operation log clearing by [@wu-clan](https://github.com/wu-clan) in [#728](https://github.com/fastapi-practices/fastapi_best_architecture/pull/728)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.6.0...v1.7.0
|
||||||
|
|
||||||
|
[Changes][v1.7.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.6.0"></a>
|
||||||
|
# [v1.6.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.6.0) - 2025-06-30
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.5.2 by [@wu-clan](https://github.com/wu-clan) in [#690](https://github.com/fastapi-practices/fastapi_best_architecture/pull/690)
|
||||||
|
* Optimize dict create and update logic by [@wu-clan](https://github.com/wu-clan) in [#691](https://github.com/fastapi-practices/fastapi_best_architecture/pull/691)
|
||||||
|
* Fix the OAuth2 redirect route names by [@wu-clan](https://github.com/wu-clan) in [#693](https://github.com/fastapi-practices/fastapi_best_architecture/pull/693)
|
||||||
|
* Update the SQL to adapt frontend plugin by [@wu-clan](https://github.com/wu-clan) in [#694](https://github.com/fastapi-practices/fastapi_best_architecture/pull/694)
|
||||||
|
* Update the extension plugin config by [@wu-clan](https://github.com/wu-clan) in [#695](https://github.com/fastapi-practices/fastapi_best_architecture/pull/695)
|
||||||
|
* Add the test user to SQL scripts by [@wu-clan](https://github.com/wu-clan) in [#696](https://github.com/fastapi-practices/fastapi_best_architecture/pull/696)
|
||||||
|
* Add custom CLI for service startup by [@wu-clan](https://github.com/wu-clan) in [#697](https://github.com/fastapi-practices/fastapi_best_architecture/pull/697)
|
||||||
|
* Add CLI support for plugin install by [@wu-clan](https://github.com/wu-clan) in [#698](https://github.com/fastapi-practices/fastapi_best_architecture/pull/698)
|
||||||
|
* Update the help for CLI run worker by [@wu-clan](https://github.com/wu-clan) in [#699](https://github.com/fastapi-practices/fastapi_best_architecture/pull/699)
|
||||||
|
* Optimize the installation of plugin dependencies by [@wu-clan](https://github.com/wu-clan) in [#700](https://github.com/fastapi-practices/fastapi_best_architecture/pull/700)
|
||||||
|
* Update the Dockerfile to adapt latest code by [@wu-clan](https://github.com/wu-clan) in [#701](https://github.com/fastapi-practices/fastapi_best_architecture/pull/701)
|
||||||
|
* Update the version number to 1.6.0 by [@wu-clan](https://github.com/wu-clan) in [#702](https://github.com/fastapi-practices/fastapi_best_architecture/pull/702)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.2...v1.6.0
|
||||||
|
|
||||||
|
[Changes][v1.6.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.5.2"></a>
|
||||||
|
# [v1.5.2](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.5.2) - 2025-06-24
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.5.1 by [@wu-clan](https://github.com/wu-clan) in [#671](https://github.com/fastapi-practices/fastapi_best_architecture/pull/671)
|
||||||
|
* Fix some error class import by [@wu-clan](https://github.com/wu-clan) in [#672](https://github.com/fastapi-practices/fastapi_best_architecture/pull/672)
|
||||||
|
* Optimize routes to better align with RESTful by [@wu-clan](https://github.com/wu-clan) in [#673](https://github.com/fastapi-practices/fastapi_best_architecture/pull/673)
|
||||||
|
* Add the snowflake ID sql script by [@wu-clan](https://github.com/wu-clan) in [#675](https://github.com/fastapi-practices/fastapi_best_architecture/pull/675)
|
||||||
|
* Optimize token detection and caching logic by [@wu-clan](https://github.com/wu-clan) in [#677](https://github.com/fastapi-practices/fastapi_best_architecture/pull/677)
|
||||||
|
* Update cache cleanup for logout interface by [@wu-clan](https://github.com/wu-clan) in [#678](https://github.com/fastapi-practices/fastapi_best_architecture/pull/678)
|
||||||
|
* Add dictionary type and datas queries by [@wu-clan](https://github.com/wu-clan) in [#679](https://github.com/fastapi-practices/fastapi_best_architecture/pull/679)
|
||||||
|
* Optimize api with semantic HTTP status codes by [@downdawn](https://github.com/downdawn) in [#681](https://github.com/fastapi-practices/fastapi_best_architecture/pull/681)
|
||||||
|
* Fix the code with outdated system config by [@wu-clan](https://github.com/wu-clan) in [#683](https://github.com/fastapi-practices/fastapi_best_architecture/pull/683)
|
||||||
|
* Update dict data label column config by [@wu-clan](https://github.com/wu-clan) in [#684](https://github.com/fastapi-practices/fastapi_best_architecture/pull/684)
|
||||||
|
* Update the init test data for SQL scripts by [@wu-clan](https://github.com/wu-clan) in [#685](https://github.com/fastapi-practices/fastapi_best_architecture/pull/685)
|
||||||
|
* Simplify custom response status codes by [@wu-clan](https://github.com/wu-clan) in [#686](https://github.com/fastapi-practices/fastapi_best_architecture/pull/686)
|
||||||
|
* Optimize the zip plug-in file name parsing by [@wu-clan](https://github.com/wu-clan) in [#687](https://github.com/fastapi-practices/fastapi_best_architecture/pull/687)
|
||||||
|
* Add built-in plugin missing files by [@wu-clan](https://github.com/wu-clan) in [#688](https://github.com/fastapi-practices/fastapi_best_architecture/pull/688)
|
||||||
|
* Update the dict pagination query parameters by [@wu-clan](https://github.com/wu-clan) in [#689](https://github.com/fastapi-practices/fastapi_best_architecture/pull/689)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.1...v1.5.2
|
||||||
|
|
||||||
|
[Changes][v1.5.2]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.5.1"></a>
|
||||||
|
# [v1.5.1](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.5.1) - 2025-06-16
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.5.0 by [@wu-clan](https://github.com/wu-clan) in [#664](https://github.com/fastapi-practices/fastapi_best_architecture/pull/664)
|
||||||
|
* Fix the sidebar menu type filtering by [@wu-clan](https://github.com/wu-clan) in [#667](https://github.com/fastapi-practices/fastapi_best_architecture/pull/667)
|
||||||
|
* Bump sqlalchemy crud plus version to 1.10.0 by [@wu-clan](https://github.com/wu-clan) in [#668](https://github.com/fastapi-practices/fastapi_best_architecture/pull/668)
|
||||||
|
* Fix the postgresql sql script syntax error by [@downdawn](https://github.com/downdawn) in [#669](https://github.com/fastapi-practices/fastapi_best_architecture/pull/669)
|
||||||
|
* Add Initial Snowflake ID Support by [@downdawn](https://github.com/downdawn) in [#670](https://github.com/fastapi-practices/fastapi_best_architecture/pull/670)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.0...v1.5.1
|
||||||
|
|
||||||
|
[Changes][v1.5.1]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.5.0"></a>
|
||||||
|
# [v1.5.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.5.0) - 2025-06-09
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.4.3 by [@wu-clan](https://github.com/wu-clan) in [#651](https://github.com/fastapi-practices/fastapi_best_architecture/pull/651)
|
||||||
|
* Update OAuth2 callback interface return by [@wu-clan](https://github.com/wu-clan) in [#653](https://github.com/fastapi-practices/fastapi_best_architecture/pull/653)
|
||||||
|
* Update user email and phone operation logic by [@wu-clan](https://github.com/wu-clan) in [#654](https://github.com/fastapi-practices/fastapi_best_architecture/pull/654)
|
||||||
|
* Simplify OAuth2 model and optimize auth service by [@wu-clan](https://github.com/wu-clan) in [#655](https://github.com/fastapi-practices/fastapi_best_architecture/pull/655)
|
||||||
|
* Add OAuth2 user to auto bind a role by [@wu-clan](https://github.com/wu-clan) in [#656](https://github.com/fastapi-practices/fastapi_best_architecture/pull/656)
|
||||||
|
* Update data scope and rule to m2m by [@wu-clan](https://github.com/wu-clan) in [#657](https://github.com/fastapi-practices/fastapi_best_architecture/pull/657)
|
||||||
|
* Update code generate interface permission by [@wu-clan](https://github.com/wu-clan) in [#658](https://github.com/fastapi-practices/fastapi_best_architecture/pull/658)
|
||||||
|
* Update the plugin download interface permission by [@wu-clan](https://github.com/wu-clan) in [#659](https://github.com/fastapi-practices/fastapi_best_architecture/pull/659)
|
||||||
|
* Update auth failed default status code by [@wu-clan](https://github.com/wu-clan) in [#660](https://github.com/fastapi-practices/fastapi_best_architecture/pull/660)
|
||||||
|
* Update menu sort in init test sql by [@wu-clan](https://github.com/wu-clan) in [#661](https://github.com/fastapi-practices/fastapi_best_architecture/pull/661)
|
||||||
|
* Add data permission in init test sql by [@wu-clan](https://github.com/wu-clan) in [#662](https://github.com/fastapi-practices/fastapi_best_architecture/pull/662)
|
||||||
|
* Update the version to 1.5.0 by [@wu-clan](https://github.com/wu-clan) in [#663](https://github.com/fastapi-practices/fastapi_best_architecture/pull/663)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.3...v1.5.0
|
||||||
|
|
||||||
|
[Changes][v1.5.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.4.3"></a>
|
||||||
|
# [v1.4.3](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.4.3) - 2025-06-02
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.4.2 by [@wu-clan](https://github.com/wu-clan) in [#639](https://github.com/fastapi-practices/fastapi_best_architecture/pull/639)
|
||||||
|
* Fix the role update business variables by [@wu-clan](https://github.com/wu-clan) in [#640](https://github.com/fastapi-practices/fastapi_best_architecture/pull/640)
|
||||||
|
* Fix the menu delete interface arg description by [@wu-clan](https://github.com/wu-clan) in [#641](https://github.com/fastapi-practices/fastapi_best_architecture/pull/641)
|
||||||
|
* Fix the filter of query all menus by [@wu-clan](https://github.com/wu-clan) in [#642](https://github.com/fastapi-practices/fastapi_best_architecture/pull/642)
|
||||||
|
* Refactor routes to better align with RESTful by [@wu-clan](https://github.com/wu-clan) in [#645](https://github.com/fastapi-practices/fastapi_best_architecture/pull/645)
|
||||||
|
* Update the server startup time to string by [@wu-clan](https://github.com/wu-clan) in [#646](https://github.com/fastapi-practices/fastapi_best_architecture/pull/646)
|
||||||
|
* Add get all data scope rules interface by [@wu-clan](https://github.com/wu-clan) in [#647](https://github.com/fastapi-practices/fastapi_best_architecture/pull/647)
|
||||||
|
* Add data permission condition for filter data by [@wu-clan](https://github.com/wu-clan) in [#648](https://github.com/fastapi-practices/fastapi_best_architecture/pull/648)
|
||||||
|
* Update default value for role filter scopes by [@wu-clan](https://github.com/wu-clan) in [#649](https://github.com/fastapi-practices/fastapi_best_architecture/pull/649)
|
||||||
|
* Fix data permission condition for filter data by [@wu-clan](https://github.com/wu-clan) in [#650](https://github.com/fastapi-practices/fastapi_best_architecture/pull/650)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.2...v1.4.3
|
||||||
|
|
||||||
|
[Changes][v1.4.3]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.4.2"></a>
|
||||||
|
# [v1.4.2](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.4.2) - 2025-05-29
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.4.1 by [@wu-clan](https://github.com/wu-clan) in [#630](https://github.com/fastapi-practices/fastapi_best_architecture/pull/630)
|
||||||
|
* Update non-linked sidebar support by [@wu-clan](https://github.com/wu-clan) in [#633](https://github.com/fastapi-practices/fastapi_best_architecture/pull/633)
|
||||||
|
* Update the captcha invalidation error class by [@wu-clan](https://github.com/wu-clan) in [#634](https://github.com/fastapi-practices/fastapi_best_architecture/pull/634)
|
||||||
|
* Optimize role-related data processing performance by [@wu-clan](https://github.com/wu-clan) in [#635](https://github.com/fastapi-practices/fastapi_best_architecture/pull/635)
|
||||||
|
* Optimize install and build of plugin zip by [@wu-clan](https://github.com/wu-clan) in [#636](https://github.com/fastapi-practices/fastapi_best_architecture/pull/636)
|
||||||
|
* Fix auto-increment id for postgres init data by [@huyuwei1996](https://github.com/huyuwei1996) in [#632](https://github.com/fastapi-practices/fastapi_best_architecture/pull/632)
|
||||||
|
* Fix: prevent overwriting existing init files in code generator by [@lin-wu-1990](https://github.com/lin-wu-1990) in [#637](https://github.com/fastapi-practices/fastapi_best_architecture/pull/637)
|
||||||
|
* Simplify the user info update business by [@wu-clan](https://github.com/wu-clan) in [#638](https://github.com/fastapi-practices/fastapi_best_architecture/pull/638)
|
||||||
|
|
||||||
|
## New Contributors
|
||||||
|
* [@lin-wu-1990](https://github.com/lin-wu-1990) made their first contribution in [#637](https://github.com/fastapi-practices/fastapi_best_architecture/pull/637)
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.1...v1.4.2
|
||||||
|
|
||||||
|
[Changes][v1.4.2]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.4.1"></a>
|
||||||
|
# [v1.4.1](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.4.1) - 2025-05-25
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.4.0 by [@wu-clan](https://github.com/wu-clan) in [#621](https://github.com/fastapi-practices/fastapi_best_architecture/pull/621)
|
||||||
|
* Update the menu path and type columns by [@wu-clan](https://github.com/wu-clan) in [#622](https://github.com/fastapi-practices/fastapi_best_architecture/pull/622)
|
||||||
|
* Add the deepwiki badge to README by [@wu-clan](https://github.com/wu-clan) in [#623](https://github.com/fastapi-practices/fastapi_best_architecture/pull/623)
|
||||||
|
* Refactor the system token to online users by [@wu-clan](https://github.com/wu-clan) in [#624](https://github.com/fastapi-practices/fastapi_best_architecture/pull/624)
|
||||||
|
* Update the token check for logout interface by [@wu-clan](https://github.com/wu-clan) in [#625](https://github.com/fastapi-practices/fastapi_best_architecture/pull/625)
|
||||||
|
* Update the token decode for logout interface by [@wu-clan](https://github.com/wu-clan) in [#629](https://github.com/fastapi-practices/fastapi_best_architecture/pull/629)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.0...v1.4.1
|
||||||
|
|
||||||
|
[Changes][v1.4.1]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.4.0"></a>
|
||||||
|
# [v1.4.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.4.0) - 2025-05-22
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.3.0 by [@wu-clan](https://github.com/wu-clan) in [#605](https://github.com/fastapi-practices/fastapi_best_architecture/pull/605)
|
||||||
|
* Add new plugin status check interface by [@wu-clan](https://github.com/wu-clan) in [#606](https://github.com/fastapi-practices/fastapi_best_architecture/pull/606)
|
||||||
|
* Update the new plugin status to changed by [@wu-clan](https://github.com/wu-clan) in [#607](https://github.com/fastapi-practices/fastapi_best_architecture/pull/607)
|
||||||
|
* Fix the task result schema param type by [@wu-clan](https://github.com/wu-clan) in [#611](https://github.com/fastapi-practices/fastapi_best_architecture/pull/611)
|
||||||
|
* Fix the plugin status update logic by [@wu-clan](https://github.com/wu-clan) in [#613](https://github.com/fastapi-practices/fastapi_best_architecture/pull/613)
|
||||||
|
* Update uninstall and build plugin api method by [@wu-clan](https://github.com/wu-clan) in [#614](https://github.com/fastapi-practices/fastapi_best_architecture/pull/614)
|
||||||
|
* Fix non-asyncio nested async IO by [@wu-clan](https://github.com/wu-clan) in [#610](https://github.com/fastapi-practices/fastapi_best_architecture/pull/610)
|
||||||
|
* Update the build plugin api params by [@wu-clan](https://github.com/wu-clan) in [#615](https://github.com/fastapi-practices/fastapi_best_architecture/pull/615)
|
||||||
|
* Update uv installation in docker deploy by [@wu-clan](https://github.com/wu-clan) in [#619](https://github.com/fastapi-practices/fastapi_best_architecture/pull/619)
|
||||||
|
* Update the OAuth2 module to plugin by [@wu-clan](https://github.com/wu-clan) in [#620](https://github.com/fastapi-practices/fastapi_best_architecture/pull/620)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.3.0...v1.4.0
|
||||||
|
|
||||||
|
[Changes][v1.4.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.3.0"></a>
|
||||||
|
# [v1.3.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.3.0) - 2025-05-16
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.2.0 by [@wu-clan](https://github.com/wu-clan) in [#598](https://github.com/fastapi-practices/fastapi_best_architecture/pull/598)
|
||||||
|
* Simplify apps and plugins config method by [@wu-clan](https://github.com/wu-clan) in [#600](https://github.com/fastapi-practices/fastapi_best_architecture/pull/600)
|
||||||
|
* Add plugin info config and interfaces by [@wu-clan](https://github.com/wu-clan) in [#601](https://github.com/fastapi-practices/fastapi_best_architecture/pull/601)
|
||||||
|
* Fix the fastapi cli startup event loop by [@wu-clan](https://github.com/wu-clan) in [#602](https://github.com/fastapi-practices/fastapi_best_architecture/pull/602)
|
||||||
|
* Optimize the zip plugin install logic by [@wu-clan](https://github.com/wu-clan) in [#603](https://github.com/fastapi-practices/fastapi_best_architecture/pull/603)
|
||||||
|
* Update the casbin RBAC module path by [@wu-clan](https://github.com/wu-clan) in [#604](https://github.com/fastapi-practices/fastapi_best_architecture/pull/604)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.2.0...v1.3.0
|
||||||
|
|
||||||
|
[Changes][v1.3.0]
|
||||||
|
|
||||||
|
|
||||||
|
<a id="v1.2.0"></a>
|
||||||
|
# [v1.2.0](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.2.0) - 2025-05-01
|
||||||
|
|
||||||
|
## What's Changed
|
||||||
|
* Update changelog for v1.1.2 by [@wu-clan](https://github.com/wu-clan) in [#589](https://github.com/fastapi-practices/fastapi_best_architecture/pull/589)
|
||||||
|
* Update code generator table columns by [@wu-clan](https://github.com/wu-clan) in [#590](https://github.com/fastapi-practices/fastapi_best_architecture/pull/590)
|
||||||
|
* Update the default RBAC solution by [@wu-clan](https://github.com/wu-clan) in [#593](https://github.com/fastapi-practices/fastapi_best_architecture/pull/593)
|
||||||
|
* Optimize the server information retrieval by [@wu-clan](https://github.com/wu-clan) in [#595](https://github.com/fastapi-practices/fastapi_best_architecture/pull/595)
|
||||||
|
* Refactor the data rule to scope rule by [@wu-clan](https://github.com/wu-clan) in [#596](https://github.com/fastapi-practices/fastapi_best_architecture/pull/596)
|
||||||
|
* Update the SQL script for creat tables by [@wu-clan](https://github.com/wu-clan) in [#597](https://github.com/fastapi-practices/fastapi_best_architecture/pull/597)
|
||||||
|
|
||||||
|
|
||||||
|
**Full Changelog**: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.2...v1.2.0
|
||||||
|
|
||||||
|
[Changes][v1.2.0]
|
||||||
|
|
||||||
|
|
||||||
<a id="v1.1.2"></a>
|
<a id="v1.1.2"></a>
|
||||||
# [v1.1.2](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.1.2) - 2025-04-23
|
# [v1.1.2](https://github.com/fastapi-practices/fastapi_best_architecture/releases/tag/v1.1.2) - 2025-04-23
|
||||||
|
|
||||||
@@ -477,6 +758,18 @@
|
|||||||
[Changes][v1.0.0]
|
[Changes][v1.0.0]
|
||||||
|
|
||||||
|
|
||||||
|
[v1.8.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.7.0...v1.8.0
|
||||||
|
[v1.7.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.6.0...v1.7.0
|
||||||
|
[v1.6.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.2...v1.6.0
|
||||||
|
[v1.5.2]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.1...v1.5.2
|
||||||
|
[v1.5.1]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.5.0...v1.5.1
|
||||||
|
[v1.5.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.3...v1.5.0
|
||||||
|
[v1.4.3]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.2...v1.4.3
|
||||||
|
[v1.4.2]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.1...v1.4.2
|
||||||
|
[v1.4.1]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.4.0...v1.4.1
|
||||||
|
[v1.4.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.3.0...v1.4.0
|
||||||
|
[v1.3.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.2.0...v1.3.0
|
||||||
|
[v1.2.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.2...v1.2.0
|
||||||
[v1.1.2]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.1...v1.1.2
|
[v1.1.2]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.1...v1.1.2
|
||||||
[v1.1.1]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.0...v1.1.1
|
[v1.1.1]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.1.0...v1.1.1
|
||||||
[v1.1.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.0.5...v1.1.0
|
[v1.1.0]: https://github.com/fastapi-practices/fastapi_best_architecture/compare/v1.0.5...v1.1.0
|
||||||
|
|||||||
+10
-18
@@ -2,8 +2,7 @@
|
|||||||
ARG SERVER_TYPE=fastapi_server
|
ARG SERVER_TYPE=fastapi_server
|
||||||
|
|
||||||
# === Python environment from uv ===
|
# === Python environment from uv ===
|
||||||
FROM python:3.10-slim AS builder
|
FROM ghcr.io/astral-sh/uv:python3.10-bookworm-slim AS builder
|
||||||
COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
|
|
||||||
|
|
||||||
# Used for build Python packages
|
# Used for build Python packages
|
||||||
RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list.d/debian.sources \
|
RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list.d/debian.sources \
|
||||||
@@ -11,6 +10,10 @@ RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list.d/debi
|
|||||||
&& apt-get install -y --no-install-recommends gcc python3-dev \
|
&& apt-get install -y --no-install-recommends gcc python3-dev \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
COPY . /fba
|
||||||
|
|
||||||
|
WORKDIR /fba
|
||||||
|
|
||||||
# Configure uv environment
|
# Configure uv environment
|
||||||
ENV UV_COMPILE_BYTECODE=1 \
|
ENV UV_COMPILE_BYTECODE=1 \
|
||||||
UV_NO_CACHE=1 \
|
UV_NO_CACHE=1 \
|
||||||
@@ -19,49 +22,38 @@ ENV UV_COMPILE_BYTECODE=1 \
|
|||||||
|
|
||||||
# Install dependencies with cache
|
# Install dependencies with cache
|
||||||
RUN --mount=type=cache,target=/root/.cache/uv \
|
RUN --mount=type=cache,target=/root/.cache/uv \
|
||||||
--mount=type=bind,source=uv.lock,target=uv.lock \
|
|
||||||
--mount=type=bind,source=pyproject.toml,target=pyproject.toml \
|
|
||||||
uv sync --frozen --no-default-groups --group server
|
uv sync --frozen --no-default-groups --group server
|
||||||
|
|
||||||
# === Runtime base server image ===
|
# === Runtime base server image ===
|
||||||
FROM python:3.10-slim AS base_server
|
FROM python:3.10-slim AS base_server
|
||||||
|
|
||||||
SHELL ["/bin/bash", "-c"]
|
|
||||||
|
|
||||||
RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list.d/debian.sources \
|
RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list.d/debian.sources \
|
||||||
&& apt-get update \
|
&& apt-get update \
|
||||||
&& apt-get install -y --no-install-recommends supervisor \
|
&& apt-get install -y --no-install-recommends supervisor \
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
COPY . /fba
|
COPY --from=builder /fba /fba
|
||||||
|
|
||||||
COPY --from=builder /usr/local /usr/local
|
COPY --from=builder /usr/local /usr/local
|
||||||
|
|
||||||
# Install plugin dependencies
|
COPY deploy/backend/supervisord.conf /etc/supervisor/supervisord.conf
|
||||||
WORKDIR /fba
|
|
||||||
ENV PYTHONPATH=/fba
|
WORKDIR /fba/backend
|
||||||
RUN python3 backend/scripts/init_plugin.py
|
|
||||||
|
|
||||||
# === FastAPI server image ===
|
# === FastAPI server image ===
|
||||||
FROM base_server AS fastapi_server
|
FROM base_server AS fastapi_server
|
||||||
|
|
||||||
WORKDIR /fba
|
|
||||||
|
|
||||||
COPY deploy/backend/supervisord.conf /etc/supervisor/supervisord.conf
|
|
||||||
COPY deploy/backend/fastapi_server.conf /etc/supervisor/conf.d/
|
COPY deploy/backend/fastapi_server.conf /etc/supervisor/conf.d/
|
||||||
|
|
||||||
RUN mkdir -p /var/log/fastapi_server
|
RUN mkdir -p /var/log/fastapi_server
|
||||||
|
|
||||||
EXPOSE 8001
|
EXPOSE 8001
|
||||||
|
|
||||||
CMD ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port","8000"]
|
CMD ["/usr/local/bin/granian", "main:app", "--interface", "asgi", "--host", "0.0.0.0", "--port","8000"]
|
||||||
|
|
||||||
# === Celery server image ===
|
# === Celery server image ===
|
||||||
FROM base_server AS celery
|
FROM base_server AS celery
|
||||||
|
|
||||||
WORKDIR /fba/backend/
|
|
||||||
|
|
||||||
COPY deploy/backend/supervisord.conf /etc/supervisor/supervisord.conf
|
|
||||||
COPY deploy/backend/celery.conf /etc/supervisor/conf.d/
|
COPY deploy/backend/celery.conf /etc/supervisor/conf.d/
|
||||||
|
|
||||||
RUN mkdir -p /var/log/celery
|
RUN mkdir -p /var/log/celery
|
||||||
|
|||||||
@@ -4,11 +4,9 @@
|
|||||||
|
|
||||||
# FastAPI Best Architecture
|
# FastAPI Best Architecture
|
||||||
|
|
||||||
English | [简体中文](./README.zh-CN.md)
|
|
||||||
|
|
||||||
Enterprise-level backend architecture solution
|
Enterprise-level backend architecture solution
|
||||||
|
|
||||||
**🔥Continuously updated and maintained🔥**
|
English | [简体中文](./README.zh-CN.md)
|
||||||
|
|
||||||
[](https://github.com/fastapi-practices/fastapi_best_architecture/blob/master/LICENSE)
|
[](https://github.com/fastapi-practices/fastapi_best_architecture/blob/master/LICENSE)
|
||||||
[](https://www.python.org/downloads/)
|
[](https://www.python.org/downloads/)
|
||||||
@@ -21,12 +19,10 @@ Enterprise-level backend architecture solution
|
|||||||

|

|
||||||
[](https://discord.com/invite/yNN3wTbVAC)
|
[](https://discord.com/invite/yNN3wTbVAC)
|
||||||

|

|
||||||
|
[](https://deepwiki.com/fastapi-practices/fastapi_best_architecture)
|
||||||
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
> [!NOTE]
|
|
||||||
> This repository as a template library open to any person or enterprise can be used for free!
|
|
||||||
|
|
||||||
## Pseudo 3-tier architecture
|
## Pseudo 3-tier architecture
|
||||||
|
|
||||||
The mvc architecture is a common design pattern in python web, but the 3-tier architecture is even more fascinating
|
The mvc architecture is a common design pattern in python web, but the 3-tier architecture is even more fascinating
|
||||||
@@ -45,43 +41,7 @@ pattern, use templates to transform it to your heart's content!
|
|||||||
| data access | dao / mapper | crud |
|
| data access | dao / mapper | crud |
|
||||||
| model | model / entity | model |
|
| model | model / entity | model |
|
||||||
|
|
||||||
## Features
|
## Help
|
||||||
|
|
||||||
- [x] Global FastAPI PEP 593 Annotated parameter style
|
|
||||||
- [x] Comprehensive async/await + asgiref asynchronous design
|
|
||||||
- [x] Adheres to RESTful API specifications
|
|
||||||
- [x] Uses SQLAlchemy 2.0 with new syntax
|
|
||||||
- [x] Uses Pydantic v2 version
|
|
||||||
- [x] Implements role-menu RBAC access control
|
|
||||||
- [x] Integrates Casbin RBAC access control
|
|
||||||
- [x] Supports Celery asynchronous tasks
|
|
||||||
- [x] Custom-developed JWT authentication middleware
|
|
||||||
- [x] Supports global custom time zones
|
|
||||||
- [x] Supports Docker / Docker-compose deployment
|
|
||||||
- [x] Integrates Pytest unit testing
|
|
||||||
|
|
||||||
## Built-in Functions
|
|
||||||
|
|
||||||
- [x] User Management: Assign roles and permissions
|
|
||||||
- [x] Department Management: Configure organizational structure (company, department, team, etc.)
|
|
||||||
- [x] Menu Management: Set up menus and button-level permissions
|
|
||||||
- [x] Role Management: Configure roles, assign menus and permissions
|
|
||||||
- [x] Dictionary Management: Maintain common parameters and configurations
|
|
||||||
- [x] Parameter Management: Dynamically configure commonly used system parameters
|
|
||||||
- [x] Notification Announcements: Publish and maintain system notification and announcement information
|
|
||||||
- [x] Token Management: Detect online status, support forced logout
|
|
||||||
- [x] Multi-device Login: Support one-click switching between multi-device login modes
|
|
||||||
- [x] OAuth 2.0: Built-in custom-developed OAuth 2.0 authorization login
|
|
||||||
- [x] Plugin System: Hot-swappable plugin design to reduce coupling
|
|
||||||
- [x] Scheduled Tasks: Support scheduled, asynchronous tasks, and function calls
|
|
||||||
- [x] Code Generation: Automatically generate code with preview, write, and download support
|
|
||||||
- [x] Operation Logs: Record and query normal and abnormal operations
|
|
||||||
- [x] Login Logs: Record and query normal and abnormal logins
|
|
||||||
- [x] Cache Monitoring: Query system cache information and command statistics
|
|
||||||
- [x] Service Monitoring: View server hardware information and status
|
|
||||||
- [x] API Documentation: Automatically generate online interactive API documentation
|
|
||||||
|
|
||||||
## Development and deployment
|
|
||||||
|
|
||||||
For more details, please check
|
For more details, please check
|
||||||
the [official documentation](https://fastapi-practices.github.io/fastapi_best_architecture_docs/)
|
the [official documentation](https://fastapi-practices.github.io/fastapi_best_architecture_docs/)
|
||||||
|
|||||||
+3
-43
@@ -4,11 +4,9 @@
|
|||||||
|
|
||||||
# FastAPI Best Architecture
|
# FastAPI Best Architecture
|
||||||
|
|
||||||
简体中文 | [English](./README.md)
|
|
||||||
|
|
||||||
企业级后端架构解决方案
|
企业级后端架构解决方案
|
||||||
|
|
||||||
**🔥持续更新维护中🔥**
|
简体中文 | [English](./README.md)
|
||||||
|
|
||||||
[](https://github.com/fastapi-practices/fastapi_best_architecture/blob/master/LICENSE)
|
[](https://github.com/fastapi-practices/fastapi_best_architecture/blob/master/LICENSE)
|
||||||
[](https://www.python.org/downloads/)
|
[](https://www.python.org/downloads/)
|
||||||
@@ -21,12 +19,10 @@
|
|||||||

|

|
||||||
[](https://discord.com/invite/yNN3wTbVAC)
|
[](https://discord.com/invite/yNN3wTbVAC)
|
||||||

|

|
||||||
|
[](https://deepwiki.com/fastapi-practices/fastapi_best_architecture)
|
||||||
|
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
> [!NOTE]
|
|
||||||
> 此仓库作为模板库公开,任何人或企业均可免费使用!
|
|
||||||
|
|
||||||
## 伪三层架构
|
## 伪三层架构
|
||||||
|
|
||||||
mvc 架构作为常规设计模式,在 python web 中也很常见,但是三层架构更令人着迷
|
mvc 架构作为常规设计模式,在 python web 中也很常见,但是三层架构更令人着迷
|
||||||
@@ -43,43 +39,7 @@ mvc 架构作为常规设计模式,在 python web 中也很常见,但是三
|
|||||||
| 数据访问 | dao / mapper | crud |
|
| 数据访问 | dao / mapper | crud |
|
||||||
| 模型 | model / entity | model |
|
| 模型 | model / entity | model |
|
||||||
|
|
||||||
## 特性
|
## 帮助
|
||||||
|
|
||||||
- [x] 全局 FastAPI PEP 593 Annotated 参数风格
|
|
||||||
- [x] 全面 async/await + asgiref 异步设计
|
|
||||||
- [x] 遵循 RESTful API 规范
|
|
||||||
- [x] 使用 SQLAlchemy 2.0 全新语法
|
|
||||||
- [x] 使用 Pydantic v2 版本
|
|
||||||
- [x] 实现角色菜单 RBAC 访问控制
|
|
||||||
- [x] 集成 Casbin RBAC 访问控制
|
|
||||||
- [x] 支持 Celery 异步任务
|
|
||||||
- [x] 自研 JWT 认证中间件
|
|
||||||
- [x] 支持全局自定义时间时区
|
|
||||||
- [x] 支持 Docker / Docker-compose 部署
|
|
||||||
- [x] 集成 Pytest 单元测试
|
|
||||||
|
|
||||||
## 内置功能
|
|
||||||
|
|
||||||
- [x] 用户管理:分配角色和权限
|
|
||||||
- [x] 部门管理:配置组织架构(公司、部门、小组等)
|
|
||||||
- [x] 菜单管理:设置菜单及按钮级权限
|
|
||||||
- [x] 角色管理:配置角色、分配菜单和权限
|
|
||||||
- [x] 字典管理:维护常用参数和配置
|
|
||||||
- [x] 参数管理:系统常用参数动态配置
|
|
||||||
- [x] 通知公告:发布和维护系统通知公告信息
|
|
||||||
- [x] 令牌管理:检测在线状态,支持强制下线
|
|
||||||
- [x] 多端登录:支持一键切换多端登录模式
|
|
||||||
- [x] OAuth 2.0:内置自研 OAuth 2.0 授权登录
|
|
||||||
- [x] 插件系统:热插拔插件设计,降低耦合
|
|
||||||
- [x] 定时任务:支持定时,异步任务及函数调用
|
|
||||||
- [x] 代码生成:自动生成代码,支持预览、写入和下载
|
|
||||||
- [x] 操作日志:记录和查询正常和异常操作
|
|
||||||
- [x] 登录日志:记录和查询正常和异常登录
|
|
||||||
- [x] 缓存监控:查询系统缓存信息和命令统计
|
|
||||||
- [x] 服务监控:查看服务器硬件信息和状态
|
|
||||||
- [x] 接口文档:自动生成在线交互式 API 文档
|
|
||||||
|
|
||||||
## 开发部署
|
|
||||||
|
|
||||||
更多详情请查看 [官方文档](https://fastapi-practices.github.io/fastapi_best_architecture_docs/)
|
更多详情请查看 [官方文档](https://fastapi-practices.github.io/fastapi_best_architecture_docs/)
|
||||||
|
|
||||||
|
|||||||
+16
-13
@@ -1,4 +1,4 @@
|
|||||||
# Env: dev、pro
|
# Env
|
||||||
ENVIRONMENT='dev'
|
ENVIRONMENT='dev'
|
||||||
# Database
|
# Database
|
||||||
DATABASE_TYPE='mysql'
|
DATABASE_TYPE='mysql'
|
||||||
@@ -15,18 +15,21 @@ REDIS_DATABASE=0
|
|||||||
TOKEN_SECRET_KEY='1VkVF75nsNABBjK_7-qz7GtzNy3AMvktc9TCPwKczCk'
|
TOKEN_SECRET_KEY='1VkVF75nsNABBjK_7-qz7GtzNy3AMvktc9TCPwKczCk'
|
||||||
# Opera Log
|
# Opera Log
|
||||||
OPERA_LOG_ENCRYPT_SECRET_KEY='d77b25790a804c2b4a339dd0207941e4cefa5751935a33735bc73bb7071a005b'
|
OPERA_LOG_ENCRYPT_SECRET_KEY='d77b25790a804c2b4a339dd0207941e4cefa5751935a33735bc73bb7071a005b'
|
||||||
# Admin
|
# [ App ] task
|
||||||
# OAuth2
|
|
||||||
OAUTH2_GITHUB_CLIENT_ID='test'
|
|
||||||
OAUTH2_GITHUB_CLIENT_SECRET='test'
|
|
||||||
OAUTH2_LINUX_DO_CLIENT_ID='test'
|
|
||||||
OAUTH2_LINUX_DO_CLIENT_SECRET='test'
|
|
||||||
# Task
|
|
||||||
# Celery
|
# Celery
|
||||||
CELERY_BROKER_REDIS_DATABASE=1
|
CELERY_BROKER_REDIS_DATABASE=1
|
||||||
CELERY_BACKEND_REDIS_DATABASE=2
|
|
||||||
# Rabbitmq
|
# Rabbitmq
|
||||||
RABBITMQ_HOST='127.0.0.1'
|
CELERY_RABBITMQ_HOST='127.0.0.1'
|
||||||
RABBITMQ_PORT=5672
|
CELERY_RABBITMQ_PORT=5672
|
||||||
RABBITMQ_USERNAME='guest'
|
CELERY_RABBITMQ_USERNAME='guest'
|
||||||
RABBITMQ_PASSWORD='guest'
|
CELERY_RABBITMQ_PASSWORD='guest'
|
||||||
|
# [ Plugin ] oauth2
|
||||||
|
OAUTH2_GITHUB_CLIENT_ID='test'
|
||||||
|
OAUTH2_GITHUB_CLIENT_SECRET='test'
|
||||||
|
OAUTH2_GOOGLE_CLIENT_ID='test'
|
||||||
|
OAUTH2_GOOGLE_CLIENT_SECRET='test'
|
||||||
|
OAUTH2_LINUX_DO_CLIENT_ID='test'
|
||||||
|
OAUTH2_LINUX_DO_CLIENT_SECRET='test'
|
||||||
|
# [ Plugin ] email
|
||||||
|
EMAIL_USERNAME=''
|
||||||
|
EMAIL_PASSWORD=''
|
||||||
|
|||||||
+1
-1
@@ -72,7 +72,7 @@
|
|||||||
>
|
>
|
||||||
> It is recommended to execute under the backend directory, and chmod authorization may be required
|
> It is recommended to execute under the backend directory, and chmod authorization may be required
|
||||||
|
|
||||||
- `pre_start.sh`: Perform automatic database migration and create database tables
|
- `pre_start.sh`: Perform automatic database migration
|
||||||
|
|
||||||
- `celery-start.sh`: For celery docker script, implementation is not recommended
|
- `celery-start.sh`: For celery docker script, implementation is not recommended
|
||||||
|
|
||||||
|
|||||||
@@ -1,2 +1,9 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
from backend.common.i18n import i18n
|
||||||
|
|
||||||
|
__version__ = '1.8.0'
|
||||||
|
|
||||||
|
|
||||||
|
# 初始化 i18n
|
||||||
|
i18n.load_locales()
|
||||||
|
|||||||
+4
-12
@@ -3,7 +3,6 @@
|
|||||||
# ruff: noqa: F403, F401, I001, RUF100
|
# ruff: noqa: F403, F401, I001, RUF100
|
||||||
import asyncio
|
import asyncio
|
||||||
import os
|
import os
|
||||||
import sys
|
|
||||||
from logging.config import fileConfig
|
from logging.config import fileConfig
|
||||||
|
|
||||||
from alembic import context
|
from alembic import context
|
||||||
@@ -11,23 +10,16 @@ from sqlalchemy import pool
|
|||||||
from sqlalchemy.engine import Connection
|
from sqlalchemy.engine import Connection
|
||||||
from sqlalchemy.ext.asyncio import async_engine_from_config
|
from sqlalchemy.ext.asyncio import async_engine_from_config
|
||||||
|
|
||||||
sys.path.append('../')
|
from backend.app import get_app_models
|
||||||
|
|
||||||
from backend.common.model import MappedBase
|
from backend.common.model import MappedBase
|
||||||
from backend.core import path_conf
|
from backend.core import path_conf
|
||||||
from backend.database.db import SQLALCHEMY_DATABASE_URL
|
from backend.database.db import SQLALCHEMY_DATABASE_URL
|
||||||
from backend.plugin.tools import get_plugin_models
|
from backend.plugin.tools import get_plugin_models
|
||||||
|
|
||||||
# import your new model here
|
# import models
|
||||||
from backend.app.admin.model import * # noqa: F401
|
for cls in get_app_models() + get_plugin_models():
|
||||||
from backend.plugin.code_generator.model import * # noqa: F401
|
|
||||||
|
|
||||||
# import plugin model
|
|
||||||
for cls in get_plugin_models():
|
|
||||||
class_name = cls.__name__
|
class_name = cls.__name__
|
||||||
if class_name in globals():
|
if class_name not in globals():
|
||||||
print(f'\nWarning: Class "{class_name}" already exists in global namespace.')
|
|
||||||
else:
|
|
||||||
globals()[class_name] = cls
|
globals()[class_name] = cls
|
||||||
|
|
||||||
if not os.path.exists(path_conf.ALEMBIC_VERSION_DIR):
|
if not os.path.exists(path_conf.ALEMBIC_VERSION_DIR):
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ Create Date: ${create_date}
|
|||||||
"""
|
"""
|
||||||
from alembic import op
|
from alembic import op
|
||||||
import sqlalchemy as sa
|
import sqlalchemy as sa
|
||||||
|
import backend.common.model
|
||||||
${imports if imports else ""}
|
${imports if imports else ""}
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
# revision identifiers, used by Alembic.
|
||||||
|
|||||||
@@ -1,2 +1,35 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
import os.path
|
||||||
|
|
||||||
|
from backend.core.path_conf import BASE_PATH
|
||||||
|
from backend.utils.import_parse import get_model_object
|
||||||
|
|
||||||
|
|
||||||
|
def get_app_models() -> list[type]:
|
||||||
|
"""获取 app 所有模型类"""
|
||||||
|
app_path = os.path.join(BASE_PATH, 'app')
|
||||||
|
list_dirs = os.listdir(app_path)
|
||||||
|
|
||||||
|
apps = []
|
||||||
|
|
||||||
|
for d in list_dirs:
|
||||||
|
if os.path.isdir(os.path.join(app_path, d)) and d != '__pycache__':
|
||||||
|
apps.append(d)
|
||||||
|
|
||||||
|
objs = []
|
||||||
|
|
||||||
|
for app in apps:
|
||||||
|
module_path = f'backend.app.{app}.model'
|
||||||
|
obj = get_model_object(module_path)
|
||||||
|
if obj:
|
||||||
|
objs.append(obj)
|
||||||
|
|
||||||
|
return objs
|
||||||
|
|
||||||
|
|
||||||
|
# import all app models for auto create db tables
|
||||||
|
for cls in get_app_models():
|
||||||
|
class_name = cls.__name__
|
||||||
|
if class_name not in globals():
|
||||||
|
globals()[class_name] = cls
|
||||||
|
|||||||
@@ -5,14 +5,12 @@ from fastapi import APIRouter
|
|||||||
from backend.app.admin.api.v1.auth import router as auth_router
|
from backend.app.admin.api.v1.auth import router as auth_router
|
||||||
from backend.app.admin.api.v1.log import router as log_router
|
from backend.app.admin.api.v1.log import router as log_router
|
||||||
from backend.app.admin.api.v1.monitor import router as monitor_router
|
from backend.app.admin.api.v1.monitor import router as monitor_router
|
||||||
from backend.app.admin.api.v1.oauth2 import router as oauth2_router
|
|
||||||
from backend.app.admin.api.v1.sys import router as sys_router
|
from backend.app.admin.api.v1.sys import router as sys_router
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
|
|
||||||
v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH)
|
v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH)
|
||||||
|
|
||||||
v1.include_router(auth_router)
|
v1.include_router(auth_router)
|
||||||
v1.include_router(oauth2_router)
|
|
||||||
v1.include_router(sys_router)
|
v1.include_router(sys_router)
|
||||||
v1.include_router(log_router)
|
v1.include_router(log_router)
|
||||||
v1.include_router(monitor_router)
|
v1.include_router(monitor_router)
|
||||||
|
|||||||
@@ -8,4 +8,4 @@ from backend.app.admin.api.v1.auth.captcha import router as captcha_router
|
|||||||
router = APIRouter(prefix='/auth')
|
router = APIRouter(prefix='/auth')
|
||||||
|
|
||||||
router.include_router(auth_router, tags=['授权'])
|
router.include_router(auth_router, tags=['授权'])
|
||||||
router.include_router(captcha_router, prefix='/captcha', tags=['验证码'])
|
router.include_router(captcha_router, tags=['验证码'])
|
||||||
|
|||||||
@@ -17,9 +17,9 @@ router = APIRouter()
|
|||||||
|
|
||||||
|
|
||||||
@router.post('/login/swagger', summary='swagger 调试专用', description='用于快捷获取 token 进行 swagger 认证')
|
@router.post('/login/swagger', summary='swagger 调试专用', description='用于快捷获取 token 进行 swagger 认证')
|
||||||
async def swagger_login(obj: Annotated[HTTPBasicCredentials, Depends()]) -> GetSwaggerToken:
|
async def login_swagger(obj: Annotated[HTTPBasicCredentials, Depends()]) -> GetSwaggerToken:
|
||||||
token, user = await auth_service.swagger_login(obj=obj)
|
token, user = await auth_service.swagger_login(obj=obj)
|
||||||
return GetSwaggerToken(access_token=token, user=user) # type: ignore
|
return GetSwaggerToken(access_token=token, user=user)
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
@@ -28,20 +28,26 @@ async def swagger_login(obj: Annotated[HTTPBasicCredentials, Depends()]) -> GetS
|
|||||||
description='json 格式登录, 仅支持在第三方api工具调试, 例如: postman',
|
description='json 格式登录, 仅支持在第三方api工具调试, 例如: postman',
|
||||||
dependencies=[Depends(RateLimiter(times=5, minutes=1))],
|
dependencies=[Depends(RateLimiter(times=5, minutes=1))],
|
||||||
)
|
)
|
||||||
async def user_login(
|
async def login(
|
||||||
request: Request, response: Response, obj: AuthLoginParam, background_tasks: BackgroundTasks
|
request: Request, response: Response, obj: AuthLoginParam, background_tasks: BackgroundTasks
|
||||||
) -> ResponseSchemaModel[GetLoginToken]:
|
) -> ResponseSchemaModel[GetLoginToken]:
|
||||||
data = await auth_service.login(request=request, response=response, obj=obj, background_tasks=background_tasks)
|
data = await auth_service.login(request=request, response=response, obj=obj, background_tasks=background_tasks)
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.post('/token/new', summary='创建新 token')
|
@router.get('/codes', summary='获取所有授权码', description='适配 vben admin v5', dependencies=[DependsJwtAuth])
|
||||||
async def create_new_token(request: Request) -> ResponseSchemaModel[GetNewToken]:
|
async def get_codes(request: Request) -> ResponseSchemaModel[list[str]]:
|
||||||
data = await auth_service.new_token(request=request)
|
codes = await auth_service.get_codes(request=request)
|
||||||
|
return response_base.success(data=codes)
|
||||||
|
|
||||||
|
|
||||||
|
@router.post('/tokens', summary='刷新 token')
|
||||||
|
async def refresh_token(request: Request) -> ResponseSchemaModel[GetNewToken]:
|
||||||
|
data = await auth_service.refresh_token(request=request)
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.post('/logout', summary='用户登出', dependencies=[DependsJwtAuth])
|
@router.post('/logout', summary='用户登出')
|
||||||
async def user_logout(request: Request, response: Response) -> ResponseModel:
|
async def logout(request: Request, response: Response) -> ResponseModel:
|
||||||
await auth_service.logout(request=request, response=response)
|
await auth_service.logout(request=request, response=response)
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
|
|||||||
@@ -5,16 +5,16 @@ from fastapi import APIRouter, Depends, Request
|
|||||||
from fastapi_limiter.depends import RateLimiter
|
from fastapi_limiter.depends import RateLimiter
|
||||||
from starlette.concurrency import run_in_threadpool
|
from starlette.concurrency import run_in_threadpool
|
||||||
|
|
||||||
from backend.app.admin.conf import admin_settings
|
|
||||||
from backend.app.admin.schema.captcha import GetCaptchaDetail
|
from backend.app.admin.schema.captcha import GetCaptchaDetail
|
||||||
from backend.common.response.response_schema import ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseSchemaModel, response_base
|
||||||
|
from backend.core.conf import settings
|
||||||
from backend.database.redis import redis_client
|
from backend.database.redis import redis_client
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
'',
|
'/captcha',
|
||||||
summary='获取登录验证码',
|
summary='获取登录验证码',
|
||||||
dependencies=[Depends(RateLimiter(times=5, seconds=10))],
|
dependencies=[Depends(RateLimiter(times=5, seconds=10))],
|
||||||
)
|
)
|
||||||
@@ -26,9 +26,9 @@ async def get_captcha(request: Request) -> ResponseSchemaModel[GetCaptchaDetail]
|
|||||||
img, code = await run_in_threadpool(img_captcha, img_byte=img_type)
|
img, code = await run_in_threadpool(img_captcha, img_byte=img_type)
|
||||||
ip = request.state.ip
|
ip = request.state.ip
|
||||||
await redis_client.set(
|
await redis_client.set(
|
||||||
f'{admin_settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{ip}',
|
f'{settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{ip}',
|
||||||
code,
|
code,
|
||||||
ex=admin_settings.CAPTCHA_LOGIN_EXPIRE_SECONDS,
|
ex=settings.CAPTCHA_LOGIN_EXPIRE_SECONDS,
|
||||||
)
|
)
|
||||||
data = GetCaptchaDetail(image_type=img_type, image=img)
|
data = GetCaptchaDetail(image_type=img_type, image=img)
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from typing import Annotated
|
|||||||
|
|
||||||
from fastapi import APIRouter, Depends, Query
|
from fastapi import APIRouter, Depends, Query
|
||||||
|
|
||||||
from backend.app.admin.schema.login_log import GetLoginLogDetail
|
from backend.app.admin.schema.login_log import DeleteLoginLogParam, GetLoginLogDetail
|
||||||
from backend.app.admin.service.login_log_service import login_log_service
|
from backend.app.admin.service.login_log_service import login_log_service
|
||||||
from backend.common.pagination import DependsPagination, PageData, paging_data
|
from backend.common.pagination import DependsPagination, PageData, paging_data
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
@@ -24,7 +24,7 @@ router = APIRouter()
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_login_logs(
|
async def get_login_logs_paged(
|
||||||
db: CurrentSession,
|
db: CurrentSession,
|
||||||
username: Annotated[str | None, Query(description='用户名')] = None,
|
username: Annotated[str | None, Query(description='用户名')] = None,
|
||||||
status: Annotated[int | None, Query(description='状态')] = None,
|
status: Annotated[int | None, Query(description='状态')] = None,
|
||||||
@@ -43,8 +43,8 @@ async def get_pagination_login_logs(
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_login_log(pk: Annotated[list[int], Query(description='登录日志 ID 列表')]) -> ResponseModel:
|
async def delete_login_logs(obj: DeleteLoginLogParam) -> ResponseModel:
|
||||||
count = await login_log_service.delete(pk=pk)
|
count = await login_log_service.delete(obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
@@ -54,12 +54,10 @@ async def delete_login_log(pk: Annotated[list[int], Query(description='登录日
|
|||||||
'/all',
|
'/all',
|
||||||
summary='清空登录日志',
|
summary='清空登录日志',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('log:login:empty')),
|
Depends(RequestPermission('log:login:clear')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_all_login_logs() -> ResponseModel:
|
async def delete_all_login_logs() -> ResponseModel:
|
||||||
count = await login_log_service.delete_all()
|
await login_log_service.delete_all()
|
||||||
if count > 0:
|
return response_base.success()
|
||||||
return response_base.success()
|
|
||||||
return response_base.fail()
|
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from typing import Annotated
|
|||||||
|
|
||||||
from fastapi import APIRouter, Depends, Query
|
from fastapi import APIRouter, Depends, Query
|
||||||
|
|
||||||
from backend.app.admin.schema.opera_log import GetOperaLogDetail
|
from backend.app.admin.schema.opera_log import DeleteOperaLogParam, GetOperaLogDetail
|
||||||
from backend.app.admin.service.opera_log_service import opera_log_service
|
from backend.app.admin.service.opera_log_service import opera_log_service
|
||||||
from backend.common.pagination import DependsPagination, PageData, paging_data
|
from backend.common.pagination import DependsPagination, PageData, paging_data
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
@@ -24,7 +24,7 @@ router = APIRouter()
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_opera_logs(
|
async def get_opera_logs_paged(
|
||||||
db: CurrentSession,
|
db: CurrentSession,
|
||||||
username: Annotated[str | None, Query(description='用户名')] = None,
|
username: Annotated[str | None, Query(description='用户名')] = None,
|
||||||
status: Annotated[int | None, Query(description='状态')] = None,
|
status: Annotated[int | None, Query(description='状态')] = None,
|
||||||
@@ -43,8 +43,8 @@ async def get_pagination_opera_logs(
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_opera_log(pk: Annotated[list[int], Query(description='操作日志 ID 列表')]) -> ResponseModel:
|
async def delete_opera_logs(obj: DeleteOperaLogParam) -> ResponseModel:
|
||||||
count = await opera_log_service.delete(pk=pk)
|
count = await opera_log_service.delete(obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
@@ -54,12 +54,10 @@ async def delete_opera_log(pk: Annotated[list[int], Query(description='操作日
|
|||||||
'/all',
|
'/all',
|
||||||
summary='清空操作日志',
|
summary='清空操作日志',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('log:opera:empty')),
|
Depends(RequestPermission('log:opera:clear')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_all_opera_logs() -> ResponseModel:
|
async def delete_all_opera_logs() -> ResponseModel:
|
||||||
count = await opera_log_service.delete_all()
|
await opera_log_service.delete_all()
|
||||||
if count > 0:
|
return response_base.success()
|
||||||
return response_base.success()
|
|
||||||
return response_base.fail()
|
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from fastapi import APIRouter
|
from fastapi import APIRouter
|
||||||
|
|
||||||
|
from backend.app.admin.api.v1.monitor.online import router as token_router
|
||||||
from backend.app.admin.api.v1.monitor.redis import router as redis_router
|
from backend.app.admin.api.v1.monitor.redis import router as redis_router
|
||||||
from backend.app.admin.api.v1.monitor.server import router as server_router
|
from backend.app.admin.api.v1.monitor.server import router as server_router
|
||||||
|
|
||||||
@@ -9,3 +10,4 @@ router = APIRouter(prefix='/monitors')
|
|||||||
|
|
||||||
router.include_router(redis_router, prefix='/redis', tags=['redis监控'])
|
router.include_router(redis_router, prefix='/redis', tags=['redis监控'])
|
||||||
router.include_router(server_router, prefix='/server', tags=['服务器监控'])
|
router.include_router(server_router, prefix='/server', tags=['服务器监控'])
|
||||||
|
router.include_router(token_router, prefix='/sessions', tags=['会话监控'])
|
||||||
|
|||||||
+13
-10
@@ -6,7 +6,7 @@ from typing import Annotated
|
|||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Depends, Path, Query, Request
|
||||||
|
|
||||||
from backend.app.admin.schema.token import GetTokenDetail, KickOutToken
|
from backend.app.admin.schema.token import GetTokenDetail
|
||||||
from backend.common.enums import StatusType
|
from backend.common.enums import StatusType
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth, jwt_decode, revoke_token, superuser_verify
|
from backend.common.security.jwt import DependsJwtAuth, jwt_decode, revoke_token, superuser_verify
|
||||||
@@ -18,8 +18,8 @@ from backend.database.redis import redis_client
|
|||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get('', summary='获取令牌列表', dependencies=[DependsJwtAuth])
|
@router.get('', summary='获取在线用户', dependencies=[DependsJwtAuth])
|
||||||
async def get_tokens(
|
async def get_sessions(
|
||||||
username: Annotated[str | None, Query(description='用户名')] = None,
|
username: Annotated[str | None, Query(description='用户名')] = None,
|
||||||
) -> ResponseSchemaModel[list[GetTokenDetail]]:
|
) -> ResponseSchemaModel[list[GetTokenDetail]]:
|
||||||
token_keys = await redis_client.keys(f'{settings.TOKEN_REDIS_PREFIX}:*')
|
token_keys = await redis_client.keys(f'{settings.TOKEN_REDIS_PREFIX}:*')
|
||||||
@@ -44,9 +44,10 @@ async def get_tokens(
|
|||||||
for key in token_keys:
|
for key in token_keys:
|
||||||
token = await redis_client.get(key)
|
token = await redis_client.get(key)
|
||||||
token_payload = jwt_decode(token)
|
token_payload = jwt_decode(token)
|
||||||
|
user_id = token_payload.id
|
||||||
session_uuid = token_payload.session_uuid
|
session_uuid = token_payload.session_uuid
|
||||||
token_detail = GetTokenDetail(
|
token_detail = GetTokenDetail(
|
||||||
id=token_payload.id,
|
id=user_id,
|
||||||
session_uuid=session_uuid,
|
session_uuid=session_uuid,
|
||||||
username='未知',
|
username='未知',
|
||||||
nickname='未知',
|
nickname='未知',
|
||||||
@@ -58,7 +59,7 @@ async def get_tokens(
|
|||||||
last_login_time='未知',
|
last_login_time='未知',
|
||||||
expire_time=token_payload.expire_time,
|
expire_time=token_payload.expire_time,
|
||||||
)
|
)
|
||||||
extra_info = await redis_client.get(f'{settings.TOKEN_EXTRA_INFO_REDIS_PREFIX}:{session_uuid}')
|
extra_info = await redis_client.get(f'{settings.TOKEN_EXTRA_INFO_REDIS_PREFIX}:{user_id}:{session_uuid}')
|
||||||
if extra_info:
|
if extra_info:
|
||||||
extra_info = json.loads(extra_info)
|
extra_info = json.loads(extra_info)
|
||||||
# 排除 swagger 登录生成的 token
|
# 排除 swagger 登录生成的 token
|
||||||
@@ -75,15 +76,17 @@ async def get_tokens(
|
|||||||
|
|
||||||
@router.delete(
|
@router.delete(
|
||||||
'/{pk}',
|
'/{pk}',
|
||||||
summary='踢下线',
|
summary='强制下线',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:token:kick')),
|
Depends(RequestPermission('sys:session:delete')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def kick_out(
|
async def delete_session(
|
||||||
request: Request, pk: Annotated[int, Path(description='用户 ID')], obj: KickOutToken
|
request: Request,
|
||||||
|
pk: Annotated[int, Path(description='用户 ID')],
|
||||||
|
session_uuid: Annotated[str, Query(description='会话 UUID')],
|
||||||
) -> ResponseModel:
|
) -> ResponseModel:
|
||||||
superuser_verify(request)
|
superuser_verify(request)
|
||||||
await revoke_token(str(pk), obj.session_uuid)
|
await revoke_token(pk, session_uuid)
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
@@ -1,23 +1,15 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from fastapi import APIRouter, Depends
|
from fastapi import APIRouter
|
||||||
|
|
||||||
from backend.common.response.response_schema import ResponseModel, response_base
|
from backend.common.response.response_schema import ResponseModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
from backend.common.security.permission import RequestPermission
|
|
||||||
from backend.utils.redis_info import redis_info
|
from backend.utils.redis_info import redis_info
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get('', summary='redis 监控', dependencies=[DependsJwtAuth])
|
||||||
'',
|
|
||||||
summary='redis 监控',
|
|
||||||
dependencies=[
|
|
||||||
Depends(RequestPermission('sys:monitor:redis')),
|
|
||||||
DependsJwtAuth,
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def get_redis_info() -> ResponseModel:
|
async def get_redis_info() -> ResponseModel:
|
||||||
data = {
|
data = {
|
||||||
'info': await redis_info.get_info(),
|
'info': await redis_info.get_info(),
|
||||||
|
|||||||
@@ -1,24 +1,16 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from fastapi import APIRouter, Depends
|
from fastapi import APIRouter
|
||||||
from starlette.concurrency import run_in_threadpool
|
from starlette.concurrency import run_in_threadpool
|
||||||
|
|
||||||
from backend.common.response.response_schema import ResponseModel, response_base
|
from backend.common.response.response_schema import ResponseModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
from backend.common.security.permission import RequestPermission
|
|
||||||
from backend.utils.server_info import server_info
|
from backend.utils.server_info import server_info
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get('', summary='server 监控', dependencies=[DependsJwtAuth])
|
||||||
'',
|
|
||||||
summary='server 监控',
|
|
||||||
dependencies=[
|
|
||||||
Depends(RequestPermission('sys:monitor:server')),
|
|
||||||
DependsJwtAuth,
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def get_server_info() -> ResponseModel:
|
async def get_server_info() -> ResponseModel:
|
||||||
data = {
|
data = {
|
||||||
# 扔到线程池,避免阻塞
|
# 扔到线程池,避免阻塞
|
||||||
|
|||||||
@@ -1,11 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from fastapi import APIRouter
|
|
||||||
|
|
||||||
from backend.app.admin.api.v1.oauth2.github import router as github_router
|
|
||||||
from backend.app.admin.api.v1.oauth2.linux_do import router as linux_do_router
|
|
||||||
|
|
||||||
router = APIRouter(prefix='/oauth2')
|
|
||||||
|
|
||||||
router.include_router(github_router, prefix='/github', tags=['GitHub OAuth2'])
|
|
||||||
router.include_router(linux_do_router, prefix='/linux-do', tags=['LinuxDo OAuth2'])
|
|
||||||
@@ -1,15 +1,14 @@
|
|||||||
#!/usr/bin/env python3
|
# !/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from fastapi import APIRouter
|
from fastapi import APIRouter
|
||||||
|
|
||||||
from backend.app.admin.api.v1.sys.data_rule import router as data_rule_router
|
from backend.app.admin.api.v1.sys.data_rule import router as data_rule_router
|
||||||
from backend.app.admin.api.v1.sys.data_scope import router as data_scope_router
|
from backend.app.admin.api.v1.sys.data_scope import router as data_scope_router
|
||||||
from backend.app.admin.api.v1.sys.dept import router as dept_router
|
from backend.app.admin.api.v1.sys.dept import router as dept_router
|
||||||
|
from backend.app.admin.api.v1.sys.files import router as file_router
|
||||||
from backend.app.admin.api.v1.sys.menu import router as menu_router
|
from backend.app.admin.api.v1.sys.menu import router as menu_router
|
||||||
from backend.app.admin.api.v1.sys.plugin import router as plugin_router
|
from backend.app.admin.api.v1.sys.plugin import router as plugin_router
|
||||||
from backend.app.admin.api.v1.sys.role import router as role_router
|
from backend.app.admin.api.v1.sys.role import router as role_router
|
||||||
from backend.app.admin.api.v1.sys.token import router as token_router
|
|
||||||
from backend.app.admin.api.v1.sys.upload import router as upload_router
|
|
||||||
from backend.app.admin.api.v1.sys.user import router as user_router
|
from backend.app.admin.api.v1.sys.user import router as user_router
|
||||||
|
|
||||||
router = APIRouter(prefix='/sys')
|
router = APIRouter(prefix='/sys')
|
||||||
@@ -20,6 +19,5 @@ router.include_router(role_router, prefix='/roles', tags=['系统角色'])
|
|||||||
router.include_router(user_router, prefix='/users', tags=['系统用户'])
|
router.include_router(user_router, prefix='/users', tags=['系统用户'])
|
||||||
router.include_router(data_rule_router, prefix='/data-rules', tags=['系统数据规则'])
|
router.include_router(data_rule_router, prefix='/data-rules', tags=['系统数据规则'])
|
||||||
router.include_router(data_scope_router, prefix='/data-scopes', tags=['系统数据范围'])
|
router.include_router(data_scope_router, prefix='/data-scopes', tags=['系统数据范围'])
|
||||||
router.include_router(token_router, prefix='/tokens', tags=['系统令牌'])
|
router.include_router(file_router, prefix='/files', tags=['系统文件'])
|
||||||
router.include_router(upload_router, prefix='/upload', tags=['系统上传'])
|
router.include_router(plugin_router, prefix='/plugins', tags=['系统插件'])
|
||||||
router.include_router(plugin_router, prefix='/plugin', tags=['系统插件'])
|
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ from fastapi import APIRouter, Depends, Path, Query
|
|||||||
|
|
||||||
from backend.app.admin.schema.data_rule import (
|
from backend.app.admin.schema.data_rule import (
|
||||||
CreateDataRuleParam,
|
CreateDataRuleParam,
|
||||||
|
DeleteDataRuleParam,
|
||||||
GetDataRuleColumnDetail,
|
GetDataRuleColumnDetail,
|
||||||
GetDataRuleDetail,
|
GetDataRuleDetail,
|
||||||
UpdateDataRuleParam,
|
UpdateDataRuleParam,
|
||||||
@@ -27,7 +28,7 @@ async def get_data_rule_models() -> ResponseSchemaModel[list[str]]:
|
|||||||
return response_base.success(data=models)
|
return response_base.success(data=models)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/model/{model}/columns', summary='获取数据规则可用模型列', dependencies=[DependsJwtAuth])
|
@router.get('/models/{model}/columns', summary='获取数据规则可用模型列', dependencies=[DependsJwtAuth])
|
||||||
async def get_data_rule_model_columns(
|
async def get_data_rule_model_columns(
|
||||||
model: Annotated[str, Path(description='模型名称')],
|
model: Annotated[str, Path(description='模型名称')],
|
||||||
) -> ResponseSchemaModel[list[GetDataRuleColumnDetail]]:
|
) -> ResponseSchemaModel[list[GetDataRuleColumnDetail]]:
|
||||||
@@ -57,7 +58,7 @@ async def get_data_rule(
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_data_rules(
|
async def get_data_rules_paged(
|
||||||
db: CurrentSession, name: Annotated[str | None, Query(description='规则名称')] = None
|
db: CurrentSession, name: Annotated[str | None, Query(description='规则名称')] = None
|
||||||
) -> ResponseSchemaModel[PageData[GetDataRuleDetail]]:
|
) -> ResponseSchemaModel[PageData[GetDataRuleDetail]]:
|
||||||
data_rule_select = await data_rule_service.get_select(name=name)
|
data_rule_select = await data_rule_service.get_select(name=name)
|
||||||
@@ -103,8 +104,8 @@ async def update_data_rule(
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_data_rule(pk: Annotated[list[int], Query(description='数据规则 ID 列表')]) -> ResponseModel:
|
async def delete_data_rules(obj: DeleteDataRuleParam) -> ResponseModel:
|
||||||
count = await data_rule_service.delete(pk=pk)
|
count = await data_rule_service.delete(obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ from fastapi import APIRouter, Depends, Path, Query
|
|||||||
|
|
||||||
from backend.app.admin.schema.data_scope import (
|
from backend.app.admin.schema.data_scope import (
|
||||||
CreateDataScopeParam,
|
CreateDataScopeParam,
|
||||||
|
DeleteDataScopeParam,
|
||||||
GetDataScopeDetail,
|
GetDataScopeDetail,
|
||||||
GetDataScopeWithRelationDetail,
|
GetDataScopeWithRelationDetail,
|
||||||
UpdateDataScopeParam,
|
UpdateDataScopeParam,
|
||||||
@@ -22,6 +23,12 @@ from backend.database.db import CurrentSession
|
|||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/all', summary='获取所有数据范围', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_all_data_scope() -> ResponseSchemaModel[list[GetDataScopeDetail]]:
|
||||||
|
data = await data_scope_service.get_all()
|
||||||
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{pk}', summary='获取数据范围详情', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}', summary='获取数据范围详情', dependencies=[DependsJwtAuth])
|
||||||
async def get_data_scope(
|
async def get_data_scope(
|
||||||
pk: Annotated[int, Path(description='数据范围 ID')],
|
pk: Annotated[int, Path(description='数据范围 ID')],
|
||||||
@@ -46,7 +53,7 @@ async def get_data_scope_rules(
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_data_scopes(
|
async def get_data_scopes_paged(
|
||||||
db: CurrentSession,
|
db: CurrentSession,
|
||||||
name: Annotated[str | None, Query(description='范围名称')] = None,
|
name: Annotated[str | None, Query(description='范围名称')] = None,
|
||||||
status: Annotated[int | None, Query(description='状态')] = None,
|
status: Annotated[int | None, Query(description='状态')] = None,
|
||||||
@@ -111,8 +118,8 @@ async def update_data_scope_rules(
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_data_scope(pk: Annotated[list[int], Query(description='数据范围 ID 列表')]) -> ResponseModel:
|
async def delete_data_scopes(obj: DeleteDataScopeParam) -> ResponseModel:
|
||||||
count = await data_scope_service.delete(pk=pk)
|
count = await data_scope_service.delete(obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|||||||
@@ -1,10 +1,10 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Annotated, Any
|
from typing import Annotated
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Depends, Path, Query, Request
|
||||||
|
|
||||||
from backend.app.admin.schema.dept import CreateDeptParam, GetDeptDetail, UpdateDeptParam
|
from backend.app.admin.schema.dept import CreateDeptParam, GetDeptDetail, GetDeptTree, UpdateDeptParam
|
||||||
from backend.app.admin.service.dept_service import dept_service
|
from backend.app.admin.service.dept_service import dept_service
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
@@ -20,15 +20,15 @@ async def get_dept(pk: Annotated[int, Path(description='部门 ID')]) -> Respons
|
|||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.get('', summary='获取所有部门展示树', dependencies=[DependsJwtAuth])
|
@router.get('', summary='获取部门树', dependencies=[DependsJwtAuth])
|
||||||
async def get_all_depts(
|
async def get_dept_tree(
|
||||||
request: Request,
|
request: Request,
|
||||||
name: Annotated[str | None, Query(description='部门名称')] = None,
|
name: Annotated[str | None, Query(description='部门名称')] = None,
|
||||||
leader: Annotated[str | None, Query(description='部门负责人')] = None,
|
leader: Annotated[str | None, Query(description='部门负责人')] = None,
|
||||||
phone: Annotated[str | None, Query(description='联系电话')] = None,
|
phone: Annotated[str | None, Query(description='联系电话')] = None,
|
||||||
status: Annotated[int | None, Query(description='状态')] = None,
|
status: Annotated[int | None, Query(description='状态')] = None,
|
||||||
) -> ResponseSchemaModel[list[dict[str, Any]]]:
|
) -> ResponseSchemaModel[list[GetDeptTree]]:
|
||||||
dept = await dept_service.get_dept_tree(request=request, name=name, leader=leader, phone=phone, status=status)
|
dept = await dept_service.get_tree(request=request, name=name, leader=leader, phone=phone, status=status)
|
||||||
return response_base.success(data=dept)
|
return response_base.success(data=dept)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,27 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Annotated
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, File, UploadFile
|
||||||
|
|
||||||
|
from backend.common.dataclasses import UploadUrl
|
||||||
|
from backend.common.response.response_schema import ResponseSchemaModel, response_base
|
||||||
|
from backend.common.security.permission import RequestPermission
|
||||||
|
from backend.common.security.rbac import DependsRBAC
|
||||||
|
from backend.utils.file_ops import upload_file, upload_file_verify
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.post(
|
||||||
|
'/upload',
|
||||||
|
summary='文件上传',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:file:upload')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def upload_files(file: Annotated[UploadFile, File()]) -> ResponseSchemaModel[UploadUrl]:
|
||||||
|
upload_file_verify(file)
|
||||||
|
filename = await upload_file(file)
|
||||||
|
return response_base.success(data={'url': f'/static/upload/{filename}'})
|
||||||
@@ -4,7 +4,7 @@ from typing import Annotated, Any
|
|||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Depends, Path, Query, Request
|
||||||
|
|
||||||
from backend.app.admin.schema.menu import CreateMenuParam, GetMenuDetail, UpdateMenuParam
|
from backend.app.admin.schema.menu import CreateMenuParam, GetMenuDetail, GetMenuTree, UpdateMenuParam
|
||||||
from backend.app.admin.service.menu_service import menu_service
|
from backend.app.admin.service.menu_service import menu_service
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
@@ -14,9 +14,9 @@ from backend.common.security.rbac import DependsRBAC
|
|||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get('/sidebar', summary='获取用户菜单侧边栏', description='适配 vben5', dependencies=[DependsJwtAuth])
|
@router.get('/sidebar', summary='获取用户菜单侧边栏', description='已适配 vben admin v5', dependencies=[DependsJwtAuth])
|
||||||
async def get_user_sidebar(request: Request) -> ResponseSchemaModel[list[dict[str, Any]]]:
|
async def get_user_sidebar(request: Request) -> ResponseSchemaModel[list[dict[str, Any] | None]]:
|
||||||
menu = await menu_service.get_user_menu_tree(request=request)
|
menu = await menu_service.get_sidebar(request=request)
|
||||||
return response_base.success(data=menu)
|
return response_base.success(data=menu)
|
||||||
|
|
||||||
|
|
||||||
@@ -26,12 +26,12 @@ async def get_menu(pk: Annotated[int, Path(description='菜单 ID')]) -> Respons
|
|||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.get('', summary='获取所有菜单展示树', dependencies=[DependsJwtAuth])
|
@router.get('', summary='获取菜单树', dependencies=[DependsJwtAuth])
|
||||||
async def get_all_menus(
|
async def get_menu_tree(
|
||||||
title: Annotated[str | None, Query(description='菜单标题')] = None,
|
title: Annotated[str | None, Query(description='菜单标题')] = None,
|
||||||
status: Annotated[int | None, Query(description='状体')] = None,
|
status: Annotated[int | None, Query(description='状体')] = None,
|
||||||
) -> ResponseSchemaModel[list[dict[str, Any]]]:
|
) -> ResponseSchemaModel[list[GetMenuTree]]:
|
||||||
menu = await menu_service.get_menu_tree(title=title, status=status)
|
menu = await menu_service.get_tree(title=title, status=status)
|
||||||
return response_base.success(data=menu)
|
return response_base.success(data=menu)
|
||||||
|
|
||||||
|
|
||||||
@@ -71,7 +71,7 @@ async def update_menu(pk: Annotated[int, Path(description='菜单 ID')], obj: Up
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_menu(pk: Annotated[int, Path(description='菜单 ID 列表')]) -> ResponseModel:
|
async def delete_menu(pk: Annotated[int, Path(description='菜单 ID')]) -> ResponseModel:
|
||||||
count = await menu_service.delete(pk=pk)
|
count = await menu_service.delete(pk=pk)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
|
|||||||
@@ -1,98 +1,88 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import io
|
from typing import Annotated, Any
|
||||||
import os.path
|
|
||||||
import zipfile
|
|
||||||
|
|
||||||
from typing import Annotated
|
from fastapi import APIRouter, Depends, File, Path, UploadFile
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, File, UploadFile
|
|
||||||
from fastapi.params import Query
|
from fastapi.params import Query
|
||||||
from starlette.responses import StreamingResponse
|
from starlette.responses import StreamingResponse
|
||||||
|
|
||||||
from backend.common.exception import errors
|
from backend.app.admin.service.plugin_service import plugin_service
|
||||||
from backend.common.response.response_schema import ResponseModel, response_base
|
from backend.common.enums import PluginType
|
||||||
|
from backend.common.response.response_code import CustomResponse
|
||||||
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
from backend.common.security.permission import RequestPermission
|
from backend.common.security.permission import RequestPermission
|
||||||
from backend.common.security.rbac import DependsRBAC
|
from backend.common.security.rbac import DependsRBAC
|
||||||
from backend.core.path_conf import PLUGIN_DIR
|
|
||||||
from backend.plugin.tools import install_requirements_async
|
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('', summary='获取所有插件', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_all_plugins() -> ResponseSchemaModel[list[dict[str, Any]]]:
|
||||||
|
plugins = await plugin_service.get_all()
|
||||||
|
return response_base.success(data=plugins)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/changed', summary='是否存在插件变更', dependencies=[DependsJwtAuth])
|
||||||
|
async def plugin_changed() -> ResponseSchemaModel[bool]:
|
||||||
|
plugins = await plugin_service.changed()
|
||||||
|
return response_base.success(data=bool(plugins))
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
@router.post(
|
||||||
'/install',
|
'',
|
||||||
summary='安装插件',
|
summary='安装插件',
|
||||||
description='需使用插件 zip 压缩包进行安装',
|
description='使用插件 zip 压缩包或 git 仓库地址进行安装',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:plugin:install')),
|
Depends(RequestPermission('sys:plugin:install')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def install_plugin(file: Annotated[UploadFile, File()]) -> ResponseModel:
|
async def install_plugin(
|
||||||
contents = await file.read()
|
type: Annotated[PluginType, Query(description='插件类型')],
|
||||||
file_bytes = io.BytesIO(contents)
|
file: Annotated[UploadFile | None, File()] = None,
|
||||||
if not zipfile.is_zipfile(file_bytes):
|
repo_url: Annotated[str | None, Query(description='插件 git 仓库地址')] = None,
|
||||||
raise errors.ForbiddenError(msg='插件压缩包格式非法')
|
) -> ResponseModel:
|
||||||
with zipfile.ZipFile(file_bytes) as zf:
|
plugin_name = await plugin_service.install(type=type, file=file, repo_url=repo_url)
|
||||||
# 校验压缩包
|
return response_base.success(
|
||||||
plugin_dir_in_zip = f'{file.filename[:-4]}/backend/plugin/'
|
res=CustomResponse(
|
||||||
members_in_plugin_dir = [name for name in zf.namelist() if name.startswith(plugin_dir_in_zip)]
|
code=200, msg=f'插件 {plugin_name} 安装成功,请根据插件说明(README.md)进行相关配置并重启服务'
|
||||||
if not members_in_plugin_dir:
|
)
|
||||||
raise errors.ForbiddenError(msg='插件压缩包内容非法')
|
)
|
||||||
plugin_name = members_in_plugin_dir[1].replace(plugin_dir_in_zip, '').replace('/', '')
|
|
||||||
if (
|
|
||||||
len(members_in_plugin_dir) <= 3
|
|
||||||
or f'{plugin_dir_in_zip}{plugin_name}/plugin.toml' not in members_in_plugin_dir
|
|
||||||
or f'{plugin_dir_in_zip}{plugin_name}/README.md' not in members_in_plugin_dir
|
|
||||||
):
|
|
||||||
raise errors.ForbiddenError(msg='插件压缩包内缺少必要文件')
|
|
||||||
|
|
||||||
# 插件是否可安装
|
|
||||||
full_plugin_path = os.path.join(PLUGIN_DIR, plugin_name)
|
|
||||||
if os.path.exists(full_plugin_path):
|
|
||||||
raise errors.ForbiddenError(msg='此插件已安装')
|
|
||||||
else:
|
|
||||||
os.makedirs(full_plugin_path, exist_ok=True)
|
|
||||||
|
|
||||||
# 解压(安装)
|
|
||||||
members = []
|
|
||||||
for member in zf.infolist():
|
|
||||||
if member.filename.startswith(plugin_dir_in_zip):
|
|
||||||
new_filename = member.filename.replace(plugin_dir_in_zip, '')
|
|
||||||
if new_filename:
|
|
||||||
member.filename = new_filename
|
|
||||||
members.append(member)
|
|
||||||
zf.extractall(PLUGIN_DIR, members)
|
|
||||||
if os.path.exists(os.path.join(full_plugin_path, 'requirements.txt')):
|
|
||||||
await install_requirements_async()
|
|
||||||
|
|
||||||
return response_base.success()
|
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
@router.delete(
|
||||||
'/zip',
|
'/{plugin}',
|
||||||
summary='打包插件',
|
summary='卸载插件',
|
||||||
|
description='此操作会直接删除插件依赖,但不会直接删除插件,而是将插件移动到备份目录',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:plugin:zip')),
|
Depends(RequestPermission('sys:plugin:uninstall')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def build_plugin(plugin: Annotated[str, Query(description='插件名称')]) -> StreamingResponse:
|
async def uninstall_plugin(plugin: Annotated[str, Path(description='插件名称')]) -> ResponseModel:
|
||||||
plugin_dir = os.path.join(PLUGIN_DIR, plugin)
|
await plugin_service.uninstall(plugin=plugin)
|
||||||
if not os.path.exists(plugin_dir):
|
return response_base.success(
|
||||||
raise errors.ForbiddenError(msg='插件不存在')
|
res=CustomResponse(code=200, msg=f'插件 {plugin} 卸载成功,请根据插件说明(README.md)移除相关配置并重启服务')
|
||||||
|
)
|
||||||
|
|
||||||
bio = io.BytesIO()
|
|
||||||
with zipfile.ZipFile(bio, 'w') as zf:
|
|
||||||
for root, dirs, files in os.walk(plugin_dir):
|
|
||||||
dirs[:] = [d for d in dirs if d != '__pycache__']
|
|
||||||
for file in files:
|
|
||||||
file_path = os.path.join(root, file)
|
|
||||||
arcname = os.path.relpath(file_path, start=plugin_dir)
|
|
||||||
zf.write(file_path, arcname)
|
|
||||||
|
|
||||||
bio.seek(0)
|
@router.put(
|
||||||
|
'/{plugin}/status',
|
||||||
|
summary='更新插件状态',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:plugin:edit')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def update_plugin_status(plugin: Annotated[str, Path(description='插件名称')]) -> ResponseModel:
|
||||||
|
await plugin_service.update_status(plugin=plugin)
|
||||||
|
return response_base.success()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/{plugin}', summary='下载插件', dependencies=[DependsJwtAuth])
|
||||||
|
async def download_plugin(plugin: Annotated[str, Path(description='插件名称')]) -> StreamingResponse:
|
||||||
|
bio = await plugin_service.build(plugin=plugin)
|
||||||
return StreamingResponse(
|
return StreamingResponse(
|
||||||
bio,
|
bio,
|
||||||
media_type='application/x-zip-compressed',
|
media_type='application/x-zip-compressed',
|
||||||
|
|||||||
@@ -1,11 +1,13 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Annotated, Any
|
from typing import Annotated
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query
|
from fastapi import APIRouter, Depends, Path, Query
|
||||||
|
|
||||||
|
from backend.app.admin.schema.menu import GetMenuTree
|
||||||
from backend.app.admin.schema.role import (
|
from backend.app.admin.schema.role import (
|
||||||
CreateRoleParam,
|
CreateRoleParam,
|
||||||
|
DeleteRoleParam,
|
||||||
GetRoleDetail,
|
GetRoleDetail,
|
||||||
GetRoleWithRelationDetail,
|
GetRoleWithRelationDetail,
|
||||||
UpdateRoleMenuParam,
|
UpdateRoleMenuParam,
|
||||||
@@ -29,32 +31,22 @@ async def get_all_roles() -> ResponseSchemaModel[list[GetRoleDetail]]:
|
|||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{pk}/all', summary='获取用户所有角色', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}/menus', summary='获取角色菜单树', dependencies=[DependsJwtAuth])
|
||||||
async def get_user_all_roles(
|
async def get_role_menu_tree(
|
||||||
pk: Annotated[int, Path(description='用户 ID')],
|
|
||||||
) -> ResponseSchemaModel[list[GetRoleDetail]]:
|
|
||||||
data = await role_service.get_users(pk=pk)
|
|
||||||
return response_base.success(data=data)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{pk}/menus', summary='获取角色所有菜单', dependencies=[DependsJwtAuth])
|
|
||||||
async def get_role_all_menus(
|
|
||||||
pk: Annotated[int, Path(description='角色 ID')],
|
pk: Annotated[int, Path(description='角色 ID')],
|
||||||
) -> ResponseSchemaModel[list[dict[str, Any]]]:
|
) -> ResponseSchemaModel[list[GetMenuTree] | None]:
|
||||||
menu = await role_service.get_menu_tree(pk=pk)
|
menu = await role_service.get_menu_tree(pk=pk)
|
||||||
return response_base.success(data=menu)
|
return response_base.success(data=menu)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{pk}/scopes', summary='获取角色所有数据范围', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}/scopes', summary='获取角色所有数据范围', dependencies=[DependsJwtAuth])
|
||||||
async def get_role_all_scopes(pk: Annotated[int, Path(description='角色 ID')]) -> ResponseSchemaModel[list[int]]:
|
async def get_role_scopes(pk: Annotated[int, Path(description='角色 ID')]) -> ResponseSchemaModel[list[int]]:
|
||||||
rule = await role_service.get_scopes(pk=pk)
|
rule = await role_service.get_scopes(pk=pk)
|
||||||
return response_base.success(data=rule)
|
return response_base.success(data=rule)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{pk}', summary='获取角色详情', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}', summary='获取角色详情', dependencies=[DependsJwtAuth])
|
||||||
async def get_role(
|
async def get_role(pk: Annotated[int, Path(description='角色 ID')]) -> ResponseSchemaModel[GetRoleWithRelationDetail]:
|
||||||
pk: Annotated[int, Path(description='角色 ID')],
|
|
||||||
) -> ResponseSchemaModel[GetRoleWithRelationDetail]:
|
|
||||||
data = await role_service.get(pk=pk)
|
data = await role_service.get(pk=pk)
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
@@ -67,7 +59,7 @@ async def get_role(
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_roles(
|
async def get_roles_paged(
|
||||||
db: CurrentSession,
|
db: CurrentSession,
|
||||||
name: Annotated[str | None, Query(description='角色名称')] = None,
|
name: Annotated[str | None, Query(description='角色名称')] = None,
|
||||||
status: Annotated[int | None, Query(description='状态')] = None,
|
status: Annotated[int | None, Query(description='状态')] = None,
|
||||||
@@ -106,7 +98,7 @@ async def update_role(pk: Annotated[int, Path(description='角色 ID')], obj: Up
|
|||||||
|
|
||||||
|
|
||||||
@router.put(
|
@router.put(
|
||||||
'/{pk}/menu',
|
'/{pk}/menus',
|
||||||
summary='更新角色菜单',
|
summary='更新角色菜单',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:role:menu:edit')),
|
Depends(RequestPermission('sys:role:menu:edit')),
|
||||||
@@ -123,7 +115,7 @@ async def update_role_menus(
|
|||||||
|
|
||||||
|
|
||||||
@router.put(
|
@router.put(
|
||||||
'/{pk}/scope',
|
'/{pk}/scopes',
|
||||||
summary='更新角色数据范围',
|
summary='更新角色数据范围',
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:role:scope:edit')),
|
Depends(RequestPermission('sys:role:scope:edit')),
|
||||||
@@ -147,8 +139,8 @@ async def update_role_scopes(
|
|||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_role(pk: Annotated[list[int], Query(description='角色 ID 列表')]) -> ResponseModel:
|
async def delete_roles(obj: DeleteRoleParam) -> ResponseModel:
|
||||||
count = await role_service.delete(pk=pk)
|
count = await role_service.delete(obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|||||||
@@ -1,27 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from typing import Annotated
|
|
||||||
|
|
||||||
from fastapi import APIRouter, File, UploadFile
|
|
||||||
|
|
||||||
from backend.common.dataclasses import UploadUrl
|
|
||||||
from backend.common.enums import FileType
|
|
||||||
from backend.common.response.response_schema import ResponseSchemaModel, response_base
|
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
|
||||||
from backend.utils.file_ops import file_verify, upload_file
|
|
||||||
|
|
||||||
router = APIRouter()
|
|
||||||
|
|
||||||
|
|
||||||
@router.post('/image', summary='上传图片', dependencies=[DependsJwtAuth])
|
|
||||||
async def upload_image(file: Annotated[UploadFile, File()]) -> ResponseSchemaModel[UploadUrl]:
|
|
||||||
file_verify(file, FileType.image)
|
|
||||||
filename = await upload_file(file)
|
|
||||||
return response_base.success(data={'url': f'/static/upload/{filename}'})
|
|
||||||
|
|
||||||
|
|
||||||
@router.post('/video', summary='上传视频', dependencies=[DependsJwtAuth])
|
|
||||||
async def upload_video(file: Annotated[UploadFile, File()]) -> ResponseSchemaModel[UploadUrl]:
|
|
||||||
file_verify(file, FileType.video)
|
|
||||||
filename = await upload_file(file)
|
|
||||||
return response_base.success(data={'url': f'/static/upload/{filename}'})
|
|
||||||
@@ -2,19 +2,18 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Annotated
|
from typing import Annotated
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path, Query, Request
|
from fastapi import APIRouter, Body, Depends, Path, Query, Request
|
||||||
|
|
||||||
|
from backend.app.admin.schema.role import GetRoleDetail
|
||||||
from backend.app.admin.schema.user import (
|
from backend.app.admin.schema.user import (
|
||||||
AddUserParam,
|
AddUserParam,
|
||||||
AvatarParam,
|
|
||||||
GetCurrentUserInfoWithRelationDetail,
|
GetCurrentUserInfoWithRelationDetail,
|
||||||
GetUserInfoWithRelationDetail,
|
GetUserInfoWithRelationDetail,
|
||||||
RegisterUserParam,
|
|
||||||
ResetPasswordParam,
|
ResetPasswordParam,
|
||||||
UpdateUserParam,
|
UpdateUserParam,
|
||||||
UpdateUserRoleParam,
|
|
||||||
)
|
)
|
||||||
from backend.app.admin.service.user_service import user_service
|
from backend.app.admin.service.user_service import user_service
|
||||||
|
from backend.common.enums import UserPermissionType
|
||||||
from backend.common.pagination import DependsPagination, PageData, paging_data
|
from backend.common.pagination import DependsPagination, PageData, paging_data
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
@@ -25,74 +24,24 @@ from backend.database.db import CurrentSession
|
|||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.post('/register', summary='注册用户')
|
|
||||||
async def register_user(obj: RegisterUserParam) -> ResponseModel:
|
|
||||||
await user_service.register(obj=obj)
|
|
||||||
return response_base.success()
|
|
||||||
|
|
||||||
|
|
||||||
@router.post('/add', summary='添加用户', dependencies=[DependsRBAC])
|
|
||||||
async def add_user(request: Request, obj: AddUserParam) -> ResponseSchemaModel[GetUserInfoWithRelationDetail]:
|
|
||||||
await user_service.add(request=request, obj=obj)
|
|
||||||
data = await user_service.get_userinfo(username=obj.username)
|
|
||||||
return response_base.success(data=data)
|
|
||||||
|
|
||||||
|
|
||||||
@router.post('/password/reset', summary='密码重置', dependencies=[DependsJwtAuth])
|
|
||||||
async def password_reset(request: Request, obj: ResetPasswordParam) -> ResponseModel:
|
|
||||||
count = await user_service.pwd_reset(request=request, obj=obj)
|
|
||||||
if count > 0:
|
|
||||||
return response_base.success()
|
|
||||||
return response_base.fail()
|
|
||||||
|
|
||||||
|
|
||||||
@router.get('/me', summary='获取当前用户信息', dependencies=[DependsJwtAuth])
|
@router.get('/me', summary='获取当前用户信息', dependencies=[DependsJwtAuth])
|
||||||
async def get_current_user(request: Request) -> ResponseSchemaModel[GetCurrentUserInfoWithRelationDetail]:
|
async def get_current_user(request: Request) -> ResponseSchemaModel[GetCurrentUserInfoWithRelationDetail]:
|
||||||
data = request.user.model_dump()
|
data = request.user.model_dump()
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.get('/{username}', summary='查看用户信息', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}', summary='获取用户信息', dependencies=[DependsJwtAuth])
|
||||||
async def get_user(
|
async def get_userinfo(
|
||||||
username: Annotated[str, Path(description='用户名')],
|
pk: Annotated[int, Path(description='用户 ID')],
|
||||||
) -> ResponseSchemaModel[GetUserInfoWithRelationDetail]:
|
) -> ResponseSchemaModel[GetUserInfoWithRelationDetail]:
|
||||||
data = await user_service.get_userinfo(username=username)
|
data = await user_service.get_userinfo(pk=pk)
|
||||||
return response_base.success(data=data)
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{username}', summary='更新用户信息', dependencies=[DependsJwtAuth])
|
@router.get('/{pk}/roles', summary='获取用户所有角色', dependencies=[DependsJwtAuth])
|
||||||
async def update_user(
|
async def get_user_roles(pk: Annotated[int, Path(description='用户 ID')]) -> ResponseSchemaModel[list[GetRoleDetail]]:
|
||||||
request: Request, username: Annotated[str, Path(description='用户名')], obj: UpdateUserParam
|
data = await user_service.get_roles(pk=pk)
|
||||||
) -> ResponseModel:
|
return response_base.success(data=data)
|
||||||
count = await user_service.update(request=request, username=username, obj=obj)
|
|
||||||
if count > 0:
|
|
||||||
return response_base.success()
|
|
||||||
return response_base.fail()
|
|
||||||
|
|
||||||
|
|
||||||
@router.put(
|
|
||||||
'/{username}/role',
|
|
||||||
summary='更新用户角色',
|
|
||||||
dependencies=[
|
|
||||||
Depends(RequestPermission('sys:user:role:edit')),
|
|
||||||
DependsRBAC,
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def update_user_role(
|
|
||||||
request: Request, username: Annotated[str, Path(description='用户名')], obj: UpdateUserRoleParam
|
|
||||||
) -> ResponseModel:
|
|
||||||
await user_service.update_roles(request=request, username=username, obj=obj)
|
|
||||||
return response_base.success()
|
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{username}/avatar', summary='更新头像', dependencies=[DependsJwtAuth])
|
|
||||||
async def update_avatar(
|
|
||||||
request: Request, username: Annotated[str, Path(description='用户名')], avatar: AvatarParam
|
|
||||||
) -> ResponseModel:
|
|
||||||
count = await user_service.update_avatar(request=request, username=username, avatar=avatar)
|
|
||||||
if count > 0:
|
|
||||||
return response_base.success()
|
|
||||||
return response_base.fail()
|
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
@router.get(
|
||||||
@@ -103,7 +52,7 @@ async def update_avatar(
|
|||||||
DependsPagination,
|
DependsPagination,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def get_pagination_users(
|
async def get_users_paged(
|
||||||
db: CurrentSession,
|
db: CurrentSession,
|
||||||
dept: Annotated[int | None, Query(description='部门 ID')] = None,
|
dept: Annotated[int | None, Query(description='部门 ID')] = None,
|
||||||
username: Annotated[str | None, Query(description='用户名')] = None,
|
username: Annotated[str | None, Query(description='用户名')] = None,
|
||||||
@@ -115,49 +64,97 @@ async def get_pagination_users(
|
|||||||
return response_base.success(data=page_data)
|
return response_base.success(data=page_data)
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{pk}/super', summary='修改用户超级权限', dependencies=[DependsRBAC])
|
@router.post('', summary='创建用户', dependencies=[DependsRBAC])
|
||||||
async def super_set(request: Request, pk: Annotated[int, Path(description='用户 ID')]) -> ResponseModel:
|
async def create_user(request: Request, obj: AddUserParam) -> ResponseSchemaModel[GetUserInfoWithRelationDetail]:
|
||||||
count = await user_service.update_permission(request=request, pk=pk)
|
await user_service.create(request=request, obj=obj)
|
||||||
|
data = await user_service.get_userinfo(username=obj.username)
|
||||||
|
return response_base.success(data=data)
|
||||||
|
|
||||||
|
|
||||||
|
@router.put('/{pk}', summary='更新用户信息', dependencies=[DependsRBAC])
|
||||||
|
async def update_user(
|
||||||
|
request: Request, pk: Annotated[int, Path(description='用户 ID')], obj: UpdateUserParam
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.update(request=request, pk=pk, obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{pk}/staff', summary='修改用户后台登录权限', dependencies=[DependsRBAC])
|
@router.put('/{pk}/permissions', summary='更新用户权限', dependencies=[DependsRBAC])
|
||||||
async def staff_set(request: Request, pk: Annotated[int, Path(description='用户 ID')]) -> ResponseModel:
|
async def update_user_permission(
|
||||||
count = await user_service.update_staff(request=request, pk=pk)
|
request: Request,
|
||||||
|
pk: Annotated[int, Path(description='用户 ID')],
|
||||||
|
type: Annotated[UserPermissionType, Query(description='权限类型')],
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.update_permission(request=request, pk=pk, type=type)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{pk}/status', summary='修改用户状态', dependencies=[DependsRBAC])
|
@router.put('/me/password', summary='更新当前用户密码', dependencies=[DependsJwtAuth])
|
||||||
async def status_set(request: Request, pk: Annotated[int, Path(description='用户 ID')]) -> ResponseModel:
|
async def update_user_password(request: Request, obj: ResetPasswordParam) -> ResponseModel:
|
||||||
count = await user_service.update_status(request=request, pk=pk)
|
count = await user_service.update_password(request=request, obj=obj)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
@router.put('/{pk}/multi', summary='修改用户多端登录状态', dependencies=[DependsRBAC])
|
@router.put('/{pk}/password', summary='重置用户密码', dependencies=[DependsRBAC])
|
||||||
async def multi_set(request: Request, pk: Annotated[int, Path(description='用户 ID')]) -> ResponseModel:
|
async def reset_user_password(
|
||||||
count = await user_service.update_multi_login(request=request, pk=pk)
|
request: Request,
|
||||||
|
pk: Annotated[int, Path(description='用户 ID')],
|
||||||
|
password: Annotated[str, Body(embed=True, description='新密码')],
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.reset_password(request=request, pk=pk, password=password)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put('/me/nickname', summary='更新当前用户昵称', dependencies=[DependsJwtAuth])
|
||||||
|
async def update_user_nickname(
|
||||||
|
request: Request, nickname: Annotated[str, Body(embed=True, description='用户昵称')]
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.update_nickname(request=request, nickname=nickname)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put('/me/avatar', summary='更新当前用户头像', dependencies=[DependsJwtAuth])
|
||||||
|
async def update_user_avatar(
|
||||||
|
request: Request, avatar: Annotated[str, Body(embed=True, description='用户头像地址')]
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.update_avatar(request=request, avatar=avatar)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put('/me/email', summary='更新当前用户邮箱', dependencies=[DependsJwtAuth])
|
||||||
|
async def update_user_email(
|
||||||
|
request: Request,
|
||||||
|
captcha: Annotated[str, Body(embed=True, description='邮箱验证码')],
|
||||||
|
email: Annotated[str, Body(embed=True, description='用户邮箱')],
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await user_service.update_email(request=request, captcha=captcha, email=email)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
@router.delete(
|
@router.delete(
|
||||||
path='/{username}',
|
path='/{pk}',
|
||||||
summary='用户注销',
|
summary='删除用户',
|
||||||
description='用户注销 != 用户登出,注销之后用户将从数据库删除',
|
|
||||||
dependencies=[
|
dependencies=[
|
||||||
Depends(RequestPermission('sys:user:del')),
|
Depends(RequestPermission('sys:user:del')),
|
||||||
DependsRBAC,
|
DependsRBAC,
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
async def delete_user(username: Annotated[str, Path(description='用户名')]) -> ResponseModel:
|
async def delete_user(pk: Annotated[int, Path(description='用户 ID')]) -> ResponseModel:
|
||||||
count = await user_service.delete(username=username)
|
count = await user_service.delete(pk=pk)
|
||||||
if count > 0:
|
if count > 0:
|
||||||
return response_base.success()
|
return response_base.success()
|
||||||
return response_base.fail()
|
return response_base.fail()
|
||||||
|
|||||||
@@ -1,35 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from functools import lru_cache
|
|
||||||
|
|
||||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
||||||
|
|
||||||
from backend.core.path_conf import BASE_PATH
|
|
||||||
|
|
||||||
|
|
||||||
class AdminSettings(BaseSettings):
|
|
||||||
"""Admin 配置"""
|
|
||||||
|
|
||||||
model_config = SettingsConfigDict(env_file=f'{BASE_PATH}/.env', env_file_encoding='utf-8', extra='ignore')
|
|
||||||
|
|
||||||
# .env OAuth2
|
|
||||||
OAUTH2_GITHUB_CLIENT_ID: str
|
|
||||||
OAUTH2_GITHUB_CLIENT_SECRET: str
|
|
||||||
OAUTH2_LINUX_DO_CLIENT_ID: str
|
|
||||||
OAUTH2_LINUX_DO_CLIENT_SECRET: str
|
|
||||||
|
|
||||||
# OAuth2
|
|
||||||
OAUTH2_FRONTEND_REDIRECT_URI: str = 'http://localhost:5173/oauth2/callback'
|
|
||||||
|
|
||||||
# 验证码
|
|
||||||
CAPTCHA_LOGIN_REDIS_PREFIX: str = 'fba:login:captcha'
|
|
||||||
CAPTCHA_LOGIN_EXPIRE_SECONDS: int = 60 * 5 # 3 分钟
|
|
||||||
|
|
||||||
|
|
||||||
@lru_cache
|
|
||||||
def get_admin_settings() -> AdminSettings:
|
|
||||||
"""获取 admin 配置"""
|
|
||||||
return AdminSettings()
|
|
||||||
|
|
||||||
|
|
||||||
admin_settings = get_admin_settings()
|
|
||||||
@@ -2,9 +2,8 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Sequence
|
from typing import Sequence
|
||||||
|
|
||||||
from sqlalchemy import Select, and_, desc, select
|
from sqlalchemy import Select
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import noload
|
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
from backend.app.admin.model import DataRule
|
from backend.app.admin.model import DataRule
|
||||||
@@ -31,16 +30,12 @@ class CRUDDataRule(CRUDPlus[DataRule]):
|
|||||||
:param name: 规则名称
|
:param name: 规则名称
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).options(noload(self.model.scope)).order_by(desc(self.model.created_time))
|
filters = {}
|
||||||
|
|
||||||
filters = []
|
|
||||||
if name is not None:
|
if name is not None:
|
||||||
filters.append(self.model.name.like(f'%{name}%'))
|
filters['name__like'] = f'%{name}%'
|
||||||
|
|
||||||
if filters:
|
return await self.select_order('id', load_strategies={'scopes': 'noload'}, **filters)
|
||||||
stmt = stmt.where(and_(*filters))
|
|
||||||
|
|
||||||
return stmt
|
|
||||||
|
|
||||||
async def get_by_name(self, db: AsyncSession, name: str) -> DataRule | None:
|
async def get_by_name(self, db: AsyncSession, name: str) -> DataRule | None:
|
||||||
"""
|
"""
|
||||||
@@ -82,15 +77,15 @@ class CRUDDataRule(CRUDPlus[DataRule]):
|
|||||||
"""
|
"""
|
||||||
return await self.update_model(db, pk, obj)
|
return await self.update_model(db, pk, obj)
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, pk: list[int]) -> int:
|
async def delete(self, db: AsyncSession, pks: list[int]) -> int:
|
||||||
"""
|
"""
|
||||||
删除规则
|
批量删除规则
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param pk: 规则 ID 列表
|
:param pks: 规则 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pk)
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pks)
|
||||||
|
|
||||||
|
|
||||||
data_rule_dao: CRUDDataRule = CRUDDataRule(DataRule)
|
data_rule_dao: CRUDDataRule = CRUDDataRule(DataRule)
|
||||||
|
|||||||
@@ -1,8 +1,9 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from sqlalchemy import Select, and_, desc, select
|
from typing import Sequence
|
||||||
|
|
||||||
|
from sqlalchemy import Select, select
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import noload, selectinload
|
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
from backend.app.admin.model import DataRule, DataScope
|
from backend.app.admin.model import DataRule, DataScope
|
||||||
@@ -40,9 +41,16 @@ class CRUDDataScope(CRUDPlus[DataScope]):
|
|||||||
:param pk: 范围 ID
|
:param pk: 范围 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).options(selectinload(self.model.rules)).where(self.model.id == pk)
|
return await self.select_model(db, pk, load_strategies=['rules'])
|
||||||
data_scope = await db.execute(stmt)
|
|
||||||
return data_scope.scalars().first()
|
async def get_all(self, db: AsyncSession) -> Sequence[DataScope]:
|
||||||
|
"""
|
||||||
|
获取所有数据范围
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.select_models(db)
|
||||||
|
|
||||||
async def get_list(self, name: str | None, status: int | None) -> Select:
|
async def get_list(self, name: str | None, status: int | None) -> Select:
|
||||||
"""
|
"""
|
||||||
@@ -52,22 +60,14 @@ class CRUDDataScope(CRUDPlus[DataScope]):
|
|||||||
:param status: 范围状态
|
:param status: 范围状态
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = (
|
filters = {}
|
||||||
select(self.model)
|
|
||||||
.options(noload(self.model.rules), noload(self.model.roles))
|
|
||||||
.order_by(desc(self.model.created_time))
|
|
||||||
)
|
|
||||||
|
|
||||||
filters = []
|
|
||||||
if name is not None:
|
if name is not None:
|
||||||
filters.append(self.model.name.like(f'%{name}%'))
|
filters['name__like'] = f'%{name}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.append(self.model.status == status)
|
filters['status'] = status
|
||||||
|
|
||||||
if filters:
|
return await self.select_order('id', load_strategies={'rules': 'noload', 'roles': 'noload'}, **filters)
|
||||||
stmt = stmt.where(and_(*filters))
|
|
||||||
|
|
||||||
return stmt
|
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: CreateDataScopeParam) -> None:
|
async def create(self, db: AsyncSession, obj: CreateDataScopeParam) -> None:
|
||||||
"""
|
"""
|
||||||
@@ -105,15 +105,15 @@ class CRUDDataScope(CRUDPlus[DataScope]):
|
|||||||
current_data_scope.rules = rules.scalars().all()
|
current_data_scope.rules = rules.scalars().all()
|
||||||
return len(current_data_scope.rules)
|
return len(current_data_scope.rules)
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, pk: list[int]) -> int:
|
async def delete(self, db: AsyncSession, pks: list[int]) -> int:
|
||||||
"""
|
"""
|
||||||
删除数据范围
|
批量删除数据范围
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param pk: 范围 ID 列表
|
:param pks: 范围 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pk)
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pks)
|
||||||
|
|
||||||
|
|
||||||
data_scope_dao: CRUDDataScope = CRUDDataScope(DataScope)
|
data_scope_dao: CRUDDataScope = CRUDDataScope(DataScope)
|
||||||
|
|||||||
@@ -3,9 +3,7 @@
|
|||||||
from typing import Sequence
|
from typing import Sequence
|
||||||
|
|
||||||
from fastapi import Request
|
from fastapi import Request
|
||||||
from sqlalchemy import select
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import selectinload
|
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
from backend.app.admin.model import Dept
|
from backend.app.admin.model import Dept
|
||||||
@@ -56,16 +54,19 @@ class CRUDDept(CRUDPlus[Dept]):
|
|||||||
:param status: 部门状态
|
:param status: 部门状态
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
filters = {'del_flag__eq': 0}
|
filters = {'del_flag': 0}
|
||||||
|
|
||||||
if name is not None:
|
if name is not None:
|
||||||
filters.update(name__like=f'%{name}%')
|
filters['name__like'] = f'%{name}%'
|
||||||
if leader is not None:
|
if leader is not None:
|
||||||
filters.update(leader__like=f'%{leader}%')
|
filters['leader__like'] = f'%{leader}%'
|
||||||
if phone is not None:
|
if phone is not None:
|
||||||
filters.update(phone__startswith=phone)
|
filters['phone__startswith'] = phone
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.update(status=status)
|
filters['status'] = status
|
||||||
return await self.select_models_order(db, 'sort', None, await filter_data_permission(db, request), **filters)
|
|
||||||
|
data_filtered = await filter_data_permission(db, request)
|
||||||
|
return await self.select_models_order(db, 'sort', 'desc', data_filtered, **filters)
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: CreateDeptParam) -> None:
|
async def create(self, db: AsyncSession, obj: CreateDeptParam) -> None:
|
||||||
"""
|
"""
|
||||||
@@ -106,9 +107,7 @@ class CRUDDept(CRUDPlus[Dept]):
|
|||||||
:param dept_id: 部门 ID
|
:param dept_id: 部门 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).options(selectinload(self.model.users)).where(self.model.id == dept_id)
|
return await self.select_model(db, dept_id, load_strategies=['users'])
|
||||||
result = await db.execute(stmt)
|
|
||||||
return result.scalars().first()
|
|
||||||
|
|
||||||
async def get_children(self, db: AsyncSession, dept_id: int) -> Sequence[Dept | None]:
|
async def get_children(self, db: AsyncSession, dept_id: int) -> Sequence[Dept | None]:
|
||||||
"""
|
"""
|
||||||
@@ -118,9 +117,7 @@ class CRUDDept(CRUDPlus[Dept]):
|
|||||||
:param dept_id: 部门 ID
|
:param dept_id: 部门 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).where(self.model.parent_id == dept_id, self.model.del_flag == 0)
|
return await self.select_models(db, parent_id=dept_id, del_flag=0)
|
||||||
result = await db.execute(stmt)
|
|
||||||
return result.scalars().all()
|
|
||||||
|
|
||||||
|
|
||||||
dept_dao: CRUDDept = CRUDDept(Dept)
|
dept_dao: CRUDDept = CRUDDept(Dept)
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from sqlalchemy import Select
|
from sqlalchemy import Select
|
||||||
|
from sqlalchemy import delete as sa_delete
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
@@ -21,12 +22,14 @@ class CRUDLoginLog(CRUDPlus[LoginLog]):
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
filters = {}
|
filters = {}
|
||||||
|
|
||||||
if username is not None:
|
if username is not None:
|
||||||
filters.update(username__like=f'%{username}%')
|
filters['username__like'] = f'%{username}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.update(status=status)
|
filters['status'] = status
|
||||||
if ip is not None:
|
if ip is not None:
|
||||||
filters.update(ip__like=f'%{ip}%')
|
filters['ip__like'] = f'%{ip}%'
|
||||||
|
|
||||||
return await self.select_order('created_time', 'desc', **filters)
|
return await self.select_order('created_time', 'desc', **filters)
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: CreateLoginLogParam) -> None:
|
async def create(self, db: AsyncSession, obj: CreateLoginLogParam) -> None:
|
||||||
@@ -39,24 +42,25 @@ class CRUDLoginLog(CRUDPlus[LoginLog]):
|
|||||||
"""
|
"""
|
||||||
await self.create_model(db, obj, commit=True)
|
await self.create_model(db, obj, commit=True)
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, pk: list[int]) -> int:
|
async def delete(self, db: AsyncSession, pks: list[int]) -> int:
|
||||||
"""
|
"""
|
||||||
删除登录日志
|
批量删除登录日志
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param pk: 登录日志 ID 列表
|
:param pks: 登录日志 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pk)
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pks)
|
||||||
|
|
||||||
async def delete_all(self, db: AsyncSession) -> int:
|
@staticmethod
|
||||||
|
async def delete_all(db: AsyncSession) -> None:
|
||||||
"""
|
"""
|
||||||
删除所有日志
|
删除所有日志
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True)
|
await db.execute(sa_delete(LoginLog))
|
||||||
|
|
||||||
|
|
||||||
login_log_dao: CRUDLoginLog = CRUDLoginLog(LoginLog)
|
login_log_dao: CRUDLoginLog = CRUDLoginLog(LoginLog)
|
||||||
|
|||||||
@@ -2,9 +2,7 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Sequence
|
from typing import Sequence
|
||||||
|
|
||||||
from sqlalchemy import and_, asc, select
|
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import selectinload
|
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
from backend.app.admin.model import Menu
|
from backend.app.admin.model import Menu
|
||||||
@@ -44,28 +42,28 @@ class CRUDMenu(CRUDPlus[Menu]):
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
filters = {}
|
filters = {}
|
||||||
|
|
||||||
if title is not None:
|
if title is not None:
|
||||||
filters.update(title=f'%{title}%')
|
filters['title__like'] = f'%{title}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.update(status=status)
|
filters['status'] = status
|
||||||
|
|
||||||
return await self.select_models_order(db, 'sort', **filters)
|
return await self.select_models_order(db, 'sort', **filters)
|
||||||
|
|
||||||
async def get_role_menus(self, db: AsyncSession, superuser: bool, menu_ids: list[int]) -> Sequence[Menu]:
|
async def get_sidebar(self, db: AsyncSession, menu_ids: list[int] | None) -> Sequence[Menu]:
|
||||||
"""
|
"""
|
||||||
获取角色菜单列表
|
获取用户的菜单侧边栏
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param superuser: 是否超级管理员
|
|
||||||
:param menu_ids: 菜单 ID 列表
|
:param menu_ids: 菜单 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).order_by(asc(self.model.sort))
|
filters = {'type__in': [0, 1, 3, 4]}
|
||||||
filters = [self.model.type.in_([0, 1])]
|
|
||||||
if not superuser:
|
if menu_ids:
|
||||||
filters.append(self.model.id.in_(menu_ids))
|
filters['id__in'] = menu_ids
|
||||||
stmt = stmt.where(and_(*filters))
|
|
||||||
menu = await db.execute(stmt)
|
return await self.select_models_order(db, 'sort', 'asc', **filters)
|
||||||
return menu.scalars().all()
|
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: CreateMenuParam) -> None:
|
async def create(self, db: AsyncSession, obj: CreateMenuParam) -> None:
|
||||||
"""
|
"""
|
||||||
@@ -106,9 +104,7 @@ class CRUDMenu(CRUDPlus[Menu]):
|
|||||||
:param menu_id: 菜单 ID
|
:param menu_id: 菜单 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).options(selectinload(self.model.children)).where(self.model.id == menu_id)
|
menu = await self.select_model(db, menu_id, load_strategies=['children'])
|
||||||
result = await db.execute(stmt)
|
|
||||||
menu = result.scalars().first()
|
|
||||||
return menu.children
|
return menu.children
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from sqlalchemy import Select
|
from sqlalchemy import Select
|
||||||
|
from sqlalchemy import delete as sa_delete
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
@@ -21,12 +22,14 @@ class CRUDOperaLogDao(CRUDPlus[OperaLog]):
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
filters = {}
|
filters = {}
|
||||||
|
|
||||||
if username is not None:
|
if username is not None:
|
||||||
filters.update(username__like=f'%{username}%')
|
filters['username__like'] = f'%{username}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.update(status=status)
|
filters['status__eq'] = status
|
||||||
if ip is not None:
|
if ip is not None:
|
||||||
filters.update(ip__like=f'%{ip}%')
|
filters['ip__like'] = f'%{ip}%'
|
||||||
|
|
||||||
return await self.select_order('created_time', 'desc', **filters)
|
return await self.select_order('created_time', 'desc', **filters)
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: CreateOperaLogParam) -> None:
|
async def create(self, db: AsyncSession, obj: CreateOperaLogParam) -> None:
|
||||||
@@ -34,29 +37,40 @@ class CRUDOperaLogDao(CRUDPlus[OperaLog]):
|
|||||||
创建操作日志
|
创建操作日志
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param obj: 创建操作日志参数
|
:param obj: 操作日志创建参数
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
await self.create_model(db, obj)
|
await self.create_model(db, obj)
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, pk: list[int]) -> int:
|
async def bulk_create(self, db: AsyncSession, objs: list[CreateOperaLogParam]) -> None:
|
||||||
"""
|
"""
|
||||||
删除操作日志
|
批量创建操作日志
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param pk: 操作日志 ID 列表
|
:param objs: 操作日志创建参数列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pk)
|
await self.create_models(db, objs)
|
||||||
|
|
||||||
async def delete_all(self, db: AsyncSession) -> int:
|
async def delete(self, db: AsyncSession, pks: list[int]) -> int:
|
||||||
|
"""
|
||||||
|
批量删除操作日志
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pks: 操作日志 ID 列表
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pks)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def delete_all(db: AsyncSession) -> None:
|
||||||
"""
|
"""
|
||||||
删除所有日志
|
删除所有日志
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True)
|
await db.execute(sa_delete(OperaLog))
|
||||||
|
|
||||||
|
|
||||||
opera_log_dao: CRUDOperaLogDao = CRUDOperaLogDao(OperaLog)
|
opera_log_dao: CRUDOperaLogDao = CRUDOperaLogDao(OperaLog)
|
||||||
|
|||||||
@@ -2,12 +2,11 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Sequence
|
from typing import Sequence
|
||||||
|
|
||||||
from sqlalchemy import Select, and_, desc, select
|
from sqlalchemy import Select, select
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import noload, selectinload
|
|
||||||
from sqlalchemy_crud_plus import CRUDPlus
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
from backend.app.admin.model import DataScope, Menu, Role, User
|
from backend.app.admin.model import DataScope, Menu, Role
|
||||||
from backend.app.admin.schema.role import (
|
from backend.app.admin.schema.role import (
|
||||||
CreateRoleParam,
|
CreateRoleParam,
|
||||||
UpdateRoleMenuParam,
|
UpdateRoleMenuParam,
|
||||||
@@ -37,13 +36,7 @@ class CRUDRole(CRUDPlus[Role]):
|
|||||||
:param role_id: 角色 ID
|
:param role_id: 角色 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = (
|
return await self.select_model(db, role_id, load_strategies=['menus', 'scopes'])
|
||||||
select(self.model)
|
|
||||||
.options(selectinload(self.model.menus), selectinload(self.model.scopes))
|
|
||||||
.where(self.model.id == role_id)
|
|
||||||
)
|
|
||||||
role = await db.execute(stmt)
|
|
||||||
return role.scalars().first()
|
|
||||||
|
|
||||||
async def get_all(self, db: AsyncSession) -> Sequence[Role]:
|
async def get_all(self, db: AsyncSession) -> Sequence[Role]:
|
||||||
"""
|
"""
|
||||||
@@ -54,18 +47,6 @@ class CRUDRole(CRUDPlus[Role]):
|
|||||||
"""
|
"""
|
||||||
return await self.select_models(db)
|
return await self.select_models(db)
|
||||||
|
|
||||||
async def get_users(self, db: AsyncSession, user_id: int) -> Sequence[Role]:
|
|
||||||
"""
|
|
||||||
获取用户角色列表
|
|
||||||
|
|
||||||
:param db: 数据库会话
|
|
||||||
:param user_id: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
stmt = select(self.model).join(self.model.users).where(User.id == user_id)
|
|
||||||
roles = await db.execute(stmt)
|
|
||||||
return roles.scalars().all()
|
|
||||||
|
|
||||||
async def get_list(self, name: str | None, status: int | None) -> Select:
|
async def get_list(self, name: str | None, status: int | None) -> Select:
|
||||||
"""
|
"""
|
||||||
获取角色列表
|
获取角色列表
|
||||||
@@ -74,22 +55,23 @@ class CRUDRole(CRUDPlus[Role]):
|
|||||||
:param status: 角色状态
|
:param status: 角色状态
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = (
|
|
||||||
select(self.model)
|
|
||||||
.options(noload(self.model.users), noload(self.model.menus), noload(self.model.scopes))
|
|
||||||
.order_by(desc(self.model.created_time))
|
|
||||||
)
|
|
||||||
|
|
||||||
filters = []
|
filters = {}
|
||||||
|
|
||||||
if name is not None:
|
if name is not None:
|
||||||
filters.append(self.model.name.like(f'%{name}%'))
|
filters['name__like'] = f'%{name}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.append(self.model.status == status)
|
filters['status'] = status
|
||||||
|
|
||||||
if filters:
|
return await self.select_order(
|
||||||
stmt = stmt.where(and_(*filters))
|
'id',
|
||||||
|
load_strategies={
|
||||||
return stmt
|
'users': 'noload',
|
||||||
|
'menus': 'noload',
|
||||||
|
'scopes': 'noload',
|
||||||
|
},
|
||||||
|
**filters,
|
||||||
|
)
|
||||||
|
|
||||||
async def get_by_name(self, db: AsyncSession, name: str) -> Role | None:
|
async def get_by_name(self, db: AsyncSession, name: str) -> Role | None:
|
||||||
"""
|
"""
|
||||||
@@ -152,15 +134,15 @@ class CRUDRole(CRUDPlus[Role]):
|
|||||||
current_role.scopes = scopes.scalars().all()
|
current_role.scopes = scopes.scalars().all()
|
||||||
return len(current_role.scopes)
|
return len(current_role.scopes)
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, role_id: list[int]) -> int:
|
async def delete(self, db: AsyncSession, role_ids: list[int]) -> int:
|
||||||
"""
|
"""
|
||||||
删除角色
|
批量删除角色
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param role_id: 角色 ID 列表
|
:param role_ids: 角色 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.delete_model_by_column(db, allow_multiple=True, id__in=role_id)
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=role_ids)
|
||||||
|
|
||||||
|
|
||||||
role_dao: CRUDRole = CRUDRole(Role)
|
role_dao: CRUDRole = CRUDRole(Role)
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import bcrypt
|
import bcrypt
|
||||||
|
|
||||||
from sqlalchemy import and_, desc, select
|
from sqlalchemy import select
|
||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from sqlalchemy.orm import noload, selectinload
|
from sqlalchemy.orm import noload, selectinload
|
||||||
from sqlalchemy.sql import Select
|
from sqlalchemy.sql import Select
|
||||||
@@ -10,11 +10,9 @@ from sqlalchemy_crud_plus import CRUDPlus
|
|||||||
|
|
||||||
from backend.app.admin.model import Dept, Role, User
|
from backend.app.admin.model import Dept, Role, User
|
||||||
from backend.app.admin.schema.user import (
|
from backend.app.admin.schema.user import (
|
||||||
|
AddOAuth2UserParam,
|
||||||
AddUserParam,
|
AddUserParam,
|
||||||
AvatarParam,
|
|
||||||
RegisterUserParam,
|
|
||||||
UpdateUserParam,
|
UpdateUserParam,
|
||||||
UpdateUserRoleParam,
|
|
||||||
)
|
)
|
||||||
from backend.common.security.jwt import get_hash_password
|
from backend.common.security.jwt import get_hash_password
|
||||||
from backend.utils.timezone import timezone
|
from backend.utils.timezone import timezone
|
||||||
@@ -63,26 +61,6 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
"""
|
"""
|
||||||
return await self.update_model_by_column(db, {'last_login_time': timezone.now()}, username=username)
|
return await self.update_model_by_column(db, {'last_login_time': timezone.now()}, username=username)
|
||||||
|
|
||||||
async def create(self, db: AsyncSession, obj: RegisterUserParam, *, social: bool = False) -> None:
|
|
||||||
"""
|
|
||||||
创建用户
|
|
||||||
|
|
||||||
:param db: 数据库会话
|
|
||||||
:param obj: 注册用户参数
|
|
||||||
:param social: 是否社交用户
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
if not social:
|
|
||||||
salt = bcrypt.gensalt()
|
|
||||||
obj.password = get_hash_password(obj.password, salt)
|
|
||||||
dict_obj = obj.model_dump()
|
|
||||||
dict_obj.update({'is_staff': True, 'salt': salt})
|
|
||||||
else:
|
|
||||||
dict_obj = obj.model_dump()
|
|
||||||
dict_obj.update({'is_staff': True, 'salt': None})
|
|
||||||
new_user = self.model(**dict_obj)
|
|
||||||
db.add(new_user)
|
|
||||||
|
|
||||||
async def add(self, db: AsyncSession, obj: AddUserParam) -> None:
|
async def add(self, db: AsyncSession, obj: AddUserParam) -> None:
|
||||||
"""
|
"""
|
||||||
添加用户
|
添加用户
|
||||||
@@ -97,14 +75,31 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
dict_obj.update({'salt': salt})
|
dict_obj.update({'salt': salt})
|
||||||
new_user = self.model(**dict_obj)
|
new_user = self.model(**dict_obj)
|
||||||
|
|
||||||
role_list = []
|
stmt = select(Role).where(Role.id.in_(obj.roles))
|
||||||
for role_id in obj.roles:
|
roles = await db.execute(stmt)
|
||||||
role_list.append(await db.get(Role, role_id))
|
new_user.roles = roles.scalars().all()
|
||||||
new_user.roles.extend(role_list)
|
|
||||||
|
|
||||||
db.add(new_user)
|
db.add(new_user)
|
||||||
|
|
||||||
async def update_userinfo(self, db: AsyncSession, input_user: int, obj: UpdateUserParam) -> int:
|
async def add_by_oauth2(self, db: AsyncSession, obj: AddOAuth2UserParam) -> None:
|
||||||
|
"""
|
||||||
|
通过 OAuth2 添加用户
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param obj: 注册用户参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
dict_obj = obj.model_dump()
|
||||||
|
dict_obj.update({'is_staff': True, 'salt': None})
|
||||||
|
new_user = self.model(**dict_obj)
|
||||||
|
|
||||||
|
stmt = select(Role)
|
||||||
|
role = await db.execute(stmt)
|
||||||
|
new_user.roles = [role.scalars().first()] # 默认绑定第一个角色
|
||||||
|
|
||||||
|
db.add(new_user)
|
||||||
|
|
||||||
|
async def update(self, db: AsyncSession, input_user: User, obj: UpdateUserParam) -> int:
|
||||||
"""
|
"""
|
||||||
更新用户信息
|
更新用户信息
|
||||||
|
|
||||||
@@ -113,36 +108,47 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
:param obj: 更新用户参数
|
:param obj: 更新用户参数
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.update_model(db, input_user, obj)
|
role_ids = obj.roles
|
||||||
|
del obj.roles
|
||||||
|
count = await self.update_model(db, input_user.id, obj)
|
||||||
|
|
||||||
@staticmethod
|
stmt = select(Role).where(Role.id.in_(role_ids))
|
||||||
async def update_role(db: AsyncSession, input_user: User, obj: UpdateUserRoleParam) -> None:
|
roles = await db.execute(stmt)
|
||||||
|
input_user.roles = roles.scalars().all()
|
||||||
|
return count
|
||||||
|
|
||||||
|
async def update_nickname(self, db: AsyncSession, user_id: int, nickname: str) -> int:
|
||||||
"""
|
"""
|
||||||
更新用户角色
|
更新用户昵称
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param input_user: 用户对象
|
:param user_id: 用户 ID
|
||||||
:param obj: 更新角色参数
|
:param nickname: 用户昵称
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
for i in list(input_user.roles):
|
return await self.update_model(db, user_id, {'nickname': nickname})
|
||||||
input_user.roles.remove(i)
|
|
||||||
|
|
||||||
role_list = []
|
async def update_avatar(self, db: AsyncSession, user_id: int, avatar: str) -> int:
|
||||||
for role_id in obj.roles:
|
|
||||||
role_list.append(await db.get(Role, role_id))
|
|
||||||
input_user.roles.extend(role_list)
|
|
||||||
|
|
||||||
async def update_avatar(self, db: AsyncSession, input_user: int, avatar: AvatarParam) -> int:
|
|
||||||
"""
|
"""
|
||||||
更新用户头像
|
更新用户头像
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param input_user: 用户 ID
|
:param user_id: 用户 ID
|
||||||
:param avatar: 头像地址
|
:param avatar: 头像地址
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.update_model(db, input_user, {'avatar': str(avatar.url)})
|
return await self.update_model(db, user_id, {'avatar': avatar})
|
||||||
|
|
||||||
|
async def update_email(self, db: AsyncSession, user_id: int, email: str) -> int:
|
||||||
|
"""
|
||||||
|
更新用户邮箱
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param user_id: 用户 ID
|
||||||
|
:param email: 邮箱
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.update_model(db, user_id, {'email': email})
|
||||||
|
|
||||||
async def delete(self, db: AsyncSession, user_id: int) -> int:
|
async def delete(self, db: AsyncSession, user_id: int) -> int:
|
||||||
"""
|
"""
|
||||||
@@ -156,7 +162,7 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
|
|
||||||
async def check_email(self, db: AsyncSession, email: str) -> User | None:
|
async def check_email(self, db: AsyncSession, email: str) -> User | None:
|
||||||
"""
|
"""
|
||||||
检查邮箱是否已被注册
|
检查邮箱是否已被绑定
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param email: 电子邮箱
|
:param email: 电子邮箱
|
||||||
@@ -164,16 +170,18 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
"""
|
"""
|
||||||
return await self.select_model_by_column(db, email=email)
|
return await self.select_model_by_column(db, email=email)
|
||||||
|
|
||||||
async def reset_password(self, db: AsyncSession, pk: int, new_pwd: str) -> int:
|
async def reset_password(self, db: AsyncSession, pk: int, password: str) -> int:
|
||||||
"""
|
"""
|
||||||
重置用户密码
|
重置用户密码
|
||||||
|
|
||||||
:param db: 数据库会话
|
:param db: 数据库会话
|
||||||
:param pk: 用户 ID
|
:param pk: 用户 ID
|
||||||
:param new_pwd: 新密码(已加密)
|
:param password: 新密码
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
return await self.update_model(db, pk, {'password': new_pwd})
|
salt = bcrypt.gensalt()
|
||||||
|
new_pwd = get_hash_password(password, salt)
|
||||||
|
return await self.update_model(db, pk, {'password': new_pwd, 'salt': salt})
|
||||||
|
|
||||||
async def get_list(self, dept: int | None, username: str | None, phone: str | None, status: int | None) -> Select:
|
async def get_list(self, dept: int | None, username: str | None, phone: str | None, status: int | None) -> Select:
|
||||||
"""
|
"""
|
||||||
@@ -185,74 +193,26 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
:param status: 用户状态
|
:param status: 用户状态
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = (
|
filters = {}
|
||||||
select(self.model)
|
|
||||||
.options(
|
|
||||||
selectinload(self.model.dept).options(noload(Dept.parent), noload(Dept.children), noload(Dept.users)),
|
|
||||||
noload(self.model.socials),
|
|
||||||
selectinload(self.model.roles).options(noload(Role.users), noload(Role.menus), noload(Role.rules)),
|
|
||||||
)
|
|
||||||
.order_by(desc(self.model.join_time))
|
|
||||||
)
|
|
||||||
|
|
||||||
filters = []
|
|
||||||
if dept:
|
if dept:
|
||||||
filters.append(self.model.dept_id == dept)
|
filters['dept_id'] = dept
|
||||||
if username:
|
if username:
|
||||||
filters.append(self.model.username.like(f'%{username}%'))
|
filters['username__like'] = f'%{username}%'
|
||||||
if phone:
|
if phone:
|
||||||
filters.append(self.model.phone.like(f'%{phone}%'))
|
filters['phone_like'] = f'%{phone}%'
|
||||||
if status is not None:
|
if status is not None:
|
||||||
filters.append(self.model.status == status)
|
filters['status'] = status
|
||||||
|
|
||||||
if filters:
|
return await self.select_order(
|
||||||
stmt = stmt.where(and_(*filters))
|
'id',
|
||||||
|
'desc',
|
||||||
return stmt
|
load_options=[
|
||||||
|
selectinload(self.model.dept).options(noload(Dept.parent), noload(Dept.children), noload(Dept.users)),
|
||||||
async def get_super(self, db: AsyncSession, user_id: int) -> bool:
|
selectinload(self.model.roles).options(noload(Role.users), noload(Role.menus), noload(Role.scopes)),
|
||||||
"""
|
],
|
||||||
获取用户是否为超级管理员
|
**filters,
|
||||||
|
)
|
||||||
:param db: 数据库会话
|
|
||||||
:param user_id: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
user = await self.get(db, user_id)
|
|
||||||
return user.is_superuser
|
|
||||||
|
|
||||||
async def get_staff(self, db: AsyncSession, user_id: int) -> bool:
|
|
||||||
"""
|
|
||||||
获取用户是否可以登录后台
|
|
||||||
|
|
||||||
:param db: 数据库会话
|
|
||||||
:param user_id: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
user = await self.get(db, user_id)
|
|
||||||
return user.is_staff
|
|
||||||
|
|
||||||
async def get_status(self, db: AsyncSession, user_id: int) -> int:
|
|
||||||
"""
|
|
||||||
获取用户状态
|
|
||||||
|
|
||||||
:param db: 数据库会话
|
|
||||||
:param user_id: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
user = await self.get(db, user_id)
|
|
||||||
return user.status
|
|
||||||
|
|
||||||
async def get_multi_login(self, db: AsyncSession, user_id: int) -> bool:
|
|
||||||
"""
|
|
||||||
获取用户是否允许多端登录
|
|
||||||
|
|
||||||
:param db: 数据库会话
|
|
||||||
:param user_id: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
user = await self.get(db, user_id)
|
|
||||||
return user.is_multi_login
|
|
||||||
|
|
||||||
async def set_super(self, db: AsyncSession, user_id: int, is_super: bool) -> int:
|
async def set_super(self, db: AsyncSession, user_id: int, is_super: bool) -> int:
|
||||||
"""
|
"""
|
||||||
@@ -309,22 +269,19 @@ class CRUDUser(CRUDPlus[User]):
|
|||||||
:param username: 用户名
|
:param username: 用户名
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
stmt = select(self.model).options(
|
filters = {}
|
||||||
selectinload(self.model.dept),
|
|
||||||
selectinload(self.model.roles).options(selectinload(Role.menus), selectinload(Role.scopes)),
|
|
||||||
)
|
|
||||||
|
|
||||||
filters = []
|
|
||||||
if user_id:
|
if user_id:
|
||||||
filters.append(self.model.id == user_id)
|
filters['id'] = user_id
|
||||||
if username:
|
if username:
|
||||||
filters.append(self.model.username == username)
|
filters['username'] = username
|
||||||
|
|
||||||
if filters:
|
return await self.select_model_by_column(
|
||||||
stmt = stmt.where(and_(*filters))
|
db,
|
||||||
|
load_options=[selectinload(self.model.roles).options(selectinload(Role.menus), selectinload(Role.scopes))],
|
||||||
user = await db.execute(stmt)
|
load_strategies=['dept'],
|
||||||
return user.scalars().first()
|
**filters,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
user_dao: CRUDUser = CRUDUser(User)
|
user_dao: CRUDUser = CRUDUser(User)
|
||||||
|
|||||||
@@ -8,4 +8,3 @@ from backend.app.admin.model.menu import Menu
|
|||||||
from backend.app.admin.model.opera_log import OperaLog
|
from backend.app.admin.model.opera_log import OperaLog
|
||||||
from backend.app.admin.model.role import Role
|
from backend.app.admin.model.role import Role
|
||||||
from backend.app.admin.model.user import User
|
from backend.app.admin.model.user import User
|
||||||
from backend.app.admin.model.user_social import UserSocial
|
|
||||||
|
|||||||
@@ -4,9 +4,10 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from typing import TYPE_CHECKING
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
from sqlalchemy import ForeignKey, String
|
from sqlalchemy import String
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
|
|
||||||
|
from backend.app.admin.model.m2m import sys_data_scope_rule
|
||||||
from backend.common.model import Base, id_key
|
from backend.common.model import Base, id_key
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
@@ -28,8 +29,5 @@ class DataRule(Base):
|
|||||||
)
|
)
|
||||||
value: Mapped[str] = mapped_column(String(255), comment='规则值')
|
value: Mapped[str] = mapped_column(String(255), comment='规则值')
|
||||||
|
|
||||||
# 数据范围规则一对多
|
# 数据范围规则多对多
|
||||||
scope_id: Mapped[int | None] = mapped_column(
|
scopes: Mapped[list[DataScope]] = relationship(init=False, secondary=sys_data_scope_rule, back_populates='rules')
|
||||||
ForeignKey('sys_data_scope.id', ondelete='SET NULL'), default=None, comment='数据范围关联 ID'
|
|
||||||
)
|
|
||||||
scope: Mapped[DataScope] = relationship(init=False, back_populates='rules')
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING
|
|||||||
from sqlalchemy import String
|
from sqlalchemy import String
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
|
|
||||||
from backend.app.admin.model.m2m import sys_role_data_scope
|
from backend.app.admin.model.m2m import sys_data_scope_rule, sys_role_data_scope
|
||||||
from backend.common.model import Base, id_key
|
from backend.common.model import Base, id_key
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
@@ -23,8 +23,8 @@ class DataScope(Base):
|
|||||||
name: Mapped[str] = mapped_column(String(50), unique=True, comment='名称')
|
name: Mapped[str] = mapped_column(String(50), unique=True, comment='名称')
|
||||||
status: Mapped[int] = mapped_column(default=1, comment='状态(0停用 1正常)')
|
status: Mapped[int] = mapped_column(default=1, comment='状态(0停用 1正常)')
|
||||||
|
|
||||||
# 数据范围规则一对多
|
# 数据范围规则多对多
|
||||||
rules: Mapped[list[DataRule]] = relationship(init=False, back_populates='scope')
|
rules: Mapped[list[DataRule]] = relationship(init=False, secondary=sys_data_scope_rule, back_populates='scopes')
|
||||||
|
|
||||||
# 角色数据范围多对多
|
# 角色数据范围多对多
|
||||||
roles: Mapped[list[Role]] = relationship(init=False, secondary=sys_role_data_scope, back_populates='scopes')
|
roles: Mapped[list[Role]] = relationship(init=False, secondary=sys_role_data_scope, back_populates='scopes')
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from typing import TYPE_CHECKING, Optional
|
from typing import TYPE_CHECKING, Optional
|
||||||
|
|
||||||
from sqlalchemy import Boolean, ForeignKey, String
|
from sqlalchemy import BigInteger, Boolean, ForeignKey, String
|
||||||
from sqlalchemy.dialects.postgresql import INTEGER
|
from sqlalchemy.dialects.postgresql import INTEGER
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
|
|
||||||
@@ -32,7 +32,7 @@ class Dept(Base):
|
|||||||
|
|
||||||
# 父级部门一对多
|
# 父级部门一对多
|
||||||
parent_id: Mapped[int | None] = mapped_column(
|
parent_id: Mapped[int | None] = mapped_column(
|
||||||
ForeignKey('sys_dept.id', ondelete='SET NULL'), default=None, index=True, comment='父部门ID'
|
BigInteger, ForeignKey('sys_dept.id', ondelete='SET NULL'), default=None, index=True, comment='父部门ID'
|
||||||
)
|
)
|
||||||
parent: Mapped[Optional['Dept']] = relationship(init=False, back_populates='children', remote_side=[id])
|
parent: Mapped[Optional['Dept']] = relationship(init=False, back_populates='children', remote_side=[id])
|
||||||
children: Mapped[Optional[list['Dept']]] = relationship(init=False, back_populates='parent')
|
children: Mapped[Optional[list['Dept']]] = relationship(init=False, back_populates='parent')
|
||||||
|
|||||||
@@ -2,12 +2,12 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from sqlalchemy import DateTime, String
|
from sqlalchemy import String
|
||||||
from sqlalchemy.dialects.mysql import LONGTEXT
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
||||||
from sqlalchemy.dialects.postgresql import TEXT
|
from sqlalchemy.dialects.postgresql import TEXT
|
||||||
from sqlalchemy.orm import Mapped, mapped_column
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
|
||||||
from backend.common.model import DataClassBase, id_key
|
from backend.common.model import DataClassBase, TimeZone, id_key
|
||||||
from backend.utils.timezone import timezone
|
from backend.utils.timezone import timezone
|
||||||
|
|
||||||
|
|
||||||
@@ -29,7 +29,7 @@ class LoginLog(DataClassBase):
|
|||||||
browser: Mapped[str | None] = mapped_column(String(50), comment='浏览器')
|
browser: Mapped[str | None] = mapped_column(String(50), comment='浏览器')
|
||||||
device: Mapped[str | None] = mapped_column(String(50), comment='设备')
|
device: Mapped[str | None] = mapped_column(String(50), comment='设备')
|
||||||
msg: Mapped[str] = mapped_column(LONGTEXT().with_variant(TEXT, 'postgresql'), comment='提示消息')
|
msg: Mapped[str] = mapped_column(LONGTEXT().with_variant(TEXT, 'postgresql'), comment='提示消息')
|
||||||
login_time: Mapped[datetime] = mapped_column(DateTime(timezone=True), comment='登录时间')
|
login_time: Mapped[datetime] = mapped_column(TimeZone, comment='登录时间')
|
||||||
created_time: Mapped[datetime] = mapped_column(
|
created_time: Mapped[datetime] = mapped_column(
|
||||||
DateTime(timezone=True), init=False, default_factory=timezone.now, comment='创建时间'
|
TimeZone, init=False, default_factory=timezone.now, comment='创建时间'
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,35 +1,55 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from sqlalchemy import INT, Column, ForeignKey, Integer, Table
|
from sqlalchemy import BigInteger, Column, ForeignKey, Table
|
||||||
|
|
||||||
from backend.common.model import MappedBase
|
from backend.common.model import MappedBase
|
||||||
|
|
||||||
sys_user_role = Table(
|
sys_user_role = Table(
|
||||||
'sys_user_role',
|
'sys_user_role',
|
||||||
MappedBase.metadata,
|
MappedBase.metadata,
|
||||||
Column('id', INT, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键ID'),
|
Column('id', BigInteger, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键ID'),
|
||||||
Column('user_id', Integer, ForeignKey('sys_user.id', ondelete='CASCADE'), primary_key=True, comment='用户ID'),
|
Column('user_id', BigInteger, ForeignKey('sys_user.id', ondelete='CASCADE'), primary_key=True, comment='用户ID'),
|
||||||
Column('role_id', Integer, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色ID'),
|
Column('role_id', BigInteger, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色ID'),
|
||||||
)
|
)
|
||||||
|
|
||||||
sys_role_menu = Table(
|
sys_role_menu = Table(
|
||||||
'sys_role_menu',
|
'sys_role_menu',
|
||||||
MappedBase.metadata,
|
MappedBase.metadata,
|
||||||
Column('id', INT, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键ID'),
|
Column('id', BigInteger, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键ID'),
|
||||||
Column('role_id', Integer, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色ID'),
|
Column('role_id', BigInteger, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色ID'),
|
||||||
Column('menu_id', Integer, ForeignKey('sys_menu.id', ondelete='CASCADE'), primary_key=True, comment='菜单ID'),
|
Column('menu_id', BigInteger, ForeignKey('sys_menu.id', ondelete='CASCADE'), primary_key=True, comment='菜单ID'),
|
||||||
)
|
)
|
||||||
|
|
||||||
sys_role_data_scope = Table(
|
sys_role_data_scope = Table(
|
||||||
'sys_role_data_scope',
|
'sys_role_data_scope',
|
||||||
MappedBase.metadata,
|
MappedBase.metadata,
|
||||||
Column('id', INT, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键 ID'),
|
Column('id', BigInteger, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键 ID'),
|
||||||
Column('role_id', Integer, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色 ID'),
|
Column('role_id', BigInteger, ForeignKey('sys_role.id', ondelete='CASCADE'), primary_key=True, comment='角色 ID'),
|
||||||
Column(
|
Column(
|
||||||
'data_scope_id',
|
'data_scope_id',
|
||||||
Integer,
|
BigInteger,
|
||||||
ForeignKey('sys_data_scope.id', ondelete='CASCADE'),
|
ForeignKey('sys_data_scope.id', ondelete='CASCADE'),
|
||||||
primary_key=True,
|
primary_key=True,
|
||||||
comment='数据范围 ID',
|
comment='数据范围 ID',
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
sys_data_scope_rule = Table(
|
||||||
|
'sys_data_scope_rule',
|
||||||
|
MappedBase.metadata,
|
||||||
|
Column('id', BigInteger, primary_key=True, unique=True, index=True, autoincrement=True, comment='主键ID'),
|
||||||
|
Column(
|
||||||
|
'data_scope_id',
|
||||||
|
BigInteger,
|
||||||
|
ForeignKey('sys_data_scope.id', ondelete='CASCADE'),
|
||||||
|
primary_key=True,
|
||||||
|
comment='数据范围 ID',
|
||||||
|
),
|
||||||
|
Column(
|
||||||
|
'data_rule_id',
|
||||||
|
BigInteger,
|
||||||
|
ForeignKey('sys_data_rule.id', ondelete='CASCADE'),
|
||||||
|
primary_key=True,
|
||||||
|
comment='数据规则 ID',
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from typing import TYPE_CHECKING, Optional
|
from typing import TYPE_CHECKING, Optional
|
||||||
|
|
||||||
from sqlalchemy import ForeignKey, String
|
from sqlalchemy import BigInteger, ForeignKey, String
|
||||||
from sqlalchemy.dialects.mysql import LONGTEXT
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
||||||
from sqlalchemy.dialects.postgresql import TEXT
|
from sqlalchemy.dialects.postgresql import TEXT
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
@@ -24,10 +24,10 @@ class Menu(Base):
|
|||||||
id: Mapped[id_key] = mapped_column(init=False)
|
id: Mapped[id_key] = mapped_column(init=False)
|
||||||
title: Mapped[str] = mapped_column(String(50), comment='菜单标题')
|
title: Mapped[str] = mapped_column(String(50), comment='菜单标题')
|
||||||
name: Mapped[str] = mapped_column(String(50), comment='菜单名称')
|
name: Mapped[str] = mapped_column(String(50), comment='菜单名称')
|
||||||
path: Mapped[str] = mapped_column(String(200), comment='路由地址')
|
path: Mapped[str | None] = mapped_column(String(200), comment='路由地址')
|
||||||
sort: Mapped[int] = mapped_column(default=0, comment='排序')
|
sort: Mapped[int] = mapped_column(default=0, comment='排序')
|
||||||
icon: Mapped[str | None] = mapped_column(String(100), default=None, comment='菜单图标')
|
icon: Mapped[str | None] = mapped_column(String(100), default=None, comment='菜单图标')
|
||||||
type: Mapped[int] = mapped_column(default=0, comment='菜单类型(0目录 1菜单 2按钮)')
|
type: Mapped[int] = mapped_column(default=0, comment='菜单类型(0目录 1菜单 2按钮 3内嵌 4外链)')
|
||||||
component: Mapped[str | None] = mapped_column(String(255), default=None, comment='组件路径')
|
component: Mapped[str | None] = mapped_column(String(255), default=None, comment='组件路径')
|
||||||
perms: Mapped[str | None] = mapped_column(String(100), default=None, comment='权限标识')
|
perms: Mapped[str | None] = mapped_column(String(100), default=None, comment='权限标识')
|
||||||
status: Mapped[int] = mapped_column(default=1, comment='菜单状态(0停用 1正常)')
|
status: Mapped[int] = mapped_column(default=1, comment='菜单状态(0停用 1正常)')
|
||||||
@@ -42,7 +42,7 @@ class Menu(Base):
|
|||||||
|
|
||||||
# 父级菜单一对多
|
# 父级菜单一对多
|
||||||
parent_id: Mapped[int | None] = mapped_column(
|
parent_id: Mapped[int | None] = mapped_column(
|
||||||
ForeignKey('sys_menu.id', ondelete='SET NULL'), default=None, index=True, comment='父菜单ID'
|
BigInteger, ForeignKey('sys_menu.id', ondelete='SET NULL'), default=None, index=True, comment='父菜单ID'
|
||||||
)
|
)
|
||||||
parent: Mapped[Optional['Menu']] = relationship(init=False, back_populates='children', remote_side=[id])
|
parent: Mapped[Optional['Menu']] = relationship(init=False, back_populates='children', remote_side=[id])
|
||||||
children: Mapped[Optional[list['Menu']]] = relationship(init=False, back_populates='parent')
|
children: Mapped[Optional[list['Menu']]] = relationship(init=False, back_populates='parent')
|
||||||
|
|||||||
@@ -2,12 +2,12 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from sqlalchemy import DateTime, String
|
from sqlalchemy import String
|
||||||
from sqlalchemy.dialects.mysql import JSON, LONGTEXT
|
from sqlalchemy.dialects.mysql import JSON, LONGTEXT
|
||||||
from sqlalchemy.dialects.postgresql import TEXT
|
from sqlalchemy.dialects.postgresql import TEXT
|
||||||
from sqlalchemy.orm import Mapped, mapped_column
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
|
||||||
from backend.common.model import DataClassBase, id_key
|
from backend.common.model import DataClassBase, TimeZone, id_key
|
||||||
from backend.utils.timezone import timezone
|
from backend.utils.timezone import timezone
|
||||||
|
|
||||||
|
|
||||||
@@ -35,7 +35,7 @@ class OperaLog(DataClassBase):
|
|||||||
code: Mapped[str] = mapped_column(String(20), insert_default='200', comment='操作状态码')
|
code: Mapped[str] = mapped_column(String(20), insert_default='200', comment='操作状态码')
|
||||||
msg: Mapped[str | None] = mapped_column(LONGTEXT().with_variant(TEXT, 'postgresql'), comment='提示消息')
|
msg: Mapped[str | None] = mapped_column(LONGTEXT().with_variant(TEXT, 'postgresql'), comment='提示消息')
|
||||||
cost_time: Mapped[float] = mapped_column(insert_default=0.0, comment='请求耗时(ms)')
|
cost_time: Mapped[float] = mapped_column(insert_default=0.0, comment='请求耗时(ms)')
|
||||||
opera_time: Mapped[datetime] = mapped_column(DateTime(timezone=True), comment='操作时间')
|
opera_time: Mapped[datetime] = mapped_column(TimeZone, comment='操作时间')
|
||||||
created_time: Mapped[datetime] = mapped_column(
|
created_time: Mapped[datetime] = mapped_column(
|
||||||
DateTime(timezone=True), init=False, default_factory=timezone.now, comment='创建时间'
|
TimeZone, init=False, default_factory=timezone.now, comment='创建时间'
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -4,9 +4,9 @@ from __future__ import annotations
|
|||||||
|
|
||||||
from typing import TYPE_CHECKING
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
from sqlalchemy import String
|
from sqlalchemy import Boolean, String
|
||||||
from sqlalchemy.dialects.mysql import LONGTEXT
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
||||||
from sqlalchemy.dialects.postgresql import TEXT
|
from sqlalchemy.dialects.postgresql import INTEGER, TEXT
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
|
|
||||||
from backend.app.admin.model.m2m import sys_role_data_scope, sys_role_menu, sys_user_role
|
from backend.app.admin.model.m2m import sys_role_data_scope, sys_role_menu, sys_user_role
|
||||||
@@ -24,6 +24,9 @@ class Role(Base):
|
|||||||
id: Mapped[id_key] = mapped_column(init=False)
|
id: Mapped[id_key] = mapped_column(init=False)
|
||||||
name: Mapped[str] = mapped_column(String(20), unique=True, comment='角色名称')
|
name: Mapped[str] = mapped_column(String(20), unique=True, comment='角色名称')
|
||||||
status: Mapped[int] = mapped_column(default=1, comment='角色状态(0停用 1正常)')
|
status: Mapped[int] = mapped_column(default=1, comment='角色状态(0停用 1正常)')
|
||||||
|
is_filter_scopes: Mapped[bool] = mapped_column(
|
||||||
|
Boolean().with_variant(INTEGER, 'postgresql'), default=True, comment='过滤数据权限(0否 1是)'
|
||||||
|
)
|
||||||
remark: Mapped[str | None] = mapped_column(
|
remark: Mapped[str | None] = mapped_column(
|
||||||
LONGTEXT().with_variant(TEXT, 'postgresql'), default=None, comment='备注'
|
LONGTEXT().with_variant(TEXT, 'postgresql'), default=None, comment='备注'
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -5,17 +5,17 @@ from __future__ import annotations
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import TYPE_CHECKING
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
from sqlalchemy import VARBINARY, Boolean, DateTime, ForeignKey, String
|
from sqlalchemy import VARBINARY, Boolean, ForeignKey, String
|
||||||
from sqlalchemy.dialects.postgresql import BYTEA, INTEGER
|
from sqlalchemy.dialects.postgresql import BYTEA, INTEGER
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||||
|
|
||||||
from backend.app.admin.model.m2m import sys_user_role
|
from backend.app.admin.model.m2m import sys_user_role
|
||||||
from backend.common.model import Base, id_key
|
from backend.common.model import Base, TimeZone, id_key
|
||||||
from backend.database.db import uuid4_str
|
from backend.database.db import uuid4_str
|
||||||
from backend.utils.timezone import timezone
|
from backend.utils.timezone import timezone
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from backend.app.admin.model import Dept, Role, UserSocial
|
from backend.app.admin.model import Dept, Role
|
||||||
|
|
||||||
|
|
||||||
class User(Base):
|
class User(Base):
|
||||||
@@ -26,27 +26,25 @@ class User(Base):
|
|||||||
id: Mapped[id_key] = mapped_column(init=False)
|
id: Mapped[id_key] = mapped_column(init=False)
|
||||||
uuid: Mapped[str] = mapped_column(String(50), init=False, default_factory=uuid4_str, unique=True)
|
uuid: Mapped[str] = mapped_column(String(50), init=False, default_factory=uuid4_str, unique=True)
|
||||||
username: Mapped[str] = mapped_column(String(20), unique=True, index=True, comment='用户名')
|
username: Mapped[str] = mapped_column(String(20), unique=True, index=True, comment='用户名')
|
||||||
nickname: Mapped[str] = mapped_column(String(20), unique=True, comment='昵称')
|
nickname: Mapped[str] = mapped_column(String(20), comment='昵称')
|
||||||
password: Mapped[str | None] = mapped_column(String(255), comment='密码')
|
password: Mapped[str | None] = mapped_column(String(255), comment='密码')
|
||||||
salt: Mapped[bytes | None] = mapped_column(VARBINARY(255).with_variant(BYTEA(255), 'postgresql'), comment='加密盐')
|
salt: Mapped[bytes | None] = mapped_column(VARBINARY(255).with_variant(BYTEA(255), 'postgresql'), comment='加密盐')
|
||||||
email: Mapped[str] = mapped_column(String(50), unique=True, index=True, comment='邮箱')
|
email: Mapped[str | None] = mapped_column(String(50), default=None, unique=True, index=True, comment='邮箱')
|
||||||
|
phone: Mapped[str | None] = mapped_column(String(11), default=None, comment='手机号')
|
||||||
|
avatar: Mapped[str | None] = mapped_column(String(255), default=None, comment='头像')
|
||||||
|
status: Mapped[int] = mapped_column(default=1, index=True, comment='用户账号状态(0停用 1正常)')
|
||||||
is_superuser: Mapped[bool] = mapped_column(
|
is_superuser: Mapped[bool] = mapped_column(
|
||||||
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='超级权限(0否 1是)'
|
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='超级权限(0否 1是)'
|
||||||
)
|
)
|
||||||
is_staff: Mapped[bool] = mapped_column(
|
is_staff: Mapped[bool] = mapped_column(
|
||||||
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='后台管理登陆(0否 1是)'
|
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='后台管理登陆(0否 1是)'
|
||||||
)
|
)
|
||||||
status: Mapped[int] = mapped_column(default=1, index=True, comment='用户账号状态(0停用 1正常)')
|
|
||||||
is_multi_login: Mapped[bool] = mapped_column(
|
is_multi_login: Mapped[bool] = mapped_column(
|
||||||
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='是否重复登陆(0否 1是)'
|
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='是否重复登陆(0否 1是)'
|
||||||
)
|
)
|
||||||
avatar: Mapped[str | None] = mapped_column(String(255), default=None, comment='头像')
|
join_time: Mapped[datetime] = mapped_column(TimeZone, init=False, default_factory=timezone.now, comment='注册时间')
|
||||||
phone: Mapped[str | None] = mapped_column(String(11), default=None, comment='手机号')
|
|
||||||
join_time: Mapped[datetime] = mapped_column(
|
|
||||||
DateTime(timezone=True), init=False, default_factory=timezone.now, comment='注册时间'
|
|
||||||
)
|
|
||||||
last_login_time: Mapped[datetime | None] = mapped_column(
|
last_login_time: Mapped[datetime | None] = mapped_column(
|
||||||
DateTime(timezone=True), init=False, onupdate=timezone.now, comment='上次登录'
|
TimeZone, init=False, onupdate=timezone.now, comment='上次登录'
|
||||||
)
|
)
|
||||||
|
|
||||||
# 部门用户一对多
|
# 部门用户一对多
|
||||||
@@ -55,8 +53,5 @@ class User(Base):
|
|||||||
)
|
)
|
||||||
dept: Mapped[Dept | None] = relationship(init=False, back_populates='users')
|
dept: Mapped[Dept | None] = relationship(init=False, back_populates='users')
|
||||||
|
|
||||||
# 用户社交信息一对多
|
|
||||||
socials: Mapped[list[UserSocial]] = relationship(init=False, back_populates='user')
|
|
||||||
|
|
||||||
# 用户角色多对多
|
# 用户角色多对多
|
||||||
roles: Mapped[list[Role]] = relationship(init=False, secondary=sys_user_role, back_populates='users')
|
roles: Mapped[list[Role]] = relationship(init=False, secondary=sys_user_role, back_populates='users')
|
||||||
|
|||||||
@@ -1,33 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from __future__ import annotations
|
|
||||||
|
|
||||||
from typing import TYPE_CHECKING
|
|
||||||
|
|
||||||
from sqlalchemy import ForeignKey, String
|
|
||||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
||||||
|
|
||||||
from backend.common.model import Base, id_key
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from backend.app.admin.model import User
|
|
||||||
|
|
||||||
|
|
||||||
class UserSocial(Base):
|
|
||||||
"""用户社交表(OAuth2)"""
|
|
||||||
|
|
||||||
__tablename__ = 'sys_user_social'
|
|
||||||
|
|
||||||
id: Mapped[id_key] = mapped_column(init=False)
|
|
||||||
source: Mapped[str] = mapped_column(String(20), comment='第三方用户来源')
|
|
||||||
open_id: Mapped[str | None] = mapped_column(String(20), default=None, comment='第三方用户的 open id')
|
|
||||||
uid: Mapped[str | None] = mapped_column(String(20), default=None, comment='第三方用户的 ID')
|
|
||||||
union_id: Mapped[str | None] = mapped_column(String(20), default=None, comment='第三方用户的 union id')
|
|
||||||
scope: Mapped[str | None] = mapped_column(String(120), default=None, comment='第三方用户授予的权限')
|
|
||||||
code: Mapped[str | None] = mapped_column(String(50), default=None, comment='用户的授权 code')
|
|
||||||
|
|
||||||
# 用户社交信息一对多
|
|
||||||
user_id: Mapped[int | None] = mapped_column(
|
|
||||||
ForeignKey('sys_user.id', ondelete='SET NULL'), default=None, comment='用户关联ID'
|
|
||||||
)
|
|
||||||
user: Mapped[User | None] = relationship(init=False, back_populates='socials')
|
|
||||||
@@ -14,8 +14,8 @@ class DataRuleSchemaBase(SchemaBase):
|
|||||||
name: str = Field(description='规则名称')
|
name: str = Field(description='规则名称')
|
||||||
model: str = Field(description='模型名称')
|
model: str = Field(description='模型名称')
|
||||||
column: str = Field(description='字段名称')
|
column: str = Field(description='字段名称')
|
||||||
operator: RoleDataRuleOperatorType = Field(RoleDataRuleOperatorType.AND, description='操作符(AND/OR)')
|
operator: RoleDataRuleOperatorType = Field(description='操作符(AND/OR)')
|
||||||
expression: RoleDataRuleExpressionType = Field(RoleDataRuleExpressionType.eq, description='表达式类型')
|
expression: RoleDataRuleExpressionType = Field(description='表达式类型')
|
||||||
value: str = Field(description='规则值')
|
value: str = Field(description='规则值')
|
||||||
|
|
||||||
|
|
||||||
@@ -27,6 +27,12 @@ class UpdateDataRuleParam(DataRuleSchemaBase):
|
|||||||
"""更新数据规则参数"""
|
"""更新数据规则参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteDataRuleParam(SchemaBase):
|
||||||
|
"""删除数据规则参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='规则 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class GetDataRuleDetail(DataRuleSchemaBase):
|
class GetDataRuleDetail(DataRuleSchemaBase):
|
||||||
"""数据规则详情"""
|
"""数据规则详情"""
|
||||||
|
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ class DataScopeBase(SchemaBase):
|
|||||||
"""数据范围基础模型"""
|
"""数据范围基础模型"""
|
||||||
|
|
||||||
name: str = Field(description='名称')
|
name: str = Field(description='名称')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
status: StatusType = Field(description='状态')
|
||||||
|
|
||||||
|
|
||||||
class CreateDataScopeParam(DataScopeBase):
|
class CreateDataScopeParam(DataScopeBase):
|
||||||
@@ -30,6 +30,12 @@ class UpdateDataScopeRuleParam(SchemaBase):
|
|||||||
rules: list[int] = Field(description='数据规则 ID 列表')
|
rules: list[int] = Field(description='数据规则 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteDataScopeParam(SchemaBase):
|
||||||
|
"""删除数据范围参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='数据范围 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class GetDataScopeDetail(DataScopeBase):
|
class GetDataScopeDetail(DataScopeBase):
|
||||||
"""数据范围详情"""
|
"""数据范围详情"""
|
||||||
|
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ class DeptSchemaBase(SchemaBase):
|
|||||||
leader: str | None = Field(None, description='负责人')
|
leader: str | None = Field(None, description='负责人')
|
||||||
phone: CustomPhoneNumber | None = Field(None, description='联系电话')
|
phone: CustomPhoneNumber | None = Field(None, description='联系电话')
|
||||||
email: CustomEmailStr | None = Field(None, description='邮箱')
|
email: CustomEmailStr | None = Field(None, description='邮箱')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
status: StatusType = Field(description='状态')
|
||||||
|
|
||||||
|
|
||||||
class CreateDeptParam(DeptSchemaBase):
|
class CreateDeptParam(DeptSchemaBase):
|
||||||
@@ -37,3 +37,9 @@ class GetDeptDetail(DeptSchemaBase):
|
|||||||
del_flag: bool = Field(description='是否删除')
|
del_flag: bool = Field(description='是否删除')
|
||||||
created_time: datetime = Field(description='创建时间')
|
created_time: datetime = Field(description='创建时间')
|
||||||
updated_time: datetime | None = Field(None, description='更新时间')
|
updated_time: datetime | None = Field(None, description='更新时间')
|
||||||
|
|
||||||
|
|
||||||
|
class GetDeptTree(GetDeptDetail):
|
||||||
|
"""获取部门树"""
|
||||||
|
|
||||||
|
children: list['GetDeptTree'] | None = Field(None, description='子菜单')
|
||||||
|
|||||||
@@ -33,6 +33,12 @@ class UpdateLoginLogParam(LoginLogSchemaBase):
|
|||||||
"""更新登录日志参数"""
|
"""更新登录日志参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteLoginLogParam(SchemaBase):
|
||||||
|
"""删除登录日志参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='登录日志 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class GetLoginLogDetail(LoginLogSchemaBase):
|
class GetLoginLogDetail(LoginLogSchemaBase):
|
||||||
"""登录日志详情"""
|
"""登录日志详情"""
|
||||||
|
|
||||||
|
|||||||
@@ -13,16 +13,16 @@ class MenuSchemaBase(SchemaBase):
|
|||||||
|
|
||||||
title: str = Field(description='菜单标题')
|
title: str = Field(description='菜单标题')
|
||||||
name: str = Field(description='菜单名称')
|
name: str = Field(description='菜单名称')
|
||||||
path: str = Field(description='路由路径')
|
path: str | None = Field(None, description='路由地址')
|
||||||
parent_id: int | None = Field(None, description='菜单父级 ID')
|
parent_id: int | None = Field(None, description='菜单父级 ID')
|
||||||
sort: int = Field(0, ge=0, description='排序')
|
sort: int = Field(0, ge=0, description='排序')
|
||||||
icon: str | None = Field(None, description='图标')
|
icon: str | None = Field(None, description='图标')
|
||||||
type: MenuType = Field(MenuType.directory, description='菜单类型(0目录 1菜单 2按钮)')
|
type: MenuType = Field(description='菜单类型(0目录 1菜单 2按钮 3内嵌 4外链)')
|
||||||
component: str | None = Field(None, description='组件路径')
|
component: str | None = Field(None, description='组件路径')
|
||||||
perms: str | None = Field(None, description='权限标识')
|
perms: str | None = Field(None, description='权限标识')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
status: StatusType = Field(description='状态')
|
||||||
display: StatusType = Field(StatusType.enable, description='是否显示')
|
display: StatusType = Field(description='是否显示')
|
||||||
cache: StatusType = Field(StatusType.enable, description='是否缓存')
|
cache: StatusType = Field(description='是否缓存')
|
||||||
link: str | None = Field(None, description='外链地址')
|
link: str | None = Field(None, description='外链地址')
|
||||||
remark: str | None = Field(None, description='备注')
|
remark: str | None = Field(None, description='备注')
|
||||||
|
|
||||||
@@ -43,3 +43,9 @@ class GetMenuDetail(MenuSchemaBase):
|
|||||||
id: int = Field(description='菜单 ID')
|
id: int = Field(description='菜单 ID')
|
||||||
created_time: datetime = Field(description='创建时间')
|
created_time: datetime = Field(description='创建时间')
|
||||||
updated_time: datetime | None = Field(None, description='更新时间')
|
updated_time: datetime | None = Field(None, description='更新时间')
|
||||||
|
|
||||||
|
|
||||||
|
class GetMenuTree(GetMenuDetail):
|
||||||
|
"""获取菜单树"""
|
||||||
|
|
||||||
|
children: list['GetMenuTree'] | None = Field(None, description='子菜单')
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ class OperaLogSchemaBase(SchemaBase):
|
|||||||
browser: str | None = Field(None, description='浏览器')
|
browser: str | None = Field(None, description='浏览器')
|
||||||
device: str | None = Field(None, description='设备')
|
device: str | None = Field(None, description='设备')
|
||||||
args: dict[str, Any] | None = Field(None, description='请求参数')
|
args: dict[str, Any] | None = Field(None, description='请求参数')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
status: StatusType = Field(description='状态')
|
||||||
code: str = Field(description='状态码')
|
code: str = Field(description='状态码')
|
||||||
msg: str | None = Field(None, description='消息')
|
msg: str | None = Field(None, description='消息')
|
||||||
cost_time: float = Field(description='耗时')
|
cost_time: float = Field(description='耗时')
|
||||||
@@ -41,6 +41,12 @@ class UpdateOperaLogParam(OperaLogSchemaBase):
|
|||||||
"""更新操作日志参数"""
|
"""更新操作日志参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteOperaLogParam(SchemaBase):
|
||||||
|
"""删除操作日志参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='操作日志 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class GetOperaLogDetail(OperaLogSchemaBase):
|
class GetOperaLogDetail(OperaLogSchemaBase):
|
||||||
"""操作日志详情"""
|
"""操作日志详情"""
|
||||||
|
|
||||||
|
|||||||
@@ -14,7 +14,8 @@ class RoleSchemaBase(SchemaBase):
|
|||||||
"""角色基础模型"""
|
"""角色基础模型"""
|
||||||
|
|
||||||
name: str = Field(description='角色名称')
|
name: str = Field(description='角色名称')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
status: StatusType = Field(description='状态')
|
||||||
|
is_filter_scopes: bool = Field(True, description='过滤数据权限')
|
||||||
remark: str | None = Field(None, description='备注')
|
remark: str | None = Field(None, description='备注')
|
||||||
|
|
||||||
|
|
||||||
@@ -26,6 +27,12 @@ class UpdateRoleParam(RoleSchemaBase):
|
|||||||
"""更新角色参数"""
|
"""更新角色参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteRoleParam(SchemaBase):
|
||||||
|
"""删除角色参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='角色 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class UpdateRoleMenuParam(SchemaBase):
|
class UpdateRoleMenuParam(SchemaBase):
|
||||||
"""更新角色菜单参数"""
|
"""更新角色菜单参数"""
|
||||||
|
|
||||||
|
|||||||
@@ -35,12 +35,6 @@ class GetLoginToken(AccessTokenBase):
|
|||||||
user: GetUserInfoDetail = Field(description='用户信息')
|
user: GetUserInfoDetail = Field(description='用户信息')
|
||||||
|
|
||||||
|
|
||||||
class KickOutToken(SchemaBase):
|
|
||||||
"""踢出令牌"""
|
|
||||||
|
|
||||||
session_uuid: str = Field(description='会话 UUID')
|
|
||||||
|
|
||||||
|
|
||||||
class GetTokenDetail(SchemaBase):
|
class GetTokenDetail(SchemaBase):
|
||||||
"""令牌详情"""
|
"""令牌详情"""
|
||||||
|
|
||||||
|
|||||||
@@ -3,20 +3,20 @@
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from pydantic import ConfigDict, EmailStr, Field, HttpUrl, model_validator
|
from pydantic import ConfigDict, Field, HttpUrl, model_validator
|
||||||
from typing_extensions import Self
|
from typing_extensions import Self
|
||||||
|
|
||||||
from backend.app.admin.schema.dept import GetDeptDetail
|
from backend.app.admin.schema.dept import GetDeptDetail
|
||||||
from backend.app.admin.schema.role import GetRoleWithRelationDetail
|
from backend.app.admin.schema.role import GetRoleWithRelationDetail
|
||||||
from backend.common.enums import StatusType
|
from backend.common.enums import StatusType
|
||||||
from backend.common.schema import CustomPhoneNumber, SchemaBase
|
from backend.common.schema import CustomEmailStr, CustomPhoneNumber, SchemaBase
|
||||||
|
|
||||||
|
|
||||||
class AuthSchemaBase(SchemaBase):
|
class AuthSchemaBase(SchemaBase):
|
||||||
"""用户认证基础模型"""
|
"""用户认证基础模型"""
|
||||||
|
|
||||||
username: str = Field(description='用户名')
|
username: str = Field(description='用户名')
|
||||||
password: str | None = Field(description='密码')
|
password: str = Field(description='密码')
|
||||||
|
|
||||||
|
|
||||||
class AuthLoginParam(AuthSchemaBase):
|
class AuthLoginParam(AuthSchemaBase):
|
||||||
@@ -25,20 +25,23 @@ class AuthLoginParam(AuthSchemaBase):
|
|||||||
captcha: str = Field(description='验证码')
|
captcha: str = Field(description='验证码')
|
||||||
|
|
||||||
|
|
||||||
class RegisterUserParam(AuthSchemaBase):
|
|
||||||
"""用户注册参数"""
|
|
||||||
|
|
||||||
nickname: str | None = Field(None, description='昵称')
|
|
||||||
email: EmailStr = Field(examples=['user@example.com'], description='邮箱')
|
|
||||||
|
|
||||||
|
|
||||||
class AddUserParam(AuthSchemaBase):
|
class AddUserParam(AuthSchemaBase):
|
||||||
"""添加用户参数"""
|
"""添加用户参数"""
|
||||||
|
|
||||||
|
nickname: str | None = Field(None, description='昵称')
|
||||||
|
email: CustomEmailStr | None = Field(None, description='邮箱')
|
||||||
|
phone: CustomPhoneNumber | None = Field(None, description='手机号码')
|
||||||
dept_id: int = Field(description='部门 ID')
|
dept_id: int = Field(description='部门 ID')
|
||||||
roles: list[int] = Field(description='角色 ID 列表')
|
roles: list[int] = Field(description='角色 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
|
class AddOAuth2UserParam(AuthSchemaBase):
|
||||||
|
"""添加 OAuth2 用户参数"""
|
||||||
|
|
||||||
|
password: str | None = Field(None, description='密码')
|
||||||
nickname: str | None = Field(None, description='昵称')
|
nickname: str | None = Field(None, description='昵称')
|
||||||
email: EmailStr = Field(examples=['user@example.com'], description='邮箱')
|
email: CustomEmailStr | None = Field(None, description='邮箱')
|
||||||
|
avatar: HttpUrl | None = Field(None, description='头像地址')
|
||||||
|
|
||||||
|
|
||||||
class ResetPasswordParam(SchemaBase):
|
class ResetPasswordParam(SchemaBase):
|
||||||
@@ -55,26 +58,17 @@ class UserInfoSchemaBase(SchemaBase):
|
|||||||
dept_id: int | None = Field(None, description='部门 ID')
|
dept_id: int | None = Field(None, description='部门 ID')
|
||||||
username: str = Field(description='用户名')
|
username: str = Field(description='用户名')
|
||||||
nickname: str = Field(description='昵称')
|
nickname: str = Field(description='昵称')
|
||||||
email: EmailStr = Field(examples=['user@example.com'], description='邮箱')
|
avatar: HttpUrl | None = Field(None, description='头像地址')
|
||||||
|
email: CustomEmailStr | None = Field(None, description='邮箱')
|
||||||
phone: CustomPhoneNumber | None = Field(None, description='手机号')
|
phone: CustomPhoneNumber | None = Field(None, description='手机号')
|
||||||
|
|
||||||
|
|
||||||
class UpdateUserParam(UserInfoSchemaBase):
|
class UpdateUserParam(UserInfoSchemaBase):
|
||||||
"""更新用户参数"""
|
"""更新用户参数"""
|
||||||
|
|
||||||
|
|
||||||
class UpdateUserRoleParam(SchemaBase):
|
|
||||||
"""更新用户角色参数"""
|
|
||||||
|
|
||||||
roles: list[int] = Field(description='角色 ID 列表')
|
roles: list[int] = Field(description='角色 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
class AvatarParam(SchemaBase):
|
|
||||||
"""更新头像参数"""
|
|
||||||
|
|
||||||
url: HttpUrl = Field(description='头像 http 地址')
|
|
||||||
|
|
||||||
|
|
||||||
class GetUserInfoDetail(UserInfoSchemaBase):
|
class GetUserInfoDetail(UserInfoSchemaBase):
|
||||||
"""用户信息详情"""
|
"""用户信息详情"""
|
||||||
|
|
||||||
@@ -83,8 +77,7 @@ class GetUserInfoDetail(UserInfoSchemaBase):
|
|||||||
dept_id: int | None = Field(None, description='部门 ID')
|
dept_id: int | None = Field(None, description='部门 ID')
|
||||||
id: int = Field(description='用户 ID')
|
id: int = Field(description='用户 ID')
|
||||||
uuid: str = Field(description='用户 UUID')
|
uuid: str = Field(description='用户 UUID')
|
||||||
avatar: str | None = Field(None, description='头像')
|
status: StatusType = Field(description='状态')
|
||||||
status: StatusType = Field(StatusType.enable, description='状态')
|
|
||||||
is_superuser: bool = Field(description='是否超级管理员')
|
is_superuser: bool = Field(description='是否超级管理员')
|
||||||
is_staff: bool = Field(description='是否管理员')
|
is_staff: bool = Field(description='是否管理员')
|
||||||
is_multi_login: bool = Field(description='是否允许多端登录')
|
is_multi_login: bool = Field(description='是否允许多端登录')
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ from fastapi.security import HTTPBasicCredentials
|
|||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
from starlette.background import BackgroundTask, BackgroundTasks
|
from starlette.background import BackgroundTask, BackgroundTasks
|
||||||
|
|
||||||
from backend.app.admin.conf import admin_settings
|
from backend.app.admin.crud.crud_menu import menu_dao
|
||||||
from backend.app.admin.crud.crud_user import user_dao
|
from backend.app.admin.crud.crud_user import user_dao
|
||||||
from backend.app.admin.model import User
|
from backend.app.admin.model import User
|
||||||
from backend.app.admin.schema.token import GetLoginToken, GetNewToken
|
from backend.app.admin.schema.token import GetLoginToken, GetNewToken
|
||||||
@@ -13,6 +13,7 @@ from backend.app.admin.schema.user import AuthLoginParam
|
|||||||
from backend.app.admin.service.login_log_service import login_log_service
|
from backend.app.admin.service.login_log_service import login_log_service
|
||||||
from backend.common.enums import LoginLogStatusType
|
from backend.common.enums import LoginLogStatusType
|
||||||
from backend.common.exception import errors
|
from backend.common.exception import errors
|
||||||
|
from backend.common.i18n import t
|
||||||
from backend.common.log import log
|
from backend.common.log import log
|
||||||
from backend.common.response.response_code import CustomErrorCode
|
from backend.common.response.response_code import CustomErrorCode
|
||||||
from backend.common.security.jwt import (
|
from backend.common.security.jwt import (
|
||||||
@@ -33,7 +34,7 @@ class AuthService:
|
|||||||
"""认证服务类"""
|
"""认证服务类"""
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def user_verify(db: AsyncSession, username: str, password: str | None) -> User:
|
async def user_verify(db: AsyncSession, username: str, password: str) -> User:
|
||||||
"""
|
"""
|
||||||
验证用户名和密码
|
验证用户名和密码
|
||||||
|
|
||||||
@@ -67,13 +68,13 @@ class AuthService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
user = await self.user_verify(db, obj.username, obj.password)
|
user = await self.user_verify(db, obj.username, obj.password)
|
||||||
await user_dao.update_login_time(db, obj.username)
|
await user_dao.update_login_time(db, obj.username)
|
||||||
a_token = await create_access_token(
|
access_token = await create_access_token(
|
||||||
str(user.id),
|
user.id,
|
||||||
user.is_multi_login,
|
user.is_multi_login,
|
||||||
# extra info
|
# extra info
|
||||||
swagger=True,
|
swagger=True,
|
||||||
)
|
)
|
||||||
return a_token.access_token, user
|
return access_token.access_token, user
|
||||||
|
|
||||||
async def login(
|
async def login(
|
||||||
self, *, request: Request, response: Response, obj: AuthLoginParam, background_tasks: BackgroundTasks
|
self, *, request: Request, response: Response, obj: AuthLoginParam, background_tasks: BackgroundTasks
|
||||||
@@ -91,38 +92,38 @@ class AuthService:
|
|||||||
user = None
|
user = None
|
||||||
try:
|
try:
|
||||||
user = await self.user_verify(db, obj.username, obj.password)
|
user = await self.user_verify(db, obj.username, obj.password)
|
||||||
captcha_code = await redis_client.get(f'{admin_settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{request.state.ip}')
|
captcha_code = await redis_client.get(f'{settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{request.state.ip}')
|
||||||
if not captcha_code:
|
if not captcha_code:
|
||||||
raise errors.AuthorizationError(msg='验证码失效,请重新获取')
|
raise errors.RequestError(msg=t('error.captcha.expired'))
|
||||||
if captcha_code.lower() != obj.captcha.lower():
|
if captcha_code.lower() != obj.captcha.lower():
|
||||||
raise errors.CustomError(error=CustomErrorCode.CAPTCHA_ERROR)
|
raise errors.CustomError(error=CustomErrorCode.CAPTCHA_ERROR)
|
||||||
await redis_client.delete(f'{admin_settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{request.state.ip}')
|
await redis_client.delete(f'{settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{request.state.ip}')
|
||||||
await user_dao.update_login_time(db, obj.username)
|
await user_dao.update_login_time(db, obj.username)
|
||||||
await db.refresh(user)
|
await db.refresh(user)
|
||||||
a_token = await create_access_token(
|
access_token = await create_access_token(
|
||||||
str(user.id),
|
user.id,
|
||||||
user.is_multi_login,
|
user.is_multi_login,
|
||||||
# extra info
|
# extra info
|
||||||
username=user.username,
|
username=user.username,
|
||||||
nickname=user.nickname,
|
nickname=user.nickname,
|
||||||
last_login_time=timezone.t_str(user.last_login_time),
|
last_login_time=timezone.to_str(user.last_login_time),
|
||||||
ip=request.state.ip,
|
ip=request.state.ip,
|
||||||
os=request.state.os,
|
os=request.state.os,
|
||||||
browser=request.state.browser,
|
browser=request.state.browser,
|
||||||
device=request.state.device,
|
device=request.state.device,
|
||||||
)
|
)
|
||||||
r_token = await create_refresh_token(str(user.id), user.is_multi_login)
|
refresh_token = await create_refresh_token(access_token.session_uuid, user.id, user.is_multi_login)
|
||||||
response.set_cookie(
|
response.set_cookie(
|
||||||
key=settings.COOKIE_REFRESH_TOKEN_KEY,
|
key=settings.COOKIE_REFRESH_TOKEN_KEY,
|
||||||
value=r_token.refresh_token,
|
value=refresh_token.refresh_token,
|
||||||
max_age=settings.COOKIE_REFRESH_TOKEN_EXPIRE_SECONDS,
|
max_age=settings.COOKIE_REFRESH_TOKEN_EXPIRE_SECONDS,
|
||||||
expires=timezone.f_utc(r_token.refresh_token_expire_time),
|
expires=timezone.to_utc(refresh_token.refresh_token_expire_time),
|
||||||
httponly=True,
|
httponly=True,
|
||||||
)
|
)
|
||||||
except errors.NotFoundError as e:
|
except errors.NotFoundError as e:
|
||||||
log.error('登陆错误: 用户名不存在')
|
log.error('登陆错误: 用户名不存在')
|
||||||
raise errors.NotFoundError(msg=e.msg)
|
raise errors.NotFoundError(msg=e.msg)
|
||||||
except (errors.AuthorizationError, errors.CustomError) as e:
|
except (errors.RequestError, errors.CustomError) as e:
|
||||||
if not user:
|
if not user:
|
||||||
log.error('登陆错误: 用户密码有误')
|
log.error('登陆错误: 用户密码有误')
|
||||||
task = BackgroundTask(
|
task = BackgroundTask(
|
||||||
@@ -137,7 +138,7 @@ class AuthService:
|
|||||||
msg=e.msg,
|
msg=e.msg,
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
raise errors.AuthorizationError(msg=e.msg, background=task)
|
raise errors.RequestError(code=e.code, msg=e.msg, background=task)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
log.error(f'登陆错误: {e}')
|
log.error(f'登陆错误: {e}')
|
||||||
raise e
|
raise e
|
||||||
@@ -151,46 +152,72 @@ class AuthService:
|
|||||||
username=obj.username,
|
username=obj.username,
|
||||||
login_time=timezone.now(),
|
login_time=timezone.now(),
|
||||||
status=LoginLogStatusType.success.value,
|
status=LoginLogStatusType.success.value,
|
||||||
msg='登录成功',
|
msg=t('success.login.success'),
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
data = GetLoginToken(
|
data = GetLoginToken(
|
||||||
access_token=a_token.access_token,
|
access_token=access_token.access_token,
|
||||||
access_token_expire_time=a_token.access_token_expire_time,
|
access_token_expire_time=access_token.access_token_expire_time,
|
||||||
session_uuid=a_token.session_uuid,
|
session_uuid=access_token.session_uuid,
|
||||||
user=user, # type: ignore
|
user=user, # type: ignore
|
||||||
)
|
)
|
||||||
return data
|
return data
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def new_token(*, request: Request) -> GetNewToken:
|
async def get_codes(*, request: Request) -> list[str]:
|
||||||
"""
|
"""
|
||||||
获取新的访问令牌
|
获取用户权限码
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
codes = set()
|
||||||
|
if request.user.is_superuser:
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
menus = await menu_dao.get_all(db, None, None)
|
||||||
|
for menu in menus:
|
||||||
|
if menu.perms:
|
||||||
|
codes.add(*menu.perms.split(','))
|
||||||
|
else:
|
||||||
|
roles = request.user.roles
|
||||||
|
if roles:
|
||||||
|
for role in roles:
|
||||||
|
for menu in role.menus:
|
||||||
|
if menu.perms:
|
||||||
|
codes.add(*menu.perms.split(','))
|
||||||
|
|
||||||
|
return list(codes)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def refresh_token(*, request: Request) -> GetNewToken:
|
||||||
|
"""
|
||||||
|
刷新令牌
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param request: FastAPI 请求对象
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
refresh_token = request.cookies.get(settings.COOKIE_REFRESH_TOKEN_KEY)
|
refresh_token = request.cookies.get(settings.COOKIE_REFRESH_TOKEN_KEY)
|
||||||
if not refresh_token:
|
if not refresh_token:
|
||||||
raise errors.TokenError(msg='Refresh Token 已过期,请重新登录')
|
raise errors.RequestError(msg='Refresh Token 已过期,请重新登录')
|
||||||
try:
|
token_payload = jwt_decode(refresh_token)
|
||||||
user_id = jwt_decode(refresh_token).id
|
|
||||||
except Exception:
|
|
||||||
raise errors.TokenError(msg='Refresh Token 无效')
|
|
||||||
async with async_db_session() as db:
|
async with async_db_session() as db:
|
||||||
user = await user_dao.get(db, user_id)
|
user = await user_dao.get(db, token_payload.id)
|
||||||
if not user:
|
if not user:
|
||||||
raise errors.NotFoundError(msg='用户名或密码有误')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
elif not user.status:
|
elif not user.status:
|
||||||
raise errors.AuthorizationError(msg='用户已被锁定, 请联系统管理员')
|
raise errors.AuthorizationError(msg='用户已被锁定, 请联系统管理员')
|
||||||
|
if not user.is_multi_login:
|
||||||
|
if await redis_client.keys(match=f'{settings.TOKEN_REDIS_PREFIX}:{user.id}:*'):
|
||||||
|
raise errors.ForbiddenError(msg='此用户已在异地登录,请重新登录并及时修改密码')
|
||||||
new_token = await create_new_token(
|
new_token = await create_new_token(
|
||||||
user_id=str(user.id),
|
refresh_token,
|
||||||
refresh_token=refresh_token,
|
token_payload.session_uuid,
|
||||||
multi_login=user.is_multi_login,
|
user.id,
|
||||||
|
user.is_multi_login,
|
||||||
# extra info
|
# extra info
|
||||||
username=user.username,
|
username=user.username,
|
||||||
nickname=user.nickname,
|
nickname=user.nickname,
|
||||||
last_login_time=timezone.t_str(user.last_login_time),
|
last_login_time=timezone.to_str(user.last_login_time),
|
||||||
ip=request.state.ip,
|
ip=request.state.ip,
|
||||||
os=request.state.os,
|
os=request.state.os,
|
||||||
browser=request.state.browser,
|
browser=request.state.browser,
|
||||||
@@ -212,22 +239,21 @@ class AuthService:
|
|||||||
:param response: FastAPI 响应对象
|
:param response: FastAPI 响应对象
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
token = get_token(request)
|
try:
|
||||||
token_payload = jwt_decode(token)
|
token = get_token(request)
|
||||||
user_id = token_payload.id
|
token_payload = jwt_decode(token)
|
||||||
refresh_token = request.cookies.get(settings.COOKIE_REFRESH_TOKEN_KEY)
|
user_id = token_payload.id
|
||||||
response.delete_cookie(settings.COOKIE_REFRESH_TOKEN_KEY)
|
session_uuid = token_payload.session_uuid
|
||||||
if request.user.is_multi_login:
|
refresh_token = request.cookies.get(settings.COOKIE_REFRESH_TOKEN_KEY)
|
||||||
await redis_client.delete(f'{settings.TOKEN_REDIS_PREFIX}:{user_id}:{token_payload.session_uuid}')
|
except errors.TokenError:
|
||||||
if refresh_token:
|
return
|
||||||
await redis_client.delete(f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{user_id}:{refresh_token}')
|
finally:
|
||||||
else:
|
response.delete_cookie(settings.COOKIE_REFRESH_TOKEN_KEY)
|
||||||
key_prefix = [
|
|
||||||
f'{settings.TOKEN_REDIS_PREFIX}:{user_id}:',
|
await redis_client.delete(f'{settings.TOKEN_REDIS_PREFIX}:{user_id}:{session_uuid}')
|
||||||
f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{user_id}:',
|
await redis_client.delete(f'{settings.TOKEN_EXTRA_INFO_REDIS_PREFIX}:{user_id}:{session_uuid}')
|
||||||
]
|
if refresh_token:
|
||||||
for prefix in key_prefix:
|
await redis_client.delete(f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{user_id}:{refresh_token}')
|
||||||
await redis_client.delete_prefix(prefix)
|
|
||||||
|
|
||||||
|
|
||||||
auth_service: AuthService = AuthService()
|
auth_service: AuthService = AuthService()
|
||||||
|
|||||||
@@ -6,7 +6,12 @@ from sqlalchemy import Select
|
|||||||
|
|
||||||
from backend.app.admin.crud.crud_data_rule import data_rule_dao
|
from backend.app.admin.crud.crud_data_rule import data_rule_dao
|
||||||
from backend.app.admin.model import DataRule
|
from backend.app.admin.model import DataRule
|
||||||
from backend.app.admin.schema.data_rule import CreateDataRuleParam, GetDataRuleColumnDetail, UpdateDataRuleParam
|
from backend.app.admin.schema.data_rule import (
|
||||||
|
CreateDataRuleParam,
|
||||||
|
DeleteDataRuleParam,
|
||||||
|
GetDataRuleColumnDetail,
|
||||||
|
UpdateDataRuleParam,
|
||||||
|
)
|
||||||
from backend.common.exception import errors
|
from backend.common.exception import errors
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
from backend.database.db import async_db_session
|
from backend.database.db import async_db_session
|
||||||
@@ -82,7 +87,7 @@ class DataRuleService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
data_rule = await data_rule_dao.get_by_name(db, obj.name)
|
data_rule = await data_rule_dao.get_by_name(db, obj.name)
|
||||||
if data_rule:
|
if data_rule:
|
||||||
raise errors.ForbiddenError(msg='数据规则已存在')
|
raise errors.ConflictError(msg='数据规则已存在')
|
||||||
await data_rule_dao.create(db, obj)
|
await data_rule_dao.create(db, obj)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -100,20 +105,20 @@ class DataRuleService:
|
|||||||
raise errors.NotFoundError(msg='数据规则不存在')
|
raise errors.NotFoundError(msg='数据规则不存在')
|
||||||
if data_rule.name != obj.name:
|
if data_rule.name != obj.name:
|
||||||
if await data_rule_dao.get_by_name(db, obj.name):
|
if await data_rule_dao.get_by_name(db, obj.name):
|
||||||
raise errors.ForbiddenError(msg='数据规则已存在')
|
raise errors.ConflictError(msg='数据规则已存在')
|
||||||
count = await data_rule_dao.update(db, pk, obj)
|
count = await data_rule_dao.update(db, pk, obj)
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, pk: list[int]) -> int:
|
async def delete(*, obj: DeleteDataRuleParam) -> int:
|
||||||
"""
|
"""
|
||||||
删除数据规则
|
批量删除数据规则
|
||||||
|
|
||||||
:param pk: 规则 ID 列表
|
:param obj: 规则 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await data_rule_dao.delete(db, pk)
|
count = await data_rule_dao.delete(db, obj.pks)
|
||||||
return count
|
return count
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,10 +1,17 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Sequence
|
||||||
|
|
||||||
from sqlalchemy import Select
|
from sqlalchemy import Select
|
||||||
|
|
||||||
from backend.app.admin.crud.crud_data_scope import data_scope_dao
|
from backend.app.admin.crud.crud_data_scope import data_scope_dao
|
||||||
from backend.app.admin.model import DataScope
|
from backend.app.admin.model import DataScope
|
||||||
from backend.app.admin.schema.data_scope import CreateDataScopeParam, UpdateDataScopeParam, UpdateDataScopeRuleParam
|
from backend.app.admin.schema.data_scope import (
|
||||||
|
CreateDataScopeParam,
|
||||||
|
DeleteDataScopeParam,
|
||||||
|
UpdateDataScopeParam,
|
||||||
|
UpdateDataScopeRuleParam,
|
||||||
|
)
|
||||||
from backend.common.exception import errors
|
from backend.common.exception import errors
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
from backend.database.db import async_db_session
|
from backend.database.db import async_db_session
|
||||||
@@ -28,6 +35,13 @@ class DataScopeService:
|
|||||||
raise errors.NotFoundError(msg='数据范围不存在')
|
raise errors.NotFoundError(msg='数据范围不存在')
|
||||||
return data_scope
|
return data_scope
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_all() -> Sequence[DataScope]:
|
||||||
|
"""获取所有数据范围"""
|
||||||
|
async with async_db_session() as db:
|
||||||
|
data_scopes = await data_scope_dao.get_all(db)
|
||||||
|
return data_scopes
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_rules(*, pk: int) -> DataScope:
|
async def get_rules(*, pk: int) -> DataScope:
|
||||||
"""
|
"""
|
||||||
@@ -64,7 +78,7 @@ class DataScopeService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
data_scope = await data_scope_dao.get_by_name(db, obj.name)
|
data_scope = await data_scope_dao.get_by_name(db, obj.name)
|
||||||
if data_scope:
|
if data_scope:
|
||||||
raise errors.ForbiddenError(msg='数据范围已存在')
|
raise errors.ConflictError(msg='数据范围已存在')
|
||||||
await data_scope_dao.create(db, obj)
|
await data_scope_dao.create(db, obj)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -82,7 +96,7 @@ class DataScopeService:
|
|||||||
raise errors.NotFoundError(msg='数据范围不存在')
|
raise errors.NotFoundError(msg='数据范围不存在')
|
||||||
if data_scope.name != obj.name:
|
if data_scope.name != obj.name:
|
||||||
if await data_scope_dao.get_by_name(db, obj.name):
|
if await data_scope_dao.get_by_name(db, obj.name):
|
||||||
raise errors.ForbiddenError(msg='数据范围已存在')
|
raise errors.ConflictError(msg='数据范围已存在')
|
||||||
count = await data_scope_dao.update(db, pk, obj)
|
count = await data_scope_dao.update(db, pk, obj)
|
||||||
for role in await data_scope.awaitable_attrs.roles:
|
for role in await data_scope.awaitable_attrs.roles:
|
||||||
for user in await role.awaitable_attrs.users:
|
for user in await role.awaitable_attrs.users:
|
||||||
@@ -103,17 +117,17 @@ class DataScopeService:
|
|||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, pk: list[int]) -> int:
|
async def delete(*, obj: DeleteDataScopeParam) -> int:
|
||||||
"""
|
"""
|
||||||
删除数据范围
|
批量删除数据范围
|
||||||
|
|
||||||
:param pk: 范围 ID 列表
|
:param obj: 范围 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await data_scope_dao.delete(db, pk)
|
count = await data_scope_dao.delete(db, obj.pks)
|
||||||
for _pk in pk:
|
for pk in obj.pks:
|
||||||
data_rule = await data_scope_dao.get(db, _pk)
|
data_rule = await data_scope_dao.get(db, pk)
|
||||||
if data_rule:
|
if data_rule:
|
||||||
for role in await data_rule.awaitable_attrs.roles:
|
for role in await data_rule.awaitable_attrs.roles:
|
||||||
for user in await role.awaitable_attrs.users:
|
for user in await role.awaitable_attrs.users:
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ class DeptService:
|
|||||||
return dept
|
return dept
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_dept_tree(
|
async def get_tree(
|
||||||
*, request: Request, name: str | None, leader: str | None, phone: str | None, status: int | None
|
*, request: Request, name: str | None, leader: str | None, phone: str | None, status: int | None
|
||||||
) -> list[dict[str, Any]]:
|
) -> list[dict[str, Any]]:
|
||||||
"""
|
"""
|
||||||
@@ -61,7 +61,7 @@ class DeptService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
dept = await dept_dao.get_by_name(db, obj.name)
|
dept = await dept_dao.get_by_name(db, obj.name)
|
||||||
if dept:
|
if dept:
|
||||||
raise errors.ForbiddenError(msg='部门名称已存在')
|
raise errors.ConflictError(msg='部门名称已存在')
|
||||||
if obj.parent_id:
|
if obj.parent_id:
|
||||||
parent_dept = await dept_dao.get(db, obj.parent_id)
|
parent_dept = await dept_dao.get(db, obj.parent_id)
|
||||||
if not parent_dept:
|
if not parent_dept:
|
||||||
@@ -83,7 +83,7 @@ class DeptService:
|
|||||||
raise errors.NotFoundError(msg='部门不存在')
|
raise errors.NotFoundError(msg='部门不存在')
|
||||||
if dept.name != obj.name:
|
if dept.name != obj.name:
|
||||||
if await dept_dao.get_by_name(db, obj.name):
|
if await dept_dao.get_by_name(db, obj.name):
|
||||||
raise errors.ForbiddenError(msg='部门名称已存在')
|
raise errors.ConflictError(msg='部门名称已存在')
|
||||||
if obj.parent_id:
|
if obj.parent_id:
|
||||||
parent_dept = await dept_dao.get(db, obj.parent_id)
|
parent_dept = await dept_dao.get(db, obj.parent_id)
|
||||||
if not parent_dept:
|
if not parent_dept:
|
||||||
@@ -104,10 +104,10 @@ class DeptService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
dept = await dept_dao.get_with_relation(db, pk)
|
dept = await dept_dao.get_with_relation(db, pk)
|
||||||
if dept.users:
|
if dept.users:
|
||||||
raise errors.ForbiddenError(msg='部门下存在用户,无法删除')
|
raise errors.ConflictError(msg='部门下存在用户,无法删除')
|
||||||
children = await dept_dao.get_children(db, pk)
|
children = await dept_dao.get_children(db, pk)
|
||||||
if children:
|
if children:
|
||||||
raise errors.ForbiddenError(msg='部门下存在子部门,无法删除')
|
raise errors.ConflictError(msg='部门下存在子部门,无法删除')
|
||||||
count = await dept_dao.delete(db, pk)
|
count = await dept_dao.delete(db, pk)
|
||||||
for user in dept.users:
|
for user in dept.users:
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ from sqlalchemy import Select
|
|||||||
from sqlalchemy.ext.asyncio import AsyncSession
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
from backend.app.admin.crud.crud_login_log import login_log_dao
|
from backend.app.admin.crud.crud_login_log import login_log_dao
|
||||||
from backend.app.admin.schema.login_log import CreateLoginLogParam
|
from backend.app.admin.schema.login_log import CreateLoginLogParam, DeleteLoginLogParam
|
||||||
from backend.common.log import log
|
from backend.common.log import log
|
||||||
from backend.database.db import async_db_session
|
from backend.database.db import async_db_session
|
||||||
|
|
||||||
@@ -71,23 +71,22 @@ class LoginLogService:
|
|||||||
log.error(f'登录日志创建失败: {e}')
|
log.error(f'登录日志创建失败: {e}')
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, pk: list[int]) -> int:
|
async def delete(*, obj: DeleteLoginLogParam) -> int:
|
||||||
"""
|
"""
|
||||||
删除登录日志
|
批量删除登录日志
|
||||||
|
|
||||||
:param pk: 日志 ID 列表
|
:param obj: 日志 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await login_log_dao.delete(db, pk)
|
count = await login_log_dao.delete(db, obj.pks)
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete_all() -> int:
|
async def delete_all() -> None:
|
||||||
"""清空所有登录日志"""
|
"""清空所有登录日志"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await login_log_dao.delete_all(db)
|
await login_log_dao.delete_all(db)
|
||||||
return count
|
|
||||||
|
|
||||||
|
|
||||||
login_log_service: LoginLogService = LoginLogService()
|
login_log_service: LoginLogService = LoginLogService()
|
||||||
|
|||||||
@@ -32,7 +32,7 @@ class MenuService:
|
|||||||
return menu
|
return menu
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_menu_tree(*, title: str | None, status: int | None) -> list[dict[str, Any]]:
|
async def get_tree(*, title: str | None, status: int | None) -> list[dict[str, Any]]:
|
||||||
"""
|
"""
|
||||||
获取菜单树形结构
|
获取菜单树形结构
|
||||||
|
|
||||||
@@ -41,27 +41,30 @@ class MenuService:
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session() as db:
|
async with async_db_session() as db:
|
||||||
menu_select = await menu_dao.get_all(db, title=title, status=status)
|
menu_data = await menu_dao.get_all(db, title=title, status=status)
|
||||||
menu_tree = get_tree_data(menu_select)
|
menu_tree = get_tree_data(menu_data)
|
||||||
return menu_tree
|
return menu_tree
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_user_menu_tree(*, request: Request) -> list[dict[str, Any]]:
|
async def get_sidebar(*, request: Request) -> list[dict[str, Any] | None]:
|
||||||
"""
|
"""
|
||||||
获取用户的菜单树形结构
|
获取用户的菜单侧边栏
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param request: FastAPI 请求对象
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session() as db:
|
async with async_db_session() as db:
|
||||||
roles = request.user.roles
|
if request.user.is_superuser:
|
||||||
menu_ids = []
|
menu_data = await menu_dao.get_sidebar(db, None)
|
||||||
menu_tree = []
|
else:
|
||||||
if roles:
|
roles = request.user.roles
|
||||||
for role in roles:
|
menu_ids = set()
|
||||||
menu_ids.extend([menu.id for menu in role.menus])
|
if roles:
|
||||||
menu_select = await menu_dao.get_role_menus(db, request.user.is_superuser, menu_ids)
|
for role in roles:
|
||||||
menu_tree = get_vben5_tree_data(menu_select)
|
for menu in role.menus:
|
||||||
|
menu_ids.add(menu.id)
|
||||||
|
menu_data = await menu_dao.get_sidebar(db, list(menu_ids))
|
||||||
|
menu_tree = get_vben5_tree_data(menu_data)
|
||||||
return menu_tree
|
return menu_tree
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -75,7 +78,7 @@ class MenuService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
title = await menu_dao.get_by_title(db, obj.title)
|
title = await menu_dao.get_by_title(db, obj.title)
|
||||||
if title:
|
if title:
|
||||||
raise errors.ForbiddenError(msg='菜单标题已存在')
|
raise errors.ConflictError(msg='菜单标题已存在')
|
||||||
if obj.parent_id:
|
if obj.parent_id:
|
||||||
parent_menu = await menu_dao.get(db, obj.parent_id)
|
parent_menu = await menu_dao.get(db, obj.parent_id)
|
||||||
if not parent_menu:
|
if not parent_menu:
|
||||||
@@ -97,7 +100,7 @@ class MenuService:
|
|||||||
raise errors.NotFoundError(msg='菜单不存在')
|
raise errors.NotFoundError(msg='菜单不存在')
|
||||||
if menu.title != obj.title:
|
if menu.title != obj.title:
|
||||||
if await menu_dao.get_by_title(db, obj.title):
|
if await menu_dao.get_by_title(db, obj.title):
|
||||||
raise errors.ForbiddenError(msg='菜单标题已存在')
|
raise errors.ConflictError(msg='菜单标题已存在')
|
||||||
if obj.parent_id:
|
if obj.parent_id:
|
||||||
parent_menu = await menu_dao.get(db, obj.parent_id)
|
parent_menu = await menu_dao.get(db, obj.parent_id)
|
||||||
if not parent_menu:
|
if not parent_menu:
|
||||||
@@ -121,7 +124,7 @@ class MenuService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
children = await menu_dao.get_children(db, pk)
|
children = await menu_dao.get_children(db, pk)
|
||||||
if children:
|
if children:
|
||||||
raise errors.ForbiddenError(msg='菜单下存在子菜单,无法删除')
|
raise errors.ConflictError(msg='菜单下存在子菜单,无法删除')
|
||||||
menu = await menu_dao.get(db, pk)
|
menu = await menu_dao.get(db, pk)
|
||||||
count = await menu_dao.delete(db, pk)
|
count = await menu_dao.delete(db, pk)
|
||||||
if menu:
|
if menu:
|
||||||
|
|||||||
@@ -1,122 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from typing import Any
|
|
||||||
|
|
||||||
from fast_captcha import text_captcha
|
|
||||||
from fastapi import BackgroundTasks, Request, Response
|
|
||||||
|
|
||||||
from backend.app.admin.conf import admin_settings
|
|
||||||
from backend.app.admin.crud.crud_user import user_dao
|
|
||||||
from backend.app.admin.crud.crud_user_social import user_social_dao
|
|
||||||
from backend.app.admin.schema.token import GetLoginToken
|
|
||||||
from backend.app.admin.schema.user import RegisterUserParam
|
|
||||||
from backend.app.admin.schema.user_social import CreateUserSocialParam
|
|
||||||
from backend.app.admin.service.login_log_service import login_log_service
|
|
||||||
from backend.common.enums import LoginLogStatusType, UserSocialType
|
|
||||||
from backend.common.exception.errors import AuthorizationError
|
|
||||||
from backend.common.security import jwt
|
|
||||||
from backend.core.conf import settings
|
|
||||||
from backend.database.db import async_db_session
|
|
||||||
from backend.database.redis import redis_client
|
|
||||||
from backend.utils.timezone import timezone
|
|
||||||
|
|
||||||
|
|
||||||
class OAuth2Service:
|
|
||||||
"""OAuth2 认证服务类"""
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def create_with_login(
|
|
||||||
*,
|
|
||||||
request: Request,
|
|
||||||
response: Response,
|
|
||||||
background_tasks: BackgroundTasks,
|
|
||||||
user: dict[str, Any],
|
|
||||||
social: UserSocialType,
|
|
||||||
) -> GetLoginToken | None:
|
|
||||||
"""
|
|
||||||
创建 OAuth2 用户并登录
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param response: FastAPI 响应对象
|
|
||||||
:param background_tasks: FastAPI 后台任务
|
|
||||||
:param user: OAuth2 用户信息
|
|
||||||
:param social: 社交平台类型
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
# 获取 OAuth2 平台用户信息
|
|
||||||
social_id = user.get('id')
|
|
||||||
social_username = user.get('username')
|
|
||||||
if social == UserSocialType.github:
|
|
||||||
social_username = user.get('login')
|
|
||||||
social_nickname = user.get('name')
|
|
||||||
social_email = user.get('email')
|
|
||||||
if social == UserSocialType.linux_do: # 不提供明文邮箱的平台
|
|
||||||
social_email = f'{social_username}@linux.do'
|
|
||||||
if not social_email:
|
|
||||||
raise AuthorizationError(msg=f'授权失败,{social.value} 账户未绑定邮箱')
|
|
||||||
# 创建系统用户
|
|
||||||
sys_user = await user_dao.check_email(db, social_email)
|
|
||||||
if not sys_user:
|
|
||||||
sys_user = await user_dao.get_by_username(db, social_username)
|
|
||||||
if sys_user:
|
|
||||||
social_username = f'{social_username}#{text_captcha(5)}'
|
|
||||||
sys_user = await user_dao.get_by_nickname(db, social_nickname)
|
|
||||||
if sys_user:
|
|
||||||
social_username = f'{social_nickname}#{text_captcha(5)}'
|
|
||||||
new_sys_user = RegisterUserParam(
|
|
||||||
username=social_username, password=None, nickname=social_username, email=social_email
|
|
||||||
)
|
|
||||||
await user_dao.create(db, new_sys_user, social=True)
|
|
||||||
await db.flush()
|
|
||||||
sys_user = await user_dao.check_email(db, social_email)
|
|
||||||
# 绑定社交用户
|
|
||||||
sys_user_id = sys_user.id
|
|
||||||
user_social = await user_social_dao.get(db, sys_user_id, social.value)
|
|
||||||
if not user_social:
|
|
||||||
new_user_social = CreateUserSocialParam(source=social.value, uid=str(social_id), user_id=sys_user_id)
|
|
||||||
await user_social_dao.create(db, new_user_social)
|
|
||||||
# 创建 token
|
|
||||||
access_token = await jwt.create_access_token(
|
|
||||||
str(sys_user_id),
|
|
||||||
sys_user.is_multi_login,
|
|
||||||
# extra info
|
|
||||||
username=sys_user.username,
|
|
||||||
nickname=sys_user.nickname,
|
|
||||||
last_login_time=timezone.t_str(timezone.now()),
|
|
||||||
ip=request.state.ip,
|
|
||||||
os=request.state.os,
|
|
||||||
browser=request.state.browser,
|
|
||||||
device=request.state.device,
|
|
||||||
)
|
|
||||||
refresh_token = await jwt.create_refresh_token(str(sys_user_id), multi_login=sys_user.is_multi_login)
|
|
||||||
await user_dao.update_login_time(db, sys_user.username)
|
|
||||||
await db.refresh(sys_user)
|
|
||||||
login_log = dict(
|
|
||||||
db=db,
|
|
||||||
request=request,
|
|
||||||
user_uuid=sys_user.uuid,
|
|
||||||
username=sys_user.username,
|
|
||||||
login_time=timezone.now(),
|
|
||||||
status=LoginLogStatusType.success.value,
|
|
||||||
msg='登录成功(OAuth2)',
|
|
||||||
)
|
|
||||||
background_tasks.add_task(login_log_service.create, **login_log)
|
|
||||||
await redis_client.delete(f'{admin_settings.CAPTCHA_LOGIN_REDIS_PREFIX}:{request.state.ip}')
|
|
||||||
response.set_cookie(
|
|
||||||
key=settings.COOKIE_REFRESH_TOKEN_KEY,
|
|
||||||
value=refresh_token.refresh_token,
|
|
||||||
max_age=settings.COOKIE_REFRESH_TOKEN_EXPIRE_SECONDS,
|
|
||||||
expires=timezone.f_utc(refresh_token.refresh_token_expire_time),
|
|
||||||
httponly=True,
|
|
||||||
)
|
|
||||||
data = GetLoginToken(
|
|
||||||
access_token=access_token.access_token,
|
|
||||||
access_token_expire_time=access_token.access_token_expire_time,
|
|
||||||
user=sys_user, # type: ignore
|
|
||||||
session_uuid=access_token.session_uuid,
|
|
||||||
)
|
|
||||||
return data
|
|
||||||
|
|
||||||
|
|
||||||
oauth2_service: OAuth2Service = OAuth2Service()
|
|
||||||
@@ -3,7 +3,7 @@
|
|||||||
from sqlalchemy import Select
|
from sqlalchemy import Select
|
||||||
|
|
||||||
from backend.app.admin.crud.crud_opera_log import opera_log_dao
|
from backend.app.admin.crud.crud_opera_log import opera_log_dao
|
||||||
from backend.app.admin.schema.opera_log import CreateOperaLogParam
|
from backend.app.admin.schema.opera_log import CreateOperaLogParam, DeleteOperaLogParam
|
||||||
from backend.database.db import async_db_session
|
from backend.database.db import async_db_session
|
||||||
|
|
||||||
|
|
||||||
@@ -34,23 +34,33 @@ class OperaLogService:
|
|||||||
await opera_log_dao.create(db, obj)
|
await opera_log_dao.create(db, obj)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, pk: list[int]) -> int:
|
async def bulk_create(*, objs: list[CreateOperaLogParam]) -> None:
|
||||||
"""
|
"""
|
||||||
删除操作日志
|
批量创建操作日志
|
||||||
|
|
||||||
:param pk: 日志 ID 列表
|
:param objs: 操作日志创建参数列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await opera_log_dao.delete(db, pk)
|
await opera_log_dao.bulk_create(db, objs)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def delete(*, obj: DeleteOperaLogParam) -> int:
|
||||||
|
"""
|
||||||
|
批量删除操作日志
|
||||||
|
|
||||||
|
:param obj: 日志 ID 列表
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
count = await opera_log_dao.delete(db, obj.pks)
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete_all() -> int:
|
async def delete_all() -> None:
|
||||||
"""清空所有操作日志"""
|
"""清空所有操作日志"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await opera_log_dao.delete_all(db)
|
await opera_log_dao.delete_all(db)
|
||||||
return count
|
|
||||||
|
|
||||||
|
|
||||||
opera_log_service: OperaLogService = OperaLogService()
|
opera_log_service: OperaLogService = OperaLogService()
|
||||||
|
|||||||
@@ -0,0 +1,127 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import io
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import shutil
|
||||||
|
import zipfile
|
||||||
|
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from fastapi import UploadFile
|
||||||
|
|
||||||
|
from backend.common.enums import PluginType, StatusType
|
||||||
|
from backend.common.exception import errors
|
||||||
|
from backend.core.conf import settings
|
||||||
|
from backend.core.path_conf import PLUGIN_DIR
|
||||||
|
from backend.database.redis import redis_client
|
||||||
|
from backend.plugin.tools import uninstall_requirements_async
|
||||||
|
from backend.utils.file_ops import install_git_plugin, install_zip_plugin
|
||||||
|
from backend.utils.timezone import timezone
|
||||||
|
|
||||||
|
|
||||||
|
class PluginService:
|
||||||
|
"""插件服务类"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_all() -> list[dict[str, Any]]:
|
||||||
|
"""获取所有插件"""
|
||||||
|
keys = []
|
||||||
|
result = []
|
||||||
|
|
||||||
|
async for key in redis_client.scan_iter(f'{settings.PLUGIN_REDIS_PREFIX}:*'):
|
||||||
|
keys.append(key)
|
||||||
|
|
||||||
|
for info in await redis_client.mget(*keys):
|
||||||
|
result.append(json.loads(info))
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def changed() -> str | None:
|
||||||
|
"""检查插件是否发生变更"""
|
||||||
|
return await redis_client.get(f'{settings.PLUGIN_REDIS_PREFIX}:changed')
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def install(*, type: PluginType, file: UploadFile | None = None, repo_url: str | None = None) -> str:
|
||||||
|
"""
|
||||||
|
安装插件
|
||||||
|
|
||||||
|
:param type: 插件类型
|
||||||
|
:param file: 插件 zip 压缩包
|
||||||
|
:param repo_url: git 仓库地址
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
if type == PluginType.zip:
|
||||||
|
if not file:
|
||||||
|
raise errors.RequestError(msg='ZIP 压缩包不能为空')
|
||||||
|
return await install_zip_plugin(file)
|
||||||
|
if not repo_url:
|
||||||
|
raise errors.RequestError(msg='Git 仓库地址不能为空')
|
||||||
|
return await install_git_plugin(repo_url)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def uninstall(*, plugin: str):
|
||||||
|
"""
|
||||||
|
卸载插件
|
||||||
|
|
||||||
|
:param plugin: 插件名称
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
plugin_dir = os.path.join(PLUGIN_DIR, plugin)
|
||||||
|
if not os.path.exists(plugin_dir):
|
||||||
|
raise errors.NotFoundError(msg='插件不存在')
|
||||||
|
await uninstall_requirements_async(plugin)
|
||||||
|
bacup_dir = os.path.join(PLUGIN_DIR, f'{plugin}.{timezone.now().strftime("%Y%m%d%H%M%S")}.backup')
|
||||||
|
shutil.move(plugin_dir, bacup_dir)
|
||||||
|
await redis_client.delete(f'{settings.PLUGIN_REDIS_PREFIX}:{plugin}')
|
||||||
|
await redis_client.set(f'{settings.PLUGIN_REDIS_PREFIX}:changed', 'ture')
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update_status(*, plugin: str):
|
||||||
|
"""
|
||||||
|
更新插件状态
|
||||||
|
|
||||||
|
:param plugin: 插件名称
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
plugin_info = await redis_client.get(f'{settings.PLUGIN_REDIS_PREFIX}:{plugin}')
|
||||||
|
if not plugin_info:
|
||||||
|
raise errors.NotFoundError(msg='插件不存在')
|
||||||
|
plugin_info = json.loads(plugin_info)
|
||||||
|
|
||||||
|
# 更新持久缓存状态
|
||||||
|
new_status = (
|
||||||
|
str(StatusType.enable.value)
|
||||||
|
if plugin_info['plugin']['enable'] == str(StatusType.disable.value)
|
||||||
|
else str(StatusType.disable.value)
|
||||||
|
)
|
||||||
|
plugin_info['plugin']['enable'] = new_status
|
||||||
|
await redis_client.set(f'{settings.PLUGIN_REDIS_PREFIX}:{plugin}', json.dumps(plugin_info, ensure_ascii=False))
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def build(*, plugin: str) -> io.BytesIO:
|
||||||
|
"""
|
||||||
|
打包插件为 zip 压缩包
|
||||||
|
|
||||||
|
:param plugin: 插件名称
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
plugin_dir = os.path.join(PLUGIN_DIR, plugin)
|
||||||
|
if not os.path.exists(plugin_dir):
|
||||||
|
raise errors.NotFoundError(msg='插件不存在')
|
||||||
|
|
||||||
|
bio = io.BytesIO()
|
||||||
|
with zipfile.ZipFile(bio, 'w') as zf:
|
||||||
|
for root, dirs, files in os.walk(plugin_dir):
|
||||||
|
dirs[:] = [d for d in dirs if d != '__pycache__']
|
||||||
|
for file in files:
|
||||||
|
file_path = os.path.join(root, file)
|
||||||
|
arcname = os.path.relpath(file_path, start=plugin_dir)
|
||||||
|
zf.write(file_path, os.path.join(plugin, arcname))
|
||||||
|
|
||||||
|
bio.seek(0)
|
||||||
|
return bio
|
||||||
|
|
||||||
|
|
||||||
|
plugin_service: PluginService = PluginService()
|
||||||
@@ -10,6 +10,7 @@ from backend.app.admin.crud.crud_role import role_dao
|
|||||||
from backend.app.admin.model import Role
|
from backend.app.admin.model import Role
|
||||||
from backend.app.admin.schema.role import (
|
from backend.app.admin.schema.role import (
|
||||||
CreateRoleParam,
|
CreateRoleParam,
|
||||||
|
DeleteRoleParam,
|
||||||
UpdateRoleMenuParam,
|
UpdateRoleMenuParam,
|
||||||
UpdateRoleParam,
|
UpdateRoleParam,
|
||||||
UpdateRoleScopeParam,
|
UpdateRoleScopeParam,
|
||||||
@@ -45,18 +46,6 @@ class RoleService:
|
|||||||
roles = await role_dao.get_all(db)
|
roles = await role_dao.get_all(db)
|
||||||
return roles
|
return roles
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def get_users(*, pk: int) -> Sequence[Role]:
|
|
||||||
"""
|
|
||||||
获取用户的角色列表
|
|
||||||
|
|
||||||
:param pk: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session() as db:
|
|
||||||
roles = await role_dao.get_users(db, user_id=pk)
|
|
||||||
return roles
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_select(*, name: str | None, status: int | None) -> Select:
|
async def get_select(*, name: str | None, status: int | None) -> Select:
|
||||||
"""
|
"""
|
||||||
@@ -69,7 +58,7 @@ class RoleService:
|
|||||||
return await role_dao.get_list(name=name, status=status)
|
return await role_dao.get_list(name=name, status=status)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_menu_tree(*, pk: int) -> list[dict[str, Any]]:
|
async def get_menu_tree(*, pk: int) -> list[dict[str, Any] | None]:
|
||||||
"""
|
"""
|
||||||
获取角色的菜单树形结构
|
获取角色的菜单树形结构
|
||||||
|
|
||||||
@@ -80,9 +69,7 @@ class RoleService:
|
|||||||
role = await role_dao.get_with_relation(db, pk)
|
role = await role_dao.get_with_relation(db, pk)
|
||||||
if not role:
|
if not role:
|
||||||
raise errors.NotFoundError(msg='角色不存在')
|
raise errors.NotFoundError(msg='角色不存在')
|
||||||
menu_ids = [menu.id for menu in role.menus]
|
menu_tree = get_tree_data(role.menus) if role.menus else []
|
||||||
menu_select = await menu_dao.get_role_menus(db, False, menu_ids)
|
|
||||||
menu_tree = get_tree_data(menu_select)
|
|
||||||
return menu_tree
|
return menu_tree
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -111,7 +98,7 @@ class RoleService:
|
|||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
role = await role_dao.get_by_name(db, obj.name)
|
role = await role_dao.get_by_name(db, obj.name)
|
||||||
if role:
|
if role:
|
||||||
raise errors.ForbiddenError(msg='角色已存在')
|
raise errors.ConflictError(msg='角色已存在')
|
||||||
await role_dao.create(db, obj)
|
await role_dao.create(db, obj)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -128,9 +115,8 @@ class RoleService:
|
|||||||
if not role:
|
if not role:
|
||||||
raise errors.NotFoundError(msg='角色不存在')
|
raise errors.NotFoundError(msg='角色不存在')
|
||||||
if role.name != obj.name:
|
if role.name != obj.name:
|
||||||
role = await role_dao.get_by_name(db, obj.name)
|
if await role_dao.get_by_name(db, obj.name):
|
||||||
if role:
|
raise errors.ConflictError(msg='角色已存在')
|
||||||
raise errors.ForbiddenError(msg='角色已存在')
|
|
||||||
count = await role_dao.update(db, pk, obj)
|
count = await role_dao.update(db, pk, obj)
|
||||||
for user in await role.awaitable_attrs.users:
|
for user in await role.awaitable_attrs.users:
|
||||||
await redis_client.delete_prefix(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
await redis_client.delete_prefix(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
@@ -146,7 +132,7 @@ class RoleService:
|
|||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
role = await role_dao.get_with_relation(db, pk)
|
role = await role_dao.get(db, pk)
|
||||||
if not role:
|
if not role:
|
||||||
raise errors.NotFoundError(msg='角色不存在')
|
raise errors.NotFoundError(msg='角色不存在')
|
||||||
for menu_id in menu_ids.menus:
|
for menu_id in menu_ids.menus:
|
||||||
@@ -181,17 +167,17 @@ class RoleService:
|
|||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, pk: list[int]) -> int:
|
async def delete(*, obj: DeleteRoleParam) -> int:
|
||||||
"""
|
"""
|
||||||
删除角色
|
批量删除角色
|
||||||
|
|
||||||
:param pk: 角色 ID 列表
|
:param obj: 角色 ID 列表
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
count = await role_dao.delete(db, pk)
|
count = await role_dao.delete(db, obj.pks)
|
||||||
for _pk in pk:
|
for pk in obj.pks:
|
||||||
role = await role_dao.get(db, _pk)
|
role = await role_dao.get(db, pk)
|
||||||
if role:
|
if role:
|
||||||
for user in await role.awaitable_attrs.users:
|
for user in await role.awaitable_attrs.users:
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
|||||||
@@ -2,23 +2,24 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import random
|
import random
|
||||||
|
|
||||||
|
from typing import Sequence
|
||||||
|
|
||||||
from fastapi import Request
|
from fastapi import Request
|
||||||
from sqlalchemy import Select
|
from sqlalchemy import Select
|
||||||
|
|
||||||
from backend.app.admin.crud.crud_dept import dept_dao
|
from backend.app.admin.crud.crud_dept import dept_dao
|
||||||
from backend.app.admin.crud.crud_role import role_dao
|
from backend.app.admin.crud.crud_role import role_dao
|
||||||
from backend.app.admin.crud.crud_user import user_dao
|
from backend.app.admin.crud.crud_user import user_dao
|
||||||
from backend.app.admin.model import User
|
from backend.app.admin.model import Role, User
|
||||||
from backend.app.admin.schema.user import (
|
from backend.app.admin.schema.user import (
|
||||||
AddUserParam,
|
AddUserParam,
|
||||||
AvatarParam,
|
|
||||||
RegisterUserParam,
|
|
||||||
ResetPasswordParam,
|
ResetPasswordParam,
|
||||||
UpdateUserParam,
|
UpdateUserParam,
|
||||||
UpdateUserRoleParam,
|
|
||||||
)
|
)
|
||||||
|
from backend.common.enums import UserPermissionType
|
||||||
from backend.common.exception import errors
|
from backend.common.exception import errors
|
||||||
from backend.common.security.jwt import get_hash_password, get_token, jwt_decode, password_verify, superuser_verify
|
from backend.common.response.response_code import CustomErrorCode
|
||||||
|
from backend.common.security.jwt import get_token, jwt_decode, password_verify, superuser_verify
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
from backend.database.db import async_db_session
|
from backend.database.db import async_db_session
|
||||||
from backend.database.redis import redis_client
|
from backend.database.redis import redis_client
|
||||||
@@ -28,176 +29,33 @@ class UserService:
|
|||||||
"""用户服务类"""
|
"""用户服务类"""
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def register(*, obj: RegisterUserParam) -> None:
|
async def get_userinfo(*, pk: int | None = None, username: str | None = None) -> User:
|
||||||
"""
|
|
||||||
注册新用户
|
|
||||||
|
|
||||||
:param obj: 用户注册参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
if not obj.password:
|
|
||||||
raise errors.ForbiddenError(msg='密码为空')
|
|
||||||
username = await user_dao.get_by_username(db, obj.username)
|
|
||||||
if username:
|
|
||||||
raise errors.ForbiddenError(msg='用户已注册')
|
|
||||||
obj.nickname = obj.nickname if obj.nickname else f'#{random.randrange(10000, 88888)}'
|
|
||||||
nickname = await user_dao.get_by_nickname(db, obj.nickname)
|
|
||||||
if nickname:
|
|
||||||
raise errors.ForbiddenError(msg='昵称已注册')
|
|
||||||
email = await user_dao.check_email(db, obj.email)
|
|
||||||
if email:
|
|
||||||
raise errors.ForbiddenError(msg='邮箱已注册')
|
|
||||||
await user_dao.create(db, obj)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def add(*, request: Request, obj: AddUserParam) -> None:
|
|
||||||
"""
|
|
||||||
添加新用户
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param obj: 用户添加参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
superuser_verify(request)
|
|
||||||
username = await user_dao.get_by_username(db, obj.username)
|
|
||||||
if username:
|
|
||||||
raise errors.ForbiddenError(msg='用户已注册')
|
|
||||||
obj.nickname = obj.nickname if obj.nickname else f'#{random.randrange(88888, 99999)}'
|
|
||||||
nickname = await user_dao.get_by_nickname(db, obj.nickname)
|
|
||||||
if nickname:
|
|
||||||
raise errors.ForbiddenError(msg='昵称已注册')
|
|
||||||
if not obj.password:
|
|
||||||
raise errors.ForbiddenError(msg='密码为空')
|
|
||||||
email = await user_dao.check_email(db, obj.email)
|
|
||||||
if email:
|
|
||||||
raise errors.ForbiddenError(msg='邮箱已注册')
|
|
||||||
dept = await dept_dao.get(db, obj.dept_id)
|
|
||||||
if not dept:
|
|
||||||
raise errors.NotFoundError(msg='部门不存在')
|
|
||||||
for role_id in obj.roles:
|
|
||||||
role = await role_dao.get(db, role_id)
|
|
||||||
if not role:
|
|
||||||
raise errors.NotFoundError(msg='角色不存在')
|
|
||||||
await user_dao.add(db, obj)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def pwd_reset(*, request: Request, obj: ResetPasswordParam) -> int:
|
|
||||||
"""
|
|
||||||
重置用户密码
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param obj: 密码重置参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
user = await user_dao.get(db, request.user.id)
|
|
||||||
if not user:
|
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
|
||||||
if not password_verify(obj.old_password, user.password):
|
|
||||||
raise errors.ForbiddenError(msg='原密码错误')
|
|
||||||
if obj.new_password != obj.confirm_password:
|
|
||||||
raise errors.ForbiddenError(msg='密码输入不一致')
|
|
||||||
new_pwd = get_hash_password(obj.new_password, user.salt)
|
|
||||||
count = await user_dao.reset_password(db, request.user.id, new_pwd)
|
|
||||||
key_prefix = [
|
|
||||||
f'{settings.TOKEN_REDIS_PREFIX}:{request.user.id}',
|
|
||||||
f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{request.user.id}',
|
|
||||||
f'{settings.JWT_USER_REDIS_PREFIX}:{request.user.id}',
|
|
||||||
]
|
|
||||||
for prefix in key_prefix:
|
|
||||||
await redis_client.delete_prefix(prefix)
|
|
||||||
return count
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def get_userinfo(*, username: str) -> User:
|
|
||||||
"""
|
"""
|
||||||
获取用户信息
|
获取用户信息
|
||||||
|
|
||||||
|
:param pk: 用户 ID
|
||||||
:param username: 用户名
|
:param username: 用户名
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session() as db:
|
async with async_db_session() as db:
|
||||||
user = await user_dao.get_with_relation(db, username=username)
|
user = await user_dao.get_with_relation(db, user_id=pk, username=username)
|
||||||
if not user:
|
if not user:
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
return user
|
return user
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def update(*, request: Request, username: str, obj: UpdateUserParam) -> int:
|
async def get_roles(*, pk: int) -> Sequence[Role]:
|
||||||
"""
|
"""
|
||||||
更新用户信息
|
获取用户所有角色
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param pk: 用户 ID
|
||||||
:param username: 用户名
|
|
||||||
:param obj: 用户更新参数
|
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session() as db:
|
||||||
if request.user.username != username:
|
user = await user_dao.get_with_relation(db, user_id=pk)
|
||||||
raise errors.ForbiddenError(msg='你只能修改自己的信息')
|
|
||||||
user = await user_dao.get_with_relation(db, username=username)
|
|
||||||
if not user:
|
if not user:
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
if user.username != obj.username:
|
return user.roles
|
||||||
_username = await user_dao.get_by_username(db, obj.username)
|
|
||||||
if _username:
|
|
||||||
raise errors.ForbiddenError(msg='用户名已注册')
|
|
||||||
if user.nickname != obj.nickname:
|
|
||||||
nickname = await user_dao.get_by_nickname(db, obj.nickname)
|
|
||||||
if nickname:
|
|
||||||
raise errors.ForbiddenError(msg='昵称已注册')
|
|
||||||
if user.email != obj.email:
|
|
||||||
email = await user_dao.check_email(db, obj.email)
|
|
||||||
if email:
|
|
||||||
raise errors.ForbiddenError(msg='邮箱已注册')
|
|
||||||
count = await user_dao.update_userinfo(db, user.id, obj)
|
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
|
||||||
return count
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def update_roles(*, request: Request, username: str, obj: UpdateUserRoleParam) -> None:
|
|
||||||
"""
|
|
||||||
更新用户角色
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param username: 用户名
|
|
||||||
:param obj: 角色更新参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
if not request.user.is_superuser and request.user.username != username:
|
|
||||||
raise errors.ForbiddenError(msg='你只能修改自己的信息')
|
|
||||||
input_user = await user_dao.get_with_relation(db, username=username)
|
|
||||||
if not input_user:
|
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
|
||||||
for role_id in obj.roles:
|
|
||||||
role = await role_dao.get(db, role_id)
|
|
||||||
if not role:
|
|
||||||
raise errors.NotFoundError(msg='角色不存在')
|
|
||||||
await user_dao.update_role(db, input_user, obj)
|
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{input_user.id}')
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def update_avatar(*, request: Request, username: str, avatar: AvatarParam) -> int:
|
|
||||||
"""
|
|
||||||
更新用户头像
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param username: 用户名
|
|
||||||
:param avatar: 头像参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
if request.user.username != username:
|
|
||||||
raise errors.AuthorizationError
|
|
||||||
user = await user_dao.get_by_username(db, username)
|
|
||||||
if not user:
|
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
|
||||||
count = await user_dao.update_avatar(db, user.id, avatar)
|
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
|
||||||
return count
|
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def get_select(*, dept: int, username: str, phone: str, status: int) -> Select:
|
async def get_select(*, dept: int, username: str, phone: str, status: int) -> Select:
|
||||||
@@ -213,33 +71,122 @@ class UserService:
|
|||||||
return await user_dao.get_list(dept=dept, username=username, phone=phone, status=status)
|
return await user_dao.get_list(dept=dept, username=username, phone=phone, status=status)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def update_permission(*, request: Request, pk: int) -> int:
|
async def create(*, request: Request, obj: AddUserParam) -> None:
|
||||||
|
"""
|
||||||
|
创建用户
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:param obj: 用户添加参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
superuser_verify(request)
|
||||||
|
if await user_dao.get_by_username(db, obj.username):
|
||||||
|
raise errors.ConflictError(msg='用户名已注册')
|
||||||
|
obj.nickname = obj.nickname if obj.nickname else f'#{random.randrange(88888, 99999)}'
|
||||||
|
if not obj.password:
|
||||||
|
raise errors.RequestError(msg='密码不允许为空')
|
||||||
|
if not await dept_dao.get(db, obj.dept_id):
|
||||||
|
raise errors.NotFoundError(msg='部门不存在')
|
||||||
|
for role_id in obj.roles:
|
||||||
|
if not await role_dao.get(db, role_id):
|
||||||
|
raise errors.NotFoundError(msg='角色不存在')
|
||||||
|
await user_dao.add(db, obj)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update(*, request: Request, pk: int, obj: UpdateUserParam) -> int:
|
||||||
|
"""
|
||||||
|
更新用户信息
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:param pk: 用户 ID
|
||||||
|
:param obj: 用户更新参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
superuser_verify(request)
|
||||||
|
user = await user_dao.get_with_relation(db, user_id=pk)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
if obj.username != user.username:
|
||||||
|
if await user_dao.get_by_username(db, obj.username):
|
||||||
|
raise errors.ConflictError(msg='用户名已注册')
|
||||||
|
for role_id in obj.roles:
|
||||||
|
if not await role_dao.get(db, role_id):
|
||||||
|
raise errors.NotFoundError(msg='角色不存在')
|
||||||
|
count = await user_dao.update(db, user, obj)
|
||||||
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update_permission(*, request: Request, pk: int, type: UserPermissionType) -> int:
|
||||||
"""
|
"""
|
||||||
更新用户权限
|
更新用户权限
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param request: FastAPI 请求对象
|
||||||
:param pk: 用户 ID
|
:param pk: 用户 ID
|
||||||
|
:param type: 权限类型
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
superuser_verify(request)
|
superuser_verify(request)
|
||||||
user = await user_dao.get(db, pk)
|
match type:
|
||||||
if not user:
|
case UserPermissionType.superuser:
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
user = await user_dao.get(db, pk)
|
||||||
if pk == request.user.id:
|
if not user:
|
||||||
raise errors.ForbiddenError(msg='非法操作')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
super_status = await user_dao.get_super(db, pk)
|
if pk == request.user.id:
|
||||||
count = await user_dao.set_super(db, pk, not super_status)
|
raise errors.ForbiddenError(msg='禁止修改自身权限')
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
count = await user_dao.set_super(db, pk, not user.status)
|
||||||
return count
|
case UserPermissionType.staff:
|
||||||
|
user = await user_dao.get(db, pk)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
if pk == request.user.id:
|
||||||
|
raise errors.ForbiddenError(msg='禁止修改自身权限')
|
||||||
|
count = await user_dao.set_staff(db, pk, not user.is_staff)
|
||||||
|
case UserPermissionType.status:
|
||||||
|
user = await user_dao.get(db, pk)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
if pk == request.user.id:
|
||||||
|
raise errors.ForbiddenError(msg='禁止修改自身权限')
|
||||||
|
count = await user_dao.set_status(db, pk, 0 if user.status == 1 else 1)
|
||||||
|
case UserPermissionType.multi_login:
|
||||||
|
user = await user_dao.get(db, pk)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
multi_login = user.is_multi_login if pk != user.id else request.user.is_multi_login
|
||||||
|
new_multi_login = not multi_login
|
||||||
|
count = await user_dao.set_multi_login(db, pk, new_multi_login)
|
||||||
|
token = get_token(request)
|
||||||
|
token_payload = jwt_decode(token)
|
||||||
|
if pk == user.id:
|
||||||
|
# 系统管理员修改自身时,除当前 token 外,其他 token 失效
|
||||||
|
if not new_multi_login:
|
||||||
|
key_prefix = f'{settings.TOKEN_REDIS_PREFIX}:{user.id}'
|
||||||
|
await redis_client.delete_prefix(
|
||||||
|
key_prefix, exclude=f'{key_prefix}:{token_payload.session_uuid}'
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
# 系统管理员修改他人时,他人 token 全部失效
|
||||||
|
if not new_multi_login:
|
||||||
|
key_prefix = f'{settings.TOKEN_REDIS_PREFIX}:{user.id}'
|
||||||
|
await redis_client.delete_prefix(key_prefix)
|
||||||
|
case _:
|
||||||
|
raise errors.RequestError(msg='权限类型不存在')
|
||||||
|
|
||||||
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def update_staff(*, request: Request, pk: int) -> int:
|
async def reset_password(*, request: Request, pk: int, password: str) -> int:
|
||||||
"""
|
"""
|
||||||
更新用户职员状态
|
重置用户密码
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param request: FastAPI 请求对象
|
||||||
:param pk: 用户 ID
|
:param pk: 用户 ID
|
||||||
|
:param password: 新密码
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
@@ -247,76 +194,119 @@ class UserService:
|
|||||||
user = await user_dao.get(db, pk)
|
user = await user_dao.get(db, pk)
|
||||||
if not user:
|
if not user:
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
if pk == request.user.id:
|
count = await user_dao.reset_password(db, user.id, password)
|
||||||
raise errors.ForbiddenError(msg='非法操作')
|
key_prefix = [
|
||||||
staff_status = await user_dao.get_staff(db, pk)
|
f'{settings.TOKEN_REDIS_PREFIX}:{user.id}',
|
||||||
count = await user_dao.set_staff(db, pk, not staff_status)
|
f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{user.id}',
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}',
|
||||||
|
]
|
||||||
|
for prefix in key_prefix:
|
||||||
|
await redis_client.delete(prefix)
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def update_status(*, request: Request, pk: int) -> int:
|
async def update_nickname(*, request: Request, nickname: str) -> int:
|
||||||
"""
|
"""
|
||||||
更新用户状态
|
更新当前用户昵称
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
:param request: FastAPI 请求对象
|
||||||
:param pk: 用户 ID
|
:param nickname: 用户昵称
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
superuser_verify(request)
|
|
||||||
user = await user_dao.get(db, pk)
|
|
||||||
if not user:
|
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
|
||||||
if pk == request.user.id:
|
|
||||||
raise errors.ForbiddenError(msg='非法操作')
|
|
||||||
status = await user_dao.get_status(db, pk)
|
|
||||||
count = await user_dao.set_status(db, pk, 0 if status == 1 else 1)
|
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
|
||||||
return count
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
async def update_multi_login(*, request: Request, pk: int) -> int:
|
|
||||||
"""
|
|
||||||
更新用户多端登录状态
|
|
||||||
|
|
||||||
:param request: FastAPI 请求对象
|
|
||||||
:param pk: 用户 ID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
async with async_db_session.begin() as db:
|
|
||||||
superuser_verify(request)
|
|
||||||
user = await user_dao.get(db, pk)
|
|
||||||
if not user:
|
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
|
||||||
multi_login = await user_dao.get_multi_login(db, pk) if pk != user.id else request.user.is_multi_login
|
|
||||||
new_multi_login = not multi_login
|
|
||||||
count = await user_dao.set_multi_login(db, pk, new_multi_login)
|
|
||||||
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
|
||||||
token = get_token(request)
|
token = get_token(request)
|
||||||
token_payload = jwt_decode(token)
|
token_payload = jwt_decode(token)
|
||||||
if pk == user.id:
|
user = await user_dao.get(db, token_payload.id)
|
||||||
# 系统管理员修改自身时,除当前 token 外,其他 token 失效
|
if not user:
|
||||||
if not new_multi_login:
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
key_prefix = f'{settings.TOKEN_REDIS_PREFIX}:{user.id}'
|
count = await user_dao.update_nickname(db, token_payload.id, nickname)
|
||||||
await redis_client.delete_prefix(key_prefix, exclude=f'{key_prefix}:{token_payload.session_uuid}')
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
else:
|
|
||||||
# 系统管理员修改他人时,他人 token 全部失效
|
|
||||||
if not new_multi_login:
|
|
||||||
key_prefix = f'{settings.TOKEN_REDIS_PREFIX}:{user.id}'
|
|
||||||
await redis_client.delete_prefix(key_prefix)
|
|
||||||
return count
|
return count
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
async def delete(*, username: str) -> int:
|
async def update_avatar(*, request: Request, avatar: str) -> int:
|
||||||
|
"""
|
||||||
|
更新当前用户头像
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:param avatar: 头像地址
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
token = get_token(request)
|
||||||
|
token_payload = jwt_decode(token)
|
||||||
|
user = await user_dao.get(db, token_payload.id)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
count = await user_dao.update_avatar(db, token_payload.id, avatar)
|
||||||
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update_email(*, request: Request, captcha: str, email: str) -> int:
|
||||||
|
"""
|
||||||
|
更新当前用户邮箱
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:param captcha: 邮箱验证码
|
||||||
|
:param email: 邮箱
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
token = get_token(request)
|
||||||
|
token_payload = jwt_decode(token)
|
||||||
|
user = await user_dao.get(db, token_payload.id)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
captcha_code = await redis_client.get(f'{settings.EMAIL_CAPTCHA_REDIS_PREFIX}:{request.state.ip}')
|
||||||
|
if not captcha_code:
|
||||||
|
raise errors.RequestError(msg='验证码已失效,请重新获取')
|
||||||
|
if captcha != captcha_code:
|
||||||
|
raise errors.CustomError(error=CustomErrorCode.CAPTCHA_ERROR)
|
||||||
|
await redis_client.delete(f'{settings.EMAIL_CAPTCHA_REDIS_PREFIX}:{request.state.ip}')
|
||||||
|
count = await user_dao.update_email(db, token_payload.id, email)
|
||||||
|
await redis_client.delete(f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}')
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update_password(*, request: Request, obj: ResetPasswordParam) -> int:
|
||||||
|
"""
|
||||||
|
更新当前用户密码
|
||||||
|
|
||||||
|
:param request: FastAPI 请求对象
|
||||||
|
:param obj: 密码重置参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
token = get_token(request)
|
||||||
|
token_payload = jwt_decode(token)
|
||||||
|
user = await user_dao.get(db, token_payload.id)
|
||||||
|
if not user:
|
||||||
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
|
if not password_verify(obj.old_password, user.password):
|
||||||
|
raise errors.RequestError(msg='原密码错误')
|
||||||
|
if obj.new_password != obj.confirm_password:
|
||||||
|
raise errors.RequestError(msg='密码输入不一致')
|
||||||
|
count = await user_dao.reset_password(db, user.id, obj.new_password)
|
||||||
|
key_prefix = [
|
||||||
|
f'{settings.TOKEN_REDIS_PREFIX}:{user.id}',
|
||||||
|
f'{settings.TOKEN_REFRESH_REDIS_PREFIX}:{user.id}',
|
||||||
|
f'{settings.JWT_USER_REDIS_PREFIX}:{user.id}',
|
||||||
|
]
|
||||||
|
for prefix in key_prefix:
|
||||||
|
await redis_client.delete_prefix(prefix)
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def delete(*, pk: int) -> int:
|
||||||
"""
|
"""
|
||||||
删除用户
|
删除用户
|
||||||
|
|
||||||
:param username: 用户名
|
:param pk: 用户 ID
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
async with async_db_session.begin() as db:
|
async with async_db_session.begin() as db:
|
||||||
user = await user_dao.get_by_username(db, username)
|
user = await user_dao.get(db, pk)
|
||||||
if not user:
|
if not user:
|
||||||
raise errors.NotFoundError(msg='用户不存在')
|
raise errors.NotFoundError(msg='用户不存在')
|
||||||
count = await user_dao.delete(db, user.id)
|
count = await user_dao.delete(db, user.id)
|
||||||
|
|||||||
@@ -3,21 +3,21 @@
|
|||||||
当前任务使用 Celery
|
当前任务使用 Celery
|
||||||
实现,实施方案请查看 [#225](https://github.com/fastapi-practices/fastapi_best_architecture/discussions/225)
|
实现,实施方案请查看 [#225](https://github.com/fastapi-practices/fastapi_best_architecture/discussions/225)
|
||||||
|
|
||||||
## 添加任务
|
## 定时任务
|
||||||
|
|
||||||
> [!IMPORTANT]
|
在 `backend/app/task/tasks/beat.py` 文件内编写相关定时任务
|
||||||
> 由于 Celery 任务扫描规则,使其对任务的目录结构要求及其严格,务必在 celery_task 目录下添加任务
|
|
||||||
|
|
||||||
### 简单任务
|
### 简单任务
|
||||||
|
|
||||||
可以直接在 `tasks.py` 文件内编写相关任务代码
|
在 `backend/app/task/tasks/tasks.py` 文件内编写相关任务代码
|
||||||
|
|
||||||
### 层级任务
|
### 层级任务
|
||||||
|
|
||||||
如果你想对任务进行目录层级划分,使任务结构更加清晰,你可以新建任意目录,但必须注意的是
|
如果你想对任务进行目录层级划分,使任务结构更加清晰,你可以新建任意目录,但必须注意的是
|
||||||
|
|
||||||
1. 新建目录后,务必更新任务配置 `CELERY_TASKS_PACKAGES`,将新建目录添加到此列表
|
1. 在 `backend/app/task/tasks` 目录下新建 python 包目录
|
||||||
2. 在新建目录下,务必添加 `tasks.py` 文件,并在此文件中编写相关任务代码
|
2. 新建目录后,务必更新 `conf.py` 配置中的 `CELERY_TASKS_PACKAGES`,将新建目录模块路径添加到此列表
|
||||||
|
3. 在新建目录下,务必添加 `tasks.py` 文件,并在此文件中编写相关任务代码
|
||||||
|
|
||||||
## 消息代理
|
## 消息代理
|
||||||
|
|
||||||
|
|||||||
@@ -2,7 +2,9 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from pathlib import Path
|
from backend.core.path_conf import BASE_PATH
|
||||||
|
|
||||||
|
from .actions import * # noqa: F403
|
||||||
|
|
||||||
# 导入项目根目录
|
# 导入项目根目录
|
||||||
sys.path.append(str(Path(__file__).resolve().parent.parent.parent.parent))
|
sys.path.append(str(BASE_PATH.parent))
|
||||||
|
|||||||
@@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from starlette.concurrency import run_in_threadpool
|
||||||
|
|
||||||
|
from backend.app.task.celery import celery_app
|
||||||
|
from backend.common.socketio.server import sio
|
||||||
|
|
||||||
|
|
||||||
|
@sio.event
|
||||||
|
async def task_worker_status(sid, data):
|
||||||
|
"""任务 Worker 状态事件"""
|
||||||
|
worker = await run_in_threadpool(celery_app.control.ping)
|
||||||
|
await sio.emit('task_worker_status', worker, sid)
|
||||||
@@ -2,9 +2,13 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from fastapi import APIRouter
|
from fastapi import APIRouter
|
||||||
|
|
||||||
from backend.app.task.api.v1.task import router as task_router
|
from backend.app.task.api.v1.control import router as task_control_router
|
||||||
|
from backend.app.task.api.v1.result import router as task_result_router
|
||||||
|
from backend.app.task.api.v1.scheduler import router as task_scheduler_router
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
|
|
||||||
v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH)
|
v1 = APIRouter(prefix=f'{settings.FASTAPI_API_V1_PATH}/tasks', tags=['任务'])
|
||||||
|
|
||||||
v1.include_router(task_router, prefix='/tasks', tags=['任务'])
|
v1.include_router(task_control_router)
|
||||||
|
v1.include_router(task_result_router, prefix='/results')
|
||||||
|
v1.include_router(task_scheduler_router, prefix='/schedulers')
|
||||||
|
|||||||
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Annotated
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, Path
|
||||||
|
from starlette.concurrency import run_in_threadpool
|
||||||
|
|
||||||
|
from backend.app.task import celery_app
|
||||||
|
from backend.app.task.schema.control import TaskRegisteredDetail
|
||||||
|
from backend.common.exception import errors
|
||||||
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
|
from backend.common.security.permission import RequestPermission
|
||||||
|
from backend.common.security.rbac import DependsRBAC
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/registered', summary='获取已注册的任务', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_task_registered() -> ResponseSchemaModel[list[TaskRegisteredDetail]]:
|
||||||
|
inspector = celery_app.control.inspect(timeout=0.5)
|
||||||
|
registered = await run_in_threadpool(inspector.registered)
|
||||||
|
if not registered:
|
||||||
|
raise errors.ServerError(msg='Celery Worker 暂不可用,请稍后重试')
|
||||||
|
task_registered = []
|
||||||
|
celery_app_tasks = celery_app.tasks
|
||||||
|
for _, tasks in registered.items():
|
||||||
|
for task in tasks:
|
||||||
|
task_ins = celery_app_tasks.get(task)
|
||||||
|
if task_ins:
|
||||||
|
task_doc = task_ins.__doc__
|
||||||
|
task_registered.append({'name': task_doc or task_ins, 'task': task_ins})
|
||||||
|
else:
|
||||||
|
task_registered.append({'name': task, 'task': task})
|
||||||
|
return response_base.success(data=task_registered)
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete(
|
||||||
|
'/{task_id}/cancel',
|
||||||
|
summary='撤销任务',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:revoke')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def revoke_task(task_id: Annotated[str, Path(description='任务 UUID')]) -> ResponseModel:
|
||||||
|
workers = await run_in_threadpool(celery_app.control.ping, timeout=0.5)
|
||||||
|
if not workers:
|
||||||
|
raise errors.ServerError(msg='Celery Worker 暂不可用,请稍后重试')
|
||||||
|
celery_app.control.revoke(task_id)
|
||||||
|
return response_base.success()
|
||||||
@@ -0,0 +1,57 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Annotated
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, Path, Query
|
||||||
|
|
||||||
|
from backend.app.task.schema.result import DeleteTaskResultParam, GetTaskResultDetail
|
||||||
|
from backend.app.task.service.result_service import task_result_service
|
||||||
|
from backend.common.pagination import DependsPagination, PageData, paging_data
|
||||||
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
|
from backend.common.security.permission import RequestPermission
|
||||||
|
from backend.common.security.rbac import DependsRBAC
|
||||||
|
from backend.database.db import CurrentSession
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/{pk}', summary='获取任务结果详情', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_task_result(
|
||||||
|
pk: Annotated[int, Path(description='任务结果 ID')],
|
||||||
|
) -> ResponseSchemaModel[GetTaskResultDetail]:
|
||||||
|
result = await task_result_service.get(pk=pk)
|
||||||
|
return response_base.success(data=result)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get(
|
||||||
|
'',
|
||||||
|
summary='分页获取所有任务结果',
|
||||||
|
dependencies=[
|
||||||
|
DependsJwtAuth,
|
||||||
|
DependsPagination,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def get_task_results_paged(
|
||||||
|
db: CurrentSession,
|
||||||
|
name: Annotated[str | None, Query(description='任务名称')] = None,
|
||||||
|
task_id: Annotated[str | None, Query(description='任务 ID')] = None,
|
||||||
|
) -> ResponseSchemaModel[PageData[GetTaskResultDetail]]:
|
||||||
|
result_select = await task_result_service.get_select(name=name, task_id=task_id)
|
||||||
|
page_data = await paging_data(db, result_select)
|
||||||
|
return response_base.success(data=page_data)
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete(
|
||||||
|
'',
|
||||||
|
summary='批量删除任务结果',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:del')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def delete_task_result(obj: DeleteTaskResultParam) -> ResponseModel:
|
||||||
|
count = await task_result_service.delete(obj=obj)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
@@ -0,0 +1,121 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Annotated
|
||||||
|
|
||||||
|
from fastapi import APIRouter, Depends, Path, Query
|
||||||
|
|
||||||
|
from backend.app.task.schema.scheduler import CreateTaskSchedulerParam, GetTaskSchedulerDetail, UpdateTaskSchedulerParam
|
||||||
|
from backend.app.task.service.scheduler_service import task_scheduler_service
|
||||||
|
from backend.common.pagination import DependsPagination, PageData, paging_data
|
||||||
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
||||||
|
from backend.common.security.jwt import DependsJwtAuth
|
||||||
|
from backend.common.security.permission import RequestPermission
|
||||||
|
from backend.common.security.rbac import DependsRBAC
|
||||||
|
from backend.database.db import CurrentSession
|
||||||
|
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/all', summary='获取所有任务调度', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_all_task_schedulers() -> ResponseSchemaModel[list[GetTaskSchedulerDetail]]:
|
||||||
|
schedulers = await task_scheduler_service.get_all()
|
||||||
|
return response_base.success(data=schedulers)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get('/{pk}', summary='获取任务调度详情', dependencies=[DependsJwtAuth])
|
||||||
|
async def get_task_scheduler(
|
||||||
|
pk: Annotated[int, Path(description='任务调度 ID')],
|
||||||
|
) -> ResponseSchemaModel[GetTaskSchedulerDetail]:
|
||||||
|
task_scheduler = await task_scheduler_service.get(pk=pk)
|
||||||
|
return response_base.success(data=task_scheduler)
|
||||||
|
|
||||||
|
|
||||||
|
@router.get(
|
||||||
|
'',
|
||||||
|
summary='分页获取所有任务调度',
|
||||||
|
dependencies=[
|
||||||
|
DependsJwtAuth,
|
||||||
|
DependsPagination,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def get_task_scheduler_paged(
|
||||||
|
db: CurrentSession,
|
||||||
|
name: Annotated[int, Path(description='任务调度名称')] = None,
|
||||||
|
type: Annotated[int | None, Query(description='任务调度类型')] = None,
|
||||||
|
) -> ResponseSchemaModel[PageData[GetTaskSchedulerDetail]]:
|
||||||
|
task_scheduler_select = await task_scheduler_service.get_select(name=name, type=type)
|
||||||
|
page_data = await paging_data(db, task_scheduler_select)
|
||||||
|
return response_base.success(data=page_data)
|
||||||
|
|
||||||
|
|
||||||
|
@router.post(
|
||||||
|
'',
|
||||||
|
summary='创建任务调度',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:add')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def create_task_scheduler(obj: CreateTaskSchedulerParam) -> ResponseModel:
|
||||||
|
await task_scheduler_service.create(obj=obj)
|
||||||
|
return response_base.success()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put(
|
||||||
|
'/{pk}',
|
||||||
|
summary='更新任务调度',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:edit')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def update_task_scheduler(
|
||||||
|
pk: Annotated[int, Path(description='任务调度 ID')], obj: UpdateTaskSchedulerParam
|
||||||
|
) -> ResponseModel:
|
||||||
|
count = await task_scheduler_service.update(pk=pk, obj=obj)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.put(
|
||||||
|
'/{pk}/status',
|
||||||
|
summary='更新任务调度状态',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:edit')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def update_task_scheduler_status(pk: Annotated[int, Path(description='任务调度 ID')]) -> ResponseModel:
|
||||||
|
count = await task_scheduler_service.update_status(pk=pk)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.delete(
|
||||||
|
'/{pk}',
|
||||||
|
summary='删除任务调度',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:del')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def delete_task_scheduler(pk: Annotated[int, Path(description='任务调度 ID')]) -> ResponseModel:
|
||||||
|
count = await task_scheduler_service.delete(pk=pk)
|
||||||
|
if count > 0:
|
||||||
|
return response_base.success()
|
||||||
|
return response_base.fail()
|
||||||
|
|
||||||
|
|
||||||
|
@router.post(
|
||||||
|
'/{pk}/executions',
|
||||||
|
summary='手动执行任务',
|
||||||
|
dependencies=[
|
||||||
|
Depends(RequestPermission('sys:task:exec')),
|
||||||
|
DependsRBAC,
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def execute_task(pk: Annotated[int, Path(description='任务调度 ID')]) -> ResponseModel:
|
||||||
|
await task_scheduler_service.execute(pk=pk)
|
||||||
|
return response_base.success()
|
||||||
@@ -1,58 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from typing import Annotated
|
|
||||||
|
|
||||||
from fastapi import APIRouter, Depends, Path
|
|
||||||
|
|
||||||
from backend.app.task.schema.task import RunParam, TaskResult
|
|
||||||
from backend.app.task.service.task_service import task_service
|
|
||||||
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
|
||||||
from backend.common.security.jwt import DependsJwtAuth
|
|
||||||
from backend.common.security.permission import RequestPermission
|
|
||||||
from backend.common.security.rbac import DependsRBAC
|
|
||||||
|
|
||||||
router = APIRouter()
|
|
||||||
|
|
||||||
|
|
||||||
@router.get('', summary='获取可执行任务', dependencies=[DependsJwtAuth])
|
|
||||||
async def get_all_tasks() -> ResponseSchemaModel[list[str]]:
|
|
||||||
tasks = await task_service.get_list()
|
|
||||||
return response_base.success(data=tasks)
|
|
||||||
|
|
||||||
|
|
||||||
@router.get(
|
|
||||||
'/{tid}',
|
|
||||||
summary='获取任务详情',
|
|
||||||
deprecated=True,
|
|
||||||
description='此接口被视为作废,建议使用 flower 查看任务详情',
|
|
||||||
dependencies=[DependsJwtAuth],
|
|
||||||
)
|
|
||||||
async def get_task_detail(tid: Annotated[str, Path(description='任务 UUID')]) -> ResponseSchemaModel[TaskResult]:
|
|
||||||
status = task_service.get_detail(tid=tid)
|
|
||||||
return response_base.success(data=status)
|
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
|
||||||
'/{tid}',
|
|
||||||
summary='撤销任务',
|
|
||||||
dependencies=[
|
|
||||||
Depends(RequestPermission('sys:task:revoke')),
|
|
||||||
DependsRBAC,
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def revoke_task(tid: Annotated[str, Path(description='任务 UUID')]) -> ResponseModel:
|
|
||||||
task_service.revoke(tid=tid)
|
|
||||||
return response_base.success()
|
|
||||||
|
|
||||||
|
|
||||||
@router.post(
|
|
||||||
'',
|
|
||||||
summary='执行任务',
|
|
||||||
dependencies=[
|
|
||||||
Depends(RequestPermission('sys:task:run')),
|
|
||||||
DependsRBAC,
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def run_task(obj: RunParam) -> ResponseSchemaModel[str]:
|
|
||||||
task = task_service.run(obj=obj)
|
|
||||||
return response_base.success(data=task)
|
|
||||||
+32
-41
@@ -1,45 +1,23 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
from typing import Any
|
import os
|
||||||
|
|
||||||
import celery
|
import celery
|
||||||
import celery_aio_pool
|
import celery_aio_pool
|
||||||
|
|
||||||
from backend.app.task.conf import task_settings
|
from backend.app.task.tasks.beat import LOCAL_BEAT_SCHEDULE
|
||||||
from backend.core.conf import settings
|
from backend.core.conf import settings
|
||||||
|
from backend.core.path_conf import BASE_PATH
|
||||||
__all__ = ['celery_app']
|
|
||||||
|
|
||||||
|
|
||||||
def get_broker_url() -> str:
|
def find_task_packages():
|
||||||
"""获取消息代理 URL"""
|
packages = []
|
||||||
if task_settings.CELERY_BROKER == 'redis':
|
task_dir = os.path.join(BASE_PATH, 'app', 'task', 'tasks')
|
||||||
return (
|
for root, dirs, files in os.walk(task_dir):
|
||||||
f'redis://:{settings.REDIS_PASSWORD}@{settings.REDIS_HOST}:'
|
if 'tasks.py' in files:
|
||||||
f'{settings.REDIS_PORT}/{task_settings.CELERY_BROKER_REDIS_DATABASE}'
|
package = root.replace(str(BASE_PATH.parent) + os.path.sep, '').replace(os.path.sep, '.')
|
||||||
)
|
packages.append(package)
|
||||||
return (
|
return packages
|
||||||
f'amqp://{task_settings.RABBITMQ_USERNAME}:{task_settings.RABBITMQ_PASSWORD}@'
|
|
||||||
f'{task_settings.RABBITMQ_HOST}:{task_settings.RABBITMQ_PORT}'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def get_result_backend() -> str:
|
|
||||||
"""获取结果后端 URL"""
|
|
||||||
return (
|
|
||||||
f'redis://:{settings.REDIS_PASSWORD}@{settings.REDIS_HOST}:'
|
|
||||||
f'{settings.REDIS_PORT}/{task_settings.CELERY_BACKEND_REDIS_DATABASE}'
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def get_result_backend_transport_options() -> dict[str, Any]:
|
|
||||||
"""获取结果后端传输选项"""
|
|
||||||
return {
|
|
||||||
'global_keyprefix': task_settings.CELERY_BACKEND_REDIS_PREFIX,
|
|
||||||
'retry_policy': {
|
|
||||||
'timeout': task_settings.CELERY_BACKEND_REDIS_TIMEOUT,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
def init_celery() -> celery.Celery:
|
def init_celery() -> celery.Celery:
|
||||||
@@ -51,21 +29,34 @@ def init_celery() -> celery.Celery:
|
|||||||
celery.app.trace.build_tracer = celery_aio_pool.build_async_tracer
|
celery.app.trace.build_tracer = celery_aio_pool.build_async_tracer
|
||||||
celery.app.trace.reset_worker_optimizations()
|
celery.app.trace.reset_worker_optimizations()
|
||||||
|
|
||||||
|
# https://docs.celeryq.dev/en/stable/userguide/configuration.html
|
||||||
app = celery.Celery(
|
app = celery.Celery(
|
||||||
'fba_celery',
|
'fba_celery',
|
||||||
|
broker_url=f'redis://:{settings.REDIS_PASSWORD}@{settings.REDIS_HOST}:{settings.REDIS_PORT}/{settings.CELERY_BROKER_REDIS_DATABASE}'
|
||||||
|
if settings.CELERY_BROKER == 'redis'
|
||||||
|
else f'amqp://{settings.CELERY_RABBITMQ_USERNAME}:{settings.CELERY_RABBITMQ_PASSWORD}@{settings.CELERY_RABBITMQ_HOST}:{settings.CELERY_RABBITMQ_PORT}',
|
||||||
|
broker_connection_retry_on_startup=True,
|
||||||
|
result_backend=f'db+{settings.DATABASE_TYPE}+{"pymysql" if settings.DATABASE_TYPE == "mysql" else "psycopg"}'
|
||||||
|
f'://{settings.DATABASE_USER}:{settings.DATABASE_PASSWORD}@{settings.DATABASE_HOST}:{settings.DATABASE_PORT}/{settings.DATABASE_SCHEMA}',
|
||||||
|
result_extended=True,
|
||||||
|
database_engine_options={'echo': settings.DATABASE_ECHO},
|
||||||
|
# result_expires=0,
|
||||||
|
# beat_sync_every=1,
|
||||||
|
beat_schedule=LOCAL_BEAT_SCHEDULE,
|
||||||
|
beat_scheduler='backend.app.task.utils.schedulers:DatabaseScheduler',
|
||||||
|
task_cls='backend.app.task.tasks.base:TaskBase',
|
||||||
|
task_track_started=True,
|
||||||
enable_utc=False,
|
enable_utc=False,
|
||||||
timezone=settings.DATETIME_TIMEZONE,
|
timezone=settings.DATETIME_TIMEZONE,
|
||||||
beat_schedule=task_settings.CELERY_SCHEDULE,
|
|
||||||
broker_url=get_broker_url(),
|
|
||||||
broker_connection_retry_on_startup=True,
|
|
||||||
result_backend=get_result_backend(),
|
|
||||||
result_backend_transport_options=get_result_backend_transport_options(),
|
|
||||||
task_cls='app.task.celery_task.base:TaskBase',
|
|
||||||
task_track_started=True,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# 在 Celery 中设置此参数无效
|
||||||
|
# 参数:https://github.com/celery/celery/issues/7270
|
||||||
|
app.loader.override_backends = {'db': 'backend.app.task.database:DatabaseBackend'}
|
||||||
|
|
||||||
# 自动发现任务
|
# 自动发现任务
|
||||||
app.autodiscover_tasks(task_settings.CELERY_TASK_PACKAGES)
|
packages = find_task_packages()
|
||||||
|
app.autodiscover_tasks(packages)
|
||||||
|
|
||||||
return app
|
return app
|
||||||
|
|
||||||
|
|||||||
@@ -1,19 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from backend.app.admin.service.login_log_service import login_log_service
|
|
||||||
from backend.app.admin.service.opera_log_service import opera_log_service
|
|
||||||
from backend.app.task.celery import celery_app
|
|
||||||
|
|
||||||
|
|
||||||
@celery_app.task(name='delete_db_opera_log')
|
|
||||||
async def delete_db_opera_log() -> int:
|
|
||||||
"""自动删除数据库操作日志"""
|
|
||||||
result = await opera_log_service.delete_all()
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
@celery_app.task(name='delete_db_login_log')
|
|
||||||
async def delete_db_login_log() -> int:
|
|
||||||
"""自动删除数据库登录日志"""
|
|
||||||
result = await login_log_service.delete_all()
|
|
||||||
return result
|
|
||||||
@@ -1,12 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from anyio import sleep
|
|
||||||
|
|
||||||
from backend.app.task.celery import celery_app
|
|
||||||
|
|
||||||
|
|
||||||
@celery_app.task(name='task_demo_async')
|
|
||||||
async def task_demo_async() -> str:
|
|
||||||
"""异步示例任务,模拟耗时操作"""
|
|
||||||
await sleep(20)
|
|
||||||
return 'test async'
|
|
||||||
@@ -1,73 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from functools import lru_cache
|
|
||||||
from typing import Any, Literal
|
|
||||||
|
|
||||||
from celery.schedules import crontab
|
|
||||||
from pydantic import model_validator
|
|
||||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
||||||
|
|
||||||
from backend.core.path_conf import BASE_PATH
|
|
||||||
|
|
||||||
|
|
||||||
class TaskSettings(BaseSettings):
|
|
||||||
"""Celery 任务配置"""
|
|
||||||
|
|
||||||
model_config = SettingsConfigDict(env_file=f'{BASE_PATH}/.env', env_file_encoding='utf-8', extra='ignore')
|
|
||||||
|
|
||||||
# .env 环境
|
|
||||||
ENVIRONMENT: Literal['dev', 'pro']
|
|
||||||
|
|
||||||
# .env Redis 配置
|
|
||||||
CELERY_BROKER_REDIS_DATABASE: int
|
|
||||||
CELERY_BACKEND_REDIS_DATABASE: int
|
|
||||||
|
|
||||||
# .env RabbitMQ 配置
|
|
||||||
# docker run -d --hostname fba-mq --name fba-mq -p 5672:5672 -p 15672:15672 rabbitmq:latest
|
|
||||||
RABBITMQ_HOST: str
|
|
||||||
RABBITMQ_PORT: int
|
|
||||||
RABBITMQ_USERNAME: str
|
|
||||||
RABBITMQ_PASSWORD: str
|
|
||||||
|
|
||||||
# Celery 基础配置
|
|
||||||
CELERY_BROKER: Literal['rabbitmq', 'redis'] = 'redis'
|
|
||||||
CELERY_BACKEND_REDIS_PREFIX: str = 'fba:celery:'
|
|
||||||
CELERY_BACKEND_REDIS_TIMEOUT: int = 5
|
|
||||||
CELERY_TASK_PACKAGES: list[str] = [
|
|
||||||
'app.task.celery_task',
|
|
||||||
'app.task.celery_task.db_log',
|
|
||||||
]
|
|
||||||
CELERY_TASK_MAX_RETRIES: int = 5
|
|
||||||
|
|
||||||
# Celery 定时任务配置
|
|
||||||
CELERY_SCHEDULE: dict[str, dict[str, Any]] = {
|
|
||||||
'exec-every-10-seconds': {
|
|
||||||
'task': 'task_demo_async',
|
|
||||||
'schedule': 10,
|
|
||||||
},
|
|
||||||
'exec-every-sunday': {
|
|
||||||
'task': 'delete_db_opera_log',
|
|
||||||
'schedule': crontab('0', '0', day_of_week='6'),
|
|
||||||
},
|
|
||||||
'exec-every-15-of-month': {
|
|
||||||
'task': 'delete_db_login_log',
|
|
||||||
'schedule': crontab('0', '0', day_of_month='15'),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
@model_validator(mode='before')
|
|
||||||
@classmethod
|
|
||||||
def validate_celery_broker(cls, values: Any) -> Any:
|
|
||||||
"""生产环境强制使用 RabbitMQ 作为消息代理"""
|
|
||||||
if values['ENVIRONMENT'] == 'pro':
|
|
||||||
values['CELERY_BROKER'] = 'rabbitmq'
|
|
||||||
return values
|
|
||||||
|
|
||||||
|
|
||||||
@lru_cache
|
|
||||||
def get_task_settings() -> TaskSettings:
|
|
||||||
"""获取 Celery 任务配置"""
|
|
||||||
return TaskSettings()
|
|
||||||
|
|
||||||
|
|
||||||
task_settings = get_task_settings()
|
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from sqlalchemy import Select
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
|
from backend.app.task.model import TaskResult
|
||||||
|
|
||||||
|
|
||||||
|
class CRUDTaskResult(CRUDPlus[TaskResult]):
|
||||||
|
"""任务结果数据库操作类"""
|
||||||
|
|
||||||
|
async def get(self, db: AsyncSession, pk: int) -> TaskResult | None:
|
||||||
|
"""
|
||||||
|
获取任务结果详情
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pk: 任务 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.select_model(db, pk)
|
||||||
|
|
||||||
|
async def get_list(self, name: str | None, task_id: str | None) -> Select:
|
||||||
|
"""
|
||||||
|
获取任务结果列表
|
||||||
|
|
||||||
|
:param name: 任务名称
|
||||||
|
:param task_id: 任务 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
filters = {}
|
||||||
|
|
||||||
|
if name is not None:
|
||||||
|
filters['name__like'] = f'%{name}%'
|
||||||
|
if task_id is not None:
|
||||||
|
filters['task_id'] = task_id
|
||||||
|
|
||||||
|
return await self.select_order('id', 'desc', **filters)
|
||||||
|
|
||||||
|
async def delete(self, db: AsyncSession, pks: list[int]) -> int:
|
||||||
|
"""
|
||||||
|
批量删除任务结果
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pks: 任务结果 ID 列表
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.delete_model_by_column(db, allow_multiple=True, id__in=pks)
|
||||||
|
|
||||||
|
|
||||||
|
task_result_dao: CRUDTaskResult = CRUDTaskResult(TaskResult)
|
||||||
@@ -0,0 +1,117 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from typing import Sequence
|
||||||
|
|
||||||
|
from sqlalchemy import Select
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
from sqlalchemy_crud_plus import CRUDPlus
|
||||||
|
|
||||||
|
from backend.app.task.model import TaskScheduler
|
||||||
|
from backend.app.task.schema.scheduler import CreateTaskSchedulerParam, UpdateTaskSchedulerParam
|
||||||
|
|
||||||
|
|
||||||
|
class CRUDTaskScheduler(CRUDPlus[TaskScheduler]):
|
||||||
|
"""任务调度数据库操作类"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get(db: AsyncSession, pk: int) -> TaskScheduler | None:
|
||||||
|
"""
|
||||||
|
获取任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await task_scheduler_dao.select_model(db, pk)
|
||||||
|
|
||||||
|
async def get_all(self, db: AsyncSession) -> Sequence[TaskScheduler]:
|
||||||
|
"""
|
||||||
|
获取所有任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.select_models(db)
|
||||||
|
|
||||||
|
async def get_list(self, name: str | None, type: int | None) -> Select:
|
||||||
|
"""
|
||||||
|
获取任务调度列表
|
||||||
|
|
||||||
|
:param name: 任务调度名称
|
||||||
|
:param type: 任务调度类型
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
filters = {}
|
||||||
|
|
||||||
|
if name is not None:
|
||||||
|
filters['name__like'] = f'%{name}%'
|
||||||
|
if type is not None:
|
||||||
|
filters['type'] = type
|
||||||
|
|
||||||
|
return await self.select_order('id', **filters)
|
||||||
|
|
||||||
|
async def get_by_name(self, db: AsyncSession, name: str) -> TaskScheduler | None:
|
||||||
|
"""
|
||||||
|
通过名称获取任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param name: 任务调度名称
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await self.select_model_by_column(db, name=name)
|
||||||
|
|
||||||
|
async def create(self, db: AsyncSession, obj: CreateTaskSchedulerParam) -> None:
|
||||||
|
"""
|
||||||
|
创建任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param obj: 创建任务调度参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
await self.create_model(db, obj, flush=True)
|
||||||
|
TaskScheduler.no_changes = False
|
||||||
|
|
||||||
|
async def update(self, db: AsyncSession, pk: int, obj: UpdateTaskSchedulerParam) -> int:
|
||||||
|
"""
|
||||||
|
更新任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:param obj: 更新任务调度参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
task_scheduler = await self.get(db, pk)
|
||||||
|
for key, value in obj.model_dump(exclude_unset=True).items():
|
||||||
|
setattr(task_scheduler, key, value)
|
||||||
|
TaskScheduler.no_changes = False
|
||||||
|
return 1
|
||||||
|
|
||||||
|
async def set_status(self, db: AsyncSession, pk: int, status: bool) -> int:
|
||||||
|
"""
|
||||||
|
设置任务调度状态
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:param status: 状态
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
task_scheduler = await self.get(db, pk)
|
||||||
|
setattr(task_scheduler, 'enabled', status)
|
||||||
|
TaskScheduler.no_changes = False
|
||||||
|
return 1
|
||||||
|
|
||||||
|
async def delete(self, db: AsyncSession, pk: int) -> int:
|
||||||
|
"""
|
||||||
|
删除任务调度
|
||||||
|
|
||||||
|
:param db: 数据库会话
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
task_scheduler = await self.get(db, pk)
|
||||||
|
await db.delete(task_scheduler)
|
||||||
|
TaskScheduler.no_changes = False
|
||||||
|
return 1
|
||||||
|
|
||||||
|
|
||||||
|
task_scheduler_dao: CRUDTaskScheduler = CRUDTaskScheduler(TaskScheduler)
|
||||||
@@ -0,0 +1,171 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from celery import states
|
||||||
|
from celery.backends.base import BaseBackend
|
||||||
|
from celery.backends.database import retry, session_cleanup
|
||||||
|
from celery.exceptions import ImproperlyConfigured
|
||||||
|
from celery.utils.time import maybe_timedelta
|
||||||
|
|
||||||
|
from backend.app.task.model.result import Task, TaskExtended, TaskSet
|
||||||
|
from backend.app.task.session import SessionManager
|
||||||
|
|
||||||
|
"""
|
||||||
|
重写 from celery.backends.database 内部 DatabaseBackend 类,此类实现与模型配合不佳,导致 fba 创建表和 alembic 迁移困难
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class DatabaseBackend(BaseBackend):
|
||||||
|
"""The database result backend."""
|
||||||
|
|
||||||
|
# ResultSet.iterate should sleep this much between each pool,
|
||||||
|
# to not bombard the database with queries.
|
||||||
|
subpolling_interval = 0.5
|
||||||
|
|
||||||
|
task_cls = Task
|
||||||
|
taskset_cls = TaskSet
|
||||||
|
|
||||||
|
def __init__(self, dburi=None, engine_options=None, url=None, **kwargs):
|
||||||
|
# The `url` argument was added later and is used by
|
||||||
|
# the app to set backend by url (celery.app.backends.by_url)
|
||||||
|
super().__init__(expires_type=maybe_timedelta, url=url, **kwargs)
|
||||||
|
conf = self.app.conf
|
||||||
|
|
||||||
|
if self.extended_result:
|
||||||
|
self.task_cls = TaskExtended
|
||||||
|
|
||||||
|
self.url = url or dburi or conf.database_url
|
||||||
|
self.engine_options = dict(engine_options or {}, **conf.database_engine_options or {})
|
||||||
|
self.short_lived_sessions = kwargs.get('short_lived_sessions', conf.database_short_lived_sessions)
|
||||||
|
|
||||||
|
schemas = conf.database_table_schemas or {}
|
||||||
|
tablenames = conf.database_table_names or {}
|
||||||
|
self.task_cls.configure(schema=schemas.get('task'), name=tablenames.get('task'))
|
||||||
|
self.taskset_cls.configure(schema=schemas.get('group'), name=tablenames.get('group'))
|
||||||
|
|
||||||
|
if not self.url:
|
||||||
|
raise ImproperlyConfigured(
|
||||||
|
'Missing connection string! Do you have the database_url setting set to a real value?'
|
||||||
|
)
|
||||||
|
|
||||||
|
self.session_manager = SessionManager()
|
||||||
|
|
||||||
|
create_tables_at_setup = conf.database_create_tables_at_setup
|
||||||
|
if create_tables_at_setup is True:
|
||||||
|
self._create_tables()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def extended_result(self):
|
||||||
|
return self.app.conf.find_value_for_key('extended', 'result')
|
||||||
|
|
||||||
|
def _create_tables(self):
|
||||||
|
"""Create the task and taskset tables."""
|
||||||
|
self.ResultSession()
|
||||||
|
|
||||||
|
def ResultSession(self, session_manager=None):
|
||||||
|
if session_manager is None:
|
||||||
|
session_manager = self.session_manager
|
||||||
|
return session_manager.session_factory(
|
||||||
|
dburi=self.url, short_lived_sessions=self.short_lived_sessions, **self.engine_options
|
||||||
|
)
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _store_result(self, task_id, result, state, traceback=None, request=None, **kwargs):
|
||||||
|
"""Store return value and state of an executed task."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
task = list(session.query(self.task_cls).filter(self.task_cls.task_id == task_id))
|
||||||
|
task = task and task[0]
|
||||||
|
if not task:
|
||||||
|
task = self.task_cls(task_id)
|
||||||
|
task.task_id = task_id
|
||||||
|
session.add(task)
|
||||||
|
session.flush()
|
||||||
|
|
||||||
|
self._update_result(task, result, state, traceback=traceback, request=request)
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
def _update_result(self, task, result, state, traceback=None, request=None):
|
||||||
|
meta = self._get_result_meta(
|
||||||
|
result=result, state=state, traceback=traceback, request=request, format_date=False, encode=True
|
||||||
|
)
|
||||||
|
|
||||||
|
# Exclude the primary key id and task_id columns
|
||||||
|
# as we should not set it None
|
||||||
|
columns = [column.name for column in self.task_cls.__table__.columns if column.name not in {'id', 'task_id'}]
|
||||||
|
|
||||||
|
# Iterate through the columns name of the table
|
||||||
|
# to set the value from meta.
|
||||||
|
# If the value is not present in meta, set None
|
||||||
|
for column in columns:
|
||||||
|
value = meta.get(column)
|
||||||
|
setattr(task, column, value)
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _get_task_meta_for(self, task_id):
|
||||||
|
"""Get task meta-data for a task by id."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
task = list(session.query(self.task_cls).filter(self.task_cls.task_id == task_id))
|
||||||
|
task = task and task[0]
|
||||||
|
if not task:
|
||||||
|
task = self.task_cls(task_id)
|
||||||
|
task.status = states.PENDING
|
||||||
|
task.result = None
|
||||||
|
data = task.to_dict()
|
||||||
|
if data.get('args', None) is not None:
|
||||||
|
data['args'] = self.decode(data['args'])
|
||||||
|
if data.get('kwargs', None) is not None:
|
||||||
|
data['kwargs'] = self.decode(data['kwargs'])
|
||||||
|
return self.meta_from_decoded(data)
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _save_group(self, group_id, result):
|
||||||
|
"""Store the result of an executed group."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
group = self.taskset_cls(group_id, result)
|
||||||
|
session.add(group)
|
||||||
|
session.flush()
|
||||||
|
session.commit()
|
||||||
|
return result
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _restore_group(self, group_id):
|
||||||
|
"""Get meta-data for group by id."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
group = session.query(self.taskset_cls).filter(self.taskset_cls.taskset_id == group_id).first()
|
||||||
|
if group:
|
||||||
|
return group.to_dict()
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _delete_group(self, group_id):
|
||||||
|
"""Delete meta-data for group by id."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
session.query(self.taskset_cls).filter(self.taskset_cls.taskset_id == group_id).delete()
|
||||||
|
session.flush()
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
@retry
|
||||||
|
def _forget(self, task_id):
|
||||||
|
"""Forget about result."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
with session_cleanup(session):
|
||||||
|
session.query(self.task_cls).filter(self.task_cls.task_id == task_id).delete()
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
def cleanup(self):
|
||||||
|
"""Delete expired meta-data."""
|
||||||
|
session = self.ResultSession()
|
||||||
|
expires = self.expires
|
||||||
|
now = self.app.now()
|
||||||
|
with session_cleanup(session):
|
||||||
|
session.query(self.task_cls).filter(self.task_cls.date_done < (now - expires)).delete()
|
||||||
|
session.query(self.taskset_cls).filter(self.taskset_cls.date_done < (now - expires)).delete()
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
def __reduce__(self, args=(), kwargs=None):
|
||||||
|
kwargs = {} if not kwargs else kwargs
|
||||||
|
kwargs.update({'dburi': self.url, 'expires': self.expires, 'engine_options': self.engine_options})
|
||||||
|
return super().__reduce__(args, kwargs)
|
||||||
@@ -0,0 +1,20 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from backend.common.enums import IntEnum, StrEnum
|
||||||
|
|
||||||
|
|
||||||
|
class TaskSchedulerType(IntEnum):
|
||||||
|
"""任务调度类型"""
|
||||||
|
|
||||||
|
INTERVAL = 0
|
||||||
|
CRONTAB = 1
|
||||||
|
|
||||||
|
|
||||||
|
class PeriodType(StrEnum):
|
||||||
|
"""周期类型"""
|
||||||
|
|
||||||
|
DAYS = 'days'
|
||||||
|
HOURS = 'hours'
|
||||||
|
MINUTES = 'minutes'
|
||||||
|
SECONDS = 'seconds'
|
||||||
|
MICROSECONDS = 'microseconds'
|
||||||
@@ -0,0 +1,4 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from backend.app.task.model.result import TaskExtended as TaskResult
|
||||||
|
from backend.app.task.model.scheduler import TaskScheduler
|
||||||
@@ -0,0 +1,109 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from datetime import datetime, timezone
|
||||||
|
|
||||||
|
import sqlalchemy as sa
|
||||||
|
|
||||||
|
from celery import states
|
||||||
|
from sqlalchemy.types import PickleType
|
||||||
|
|
||||||
|
from backend.common.model import MappedBase
|
||||||
|
|
||||||
|
"""
|
||||||
|
重写 celery.backends.database.models 内部所有模型,适配 fba 创建表和 alembic 迁移
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
class Task(MappedBase):
|
||||||
|
"""Task result/status."""
|
||||||
|
|
||||||
|
__tablename__ = 'task_result'
|
||||||
|
__table_args__ = {'comment': '任务结果表'}
|
||||||
|
|
||||||
|
id = sa.Column(sa.Integer, sa.Sequence('task_id_sequence'), primary_key=True, autoincrement=True)
|
||||||
|
task_id = sa.Column(sa.String(155), unique=True)
|
||||||
|
status = sa.Column(sa.String(50), default=states.PENDING)
|
||||||
|
result = sa.Column(PickleType, nullable=True)
|
||||||
|
date_done = sa.Column(
|
||||||
|
sa.DateTime, default=datetime.now(timezone.utc), onupdate=datetime.now(timezone.utc), nullable=True
|
||||||
|
)
|
||||||
|
traceback = sa.Column(sa.Text, nullable=True)
|
||||||
|
|
||||||
|
def __init__(self, task_id):
|
||||||
|
self.task_id = task_id
|
||||||
|
|
||||||
|
def to_dict(self):
|
||||||
|
return {
|
||||||
|
'task_id': self.task_id,
|
||||||
|
'status': self.status,
|
||||||
|
'result': self.result,
|
||||||
|
'traceback': self.traceback,
|
||||||
|
'date_done': self.date_done,
|
||||||
|
}
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return '<Task {0.task_id} state: {0.status}>'.format(self)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def configure(cls, schema=None, name=None):
|
||||||
|
cls.__table__.schema = schema
|
||||||
|
cls.id.default.schema = schema
|
||||||
|
cls.__table__.name = name or cls.__tablename__
|
||||||
|
|
||||||
|
|
||||||
|
class TaskExtended(Task):
|
||||||
|
"""For the extend result."""
|
||||||
|
|
||||||
|
__tablename__ = 'task_result'
|
||||||
|
__table_args__ = {'extend_existing': True, 'comment': '任务结果表'}
|
||||||
|
|
||||||
|
name = sa.Column(sa.String(155), nullable=True)
|
||||||
|
args = sa.Column(sa.LargeBinary, nullable=True)
|
||||||
|
kwargs = sa.Column(sa.LargeBinary, nullable=True)
|
||||||
|
worker = sa.Column(sa.String(155), nullable=True)
|
||||||
|
retries = sa.Column(sa.Integer, nullable=True)
|
||||||
|
queue = sa.Column(sa.String(155), nullable=True)
|
||||||
|
|
||||||
|
def to_dict(self):
|
||||||
|
task_dict = super().to_dict()
|
||||||
|
task_dict.update({
|
||||||
|
'name': self.name,
|
||||||
|
'args': self.args,
|
||||||
|
'kwargs': self.kwargs,
|
||||||
|
'worker': self.worker,
|
||||||
|
'retries': self.retries,
|
||||||
|
'queue': self.queue,
|
||||||
|
})
|
||||||
|
return task_dict
|
||||||
|
|
||||||
|
|
||||||
|
class TaskSet(MappedBase):
|
||||||
|
"""TaskSet result."""
|
||||||
|
|
||||||
|
__tablename__ = 'task_set_result'
|
||||||
|
__table_args__ = {'comment': '任务集结果表'}
|
||||||
|
|
||||||
|
id = sa.Column(sa.Integer, sa.Sequence('taskset_id_sequence'), autoincrement=True, primary_key=True)
|
||||||
|
taskset_id = sa.Column(sa.String(155), unique=True)
|
||||||
|
result = sa.Column(PickleType, nullable=True)
|
||||||
|
date_done = sa.Column(sa.DateTime, default=datetime.now(timezone.utc), nullable=True)
|
||||||
|
|
||||||
|
def __init__(self, taskset_id, result):
|
||||||
|
self.taskset_id = taskset_id
|
||||||
|
self.result = result
|
||||||
|
|
||||||
|
def to_dict(self):
|
||||||
|
return {
|
||||||
|
'taskset_id': self.taskset_id,
|
||||||
|
'result': self.result,
|
||||||
|
'date_done': self.date_done,
|
||||||
|
}
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f'<TaskSet: {self.taskset_id}>'
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def configure(cls, schema=None, name=None):
|
||||||
|
cls.__table__.schema = schema
|
||||||
|
cls.id.default.schema = schema
|
||||||
|
cls.__table__.name = name or cls.__tablename__
|
||||||
@@ -0,0 +1,83 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import asyncio
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from sqlalchemy import (
|
||||||
|
JSON,
|
||||||
|
Boolean,
|
||||||
|
String,
|
||||||
|
event,
|
||||||
|
)
|
||||||
|
from sqlalchemy.dialects.mysql import LONGTEXT
|
||||||
|
from sqlalchemy.dialects.postgresql import INTEGER, TEXT
|
||||||
|
from sqlalchemy.orm import Mapped, mapped_column
|
||||||
|
|
||||||
|
from backend.common.exception import errors
|
||||||
|
from backend.common.model import Base, TimeZone, id_key
|
||||||
|
from backend.core.conf import settings
|
||||||
|
from backend.database.redis import redis_client
|
||||||
|
from backend.utils.timezone import timezone
|
||||||
|
|
||||||
|
|
||||||
|
class TaskScheduler(Base):
|
||||||
|
"""任务调度表"""
|
||||||
|
|
||||||
|
__tablename__ = 'task_scheduler'
|
||||||
|
|
||||||
|
id: Mapped[id_key] = mapped_column(init=False)
|
||||||
|
name: Mapped[str] = mapped_column(String(50), unique=True, comment='任务名称')
|
||||||
|
task: Mapped[str] = mapped_column(String(255), comment='要运行的 Celery 任务')
|
||||||
|
args: Mapped[str | None] = mapped_column(JSON(), comment='任务可接收的位置参数')
|
||||||
|
kwargs: Mapped[str | None] = mapped_column(JSON(), comment='任务可接收的关键字参数')
|
||||||
|
queue: Mapped[str | None] = mapped_column(String(255), comment='CELERY_TASK_QUEUES 中定义的队列')
|
||||||
|
exchange: Mapped[str | None] = mapped_column(String(255), comment='低级别 AMQP 路由的交换机')
|
||||||
|
routing_key: Mapped[str | None] = mapped_column(String(255), comment='低级别 AMQP 路由的路由密钥')
|
||||||
|
start_time: Mapped[datetime | None] = mapped_column(TimeZone, comment='任务开始触发的时间')
|
||||||
|
expire_time: Mapped[datetime | None] = mapped_column(TimeZone, comment='任务不再触发的截止时间')
|
||||||
|
expire_seconds: Mapped[int | None] = mapped_column(comment='任务不再触发的秒数时间差')
|
||||||
|
type: Mapped[int] = mapped_column(comment='调度类型(0间隔 1定时)')
|
||||||
|
interval_every: Mapped[int | None] = mapped_column(comment='任务再次运行前的间隔周期数')
|
||||||
|
interval_period: Mapped[str | None] = mapped_column(String(255), comment='任务运行之间的周期类型')
|
||||||
|
crontab: Mapped[str | None] = mapped_column(String(50), default='* * * * *', comment='任务运行的 Crontab 计划')
|
||||||
|
one_off: Mapped[bool] = mapped_column(
|
||||||
|
Boolean().with_variant(INTEGER, 'postgresql'), default=False, comment='是否仅运行一次'
|
||||||
|
)
|
||||||
|
enabled: Mapped[bool] = mapped_column(
|
||||||
|
Boolean().with_variant(INTEGER, 'postgresql'), default=True, comment='是否启用任务'
|
||||||
|
)
|
||||||
|
total_run_count: Mapped[int] = mapped_column(default=0, comment='任务触发的总次数')
|
||||||
|
last_run_time: Mapped[datetime | None] = mapped_column(TimeZone, default=None, comment='任务最后触发的时间')
|
||||||
|
remark: Mapped[str | None] = mapped_column(
|
||||||
|
LONGTEXT().with_variant(TEXT, 'postgresql'), default=None, comment='备注'
|
||||||
|
)
|
||||||
|
|
||||||
|
no_changes: bool = False
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def before_insert_or_update(mapper, connection, target):
|
||||||
|
if target.expire_seconds is not None and target.expire_time:
|
||||||
|
raise errors.ConflictError(msg='expires 和 expire_seconds 只能设置一个')
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def changed(cls, mapper, connection, target):
|
||||||
|
if not target.no_changes:
|
||||||
|
cls.update_changed(mapper, connection, target)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
async def update_changed_async(cls):
|
||||||
|
now = timezone.now()
|
||||||
|
await redis_client.set(f'{settings.CELERY_REDIS_PREFIX}:last_update', timezone.to_str(now))
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def update_changed(cls, mapper, connection, target):
|
||||||
|
asyncio.create_task(cls.update_changed_async())
|
||||||
|
|
||||||
|
|
||||||
|
# 事件监听器
|
||||||
|
event.listen(TaskScheduler, 'before_insert', TaskScheduler.before_insert_or_update)
|
||||||
|
event.listen(TaskScheduler, 'before_update', TaskScheduler.before_insert_or_update)
|
||||||
|
event.listen(TaskScheduler, 'after_insert', TaskScheduler.update_changed)
|
||||||
|
event.listen(TaskScheduler, 'after_delete', TaskScheduler.update_changed)
|
||||||
|
event.listen(TaskScheduler, 'after_update', TaskScheduler.changed)
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from backend.common.schema import SchemaBase
|
||||||
|
|
||||||
|
|
||||||
|
class TaskRegisteredDetail(SchemaBase):
|
||||||
|
name: str
|
||||||
|
task: str
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from datetime import datetime
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
from pydantic import ConfigDict, Field, field_serializer
|
||||||
|
|
||||||
|
from backend.app.task import celery_app
|
||||||
|
from backend.common.schema import SchemaBase
|
||||||
|
|
||||||
|
|
||||||
|
class TaskResultSchemaBase(SchemaBase):
|
||||||
|
"""任务结果基础模型"""
|
||||||
|
|
||||||
|
task_id: str = Field(description='任务 ID')
|
||||||
|
status: str = Field(description='执行状态')
|
||||||
|
result: Any | None = Field(description='执行结果')
|
||||||
|
date_done: datetime | None = Field(description='结束时间')
|
||||||
|
traceback: str | None = Field(description='错误回溯')
|
||||||
|
name: str | None = Field(description='任务名称')
|
||||||
|
args: bytes | None = Field(description='任务位置参数')
|
||||||
|
kwargs: bytes | None = Field(description='任务关键字参数')
|
||||||
|
worker: str | None = Field(description='运行 Worker')
|
||||||
|
retries: int | None = Field(description='重试次数')
|
||||||
|
queue: str | None = Field(description='运行队列')
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteTaskResultParam(SchemaBase):
|
||||||
|
"""删除任务结果参数"""
|
||||||
|
|
||||||
|
pks: list[int] = Field(description='任务结果 ID 列表')
|
||||||
|
|
||||||
|
|
||||||
|
class GetTaskResultDetail(TaskResultSchemaBase):
|
||||||
|
"""任务结果详情"""
|
||||||
|
|
||||||
|
model_config = ConfigDict(from_attributes=True)
|
||||||
|
|
||||||
|
id: int = Field(description='任务结果 ID')
|
||||||
|
|
||||||
|
@field_serializer('args', 'kwargs', when_used='unless-none')
|
||||||
|
def serialize_params(self, value: bytes | None, _info) -> Any:
|
||||||
|
return celery_app.backend.decode(value)
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
from pydantic import ConfigDict, Field
|
||||||
|
from pydantic.types import JsonValue
|
||||||
|
|
||||||
|
from backend.app.task.enums import PeriodType, TaskSchedulerType
|
||||||
|
from backend.common.schema import SchemaBase
|
||||||
|
|
||||||
|
|
||||||
|
class TaskSchedulerSchemeBase(SchemaBase):
|
||||||
|
"""任务调度参数"""
|
||||||
|
|
||||||
|
name: str = Field(description='任务名称')
|
||||||
|
task: str = Field(description='要运行的 Celery 任务')
|
||||||
|
args: JsonValue | None = Field(default=None, description='任务可接收的位置参数')
|
||||||
|
kwargs: JsonValue | None = Field(default=None, description='任务可接收的关键字参数')
|
||||||
|
queue: str | None = Field(default=None, description='CELERY_TASK_QUEUES 中定义的队列')
|
||||||
|
exchange: str | None = Field(default=None, description='低级别 AMQP 路由的交换机')
|
||||||
|
routing_key: str | None = Field(default=None, description='低级别 AMQP 路由的路由密钥')
|
||||||
|
start_time: datetime | None = Field(default=None, description='任务开始触发的时间')
|
||||||
|
expire_time: datetime | None = Field(default=None, description='任务不再触发的截止时间')
|
||||||
|
expire_seconds: int | None = Field(default=None, description='任务不再触发的秒数时间差')
|
||||||
|
type: TaskSchedulerType = Field(description='任务调度类型(0间隔 1定时)')
|
||||||
|
interval_every: int | None = Field(default=None, description='任务再次运行前的间隔周期数')
|
||||||
|
interval_period: PeriodType | None = Field(default=None, description='任务运行之间的周期类型')
|
||||||
|
crontab: str = Field(default='* * * * *', description='运行的 Crontab 表达式')
|
||||||
|
one_off: bool = Field(default=False, description='是否仅运行一次')
|
||||||
|
remark: str | None = Field(default=None, description='备注')
|
||||||
|
|
||||||
|
|
||||||
|
class CreateTaskSchedulerParam(TaskSchedulerSchemeBase):
|
||||||
|
"""创建任务调度参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class UpdateTaskSchedulerParam(TaskSchedulerSchemeBase):
|
||||||
|
"""更新任务调度参数"""
|
||||||
|
|
||||||
|
|
||||||
|
class GetTaskSchedulerDetail(TaskSchedulerSchemeBase):
|
||||||
|
"""任务调度详情"""
|
||||||
|
|
||||||
|
model_config = ConfigDict(from_attributes=True)
|
||||||
|
|
||||||
|
id: int = Field(description='任务调度 ID')
|
||||||
|
enabled: bool = Field(description='是否启用任务')
|
||||||
|
total_run_count: int = Field(description='已运行总次数')
|
||||||
|
last_run_time: datetime | None = Field(None, description='最后运行时间')
|
||||||
|
created_time: datetime = Field(description='创建时间')
|
||||||
|
updated_time: datetime | None = Field(None, description='更新时间')
|
||||||
@@ -1,29 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from typing import Any
|
|
||||||
|
|
||||||
from pydantic import Field
|
|
||||||
|
|
||||||
from backend.common.schema import SchemaBase
|
|
||||||
|
|
||||||
|
|
||||||
class RunParam(SchemaBase):
|
|
||||||
"""任务运行参数"""
|
|
||||||
|
|
||||||
name: str = Field(description='任务名称')
|
|
||||||
args: list[Any] | None = Field(None, description='任务函数位置参数')
|
|
||||||
kwargs: dict[str, Any] | None = Field(None, description='任务函数关键字参数')
|
|
||||||
|
|
||||||
|
|
||||||
class TaskResult(SchemaBase):
|
|
||||||
"""任务执行结果"""
|
|
||||||
|
|
||||||
result: str = Field(description='任务执行结果')
|
|
||||||
traceback: str = Field(description='错误堆栈信息')
|
|
||||||
status: str = Field(description='任务状态')
|
|
||||||
name: str = Field(description='任务名称')
|
|
||||||
args: list[Any] | None = Field(None, description='任务函数位置参数')
|
|
||||||
kwargs: dict[str, Any] | None = Field(None, description='任务函数关键字参数')
|
|
||||||
worker: str = Field(description='执行任务的 worker')
|
|
||||||
retries: int | None = Field(None, description='重试次数')
|
|
||||||
queue: str | None = Field(None, description='任务队列')
|
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from sqlalchemy import Select
|
||||||
|
|
||||||
|
from backend.app.task.crud.crud_result import task_result_dao
|
||||||
|
from backend.app.task.model import TaskResult
|
||||||
|
from backend.app.task.schema.result import DeleteTaskResultParam
|
||||||
|
from backend.common.exception import errors
|
||||||
|
from backend.database.db import async_db_session
|
||||||
|
|
||||||
|
|
||||||
|
class TaskResultService:
|
||||||
|
@staticmethod
|
||||||
|
async def get(*, pk: int) -> TaskResult:
|
||||||
|
"""
|
||||||
|
获取任务结果详情
|
||||||
|
|
||||||
|
:param pk: 任务 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session() as db:
|
||||||
|
result = await task_result_dao.get(db, pk)
|
||||||
|
if not result:
|
||||||
|
raise errors.NotFoundError(msg='任务结果不存在')
|
||||||
|
return result
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_select(*, name: str | None, task_id: str | None) -> Select:
|
||||||
|
"""
|
||||||
|
获取任务结果列表查询条件
|
||||||
|
|
||||||
|
:param name: 任务名称
|
||||||
|
:param task_id: 任务 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await task_result_dao.get_list(name, task_id)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def delete(*, obj: DeleteTaskResultParam) -> int:
|
||||||
|
"""
|
||||||
|
批量删除任务结果
|
||||||
|
|
||||||
|
:param obj: 任务结果 ID 列表
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
count = await task_result_dao.delete(db, obj.pks)
|
||||||
|
return count
|
||||||
|
|
||||||
|
|
||||||
|
task_result_service: TaskResultService = TaskResultService()
|
||||||
@@ -0,0 +1,146 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import json
|
||||||
|
|
||||||
|
from typing import Sequence
|
||||||
|
|
||||||
|
from sqlalchemy import Select
|
||||||
|
from starlette.concurrency import run_in_threadpool
|
||||||
|
|
||||||
|
from backend.app.task.celery import celery_app
|
||||||
|
from backend.app.task.crud.crud_scheduler import task_scheduler_dao
|
||||||
|
from backend.app.task.enums import TaskSchedulerType
|
||||||
|
from backend.app.task.model import TaskScheduler
|
||||||
|
from backend.app.task.schema.scheduler import CreateTaskSchedulerParam, UpdateTaskSchedulerParam
|
||||||
|
from backend.app.task.utils.tzcrontab import crontab_verify
|
||||||
|
from backend.common.exception import errors
|
||||||
|
from backend.database.db import async_db_session
|
||||||
|
|
||||||
|
|
||||||
|
class TaskSchedulerService:
|
||||||
|
"""任务调度服务类"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get(*, pk) -> TaskScheduler | None:
|
||||||
|
"""
|
||||||
|
获取任务调度详情
|
||||||
|
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session() as db:
|
||||||
|
task_scheduler = await task_scheduler_dao.get(db, pk)
|
||||||
|
if not task_scheduler:
|
||||||
|
raise errors.NotFoundError(msg='任务调度不存在')
|
||||||
|
return task_scheduler
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_all() -> Sequence[TaskScheduler]:
|
||||||
|
"""获取所有任务调度"""
|
||||||
|
async with async_db_session() as db:
|
||||||
|
task_schedulers = await task_scheduler_dao.get_all(db)
|
||||||
|
return task_schedulers
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def get_select(*, name: str | None, type: int | None) -> Select:
|
||||||
|
"""
|
||||||
|
获取任务调度列表查询条件
|
||||||
|
|
||||||
|
:param name: 任务调度名称
|
||||||
|
:param type: 任务调度类型
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
return await task_scheduler_dao.get_list(name=name, type=type)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def create(*, obj: CreateTaskSchedulerParam) -> None:
|
||||||
|
"""
|
||||||
|
创建任务调度
|
||||||
|
|
||||||
|
:param obj: 任务调度创建参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
task_scheduler = await task_scheduler_dao.get_by_name(db, obj.name)
|
||||||
|
if task_scheduler:
|
||||||
|
raise errors.ConflictError(msg='任务调度已存在')
|
||||||
|
if obj.type == TaskSchedulerType.CRONTAB:
|
||||||
|
crontab_verify(obj.crontab)
|
||||||
|
await task_scheduler_dao.create(db, obj)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update(*, pk: int, obj: UpdateTaskSchedulerParam) -> int:
|
||||||
|
"""
|
||||||
|
更新任务调度
|
||||||
|
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:param obj: 任务调度更新参数
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
task_scheduler = await task_scheduler_dao.get(db, pk)
|
||||||
|
if not task_scheduler:
|
||||||
|
raise errors.NotFoundError(msg='任务调度不存在')
|
||||||
|
if task_scheduler.name != obj.name:
|
||||||
|
if await task_scheduler_dao.get_by_name(db, obj.name):
|
||||||
|
raise errors.ConflictError(msg='任务调度已存在')
|
||||||
|
if task_scheduler.type == TaskSchedulerType.CRONTAB:
|
||||||
|
crontab_verify(obj.crontab)
|
||||||
|
count = await task_scheduler_dao.update(db, pk, obj)
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def update_status(*, pk: int) -> int:
|
||||||
|
"""
|
||||||
|
更新任务调度状态
|
||||||
|
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
task_scheduler = await task_scheduler_dao.get(db, pk)
|
||||||
|
if not task_scheduler:
|
||||||
|
raise errors.NotFoundError(msg='任务调度不存在')
|
||||||
|
count = await task_scheduler_dao.set_status(db, pk, not task_scheduler.enabled)
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def delete(*, pk) -> int:
|
||||||
|
"""
|
||||||
|
删除任务调度
|
||||||
|
|
||||||
|
:param pk: 用户 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session.begin() as db:
|
||||||
|
task_scheduler = await task_scheduler_dao.get(db, pk)
|
||||||
|
if not task_scheduler:
|
||||||
|
raise errors.NotFoundError(msg='任务调度不存在')
|
||||||
|
count = await task_scheduler_dao.delete(db, pk)
|
||||||
|
return count
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def execute(*, pk: int) -> None:
|
||||||
|
"""
|
||||||
|
执行任务
|
||||||
|
|
||||||
|
:param pk: 任务调度 ID
|
||||||
|
:return:
|
||||||
|
"""
|
||||||
|
async with async_db_session() as db:
|
||||||
|
workers = await run_in_threadpool(celery_app.control.ping, timeout=0.5)
|
||||||
|
if not workers:
|
||||||
|
raise errors.ServerError(msg='Celery Worker 暂不可用,请稍后重试')
|
||||||
|
task_scheduler = await task_scheduler_dao.get(db, pk)
|
||||||
|
if not task_scheduler:
|
||||||
|
raise errors.NotFoundError(msg='任务调度不存在')
|
||||||
|
try:
|
||||||
|
args = json.loads(task_scheduler.args) if task_scheduler.args else None
|
||||||
|
kwargs = json.loads(task_scheduler.kwargs) if task_scheduler.kwargs else None
|
||||||
|
except (TypeError, json.JSONDecodeError):
|
||||||
|
raise errors.RequestError(msg='执行失败,任务参数非法')
|
||||||
|
else:
|
||||||
|
celery_app.send_task(name=task_scheduler.task, args=args, kwargs=kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
task_scheduler_service: TaskSchedulerService = TaskSchedulerService()
|
||||||
@@ -1,73 +0,0 @@
|
|||||||
#!/usr/bin/env python3
|
|
||||||
# -*- coding: utf-8 -*-
|
|
||||||
from celery.exceptions import NotRegistered
|
|
||||||
from celery.result import AsyncResult
|
|
||||||
from starlette.concurrency import run_in_threadpool
|
|
||||||
|
|
||||||
from backend.app.task.celery import celery_app
|
|
||||||
from backend.app.task.schema.task import RunParam, TaskResult
|
|
||||||
from backend.common.exception import errors
|
|
||||||
from backend.common.exception.errors import NotFoundError
|
|
||||||
|
|
||||||
|
|
||||||
class TaskService:
|
|
||||||
@staticmethod
|
|
||||||
async def get_list() -> list[str]:
|
|
||||||
"""获取所有已注册的 Celery 任务列表"""
|
|
||||||
registered_tasks = await run_in_threadpool(celery_app.control.inspect().registered)
|
|
||||||
if not registered_tasks:
|
|
||||||
raise errors.ForbiddenError(msg='Celery 服务未启动')
|
|
||||||
tasks = list(registered_tasks.values())[0]
|
|
||||||
return tasks
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def get_detail(*, tid: str) -> TaskResult:
|
|
||||||
"""
|
|
||||||
获取指定任务的详细信息
|
|
||||||
|
|
||||||
:param tid: 任务 UUID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
result = AsyncResult(id=tid, app=celery_app)
|
|
||||||
except NotRegistered:
|
|
||||||
raise NotFoundError(msg='任务不存在')
|
|
||||||
return TaskResult(
|
|
||||||
result=result.result,
|
|
||||||
traceback=result.traceback,
|
|
||||||
status=result.state,
|
|
||||||
name=result.name,
|
|
||||||
args=result.args,
|
|
||||||
kwargs=result.kwargs,
|
|
||||||
worker=result.worker,
|
|
||||||
retries=result.retries,
|
|
||||||
queue=result.queue,
|
|
||||||
)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def revoke(*, tid: str) -> None:
|
|
||||||
"""
|
|
||||||
撤销指定的任务
|
|
||||||
|
|
||||||
:param tid: 任务 UUID
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
result = AsyncResult(id=tid, app=celery_app)
|
|
||||||
except NotRegistered:
|
|
||||||
raise NotFoundError(msg='任务不存在')
|
|
||||||
result.revoke(terminate=True)
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def run(*, obj: RunParam) -> str:
|
|
||||||
"""
|
|
||||||
运行指定的任务
|
|
||||||
|
|
||||||
:param obj: 任务运行参数
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
task: AsyncResult = celery_app.send_task(name=obj.name, args=obj.args, kwargs=obj.kwargs)
|
|
||||||
return task.task_id
|
|
||||||
|
|
||||||
|
|
||||||
task_service: TaskService = TaskService()
|
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
from celery.backends.database.session import SessionManager as CelerySessionManager
|
||||||
|
|
||||||
|
|
||||||
|
class SessionManager(CelerySessionManager):
|
||||||
|
"""
|
||||||
|
重写 celery SessionManager
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
super().__init__()
|
||||||
|
|
||||||
|
# 禁止自动创建 celery 内部定义的任务结果表
|
||||||
|
self.prepared = True
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user