diff options
| author | Kyomotoi <[email protected]> | 2023-03-07 15:35:56 +0800 |
|---|---|---|
| committer | Kyomotoi <[email protected]> | 2023-03-07 15:35:56 +0800 |
| commit | 95e7b6fe5f41a8b91a7ef6ae55ed976177d64ae9 (patch) | |
| tree | 8758ebc77818e140a23f5a7c52e8357a0ceaf98b /ATRI/plugins/setu/nsfw_checker.py | |
| parent | 917f903ef21cd528af7a52d8dbe65bac435eaa51 (diff) | |
| download | ATRI-95e7b6fe5f41a8b91a7ef6ae55ed976177d64ae9.tar.gz ATRI-95e7b6fe5f41a8b91a7ef6ae55ed976177d64ae9.tar.bz2 ATRI-95e7b6fe5f41a8b91a7ef6ae55ed976177d64ae9.zip | |
✨ 增加对涩图动图的检测开关
Diffstat (limited to 'ATRI/plugins/setu/nsfw_checker.py')
| -rw-r--r-- | ATRI/plugins/setu/nsfw_checker.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/ATRI/plugins/setu/nsfw_checker.py b/ATRI/plugins/setu/nsfw_checker.py index 53546f6..d7ad6e1 100644 --- a/ATRI/plugins/setu/nsfw_checker.py +++ b/ATRI/plugins/setu/nsfw_checker.py @@ -40,9 +40,13 @@ def prepare_image(img): return image -async def detect_image(url: str, max_size: int) -> float: +async def detect_image(url: str, max_size: int, disab_gif: bool) -> float: try: req = await request.get(url) + if itype := req.headers.get("Content-Type"): + if disab_gif and itype == "image/gif": + return 0 + except Exception: raise RequestError("Get info from download image failed!") |