def cam0ShotCmd(bot, update, args):
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
logger = getLogger("cam0ShotCmd")
p = Path("./photo_cam0.jpg")
logger.info(f"{user}: cam0 shot")
if p.exists():
if time.time() - p.stat().st_ctime < 2:
logger.info("sending cached file")
bot.send_chat_action(chat_id=update.effective_message.chat_id,
action=ChatAction.UPLOAD_PHOTO)
bot.send_photo(chat_id=chat_id, photo=p.open('rb'), timeout=600)
return
if camera_0.take_shot():
logger.info(f"{p.name} updated!")
if p.exists():
logger.info("sending file")
bot.send_chat_action(chat_id=update.effective_message.chat_id,
action=ChatAction.UPLOAD_PHOTO)
bot.send_photo(chat_id=chat_id, photo=p.open('rb'), timeout=600)
else:
_msg = f"{p.name} was not saved!"
logger.error(_msg)
bot.send_message(chat_id=chat_id, text=_msg)
else:
_msg = "Could not take photo on cam 0!"
logger.error(_msg)
bot.send_message(chat_id=chat_id, text=_msg)
def getFileCmd(bot, update, args):
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
logger = getLogger("getFileCmd")
if not args:
bot.send_message(
chat_id=chat_id,
text="Wrong command use!\nusage: /getfile <file_path>")
return
_arg = args.pop(0)
p = Path(_arg)
logger.info(f"{user}: get file {_arg}")
if p.exists():
if p.is_file():
logger.info("Sending file \"{}\"...".format(_arg))
bot.send_chat_action(chat_id=update.effective_message.chat_id,
action=ChatAction.UPLOAD_DOCUMENT)
bot.send_document(chat_id=chat_id,
document=open(_arg, 'rb'),
timeout=600)
else:
_msg = f"\"{_arg}\" is not a file!"
logger.info(_msg)
bot.send_message(chat_id=chat_id, text=_msg)
else:
_msg = f"File \"{_arg}\" not found!"
logger.info(_msg)
bot.send_message(chat_id=chat_id, text=_msg)
def getMeteoCmd(bot, update, args):
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
logger = getLogger("getMeteoCmd")
p = Path("./meteo.png")
logger.info(f"{user}: get meteo")
if p.exists():
if time.time() - p.stat().st_ctime < 3600:
logger.info("sending cached meteo chart")
bot.send_chat_action(chat_id=update.effective_message.chat_id,
action=ChatAction.UPLOAD_PHOTO)
bot.send_photo(chat_id=chat_id, photo=p.open('rb'), timeout=600)
return
if get_meteo():
logger.info(f"{p.name} updated!")
if p.exists():
logger.info("sending file")
bot.send_chat_action(chat_id=update.effective_message.chat_id,
action=ChatAction.UPLOAD_PHOTO)
bot.send_photo(chat_id=chat_id, photo=p.open('rb'), timeout=600)
else:
_msg = f"{p.name} was not saved!"
logger.error(_msg)
bot.send_message(chat_id=chat_id, text=_msg)
else:
bot.send_message(chat_id=chat_id, text="Could not get meteo info!")
logger.error(f"Could not download {p.name}!")
def userInfoCmd(bot, update, args):
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
logger = getLogger("userInfoCmd")
result = ""
logger.info(f"{user}: user info {' '.join(args)}")
if args:
found = 0
counter = 0
if "all" in args:
user_id_list = getAvailableUserInfoIDs()
result = f"Found {len(user_id_list)} user's info:\n\n"
else:
user_id_list = [int(_arg) for _arg in args if _arg.isnumeric()]
for uid in user_id_list:
counter += 1
output = getUserInfo(uid)
if not output:
result += f"{uid} - not found\n"
else:
found += 1
result += f"{output['username']}({output['id']}) {'[Bot]' if output['is_bot'] else ''}\n"
result += f"{output['first_name']} {output['last_name']}\n"
if counter < len(user_id_list):
result += 20 * "-" + "\n"
if found < len(user_id_list):
result = f"Found {found}/{len(user_id_list)} user's info:\n\n" + result
else:
user_id_list = getAvailableUserInfoIDs()
result = f"{len(user_id_list)} user info available:\n\n"
for uid in user_id_list:
result += f"{uid}\n"
logger.info("result:\n{}".format(result))
bot.send_message(chat_id=chat_id, text=result)
def printAvailableCmds(bot, update, args):
user_id = update.message.from_user.id
chat_id = update.message.chat_id
user = getUserByID(user_id)
getLogger("printAvailableCmds").info(f"{user}: print available commands")
cmds = "{}".format("".join(f"{_cmd}\n" for _cmd in CMDS.keys()))
bot.send_message(chat_id=chat_id, text=cmds)
def any_msg_handler(bot, update, user_data):
logger = getLogger("simple_msg")
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
logger.info("{}\n{}\n".format(user, update.message))
time.sleep(0.5)
bot.send_message(chat_id=chat_id, text="Hasta la vista, baby.")
def disarmCmd(bot, update, args):
global alarm
user_id = update.message.from_user.id
user = getUserByID(user_id)
chat_id = update.message.chat_id
getLogger("disarmCmd").info(f"{user}: Disarming!")
alarm.disarm()
bot.send_message(chat_id=chat_id, text="Disarmed!")
def getUsersCmd(bot, update, args):
user_id = update.message.from_user.id
chat_id = update.message.chat_id
user = getUserByID(user_id)
logger = getLogger("getUsersCmd")
logger.info(f"{user}: get user list")
user_list = getUsersIDs()
users = "{}".format("".join(f"{_uid}\n" for _uid in user_list))
bot.send_message(chat_id=chat_id, text=users)
def removeUserCmd(bot, update, args):
user_id = update.message.from_user.id
chat_id = update.message.chat_id
logger = getLogger("removeUserCmd")
if not args:
bot.send_message(chat_id=chat_id,
text="Wrong command use!\nusage: /rmuser <id>")
return
user = getUserByID(user_id)
_arg = int(args.pop(0))
logger.info(f"{user}: Removing user {_arg}...")
msg = f"User {_arg} doesn\'t exists!"
if removeUser(_arg):
msg = f"User {_arg} removed!"
logger.info(msg)
bot.send_message(chat_id=chat_id, text=msg)
def addUserCmd(bot, update, args):
user_id = update.message.from_user.id
chat_id = update.message.chat_id
logger = getLogger("addUserCmd")
if not args:
bot.send_message(chat_id=chat_id,
text="Wrong command use!\nusage: /adduser <id>")
return
user = getUserByID(user_id)
_arg = int(args.pop(0))
logger.info(f"{user}: Adding user {_arg}...")
msg = f"User {_arg} already exists!"
if addUser(_arg, "U"):
msg = f"User {_arg} added!"
logger.info(msg)
bot.send_message(chat_id=chat_id, text=msg)
def execute(bot, update, args):
global shell
user_id = update.message.from_user.id
chat_id = update.message.chat_id
logger = getLogger("execute")
if not args:
bot.send_message(chat_id=chat_id,
text="Wrong command use!\nusage: /exec <cmd>")
return
user = getUserByID(user_id)
_args = " ".join(args)
logger.info(f"{user}: execute {_args}")
if user_id not in shell.keys():
shell[user_id] = Shell()
result = shell[user_id].execute(_args)
logger.info(result)
bot.send_message(chat_id=chat_id, text=result)
def shutdown(self, bot, update):
user_id = update.message.from_user.id
user = getUserByID(user_id)
self.__logger.info(f"{user}: shutdown !!!")
update.message.reply_text('Shutting down...')
Thread(target=self.stop).start()
def restart(self, bot, update):
user_id = update.message.from_user.id
user = getUserByID(user_id)
self.__logger.info(f"{user}: restart!")
update.message.reply_text('Restarting...')
Thread(target=self.stop_and_restart).start()
def printUserID(bot, update, args):
user_id = update.message.from_user.id
chat_id = update.message.chat_id
user = getUserByID(user_id)
getLogger("printUserID").info(f"{user}: what is my ID?")
bot.send_message(chat_id=chat_id, text=user_id)