mikl/telegram-mcp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: clean up whitespace and improve code formatting using black .

Browse source
This commit is contained in:
Tim 2025-08-06 21:07:59 +03:00
parent 2dc8454a4f
commit 78a99b06f2
1 changed files with 28 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
main.py
View file

@ -190,14 +190,14 @@ def get_sender_name(message) -> str:
"""Helper function to get sender name from a message."""
if not message.sender:
return "Unknown"
# Check for group/channel title first
if hasattr(message.sender, 'title') and message.sender.title:
if hasattr(message.sender, "title") and message.sender.title:
return message.sender.title
elif hasattr(message.sender, 'first_name'):
elif hasattr(message.sender, "first_name"):
# User sender
first_name = getattr(message.sender, 'first_name', '') or ''
last_name = getattr(message.sender, 'last_name', '') or ''
first_name = getattr(message.sender, "first_name", "") or ""
last_name = getattr(message.sender, "last_name", "") or ""
full_name = f"{first_name} {last_name}".strip()
return full_name if full_name else "Unknown"
else:
@ -251,7 +251,9 @@ async def get_messages(chat_id: int, page: int = 1, page_size: int = 20) -> str:
reply_info = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
reply_info = f" | reply to {msg.reply_to.reply_to_msg_id}"
lines.append(f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}")
lines.append(
f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}"
)
return "\n".join(lines)
except Exception as e:
return log_and_format_error(
@ -423,11 +425,11 @@ async def list_messages(
lines = []
for msg in messages:
sender_name = get_sender_name(msg)
message_text = msg.message or '[Media/No text]'
message_text = msg.message or "[Media/No text]"
reply_info = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
reply_info = f" | reply to {msg.reply_to.reply_to_msg_id}"
lines.append(
f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {message_text}"
)
@ -753,7 +755,7 @@ async def get_message_context(chat_id: int, message_id: int, context_size: int =
for msg in all_messages:
sender_name = get_sender_name(msg)
highlight = " [THIS MESSAGE]" if msg.id == message_id else ""
# Check if this message is a reply and get the replied message
reply_content = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
@ -762,13 +764,15 @@ async def get_message_context(chat_id: int, message_id: int, context_size: int =
if replied_msg:
replied_sender = "Unknown"
if replied_msg.sender:
replied_sender = getattr(replied_msg.sender, "first_name", "") or getattr(
replied_msg.sender, "title", "Unknown"
)
replied_sender = getattr(
replied_msg.sender, "first_name", ""
) or getattr(replied_msg.sender, "title", "Unknown")
reply_content = f" | reply to {msg.reply_to.reply_to_msg_id}\n → Replied message: [{replied_sender}] {replied_msg.message or '[Media/No text]'}"
except Exception:
reply_content = f" | reply to {msg.reply_to.reply_to_msg_id} (original message not found)"
reply_content = (
f" | reply to {msg.reply_to.reply_to_msg_id} (original message not found)"
)
results.append(
f"ID: {msg.id} | {sender_name} | {msg.date}{highlight}{reply_content}\n{msg.message or '[Media/No text]'}\n"
)
@ -2067,7 +2071,9 @@ async def search_messages(chat_id: int, query: str, limit: int = 20) -> str:
reply_info = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
reply_info = f" | reply to {msg.reply_to.reply_to_msg_id}"
lines.append(f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}")
lines.append(
f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}"
)
return "\n".join(lines)
except Exception as e:
return log_and_format_error(
@ -2400,7 +2406,9 @@ async def get_history(chat_id: int, limit: int = 100) -> str:
reply_info = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
reply_info = f" | reply to {msg.reply_to.reply_to_msg_id}"
lines.append(f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}")
lines.append(
f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message}"
)
return "\n".join(lines)
except Exception as e:
return log_and_format_error("get_history", e, chat_id=chat_id, limit=limit)
@ -2482,8 +2490,10 @@ async def get_pinned_messages(chat_id: int) -> str:
reply_info = ""
if msg.reply_to and msg.reply_to.reply_to_msg_id:
reply_info = f" | reply to {msg.reply_to.reply_to_msg_id}"
lines.append(f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message or '[Media/No text]'}")
lines.append(
f"ID: {msg.id} | {sender_name} | Date: {msg.date}{reply_info} | Message: {msg.message or '[Media/No text]'}"
)
return "\n".join(lines)
except Exception as e:
logger.exception(f"get_pinned_messages failed (chat_id={chat_id})")