mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-25 18:04:10 +00:00
🔥 Correct Return Type
This commit is contained in:
parent
b129a791df
commit
20c1910a98
@ -1,4 +1,4 @@
|
|||||||
from typing import List, NoReturn, Optional
|
from typing import List, Optional
|
||||||
|
|
||||||
from core.base_service import BaseService
|
from core.base_service import BaseService
|
||||||
from core.services.wiki.cache import WikiCache
|
from core.services.wiki.cache import WikiCache
|
||||||
@ -19,7 +19,7 @@ class WikiService(BaseService):
|
|||||||
self._weapon_list = []
|
self._weapon_list = []
|
||||||
self.first_run = True
|
self.first_run = True
|
||||||
|
|
||||||
async def refresh_weapon(self) -> NoReturn:
|
async def refresh_weapon(self):
|
||||||
weapon_name_list = await Weapon.get_name_list()
|
weapon_name_list = await Weapon.get_name_list()
|
||||||
logger.info("一共找到 %s 把武器信息", len(weapon_name_list))
|
logger.info("一共找到 %s 把武器信息", len(weapon_name_list))
|
||||||
|
|
||||||
@ -36,7 +36,7 @@ class WikiService(BaseService):
|
|||||||
await self._cache.delete("weapon")
|
await self._cache.delete("weapon")
|
||||||
await self._cache.set("weapon", [i.json() for i in weapon_list])
|
await self._cache.set("weapon", [i.json() for i in weapon_list])
|
||||||
|
|
||||||
async def refresh_characters(self) -> NoReturn:
|
async def refresh_characters(self):
|
||||||
character_name_list = await Character.get_name_list()
|
character_name_list = await Character.get_name_list()
|
||||||
logger.info("一共找到 %s 个角色信息", len(character_name_list))
|
logger.info("一共找到 %s 个角色信息", len(character_name_list))
|
||||||
|
|
||||||
@ -53,7 +53,7 @@ class WikiService(BaseService):
|
|||||||
await self._cache.delete("characters")
|
await self._cache.delete("characters")
|
||||||
await self._cache.set("characters", [i.json() for i in character_list])
|
await self._cache.set("characters", [i.json() for i in character_list])
|
||||||
|
|
||||||
async def refresh_wiki(self) -> NoReturn:
|
async def refresh_wiki(self):
|
||||||
"""
|
"""
|
||||||
用于把Redis的缓存全部加载进Python
|
用于把Redis的缓存全部加载进Python
|
||||||
:return:
|
:return:
|
||||||
@ -65,7 +65,7 @@ class WikiService(BaseService):
|
|||||||
await self.refresh_characters()
|
await self.refresh_characters()
|
||||||
logger.info("刷新成功")
|
logger.info("刷新成功")
|
||||||
|
|
||||||
async def init(self) -> NoReturn:
|
async def init(self):
|
||||||
"""
|
"""
|
||||||
用于把Redis的缓存全部加载进Python
|
用于把Redis的缓存全部加载进Python
|
||||||
:return:
|
:return:
|
||||||
|
@ -71,7 +71,7 @@ class Queue(Generic[T]):
|
|||||||
self.closing = False
|
self.closing = False
|
||||||
self.pending = set() # type: Set[asyncio.Future[Any]]
|
self.pending = set() # type: Set[asyncio.Future[Any]]
|
||||||
|
|
||||||
def checked_call_soon_threadsafe(callback: Callable[..., None], *args: Any) -> NoReturn:
|
def checked_call_soon_threadsafe(callback: Callable[..., None], *args: Any):
|
||||||
try:
|
try:
|
||||||
self.loop.call_soon_threadsafe(callback, *args)
|
self.loop.call_soon_threadsafe(callback, *args)
|
||||||
except RuntimeError:
|
except RuntimeError:
|
||||||
@ -79,7 +79,7 @@ class Queue(Generic[T]):
|
|||||||
|
|
||||||
self._call_soon_threadsafe = checked_call_soon_threadsafe
|
self._call_soon_threadsafe = checked_call_soon_threadsafe
|
||||||
|
|
||||||
def checked_call_soon(callback: Callable[..., None], *args: Any) -> NoReturn:
|
def checked_call_soon(callback: Callable[..., None], *args: Any):
|
||||||
if not self.loop.is_closed():
|
if not self.loop.is_closed():
|
||||||
self.loop.call_soon(callback, *args)
|
self.loop.call_soon(callback, *args)
|
||||||
|
|
||||||
@ -88,7 +88,7 @@ class Queue(Generic[T]):
|
|||||||
self._sync_queue = _SyncQueueProxy(self)
|
self._sync_queue = _SyncQueueProxy(self)
|
||||||
self._async_queue = _AsyncQueueProxy(self)
|
self._async_queue = _AsyncQueueProxy(self)
|
||||||
|
|
||||||
def close(self) -> NoReturn:
|
def close(self):
|
||||||
"""关闭队列"""
|
"""关闭队列"""
|
||||||
with self.sync_mutex:
|
with self.sync_mutex:
|
||||||
self.closing = True
|
self.closing = True
|
||||||
@ -97,7 +97,7 @@ class Queue(Generic[T]):
|
|||||||
self.finished.set() # 取消堵塞全部的 async_q.join()
|
self.finished.set() # 取消堵塞全部的 async_q.join()
|
||||||
self.all_tasks_done.notify_all() # 取消堵塞全部的 sync_q.join()
|
self.all_tasks_done.notify_all() # 取消堵塞全部的 sync_q.join()
|
||||||
|
|
||||||
async def wait_closed(self) -> NoReturn:
|
async def wait_closed(self):
|
||||||
if not self.closing:
|
if not self.closing:
|
||||||
raise RuntimeError("队列已被关闭")
|
raise RuntimeError("队列已被关闭")
|
||||||
await asyncio.sleep(0)
|
await asyncio.sleep(0)
|
||||||
@ -121,32 +121,32 @@ class Queue(Generic[T]):
|
|||||||
def async_q(self) -> "_AsyncQueueProxy[T]":
|
def async_q(self) -> "_AsyncQueueProxy[T]":
|
||||||
return self._async_queue
|
return self._async_queue
|
||||||
|
|
||||||
def _init(self) -> NoReturn:
|
def _init(self):
|
||||||
self._queue = deque() # type: Deque[T]
|
self._queue = deque() # type: Deque[T]
|
||||||
|
|
||||||
def qsize(self) -> int:
|
def qsize(self) -> int:
|
||||||
return len(self._queue)
|
return len(self._queue)
|
||||||
|
|
||||||
def put(self, item: T) -> NoReturn:
|
def put(self, item: T):
|
||||||
self._queue.append(item)
|
self._queue.append(item)
|
||||||
|
|
||||||
def get(self) -> T:
|
def get(self) -> T:
|
||||||
return self._queue.popleft()
|
return self._queue.popleft()
|
||||||
|
|
||||||
def put_internal(self, item: T) -> NoReturn:
|
def put_internal(self, item: T):
|
||||||
self.put(item)
|
self.put(item)
|
||||||
self.unfinished_tasks += 1
|
self.unfinished_tasks += 1
|
||||||
self.finished.clear()
|
self.finished.clear()
|
||||||
|
|
||||||
def notify_sync_not_empty(self) -> NoReturn:
|
def notify_sync_not_empty(self):
|
||||||
def f() -> NoReturn:
|
def f():
|
||||||
with self.sync_mutex:
|
with self.sync_mutex:
|
||||||
self.sync_not_empty.notify()
|
self.sync_not_empty.notify()
|
||||||
|
|
||||||
self.loop.run_in_executor(None, f)
|
self.loop.run_in_executor(None, f)
|
||||||
|
|
||||||
def notify_sync_not_full(self) -> NoReturn:
|
def notify_sync_not_full(self):
|
||||||
def f() -> NoReturn:
|
def f():
|
||||||
with self.sync_mutex:
|
with self.sync_mutex:
|
||||||
self.sync_not_full.notify()
|
self.sync_not_full.notify()
|
||||||
|
|
||||||
@ -154,12 +154,12 @@ class Queue(Generic[T]):
|
|||||||
fut.add_done_callback(self.pending.discard)
|
fut.add_done_callback(self.pending.discard)
|
||||||
self.pending.add(fut)
|
self.pending.add(fut)
|
||||||
|
|
||||||
def notify_async_not_empty(self, *, threadsafe: bool) -> NoReturn:
|
def notify_async_not_empty(self, *, threadsafe: bool):
|
||||||
async def f() -> NoReturn:
|
async def f():
|
||||||
async with self.async_mutex:
|
async with self.async_mutex:
|
||||||
self.async_not_empty.notify()
|
self.async_not_empty.notify()
|
||||||
|
|
||||||
def task_maker() -> NoReturn:
|
def task_maker():
|
||||||
task = self.loop.create_task(f())
|
task = self.loop.create_task(f())
|
||||||
task.add_done_callback(self.pending.discard)
|
task.add_done_callback(self.pending.discard)
|
||||||
self.pending.add(task)
|
self.pending.add(task)
|
||||||
@ -169,12 +169,12 @@ class Queue(Generic[T]):
|
|||||||
else:
|
else:
|
||||||
self._call_soon(task_maker)
|
self._call_soon(task_maker)
|
||||||
|
|
||||||
def notify_async_not_full(self, *, threadsafe: bool) -> NoReturn:
|
def notify_async_not_full(self, *, threadsafe: bool):
|
||||||
async def f() -> NoReturn:
|
async def f():
|
||||||
async with self.async_mutex:
|
async with self.async_mutex:
|
||||||
self.async_not_full.notify()
|
self.async_not_full.notify()
|
||||||
|
|
||||||
def task_maker() -> NoReturn:
|
def task_maker():
|
||||||
task = self.loop.create_task(f())
|
task = self.loop.create_task(f())
|
||||||
task.add_done_callback(self.pending.discard)
|
task.add_done_callback(self.pending.discard)
|
||||||
self.pending.add(task)
|
self.pending.add(task)
|
||||||
@ -184,7 +184,7 @@ class Queue(Generic[T]):
|
|||||||
else:
|
else:
|
||||||
self._call_soon(task_maker)
|
self._call_soon(task_maker)
|
||||||
|
|
||||||
def check_closing(self) -> NoReturn:
|
def check_closing(self):
|
||||||
if self.closing:
|
if self.closing:
|
||||||
raise RuntimeError("禁止对已关闭的队列进行操作")
|
raise RuntimeError("禁止对已关闭的队列进行操作")
|
||||||
|
|
||||||
@ -204,7 +204,7 @@ class _SyncQueueProxy(SyncQueue[T]): # pylint: disable=W0212
|
|||||||
def closed(self) -> bool:
|
def closed(self) -> bool:
|
||||||
return self._parent.closed
|
return self._parent.closed
|
||||||
|
|
||||||
def task_done(self) -> NoReturn:
|
def task_done(self):
|
||||||
self._parent.check_closing()
|
self._parent.check_closing()
|
||||||
with self._parent.all_tasks_done:
|
with self._parent.all_tasks_done:
|
||||||
unfinished = self._parent.unfinished_tasks - 1
|
unfinished = self._parent.unfinished_tasks - 1
|
||||||
@ -215,7 +215,7 @@ class _SyncQueueProxy(SyncQueue[T]): # pylint: disable=W0212
|
|||||||
self._parent.loop.call_soon_threadsafe(self._parent.finished.set)
|
self._parent.loop.call_soon_threadsafe(self._parent.finished.set)
|
||||||
self._parent.unfinished_tasks = unfinished
|
self._parent.unfinished_tasks = unfinished
|
||||||
|
|
||||||
def join(self) -> NoReturn:
|
def join(self):
|
||||||
self._parent.check_closing()
|
self._parent.check_closing()
|
||||||
with self._parent.all_tasks_done:
|
with self._parent.all_tasks_done:
|
||||||
while self._parent.unfinished_tasks:
|
while self._parent.unfinished_tasks:
|
||||||
@ -237,7 +237,7 @@ class _SyncQueueProxy(SyncQueue[T]): # pylint: disable=W0212
|
|||||||
def full(self) -> bool:
|
def full(self) -> bool:
|
||||||
return 0 < self._parent.maxsize <= self._parent.qsize()
|
return 0 < self._parent.maxsize <= self._parent.qsize()
|
||||||
|
|
||||||
def put(self, item: T, block: bool = True, timeout: OptFloat = None) -> NoReturn:
|
def put(self, item: T, block: bool = True, timeout: OptFloat = None):
|
||||||
self._parent.check_closing()
|
self._parent.check_closing()
|
||||||
with self._parent.sync_not_full:
|
with self._parent.sync_not_full:
|
||||||
if self._parent.maxsize > 0:
|
if self._parent.maxsize > 0:
|
||||||
@ -285,7 +285,7 @@ class _SyncQueueProxy(SyncQueue[T]): # pylint: disable=W0212
|
|||||||
self._parent.notify_async_not_full(threadsafe=True)
|
self._parent.notify_async_not_full(threadsafe=True)
|
||||||
return item
|
return item
|
||||||
|
|
||||||
def put_nowait(self, item: T) -> NoReturn:
|
def put_nowait(self, item: T):
|
||||||
return self.put(item, block=False)
|
return self.put(item, block=False)
|
||||||
|
|
||||||
def get_nowait(self) -> T:
|
def get_nowait(self) -> T:
|
||||||
@ -346,7 +346,7 @@ class _AsyncQueueProxy(AsyncQueue[T]): # pylint: disable=W0212
|
|||||||
if locked:
|
if locked:
|
||||||
self._parent.sync_mutex.release()
|
self._parent.sync_mutex.release()
|
||||||
|
|
||||||
def put_nowait(self, item: T) -> NoReturn:
|
def put_nowait(self, item: T):
|
||||||
self._parent.check_closing()
|
self._parent.check_closing()
|
||||||
with self._parent.sync_mutex and 0 < self._parent.maxsize <= self._parent.qsize():
|
with self._parent.sync_mutex and 0 < self._parent.maxsize <= self._parent.qsize():
|
||||||
raise AsyncQueueFull
|
raise AsyncQueueFull
|
||||||
@ -391,7 +391,7 @@ class _AsyncQueueProxy(AsyncQueue[T]): # pylint: disable=W0212
|
|||||||
self._parent.notify_sync_not_full()
|
self._parent.notify_sync_not_full()
|
||||||
return item
|
return item
|
||||||
|
|
||||||
def task_done(self) -> NoReturn:
|
def task_done(self):
|
||||||
self._parent.check_closing()
|
self._parent.check_closing()
|
||||||
with self._parent.all_tasks_done:
|
with self._parent.all_tasks_done:
|
||||||
if self._parent.unfinished_tasks <= 0:
|
if self._parent.unfinished_tasks <= 0:
|
||||||
@ -413,13 +413,13 @@ class _AsyncQueueProxy(AsyncQueue[T]): # pylint: disable=W0212
|
|||||||
class PriorityQueue(Queue[T]):
|
class PriorityQueue(Queue[T]):
|
||||||
"""优先级队列"""
|
"""优先级队列"""
|
||||||
|
|
||||||
def _init(self) -> NoReturn:
|
def _init(self):
|
||||||
self._heap_queue: List[T] = []
|
self._heap_queue: List[T] = []
|
||||||
|
|
||||||
def qsize(self) -> int:
|
def qsize(self) -> int:
|
||||||
return len(self._heap_queue)
|
return len(self._heap_queue)
|
||||||
|
|
||||||
def put(self, item: T) -> NoReturn:
|
def put(self, item: T):
|
||||||
if not isinstance(item, tuple):
|
if not isinstance(item, tuple):
|
||||||
if hasattr(item, "priority"):
|
if hasattr(item, "priority"):
|
||||||
item = (int(item.priority), item)
|
item = (int(item.priority), item)
|
||||||
@ -440,7 +440,7 @@ class LifoQueue(Queue[T]):
|
|||||||
def qsize(self) -> int:
|
def qsize(self) -> int:
|
||||||
return len(self._queue)
|
return len(self._queue)
|
||||||
|
|
||||||
def put(self, item: T) -> NoReturn:
|
def put(self, item: T):
|
||||||
self._queue.append(item)
|
self._queue.append(item)
|
||||||
|
|
||||||
def get(self) -> T:
|
def get(self) -> T:
|
||||||
|
@ -16,7 +16,7 @@ class BaseQueue(Protocol[T]): # pylint: disable=W0049
|
|||||||
def closed(self) -> bool:
|
def closed(self) -> bool:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def task_done(self) -> NoReturn:
|
def task_done(self):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
def qsize(self) -> int:
|
def qsize(self) -> int:
|
||||||
@ -49,7 +49,7 @@ class SyncQueue(BaseQueue[T], Protocol[T]): # pylint: disable=W0049
|
|||||||
def closed(self) -> bool:
|
def closed(self) -> bool:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def task_done(self) -> NoReturn:
|
def task_done(self):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
def qsize(self) -> int:
|
def qsize(self) -> int:
|
||||||
|
Loading…
Reference in New Issue
Block a user