✨ 0.7.0

ArcletProject · Mar 1, 2022 · 41bcd80 · 41bcd80
1 parent 197f931
commit 41bcd80
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/arclet/alconna/manager.py b/arclet/alconna/manager.py
@@ -61,14 +61,19 @@ def register(self, command: "Alconna") -> None:
         else:
             raise DuplicateCommand("命令已存在")
 
-    def require(self, command: str) -> "Analyser":
+    def require(self, command: Union["Alconna", str]) -> "Analyser":
         """获取解析器"""
-        namespace, name = self._command_part(command)
-        try:
-            ana = self.__commands[namespace][name]
-        except KeyError:
-            raise ValueError("命令不存在")
-        return ana
+        if isinstance(command, str):
+            namespace, name = self._command_part(command)
+            try:
+                ana = self.__commands[namespace][name]
+            except KeyError:
+                raise ValueError("命令不存在")
+            return ana
+        else:
+            cid = command.name.replace(self.sign, "")
+            namespace = command.namespace
+            return self.__commands[namespace][cid]
 
     def delete(self, command: Union["Alconna", str]) -> None:
         """删除命令"""