create config

This commit is contained in:
bakatrouble 2018-11-20 23:06:07 +03:00
parent f8cd51fe16
commit 6501238970
3 changed files with 11 additions and 8 deletions

2
.gitignore vendored
View File

@ -166,3 +166,5 @@ fabric.properties
# Editor-based Rest Client # Editor-based Rest Client
.idea/httpRequests .idea/httpRequests
config.py

1
config.example.py Normal file
View File

@ -0,0 +1 @@
BOT_TOKEN = '450146961:AAGt5QRp3jS5wcHVIZOLxmqyO55iOVf6gpY'

16
main.py
View File

@ -6,21 +6,21 @@ from telegram.error import Unauthorized, TelegramError
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
from telegram import Message, Update from telegram import Message, Update
import config
from models import Subscriber from models import Subscriber
queue = [] queue = []
def go_away(bot, update: Update): def go_away(bot, update: Update):
Subscriber(user_id=str(update.message.chat_id)) # Subscriber(user_id=str(update.message.chat_id))
update.message.reply_text('Вы были добавлены') # update.message.reply_text('Вы были добавлены')
# update.message.reply_text('Пожалуйста, обратитесь к @lono_contactbot') update.message.reply_text('Пожалуйста, обратитесь к @lono_contactbot')
def unsubscribe(bot, update: Update): def unsubscribe(bot, update: Update):
Subscriber.deleteBy(user_id=str(update.message.chat_id)) Subscriber.deleteBy(user_id=str(update.message.chat_id))
update.message.reply_text('Вы были отписаны от бота. Обратитесь к @lono_contactbot за добавлением обратно.') update.message.reply_text('Вы были отписаны от бота. Обратитесь к @lono_contactbot за добавлением обратно.')
# update.message.reply_text('Пожалуйста, обратитесь к @lono_contactbot')
def msg(bot, update: Update): def msg(bot, update: Update):
@ -34,9 +34,9 @@ def task_queue(bot, job):
current_chat = str(m.chat_id) current_chat = str(m.chat_id)
uids = set(s.user_id for s in Subscriber.select()) uids = set(s.user_id for s in Subscriber.select())
if current_chat not in uids: if current_chat not in uids:
# return m.reply_text('Пожалуйста, обратитесь к @lono_contactbot') return m.reply_text('Пожалуйста, обратитесь к @lono_contactbot')
Subscriber(user_id=current_chat) # Subscriber(user_id=current_chat)
m.reply_text('Вы были добавлены') # m.reply_text('Вы были добавлены')
try: try:
uids.remove(current_chat) uids.remove(current_chat)
@ -90,7 +90,7 @@ def task_queue(bot, job):
pass pass
updater = Updater('450146961:AAGt5QRp3jS5wcHVIZOLxmqyO55iOVf6gpY', workers=4) updater = Updater(config.BOT_TOKEN, workers=4)
updater.job_queue.run_repeating(task_queue, .04) updater.job_queue.run_repeating(task_queue, .04)
updater.dispatcher.add_handler(CommandHandler('start', go_away)) updater.dispatcher.add_handler(CommandHandler('start', go_away))