diff options
| author | Kyomotoi <[email protected]> | 2022-12-02 08:52:19 +0800 |
|---|---|---|
| committer | Kyomotoi <[email protected]> | 2022-12-02 08:52:19 +0800 |
| commit | 1ee2eae499b79bf0d22212184bffd57d873826cc (patch) | |
| tree | 918453ecf521bf7266bc1eb0cdd703470d603f31 /ATRI/plugins/setu/nsfw_checker.py | |
| parent | 6c43fc43b7c408bce13775f1a1024809c66c904e (diff) | |
| download | ATRI-1ee2eae499b79bf0d22212184bffd57d873826cc.tar.gz ATRI-1ee2eae499b79bf0d22212184bffd57d873826cc.tar.bz2 ATRI-1ee2eae499b79bf0d22212184bffd57d873826cc.zip | |
🎨 重构涩图组件
Diffstat (limited to 'ATRI/plugins/setu/nsfw_checker.py')
| -rw-r--r-- | ATRI/plugins/setu/nsfw_checker.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ATRI/plugins/setu/nsfw_checker.py b/ATRI/plugins/setu/nsfw_checker.py index c0bd2ba..53546f6 100644 --- a/ATRI/plugins/setu/nsfw_checker.py +++ b/ATRI/plugins/setu/nsfw_checker.py @@ -40,14 +40,14 @@ def prepare_image(img): return image -async def detect_image(url: str, file_size: int) -> float: +async def detect_image(url: str, max_size: int) -> float: try: req = await request.get(url) except Exception: raise RequestError("Get info from download image failed!") img_byte = getsizeof(req.read()) // 1024 - if img_byte < file_size: + if img_byte < max_size: return 0 try: |