Created
June 18, 2020 20:37
-
-
Save zorbax/506bc9096459baea6a5451017ca72034 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
[credentials] | |
token = 123456:ABCDEF1234ghIkl-zyx57W2v1u123ew11 |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python3 | |
__author__ = "@zorbax" | |
import sys | |
import json | |
import requests | |
import argparse | |
from configparser import ConfigParser | |
config_file = 'config.cfg' | |
class TelegramBot: | |
def __init__(self, config): | |
self.token = self.read_token(config) | |
self.base = "https://api.telegram.org/bot{}/".format(self.token) | |
def get_updates(self): | |
getupdates = self.base + "getUpdates?timeout=100" | |
r = requests.get(getupdates) | |
return json.loads(r.content) | |
def get_chatid(self): | |
getupdates = self.base + "getUpdates?timeout=100" | |
r = requests.get(getupdates, params={'result': 'chat'}) | |
input_dict = json.loads(r.content) | |
return input_dict['result'][0]['message']['chat']['id'] | |
def send_message(self, msg, chat_id): | |
url = self.base + "sendMessage?chat_id={}&text={}".format(chat_id, msg) | |
if msg is not None: | |
requests.get(url) | |
return None | |
@staticmethod | |
def read_token(config): | |
cfg_parser = ConfigParser() | |
cfg_parser.read(config) | |
return cfg_parser.get('credentials', 'token') | |
if __name__ == '__main__': | |
parser = argparse.ArgumentParser(description='Send message to a Telegram Bot.') | |
group = parser.add_mutually_exclusive_group(required=True) | |
group.add_argument('-m', dest="msg", type=str, metavar='MESSAGE_STRING', | |
help='"The Message, like string"') | |
group.add_argument('-f', dest="msg_file", metavar='MESSAGE_FILE', | |
type=argparse.FileType('r'), help='Message as file, e.g. file.log') | |
args = parser.parse_args() | |
bot = TelegramBot(config=config_file) | |
chatid = bot.get_chatid() | |
if args.msg: | |
bot.send_message(args.msg, chatid) | |
else: | |
bot.send_message(args.msg_file.read(), chatid) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment