Created
November 17, 2018 12:16
-
-
Save clasense4/44832e8694448ea2acd93ee1b2055d40 to your computer and use it in GitHub Desktop.
Simple Telegram Bot
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 python | |
# -*- coding: utf-8 -*- | |
"""Simple Bot to reply to Telegram messages. | |
This program is dedicated to the public domain under the CC0 license. | |
This Bot uses the Updater class to handle the bot. | |
First, a few handler functions are defined. Then, those functions are passed to | |
the Dispatcher and registered at their respective places. | |
Then, the bot is started and runs until we press Ctrl-C on the command line. | |
Usage: | |
Basic Echobot example, repeats messages. | |
Press Ctrl-C on the command line or send a signal to the process to stop the | |
bot. | |
""" | |
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters | |
from telegram.ext import InlineQueryHandler | |
from telegram import InlineQueryResultArticle, InputTextMessageContent | |
import telegram | |
import logging | |
from datetime import datetime as dt | |
# Enable logging | |
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', | |
level=logging.INFO) | |
logger = logging.getLogger(__name__) | |
help_text = """Hi ! | |
Commands : | |
/help This help message | |
/register Register to this bot | |
/product Product list | |
/order Order Product | |
Example: /order beton 100 | |
/cart Product cart | |
/checkout Checkout and receive invoice | |
""" | |
products = { | |
1 : { | |
'name' : 'genteng', | |
'price' : 1000 | |
}, | |
2 : { | |
'name' : 'beton', | |
'price' : 25000000 | |
}, | |
} | |
# Define a few command handlers. These usually take the two arguments bot and | |
# update. Error handlers also receive the raised TelegramError object in error. | |
def start(bot, update): | |
"""Send a message when the command /start is issued.""" | |
update.message.reply_text(help_text) | |
def register(bot, update): | |
update.message.reply_text('Register Success.') | |
def help(bot, update): | |
"""Send a message when the command /help is issued.""" | |
update.message.reply_text(help_text) | |
def echo(bot, update): | |
"""Echo the user message.""" | |
update.message.reply_text(update.message.text) | |
def error(bot, update, error): | |
"""Log Errors caused by Updates.""" | |
logger.warning('Update "%s" caused error "%s"', update, error) | |
def unknown(bot, update): | |
"""Unknown command.""" | |
bot.send_message(chat_id=update.message.chat_id, text="Sorry, I didn't understand that command.") | |
def product(bot, update): | |
prices = '*Produk* : \n' | |
for key, value in products.items(): | |
strings = str(key) + '. ' + value['name'] + ' = ' + str(value['price']) + '\n' | |
prices += strings | |
bot.send_message( | |
chat_id=update.message.chat_id, | |
text=prices, | |
parse_mode=telegram.ParseMode.MARKDOWN | |
) | |
def search_product(products, lookup): | |
for key, value in products.items(): | |
if lookup in value['name']: | |
return True | |
return False | |
def order(bot, update, args): | |
message = '' | |
if len(args) == 2: | |
if search_product(products, args[0]): | |
message = 'Product found, creating order' | |
else: | |
message = 'Product not found' | |
bot.send_message( | |
chat_id=update.message.chat_id, | |
text=message, | |
parse_mode=telegram.ParseMode.MARKDOWN | |
) | |
else: | |
message = 'Product is not found or wrong command. Example: /order beton 100' | |
bot.send_message( | |
chat_id=update.message.chat_id, | |
text=message, | |
parse_mode=telegram.ParseMode.MARKDOWN | |
) | |
print(args) | |
def main(): | |
"""Start the bot.""" | |
# Create the EventHandler and pass it your bot's token. | |
updater = Updater("755746654:AAFpANfrruN7G8ddhhLRWibVbVs4xyvponA") | |
# Get the dispatcher to register handlers | |
dp = updater.dispatcher | |
# on different commands - answer in Telegram | |
dp.add_handler(CommandHandler("start", start)) | |
dp.add_handler(CommandHandler("register", register)) | |
dp.add_handler(CommandHandler("help", help)) | |
dp.add_handler(CommandHandler("product", product)) | |
dp.add_handler(CommandHandler("order", order, pass_args=True)) | |
# on noncommand i.e message - echo the message on Telegram | |
dp.add_handler(MessageHandler(Filters.text, echo)) | |
# unknown command | |
dp.add_handler(MessageHandler(Filters.command, unknown)) | |
# log all errors | |
dp.add_error_handler(error) | |
# Start the Bot | |
updater.start_polling() | |
# Run the bot until you press Ctrl-C or the process receives SIGINT, | |
# SIGTERM or SIGABRT. This should be used most of the time, since | |
# start_polling() is non-blocking and will stop the bot gracefully. | |
updater.idle() | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment