fix lint - type mismatch (#2117)

* fix lint

* empty commit to trigger ci flow
This commit is contained in:
ming 2024-06-09 09:18:27 -04:00 committed by GitHub
commit 7ebaf2914c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 5 deletions

View file

@ -16,7 +16,7 @@ from collections import OrderedDict, namedtuple
from http.cookies import SimpleCookie
ParsedArgs = namedtuple(
"ParsedContext",
"ParsedArgs",
[
"command",
"url",

View file

@ -1,6 +1,6 @@
from datetime import datetime
from pathlib import Path
from typing import TYPE_CHECKING, Optional, Union
from typing import TYPE_CHECKING, Optional, Union, List
import duckdb
from loguru import logger
@ -109,12 +109,21 @@ class MonitorService(Service):
return self.exec_query(query)
def delete_messages(self, message_ids: list[int]):
query = f"DELETE FROM messages WHERE index IN ({','.join(map(str, message_ids))})"
def delete_messages(self, message_ids: Union[List[int], str]):
if isinstance(message_ids, list):
# If message_ids is a list, join the string representations of the integers
ids_str = ','.join(map(str, message_ids))
elif isinstance(message_ids, str):
# If message_ids is already a string, use it directly
ids_str = message_ids
else:
raise ValueError("message_ids must be a list of integers or a string")
query = f"DELETE FROM messages WHERE index IN ({ids_str})"
return self.exec_query(query)
def update_message(self, message_id: int, **kwargs):
def update_message(self, message_id: str, **kwargs):
query = (
f"""UPDATE messages SET {', '.join(f"{k} = '{v}'" for k, v in kwargs.items())} WHERE index = {message_id}"""
)