фиксит код по правилам из flake8-type-checking

This commit is contained in:
2024-08-22 12:00:45 +03:00
parent cc7c97e962
commit 9ca5264c6d
12 changed files with 57 additions and 38 deletions

View File

@@ -1,10 +1,14 @@
import importlib
import os
import traceback
from typing import TYPE_CHECKING
from aiogram import Bot, Dispatcher
from aiogram.types import Update
if TYPE_CHECKING:
from aiogram import Bot, Dispatcher
from fastapi import FastAPI, Request
def get_module_directory(module_name):
spec = importlib.util.find_spec(module_name)
@@ -16,23 +20,15 @@ def get_module_directory(module_name):
return os.path.dirname(module_path)
try:
from fastapi import FastAPI, Request
async def register_bot_webhook(app: FastAPI, bot: Bot, dp: Dispatcher):
async def handle_webhook(request: Request):
try:
update = Update.model_validate(await request.json(), context={"bot": bot})
await dp.feed_update(bot, update)
except Exception:
traceback.print_exc()
return {"ok": False}
async def register_bot_webhook(app: FastAPI, bot: Bot, dp: Dispatcher):
async def handle_webhook(request: Request):
try:
update = Update.model_validate(
await request.json(), context={"bot": bot}
)
await dp.feed_update(bot, update)
except Exception:
traceback.print_exc()
return {"ok": False}
return {"ok": True}
return {"ok": True}
app.post("/webhook")(handle_webhook)
except ImportError:
pass
app.post("/webhook")(handle_webhook)

View File

@@ -1,9 +1,11 @@
import types
from dataclasses import dataclass
from typing import Dict, List, Optional, Union
from typing import TYPE_CHECKING, Dict, List, Optional, Union
from dataclasses_json import dataclass_json
if TYPE_CHECKING:
import types
@dataclass_json
@dataclass

View File

@@ -1,8 +1,7 @@
import inspect
import types
from typing import Any, Tuple, Union
from typing import TYPE_CHECKING, Any, Tuple, Union
from aiogram import BaseMiddleware, Router
from aiogram.fsm.context import FSMContext
from aiogram.fsm.storage.base import StorageKey
@@ -10,6 +9,9 @@ from aiogram.fsm.storage.base import StorageKey
from karkas_core.modules_system.loaders.base import DependencyInfo
from karkas_core.singleton import Singleton
if TYPE_CHECKING:
from aiogram import BaseMiddleware, Router
async def set_chat_menu_button(menu_button):
app = Singleton()

View File

@@ -1,6 +1,5 @@
import types
from _ast import AnnAssign
from typing import Any
from typing import TYPE_CHECKING, Any
from aiogram import Bot
from RestrictedPython import (
@@ -20,6 +19,9 @@ from RestrictedPython.Guards import ( # guarded_setattr,; full_write_guard,
from karkas_core.logger import log
from karkas_core.modules_system.safe.zope_guards import extra_safe_builtins
if TYPE_CHECKING:
from _ast import AnnAssign
class RestrictedPythonPolicy(RestrictingNodeTransformer):
def visit_AsyncFunctionDef(self, node):

View File

@@ -1,7 +1,11 @@
from aiogram import Bot, Dispatcher
from typing import TYPE_CHECKING
from aiogram.fsm.storage.memory import MemoryStorage
from karkas_core.modules_system import ModulesManager
if TYPE_CHECKING:
from aiogram import Bot, Dispatcher
from karkas_core.modules_system import ModulesManager
class SingletonMeta(type):