Add start command and base config

This commit is contained in:
Ceda EI 2018-10-02 20:49:46 +05:30
parent d4b71f2e80
commit 96d557b635
3 changed files with 45 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
config.py
__pycache__/

37
bot.py Normal file
View File

@ -0,0 +1,37 @@
#!/usr/bin/env python3
import logging
import telegram
from telegram.ext import Updater, CommandHandler
try:
import config
except ImportError:
print("Missing Config. Exiting.")
exit()
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \
%(message)s', level=logging.INFO)
def start(bot, update):
chat_id = update.message.chat_id
if chat_id != config.master_id:
text = "You need to host your own instance! \ Source: \
https://gitlab.com/ceda_ei/pc-monitor-bot"
bot.send_message(chat_id=chat_id, text=text)
return
custom_keyboard = [['/state', '/screenshot'], ['/lock', '/video']]
reply_markup = telegram.ReplyKeyboardMarkup(custom_keyboard)
bot.send_message(chat_id=chat_id, text="What to do?",
reply_markup=reply_markup)
updater = Updater(token=config.api_key)
dispatcher = updater.dispatcher
start_handler = CommandHandler('start', start)
dispatcher.add_handler(start_handler)
updater.start_polling()

6
sample.config.py Normal file
View File

@ -0,0 +1,6 @@
# Create a new bot by messaging @BotFather and follow the instructions
# Replace the key by the actual token recieved from BotFather
api_key = "123456789:xxxxxxxxxxxxxxxxxxxxxxxxxxxx"
# Replace 123456789 with your own user id.
master_id = 123456789