|
|
@@ -1,11 +1,12 @@
|
|
|
import telebot
|
|
|
import re
|
|
|
|
|
|
-TOKEN = 'YOUR_TOKEN'
|
|
|
+TOKEN = ''
|
|
|
bot = telebot.TeleBot(TOKEN)
|
|
|
receiveReference = False
|
|
|
reference = ''
|
|
|
-pattern = "((http|https)\:\/\/)?[a-zA-Z0-9\.\/\?\:@\-_=#]+\.([a-zA-Z]){2,6}([a-zA-Z0-9\.\&\/\?\:@\-_=#])*"
|
|
|
+pattern = "http[s]?:\/\/(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+"
|
|
|
+refPattern = "ref=.*\?"
|
|
|
|
|
|
@bot.message_handler(commands=['start'])
|
|
|
def send_welcome(message):
|
|
|
@@ -19,7 +20,7 @@ def send_help(message):
|
|
|
def send_question(message):
|
|
|
global receiveReference
|
|
|
receiveReference = True
|
|
|
- bot.reply_to(message, "tell me, what's your reference code?.")
|
|
|
+ bot.reply_to(message, "Tell me, what's your reference code?.")
|
|
|
|
|
|
@bot.message_handler()
|
|
|
def process_response(message):
|
|
|
@@ -27,11 +28,25 @@ def process_response(message):
|
|
|
if(len(reference) > 0):
|
|
|
result = re.match(pattern, message.text)
|
|
|
if(result):
|
|
|
- bot.delete_message(chat_id=message.chat.id, message_id=message.message_id)
|
|
|
- if(receiveReference):
|
|
|
+ url = result.group()
|
|
|
+ print("[INFO] Found URL match on: %s" % url)
|
|
|
+ refResult = re.search(refPattern, url)
|
|
|
+ if(refResult):
|
|
|
+ ref = refResult.group()
|
|
|
+ print("[INFO] Found reference match on: %s" % ref)
|
|
|
+ newMessage = message.text.replace(ref, "ref=%s?" % reference)
|
|
|
+ bot.send_message(message.chat.id, newMessage)
|
|
|
+ bot.delete_message(chat_id=message.chat.id, message_id=message.message_id)
|
|
|
+ else:
|
|
|
+ print("[INFO] No reference pattern matched")
|
|
|
+ else:
|
|
|
+ print("[INFO] No URL pattern matched")
|
|
|
+ elif(receiveReference):
|
|
|
reference = message.text
|
|
|
bot.reply_to(message, "OK, your reference is: %s, send me now any URL." % reference)
|
|
|
receiveReference = False
|
|
|
+ else:
|
|
|
+ print("[INFO] Receive reference disabled, try /reference")
|
|
|
|
|
|
|
|
|
try:
|
