first commit

This commit is contained in:
Guillermo 2024-09-13 04:10:03 +02:00
commit 5ee04c855d
4 changed files with 141 additions and 0 deletions

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
__pycache__

78
bot.py Normal file
View file

@ -0,0 +1,78 @@
#!/usr/bin/env python3
from telegram import Update
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters
class Butler(object):
def __init__(self, logger, token, whitelist):
self.logger = logger
self.token = token
self.whitelist = whitelist
# Create the Application and pass it your bot's token.
application = Application.builder().token(self.token).build()
# on different commands - answer in Telegram
application.add_handler(CommandHandler("start", Butler.start))
application.add_handler(CommandHandler("help", Butler.help_command))
application.add_handler(CommandHandler("stop", self.stop))
application.add_handler(CommandHandler("echo", Butler.echo))
application.add_handler(CommandHandler("who", Butler.who))
# on non command i.e message - echo the message on Telegram
# application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, self.echo))
# bot_user = asyncio.run(application.bot.get_me())
self.username = 'Vylbot' # bot_user.username
self.application = application
def get_username(self):
return self.username
def help_msg():
return """Respondo a los siguientes comandos:
/echo - Repite el texto.
/who - Información sobre ti y tu mensaje (para debugging).
"""
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
"""Send a message when the command /start is issued."""
user = update.effective_user
await update.message.reply_html(rf"Hi {user.mention_html()}!")
async def help_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
"""Send a message when the command /help is issued."""
await update.message.reply_text(Butler.help_msg())
async def echo(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
"""Echo the user message."""
echo_text = update.message.text.split(maxsplit=1)
await update.message.reply_text(echo_text[1])
async def stop(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
user = update.effective_user
self.logger.warning("I got blocked by user {} [{}]".format(user.username, user.id))
async def who(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
mssg = update.message
user = update.effective_user
chat = update.effective_chat
name = chat.effective_name
date = mssg.date.astimezone()
answer = ("You're **{name}**, with username `{username}`, and"
" id `{uid}`.\nYou're messaging in the {ctype} chat named"
" __{cname}__, with id `{cid}`, and timestamp `{tstamp}`."
).format(name=user.full_name, username=user.username,
uid=user.id, cname=name, cid=chat.id,
ctype=chat.type, tstamp=str(date))
await mssg.reply_markdown(answer)
def run(self) -> None:
"""Start the bot."""
# Run the bot until the user presses Ctrl-C
self.application.run_polling(allowed_updates=Update.ALL_TYPES)

42
main.py Normal file
View file

@ -0,0 +1,42 @@
#!/usr/bin/env python3
import logging
import argparse
from bot import Butler
coloredlogsError = None
try:
import coloredlogs
except ImportError as e:
coloredlogsError = e
def main():
parser = argparse.ArgumentParser(description='Vylion\'s butler bot.')
parser.add_argument('token', metavar='TOKEN',
help='The Bot Token to work with the Telegram Bot API')
parser.add_argument('-c', '--chats', nargs='*', default=[], metavar='[CHATS]',
help='A whitelist of allowed chats. Empty enables all.')
args = parser.parse_args()
logger = logging.getLogger(__name__)
bot = Butler(logger, args.token, args.chats)
# Enable logging
log_format = "[{}][%(asctime)s]%(name)s::%(levelname)s: %(message)s".format(bot.get_username().upper())
if coloredlogsError:
logging.basicConfig(format=log_format, level=logging.INFO)
logger.warning("Unable to load coloredlogs:")
logger.warning(coloredlogsError)
else:
coloredlogs.install(level=logging.INFO, fmt=log_format)
logger.info("Bot started up!")
bot.run()
if __name__ == '__main__':
main()

20
timezone.py Normal file
View file

@ -0,0 +1,20 @@
#!/usr/bin/env python3
from datetime import datetime, tzinfo, timedelta
class TZMadrid(tzinfo):
def utcoffset(self, dt):
return timedelta(hours=1) + self.dst(dt)
def dst(self, dt):
dston = datetime.date(dt.year, 4, 1)
dstoff = datetime.date(dt.year, 10, 27)
# Code to set dston and dstoff to the time zone's DST
# transition times based on the input dt.year, and expressed
# in standard local time.
if dston <= dt.replace(tzinfo=None) < dstoff:
return timedelta(hours=1)
else:
return timedelta(0)