swap to internal Ulid objects

revoltchat · Oct 28, 2024 · 8feaadd · 8feaadd
1 parent c6a8162
commit 8feaadd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/revolt/ext/commands/converters.py b/revolt/ext/commands/converters.py
@@ -117,15 +117,17 @@ def member_converter(arg: str, context: Context[ClientT]) -> Member:
 def int_converter(arg: str, context: Context[ClientT]) -> int:
     return int(arg)
 
-def id_converter(arg: str, context: Context[ClientT]) -> ulid.ULID:
+def id_converter(arg: str, context: Context[ClientT]) -> utils.Ulid:
     if len(arg) != 26:
         raise ValueError("An ID was not provided.")
 
     try:
-        return ulid.parse(arg)
+        ulid.parse(arg) # validate
     except Exception as err:
         raise ValueError("An invalid ID was provided.") from err
 
+    return utils.Object(arg)
+
 IdConverter = Annotated[ulid.ULID, id_converter]
 IntConverter = Annotated[int, int_converter]
 BoolConverter = Annotated[bool, bool_converter]