def replace_auth_token_view():
try:
data = json.loads(flask.request.get_data().decode('utf-8'))
except Exception as error:
logging.error('input data parse error {}'.format(error))
data = {}
name = data.get('name')
avatar = data.get('avatar')
auth_token = data.get('token')
webhook = data.get('webhook')
if auth_token is not None and name is not None and avatar is not None:
try:
viberbot_api = Api(
BotConfiguration(name=name,
avatar=avatar,
auth_token=auth_token))
viberbot_api.set_webhook(url=webhook)
resp = flask.jsonify(message='Auth token успешно заменен',
account=viberbot_api.get_account_info())
resp.status_code = 200
return resp
except Exception as error:
logging.error('replace auth token error: {}'.format(error))
resp = flask.jsonify(error='Auth token не удалось замененить ')
resp.status_code = 404
return resp
resp = flask.jsonify(message='некорректные входные данные {}'.format(
flask.request.get_data()))
resp.status_code = 404
return resp
class Viber(ChatIface):
def __init__(self, web_url, web_config=('0.0.0.0', 8080)):
super().__init__()
self.thread = None
self.token = os.environ.get('VIBER_TOKEN')
if self.token is None:
raise Exception('VIBER_TOKEN env variable is nil.')
self.bot_configuration = BotConfiguration(
name=os.environ.get('VIBER_NAME'),
avatar='https://static.thenounproject.com/png/363639-200.png',
auth_token=self.token)
self.api = None
self.app = Flask(__name__)
self.web_config = web_config
self.web_url = web_url
self.setup_routes()
self.last_message = None
self.initial_button_text = "I want to book a hotel"
def run(self):
web_config = self.web_config
self.thread = threading.Thread(target=self.start)
self.thread.start()
self.api = Api(self.bot_configuration)
logging.info("Setting viber hook to: %s", self.web_url)
evtypes = self.api.set_webhook(
self.web_url, webhook_events=['conversation_started', 'message'])
viber_account = self.api.get_account_info()
logging.info("Registered ev types: %s", evtypes)
logging.info("Account info: %s", viber_account)
def start(self):
web_config = self.web_config
self.app.run(host=web_config[0],
port=web_config[1],
debug=True,
use_reloader=False)
# Message handlers
#
# Conversation start allows only 1 message before a subscription
def _on_start(self, user_id):
viber = self.api
d = self.get_dialog(user_id)
g = d.start()
d.on_searching(
lambda details: self._callback_on_searching(user_id, details))
keyboard_msg = self.get_initial_keyboard(user_id)._keyboard
viber.send_messages(user_id, [
TextMessage(text=g, keyboard=keyboard_msg),
])
def _on_message(self, user_id, message):
# Make sure our dialog is setup.
d = self.get_dialog(user_id)
d.on_searching(
lambda details: self._callback_on_searching(user_id, details))
try:
# Process the message
is_done, rz = self.process_message(user_id, message)
r = [rx for rx in listify(rz) if rx is not None]
# Send replies
if is_done and r is not None and len(r) > 0:
last_response = r[len(r) - 1]
last_response.buttons = [self.initial_button_text]
self.send_replies(user_id, r)
except Exception as ex:
tb = traceback.format_exc()
logging.info(tb)
logging.info(ex)
self.send_replies(user_id, "Sorry can you try a different phrase?")
def _callback_on_searching(self, user_id, details):
viber = self.api
logging.info("Starting search at %s", details)
place = details['location'].capitalize()
t = details['type']
if len(t) == 0:
txtmsg = TextMessage(
text="Great. Here are a few places for you to stay in {0}..".
format(place))
viber.send_messages(user_id, [txtmsg])
else:
txtmsg = TextMessage(
text="Great. I'll start searching for {0} in {1}..".format(
'placest to stay', place))
viber.send_messages(user_id, [txtmsg])
def send_img(self,
user_id,
rep: dialog.dialog.Reply,
keyboard: KeyboardMessage = None):
viber = self.api
imglist = rep.img
if not isinstance(imglist, list):
imglist = [imglist]
img = imglist[0]
if img is None:
return
img_url = img['url']
k = None
if keyboard is not None:
k = keyboard._keyboard
pmsg = PictureMessage(text=rep.text, media=img_url, keyboard=k)
viber.send_messages(user_id, [pmsg])
def send_keyboard(self, user_id, keyboard):
viber = self.api
try:
viber.send_messages(user_id, [keyboard])
except:
tb = traceback.format_exc()
logging.info(tb)
logging.info(ex)
def send_replies(self, user_id, replies):
viber = self.api
if not isinstance(replies, list): replies = [replies]
for rep in replies:
if isinstance(rep, str):
rep = dialog.dialog.msg_reply(rep)
reply_text = rep.str()
if reply_text == "":
reply_text = "Can you try again with a different phrase?"
logging.info("REPL[%s]: %s", rep.type, rep)
self.send_reply(rep, reply_text, user_id)
def send_on_done(self, user_id):
init_kbd = self.get_initial_keyboard(user_id)
viber = self.api
viber.send_messages(user_id, [init_kbd])
def send_reply(self, rep, reply_text, user_id):
viber = self.api
if rep.type == 'place':
# Send the image and a button for the user to confirm or reject
kbd = self.get_place_keyboard(user_id, rep.buttons)
if rep.img:
self.send_img(user_id, rep, keyboard=kbd)
else:
txtmsg = TextMessage(text=str(reply_text), keyboard=kbd)
viber.send_messages(user_id, [txtmsg])
elif rep.type == 'place_list':
logging.info("Reply itinerary: %s", reply_text)
txtmsg = TextMessage(text=str(reply_text),
keyboard=buttons_to_keyboard(rep.buttons))
viber.send_messages(user_id, [txtmsg])
elif rep.type == 'interest_question':
txtmsg = TextMessage(text=str(reply_text),
keyboard=buttons_to_keyboard(rep.buttons))
viber.send_messages(user_id, [txtmsg])
else:
rkbd = buttons_to_keyboard(rep.buttons)
logging.info("KBD: %s", rkbd)
txtmsg = TextMessage(text=str(reply_text), keyboard=rkbd)
viber.send_messages(user_id, [txtmsg])
def setup_routes(self):
app = self.app
@app.route('/', methods=['POST'])
def incoming():
import json
viber = self.api
data = request.get_data()
logging.info("received request. post data: {0}".format(data))
# every viber message is signed, you can verify the signature using this method
if not viber.verify_signature(
request.get_data(),
request.headers.get('X-Viber-Content-Signature')):
return Response(status=403)
request_dict = json.loads(data)
# this library supplies a simple way to receive a request object
viber_request = viber.parse_request(data)
if isinstance(viber_request, ViberConversationStartedRequest):
user_id = viber_request.user.id
self._on_start(user_id)
elif isinstance(viber_request, ViberSeenRequest):
pass
elif isinstance(viber_request, ViberDeliveredRequest):
pass
elif isinstance(viber_request, ViberMessageRequest):
sender_id = viber_request.sender.id
message = viber_request.message
# We need unique messages
if isinstance(message, TextMessage):
ts = int(request_dict['timestamp'])
is_old = (self.last_message is not None
and (self.last_message[2] == ts)
and self.last_message[1] == sender_id)
cr_pair = (message.text, sender_id, ts)
# logging.info("MSG DUMP: %s", cr_pair)
# logging.info("LAST DUMP: %s", self.last_message)
# logging.info("DUP: %s", is_old)
if not is_old:
self.last_message = cr_pair
self._on_message(sender_id, message.text)
# self.last_message = cr_pair
elif isinstance(viber_request, ViberFailedRequest):
logging.warn(
"client failed receiving message. failure: {0}".format(
viber_request))
return Response(status=200)
@app.route('/incoming', methods=['POST'])
def incomingUrl():
logging.debug("received request. post data: {0}".format(
request.get_data()))
# handle the request here
return Response(status=200)
@app.route('/health-check', methods=['GET'])
def health():
logging.debug("Health-check")
# handle the request here
return Response(status=200)
def get_place_keyboard(self, user_id, additional_buttons=None):
import json
buttons = [{
"Columns": 2,
"Rows": 2,
"Text": "<font color=\"#494E67\">Confirm</font>",
"TextSize": "medium",
"TextHAlign": "center",
"TextVAlign": "middle",
"ActionType": "reply",
"ActionBody": "Yes",
"BgColor": "#f7bb3f",
}, {
"Columns": 2,
"Rows": 2,
"Text": "<font color=\"#494E67\">Next</font>",
"TextSize": "medium",
"TextHAlign": "center",
"TextVAlign": "middle",
"ActionType": "reply",
"ActionBody": "No",
"BgColor": "#f6f7f9",
}, {
"Columns": 2,
"Rows": 2,
"Text": "<font color=\"#494E67\">Stop</font>",
"TextSize": "medium",
"TextHAlign": "center",
"TextVAlign": "middle",
"ActionType": "reply",
"ActionBody": "Stop",
"BgColor": "#f6f7f9",
}]
if additional_buttons:
buttons.extend([{
"Columns": 2,
"Rows": 2,
"Text": "<font color=\"#494E67\">" + b + "</font>",
"ActionBody": b,
} for b in additional_buttons])
keyboard = {
"DefaultHeight": True,
"BgColor": "#FFFFFF",
"Type": "keyboard",
"Buttons": buttons
}
msg = KeyboardMessage(keyboard=keyboard)
return msg
def get_initial_keyboard(self, user_id):
keyboard = {
"Type":
"keyboard",
"Buttons": [{
"Columns":
6,
"ActionType":
"reply",
"Rows":
1,
"Text":
"<font color=\"#494E67\">" + self.initial_button_text +
"</font>",
"TextSize":
"medium",
"TextHAlign":
"center",
"TextVAlign":
"middle",
"ActionBody":
self.initial_button_text,
}]
}
msg = KeyboardMessage(keyboard=keyboard)
return msg
def set_webhook(viber):
viber.set_webhook('https://llkfncvzsx.localtunnel.me')
def add_to_cart(cart, product, count):
for item in cart.items:
if item['product']['id'] == product['id']:
item['count'] += count
cart.write_to_db()
return item
item = {"product": product, "count": int(count)}
cart.items.append(item)
cart.write_to_db()
return item
def clear_cart(cart):
cart.items = []
cart.write_to_db()
if __name__ == "__main__":
viber.set_webhook("")
scheduler = sched.scheduler(time.time, time.sleep)
scheduler.enter(5, 1, set_webhook, (viber, ))
t = threading.Thread(target=scheduler.run)
t.start()
app.run(host='0.0.0.0', port=8000, debug=True)
Меня зовут GruzBot🤖.Я помогу вам:
- вызвать грузовое такси за три простых шага🚚.
- нанять грузчиков💪."""
c.execute("INSERT INTO users(id,stage) VALUES ('%s', '1')" %
(str(viber_request.user.id)))
conn.commit()
viber.send_messages(
viber_request.user.id,
[TextMessage(text=reply, keyboard=keyboards.first)])
elif isinstance(viber_request, ViberFailedRequest):
logger.warn("client failed receiving message. failure: {0}".format(
viber_request))
return Response(status=200)
def set_webhook():
while True:
old_url = 'https://viberrooot.herokuapp.com/'
url = "https://gruzobot.herokuapp.com/"
viber.set_webhook(url)
time.sleep(3600)
if __name__ == "__main__":
viber.set_webhook("https://gruzobot.herokuapp.com/")
t = threading.Thread(target=set_webhook)
t.start()
app.run(debug=True)
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
import Constants
viber = Api(BotConfiguration(
name=Constants.BOT_NAME,
avatar=Constants.AVATAR_URL,
auth_token=Constants.AUTH_TOKEN
))
viber.set_webhook("https://f2056b99634f.ngrok.io")
url='https://%s%s' %
(config.SERVER_NAME, config.TELEGRAM_BOT_WEBHOOK),
max_connections=100)
except Exception as e:
logging.warning(e)
else:
telegram_service = None
if config.TELEGRAM_STRANGERGEO_ENABLED:
strangergeo_service = TeleBot(config.TELEGRAM_STRANGERGEO_KEY,
threaded=False)
try:
strangergeo_service.set_webhook(
url='https://%s%s' %
(config.SERVER_NAME, config.TELEGRAM_STRANGERGEO_WEBHOOK),
max_connections=100)
except Exception as e:
logging.warning(e)
else:
strangergeo_service = None
if config.VIBER_BOT_ENABLED:
viber_service = ViberApi(
ViberBotConfiguration(name=config.VIBER_BOT_NAME,
avatar='http://viber.com/avatar.jpg',
auth_token=config.VIBER_BOT_TOKEN))
viber_service.set_webhook('https://%s%s' %
(config.SERVER_NAME, config.VIBER_BOT_WEBHOOK))
else:
viber_service = None
class ViberWebhook:
def __init__(self, appConfig, dbUsers = 'vb_users.json') -> None:
http_tunnel = ngrok.connect()
self.public_url = http_tunnel.public_url.replace('http', 'https')
print('Public URL acquired: ' + self.public_url)
self.usersDb = TinyDB(dbUsers)
self.app = Flask(__name__)
self.config = appConfig
self.viber = Api(BotConfiguration(
name = self.config.getProperty('Publishers.Viber.Name'),
avatar = self.config.getProperty('Publishers.Viber.Avatar'),
auth_token = self.config.getProperty('Publishers.Viber.Token')
))
self.query = Query()
## Delayed webhook setup
scheduler = sched.scheduler(time.time, time.sleep)
scheduler.enter(5, 1, self.set_webhook, (self.viber,))
t = threading.Thread(target=scheduler.run)
t.start()
self.app.add_url_rule('/', 'incoming', self.incoming, methods=['POST'])
self.app.add_url_rule('/ctrl', '', self.control, methods=['POST', 'GET'])
self.t_webApp = threading.Thread(target=self.flaskThread)
self.t_webApp.setDaemon(True)
print("Viber worker created.")
def __del__(self):
self.usersDb.close()
def flaskThread(self):
self.app.run(host='0.0.0.0', port=80, debug=False)
def Run(self):
self.t_webApp.run()
def GetAdmins(self):
admins = self.usersDb.search(self.query.admin == '1')
return admins
def NotifyAdmins(self, admins, message):
for admin in admins:
self.viber.send_messages(admin['id'], [ TextMessage(text = message) ])
def IsAdmin(self, user_id, admins):
return next((admin for admin in admins if admin['id'] == user_id), None) != None
def Reboot():
command = "/usr/bin/sudo /sbin/shutdown -r now"
import subprocess
process = subprocess.Popen(command.split(), stdout=subprocess.PIPE)
output = process.communicate()[0]
print(output)
def RestartViber():
command = "service Viber restart"
import subprocess
process = subprocess.Popen(command.split(), stdout=subprocess.PIPE)
output = process.communicate()[0]
print(output)
def incoming(self):
admins = self.GetAdmins()
print(request.path)
viber_request = self.viber.parse_request(request.get_data().decode('utf8'))
if isinstance(viber_request, ViberMessageRequest):
message = viber_request.message
if isinstance(message, TextMessage):
is_admin = self.IsAdmin(viber_request.sender.id, admins)
if is_admin:
print("IsAdmin: True")
## HANDLE ADMIN REQUESTS
usersListStr = ''
if(message.text.strip() == "/ListUsers" and is_admin):
for user in self.usersDb.all():
usersListStr += user['name'] + '\n'
self.NotifyAdmins(admins, 'Korisnici: \n' + usersListStr)
return Response(status=200)
if(message.text.strip() == "/ListAdmins" and is_admin):
for user in self.usersDb.search(self.query.admin == '1'):
usersListStr += user['name'] + '\n'
self.NotifyAdmins(admins, 'Administratori: \n' + usersListStr)
return Response(status=200)
if(message.text.strip() == "/GetPublicURL" and is_admin):
self.NotifyAdmins(admins, 'Javna adresa: \n' + self.public_url)
return Response(status=200)
if(message.text.strip() == "/GetLocalIP" and is_admin):
self.NotifyAdmins(admins, 'Lokalna adresa: \n' + socket.gethostbyname(socket.gethostname()))
return Response(status=200)
if(message.text.strip() == "/XRebootMe" and is_admin):
self.NotifyAdmins(admins, 'Rebooting...')
self.Reboot()
return Response(status=200)
if(message.text.strip() == "/XRestartViberService" and is_admin):
self.NotifyAdmins(admins, 'Restarting Viber service...')
self.RestartViber()
return Response(status=200)
UserQ = Query()
# Handle standard requests
if message.text.strip().lower() == 'stop':
self.usersDb.update({'active': '0'}, UserQ.id == viber_request.sender.id)
else:
if len(self.usersDb.search(UserQ.id == viber_request.sender.id)) == 0:
self.usersDb.insert({'id': viber_request.sender.id, 'name': viber_request.sender.name, 'active': '1', 'admin': '0'})
else:
self.usersDb.update({'active': '1'}, UserQ.id == viber_request.sender.id)
self.viber.send_messages(viber_request.sender.id, [ TextMessage(text = 'Uspešna prijava! Pošalji STOP za odjavu.') ])
#self.viber.send_messages("/qNmzm5H8vXHIuuJAmJZvw==", [ TextMessage(text = 'Novi korisnik: ' + viber_request.sender.name) ])
self.NotifyAdmins(admins, 'Novi korisnik: ' + viber_request.sender.name)
elif isinstance(viber_request, ViberConversationStartedRequest):
UserQ = Query()
#self.viber.send_messages(viber_request.user.id, [ TextMessage(text='Za prijavu pošaljite bilo kakvu poruku.') ])
if len(self.usersDb.search(UserQ.id == viber_request.user.id)) == 0:
self.usersDb.insert({'id': viber_request.user.id, 'name': viber_request.user.name, 'active': '1', 'admin': '0'})
else:
self.usersDb.update({'active': '0'}, UserQ.id == viber_request.user.id)
elif isinstance(viber_request, ViberSubscribedRequest):
UserQ = Query()
self.viber.send_messages(viber_request.user.id, [ TextMessage(text='Za prijavu pošaljite bilo kakvu poruku.') ])
if len(self.usersDb.search(UserQ.id == viber_request.user.id)) == 0:
self.usersDb.insert({'id': viber_request.user.id, 'name': viber_request.user.name, 'active': '1', 'admin': '0'})
else:
self.usersDb.update({'active': '1'}, UserQ.id == viber_request.user.id)
elif isinstance(viber_request, ViberUnsubscribedRequest):
UserQ = Query()
self.usersDb.update({'active': '0'}, UserQ.id == viber_request.user_id)
elif isinstance(viber_request, ViberFailedRequest):
logger.warn("client failed receiving message. failure: {0}".format(viber_request))
return Response(status=200)
def control(self):
admins = self.GetAdmins()
#data = request.get_data().decode('utf8')
if(request.args.get('command') == 'users'):
if(request.args.get('a') == '0'):
usersListStr = ""
for user in self.usersDb.all():
usersListStr += user['name'] + ';'
return Response(status=200, response=usersListStr)
else:
usersListStr = ""
for user in self.usersDb.search(self.query.admin == '1'):
usersListStr += user['name'] + ';'
return Response(status=200, response=usersListStr)
def set_webhook(self, viber):
self.viber.set_webhook(self.public_url)
"port": 25565,
}
bot_configuration = BotConfiguration(
name='InvestBot',
avatar='http://viber.com/avatar.jpg',
auth_token='4643793ea227d275-7a894dbfc47308e7-8ba014f6e7d19fa8')
# Logging
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler()
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
logger.addHandler(handler)
# Bot Server
viber = Api(bot_configuration)
app = Flask(__name__)
app.run(host='0.0.0.0', port=server_configuration["port"], debug=True)
viber.set_webhook("{0}:{1}/".format(server_configuration["host"],
server_configuration["port"]))
# Simple Echo EVENT
@app.route('/incoming', methods=['POST'])
def incoming():
return echo(request, viber, logger)
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
from viberbot.api.messages.text_message import TextMessage
from viberbot.api.messages.url_message import URLMessage
from viberbot.api.viber_requests import ViberFailedRequest, ViberConversationStartedRequest
from viberbot.api.viber_requests import ViberMessageRequest
from viberbot.api.viber_requests import ViberSubscribedRequest
app = Flask(__name__)
viber = Api(
BotConfiguration(
name='Test3312bot',
avatar='',
auth_token='4b3779249f67d35f-c9f6a69dd4d7a4b2-6e59c785345fb8b2'))
viber.set_webhook('https://viberbotapp.herokuapp.com/botviber.py')
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)
handler = logging.StreamHandler()
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
logger.addHandler(handler)
@app.route('/incoming', methods=['POST'])
def incoming():
logger.debug("received request. post data: {0}".format(request.get_data()))
# handle the request here
return Response(status=200)
from http.server import HTTPServer, SimpleHTTPRequestHandler, HTTPStatus
import ssl
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
from viberbot.api.messages.text_message import TextMessage
bot_configuration = BotConfiguration(
name='AtomSpace',
avatar='https://www.google.com/url?sa=i&source=images&cd=&cad=rja&uact=8&ved=2ahUKEwjy2sPww_ffAhWMpIsKHZdkC58QjRx6BAgBEAU&url=http%3A%2F%2Fodessa.net.ua%2Fnews%2Fv-odesse-vpervye-otkroet-svoi-dveri-atom-space&psig=AOvVaw2h8rp8WdUXn9-RUbLUkNIq&ust=1547906902440090',
auth_token='4918542585e7d307-571b9f239ed4f865-17485e2f067277a3'
)
viber = Api(bot_configuration)
# given a pem file ... openssl req -new -x509 -keyout yourpemfile.pem -out yourpemfile.pem -days 365 -nodes
httpd = HTTPServer(('localhost', 4444), SimpleHTTPRequestHandler)
httpd.socket = ssl.wrap_socket (httpd.socket, server_side=True,
certfile='cert.pem', keyfile='key.pem')
viber.set_webhook('localhost:4444')
httpd.serve_forever()
class Viber(BaseMessenger):
"""
IM connector for Viber REST API
"""
def __init__(self, token: str, **kwargs):
super().__init__(token, **kwargs)
self.bot = Api(
BotConfiguration(
auth_token=token,
name=kwargs.get('name'),
avatar=kwargs.get('avatar'),
))
def enable_webhook(self, url: str, **kwargs):
return self.bot.set_webhook(url=url)
def disable_webhook(self):
return self.bot.unset_webhook()
def get_account_info(self) -> Dict[str, Any]:
# {
# "status":0,
# "status_message":"ok",
# "id":"pa:75346594275468546724",
# "name":"account name",
# "uri":"accountUri",
# "icon":"http://example.com",
# "background":"http://example.com",
# "category":"category",
# "subcategory":"sub category",
# "location":{
# "lon":0.1,
# "lat":0.2
# },
# "country":"UK",
# "webhook":"https://my.site.com",
# "event_types":[
# "delivered",
# "seen"
# ],
# "subscribers_count":35,
# "members":[
# {
# "id":"01234567890A=",
# "name":"my name",
# "avatar":"http://example.com",
# "role":"admin"
# }
# ]
# }
data = self.bot.get_account_info()
account_info = {
'id': data.get('id'),
'username': data.get('name'),
'info': data
}
return account_info
def get_user_info(self, user_id: str, **kwargs) -> Dict[str, Any]:
# {
# "status":0,
# "status_message":"ok",
# "message_token":4912661846655238145,
# "user":{
# "id":"01234567890A=",
# "name":"John McClane",
# "avatar":"http://avatar.example.com",
# "country":"UK",
# "language":"en",
# "primary_device_os":"android 7.1",
# "api_version":1,
# "viber_version":"6.5.0",
# "mcc":1,
# "mnc":1,
# "device_type":"iPhone9,4"
# }
# }
data = self.bot.get_user_details(user_id).get('user')
user_info = {
'id': data.get('id'),
'username': data.get('name'),
'info': {
'avatar': data.get('avatar'),
'country': data.get('country'),
'language': data.get('language'),
'primary_device_os': data.get('primary_device_os'),
'api_version': data.get('api_version'),
'viber_version': data.get('viber_version'),
'device_type': data.get('device_type'),
}
}
return user_info
def parse_message(self, request: Request) -> Message:
# NOTE: There is no way to get the body
# after processing the request in DRF.
# # Verify signature
# sign = request.META.get('HTTP_X_VIBER_CONTENT_SIGNATURE')
# if not self.bot.verify_signature(request.body, sign):
# raise IMApiException(f'Viber message not verified; '
# f'Data={request.data}; Sign={sign};')
# Parse message data in to viber types
vb_request = create_request(request.data)
try:
return self._get_message(vb_request)
except Exception as err:
# TODO: remove this after development
log.exception(f'Parse message error; Message={vb_request}; '
f'Error={err};')
return Message(MessageType.UNDEFINED)
@staticmethod
def _get_message(vb_request: ViberRequest) -> Message:
if isinstance(vb_request, ViberMessageRequest):
if isinstance(vb_request.message, TextMessage):
return Message(message_type=MessageType.TEXT,
message_id=vb_request.message_token,
user_id=vb_request.sender.id,
text=vb_request.message.text,
timestamp=vb_request.timestamp)
elif isinstance(vb_request.message, PictureMessage):
return Message(message_type=MessageType.PICTURE,
message_id=vb_request.message_token,
user_id=vb_request.sender.id,
image_url=vb_request.message.media,
timestamp=vb_request.timestamp)
elif isinstance(vb_request.message, VideoMessage):
return Message(message_type=MessageType.PICTURE,
message_id=vb_request.message_token,
user_id=vb_request.sender.id,
video_url=vb_request.message.media,
size=vb_request.message.size,
timestamp=vb_request.timestamp)
else:
return Message(message_type=MessageType.TEXT,
message_id=vb_request.message_token,
user_id=vb_request.sender.id,
text=vb_request.message,
timestamp=vb_request.timestamp)
elif isinstance(vb_request, ViberConversationStartedRequest):
return Message(message_type=MessageType.START,
message_id=vb_request.message_token,
user_id=vb_request.user.id,
user_name=vb_request.user.name,
timestamp=vb_request.timestamp,
context=vb_request.context)
elif isinstance(vb_request, ViberSubscribedRequest):
return Message(message_type=MessageType.SUBSCRIBED,
user_id=vb_request.user.id,
user_name=vb_request.user.name,
timestamp=vb_request.timestamp)
elif isinstance(vb_request, ViberUnsubscribedRequest):
return Message(message_type=MessageType.UNSUBSCRIBED,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp)
elif isinstance(vb_request, ViberDeliveredRequest):
return Message(message_type=MessageType.DELIVERED,
message_id=vb_request.meesage_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp)
elif isinstance(vb_request, ViberSeenRequest):
return Message(message_type=MessageType.SEEN,
message_id=vb_request.meesage_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp)
elif isinstance(vb_request, ViberFailedRequest):
log.warning(f'Client failed receiving message; Error={vb_request}')
return Message(message_type=MessageType.FAILED,
message_id=vb_request.meesage_token,
user_id=vb_request.user_id,
error=vb_request.desc)
elif vb_request.event_type == 'webhook':
return Message(message_type=MessageType.WEBHOOK,
timestamp=vb_request.timestamp)
else:
log.warning(f'VRequest Type={type(vb_request)}; '
f'Object={vb_request};')
return Message(message_type=MessageType.UNDEFINED,
timestamp=vb_request.timestamp,
event_type=vb_request.event_type)
# raise IMApiException('Failed parse message; '
# 'Request object={}'.format(viber_request))
def send_message(self,
receiver: str,
message: str,
button_list: list = None,
**kwargs) -> str:
kb = self._get_keyboard(button_list) if button_list else None
if message:
vb_message = TextMessage(text=message, keyboard=kb)
else:
vb_message = KeyboardMessage(keyboard=kb)
try:
return self.bot.send_messages(receiver, [vb_message])[0]
except Exception as err:
if str(err) == 'failed with status: 6, message: notSubscribed':
raise NotSubscribed(err)
raise MessengerException(err)
def send_file(self,
receiver: str,
file_url: str,
file_size: int,
file_name: str,
file_type: str = None,
button_list: list = None,
**kwargs) -> str:
kb = self._get_keyboard(button_list) if button_list else None
if file_type == 'image':
message = PictureMessage(media=file_url, keyboard=kb)
elif file_type == 'video':
message = VideoMessage(media=file_url, size=file_size, keyboard=kb)
else:
message = FileMessage(media=file_url,
size=file_size,
file_name=file_name,
keyboard=kb)
try:
return self.bot.send_messages(receiver, [message])[0]
except Exception as err:
if str(err) == 'failed with status: 6, message: notSubscribed':
raise NotSubscribed(err)
raise MessengerException(err)
def welcome_message(self, text: str) -> Dict[str, str]:
return {
"sender": {
"name": self.name,
"avatar": self.avatar_url
},
"type": "text",
"text": text
}
@staticmethod
def _get_keyboard(buttons: list):
if not buttons:
return None
kb = {
'Type': 'keyboard',
'BgColor': '#ffffff',
'min_api_version': 6,
'Buttons': []
}
for button in buttons:
# if not isinstance(button, Button):
# continue
_btn = {
'Columns':
2, # TODO: how is it storage in Model?
'Rows':
1,
'BgColor':
'#aaaaaa',
'ActionType':
'reply',
'ActionBody':
button.command,
'Text':
'<font color="{clr}"><b>{text}'
'</b></font>'.format(text=button.text, clr='#131313'),
'TextVAlign':
'middle',
'TextHAlign':
'center',
'TextOpacity':
60,
'TextSize':
'large',
'TextPaddings': [12, 8, 8, 20], # [up, left, right, bottom]
}
try:
if hasattr(button, 'image'):
_btn.update(
BgMedia=
f'https://bot.it-o.ru/static/img/{button.image}',
BgMediaScaleType='fill')
except IndexError:
pass
kb['Buttons'].append(_btn)
return kb
import threading
import time
import logging
import json
import requests
import random
import os
from string import Template
from .jsons_storage import *
from .language_formatter import *
viber = Api(settings.BOT_CONFIGURATION)
scheduler = sched.scheduler(time.time, time.sleep)
scheduler.enter(5, 1, lambda: viber.set_webhook('https://' + settings.ALLOWED_HOSTS[0]))
t = threading.Thread(target=scheduler.run)
t.start()
logger = logging.getLogger()
logger.setLevel(logging.DEBUG if settings.DEBUG else logging.ERROR)
handler = logging.StreamHandler()
formatter = logging.Formatter('%(levelname)s - %(asctime)s - %(name)s - %(message)s')
handler.setFormatter(formatter)
logger.addHandler(handler)
menu_button_messages = ('ask', 'ask_custom', 'call_support', 'check_status', 'switch_lang')
def check_id(id_int: list, id_str: str):
try:
id_int.append(int(id_str))
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
from viberbot.api.messages import VideoMessage
from viberbot.api.messages.text_message import TextMessage
from viberbot.api.viber_requests import ViberConversationStartedRequest
from viberbot.api.viber_requests import ViberFailedRequest
from viberbot.api.viber_requests import ViberMessageRequest
from viberbot.api.viber_requests import ViberSubscribedRequest
from viberbot.api.viber_requests import ViberUnsubscribedRequest
# 4aa8cfe4f5a7d667-98afa3e99e58bfe6-9698235b67973816
# 4aa8cfe4f5a7d667-98afa3e99e58bfe6-9698235b67973816
# 495624962167d356-894001973007218c-448b90921c20d990
viber = Api(BotConfiguration(
name='drovobot',
avatar='http://site.com/avatar.jpg',
auth_token='495624962167d356-894001973007218c-448b90921c20d990'
))
viber.set_webhook('https://neo-boudoir.ru:443/')
from Settings import TOKEN, WEBHOOK
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
bot_configuration = BotConfiguration(name='MyLearnEnglishBot4',
avatar='http://viber.com/avatar.jpg',
auth_token=TOKEN)
viber = Api(bot_configuration)
print('setting webhook: ' + WEBHOOK)
viber.set_webhook(WEBHOOK)
from flask import Flask
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
from main import *
port = int(os.environ.get("PORT", 5000))
app = Flask(__name__)
viber = Api(
BotConfiguration(
name='Bilinhual',
avatar=
'https://dl-media.viber.com/1/share/2/long/vibes/icon/image/0x0/f09b'
'/fc477e80d5306023ccf92a07170886fee98bba96aca04959700207a62cc6f09b.jpg',
auth_token='4ac9ce371e67d30e-9dc412b2061eb971-fb15ec32858c9525'))
viber.set_webhook('https://bilinhual.herokuapp.com/')
class Viber(BaseMessenger):
"""
IM connector for Viber Bot API
"""
# region Interface
def __init__(self, token: str, **kwargs):
super().__init__(token, **kwargs)
self.bot = Api(
BotConfiguration(
auth_token=token,
name=kwargs.get('name'),
avatar=kwargs.get('avatar'),
))
def enable_webhook(self, url: str, **kwargs):
return self.bot.set_webhook(url=url)
def disable_webhook(self):
return self.bot.unset_webhook()
def get_account_info(self) -> Dict[str, Any]:
# data = {
# "status":0,
# "status_message":"ok",
# "id":"pa:75346594275468546724",
# "name":"account name",
# "uri":"accountUri",
# "icon":"http://example.com",
# "background":"http://example.com",
# "category":"category",
# "subcategory":"sub category",
# "location":{
# "lon":0.1,
# "lat":0.2
# },
# "country":"UK",
# "webhook":"https://my.site.com",
# "event_types":[
# "delivered",
# "seen"
# ],
# "subscribers_count":35,
# "members":[
# {
# "id":"01234567890A=",
# "name":"my name",
# "avatar":"http://example.com",
# "role":"admin"
# }
# ]
# }
try:
data = self.bot.get_account_info()
except Exception as err:
raise MessengerException(err)
return {
'id': data.get('id'),
'username': data.get('name'),
'uri': data.get('uri'), # check this
'info': data
}
def get_user_info(self, user_id: str, **kwargs) -> Dict[str, Any]:
# data = {
# "id":"01234567890A=",
# "name":"John McClane",
# "avatar":"http://avatar.example.com",
# "country":"UK",
# "language":"en",
# "primary_device_os":"android 7.1",
# "api_version":1,
# "viber_version":"6.5.0",
# "mcc":1,
# "mnc":1,
# "device_type":"iPhone9,4"
# }
try:
data = self.bot.get_user_details(user_id)
except Exception as err:
if 'failed with status: 12' in str(err):
raise RequestsLimitExceeded(err)
raise MessengerException(err)
return {
'id': data.get('id'),
'username': data.get('name'),
'avatar': data.get('avatar'),
'info': data,
}
def parse_message(self, request: HttpRequest) -> Message:
# Verify signature
sign = request.META.get('HTTP_X_VIBER_CONTENT_SIGNATURE')
data = json.loads(request.body)
if not self.bot.verify_signature(request.body, sign):
raise MessengerException(f'Viber message not verified; '
f'Data={data}; Sign={sign};')
return self._from_viber_message(self.bot.create_request(data))
def send_message(self, receiver: str,
messages: Union[Message, List[Message]]) -> List[str]:
if isinstance(messages, MessageList):
messages = messages.as_list()
elif isinstance(messages, Message):
messages = [messages]
vb_messages = []
for message in messages:
vb_messages.append(self._to_viber_message(message))
try:
return self.bot.send_messages(receiver, vb_messages)
except Exception as err:
if 'failed with status: 6, message: notSubscribed' in str(err):
raise NotSubscribed(err)
raise MessengerException(err)
def welcome_message(self, text: str) -> Union[str, Dict[str, Any], None]:
return {
"sender": {
"name": self.name,
"avatar": self.avatar_url
},
"type": "text",
"text": text
}
# endregion
# region Help methods
@staticmethod
def _from_viber_message(vb_request: ViberRequest) -> Message:
if isinstance(vb_request, vbr.ViberMessageRequest):
assert isinstance(vb_request.message, TypedMessage)
vb_message = vb_request.message
if isinstance(vb_message, vbm.TextMessage):
if 'btn-' in vb_message.text:
return Button(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
command=vb_message.text)
return Text(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
text=vb_message.text)
elif isinstance(vb_message, vbm.PictureMessage):
return Picture(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
file_url=vb_message.media)
elif isinstance(vb_message, vbm.VideoMessage):
return Video(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
file_url=vb_message.media,
file_size=vb_message.size)
elif isinstance(vb_message, vbm.FileMessage):
return File(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
file_url=vb_message.media,
file_size=vb_message.size)
elif isinstance(vb_message, vbm.RichMediaMessage):
return RichMedia(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
text=vb_message.alt_text,
rich_media=vb_message.rich_media)
elif isinstance(vb_message, vbm.ContactMessage):
return Contact(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
contact=vb_message.contact)
elif isinstance(vb_message, vbm.LocationMessage):
return Location(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
location=vb_message.location)
elif isinstance(vb_message, vbm.URLMessage):
return Url(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
url=vb_message.media)
elif isinstance(vb_message, vbm.StickerMessage):
return Sticker(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
file_id=vb_message.sticker_id)
return Text(id=vb_request.message_token,
user_id=vb_request.sender.id,
timestamp=vb_request.timestamp,
text=str(vb_message))
elif isinstance(vb_request, vbr.ViberConversationStartedRequest):
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.START,
user_name=vb_request.user.name,
context=vb_request.context)
elif isinstance(vb_request, vbr.ViberSubscribedRequest):
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.SUBSCRIBED,
user_name=vb_request.user.name)
elif isinstance(vb_request, vbr.ViberUnsubscribedRequest):
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.UNSUBSCRIBED)
elif isinstance(vb_request, vbr.ViberDeliveredRequest):
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.DELIVERED)
elif isinstance(vb_request, vbr.ViberSeenRequest):
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.SEEN)
elif isinstance(vb_request, vbr.ViberFailedRequest):
log.warning(f'Client failed receiving message; Error={vb_request}')
return Event(id=vb_request.message_token,
user_id=vb_request.user_id,
timestamp=vb_request.timestamp,
event_type=EType.START,
context=vb_request.desc)
elif vb_request.event_type == 'webhook':
return Event(timestamp=vb_request.timestamp)
log.warning(f'ViberRequest type={type(vb_request)}; '
f'Object={vb_request};')
return Text(timestamp=vb_request.timestamp, text=str(vb_request))
def _to_viber_message(self, message: Message) -> VbMessage:
kb = self._get_keyboard(message.buttons)
if isinstance(message, Text):
return vbm.TextMessage(text=message.text, keyboard=kb)
if isinstance(message, Sticker):
return vbm.StickerMessage(sticker_id=message.file_id, keyboard=kb)
elif isinstance(message, Picture):
return vbm.PictureMessage(media=message.file_url,
text=message.text,
keyboard=kb)
elif isinstance(message, Video):
return vbm.VideoMessage(media=message.file_url,
size=message.file_size,
text=message.text,
keyboard=kb)
elif isinstance(message, (File, Audio)):
return vbm.FileMessage(media=message.file_url,
size=message.file_size or 0,
file_name=message.file_name or '',
keyboard=kb)
elif isinstance(message, Contact):
contact = message.contact
return vbm.ContactMessage(contact=contact, keyboard=kb)
elif isinstance(message, Url):
return vbm.URLMessage(media=message.url, keyboard=kb)
elif isinstance(message, Location):
location = message.location
return vbm.LocationMessage(location=location, keyboard=kb)
elif isinstance(message, RichMedia):
rich_media = message.rich_media
return vbm.RichMediaMessage(rich_media=rich_media,
alt_text=message.text,
keyboard=kb)
@staticmethod
def _get_keyboard(buttons: List[Button]) -> Optional[Dict[str, Any]]:
# TODO do refactoring
if not buttons:
return None
vb_buttons = []
for button in buttons:
# assert isinstance(button, Button), f'{button=} {type(button)}'
vb_btn = {
'Columns':
2, # TODO: how is it storage in Model?
'Rows':
1,
'BgColor':
'#aaaaaa',
'ActionType':
'reply',
'ActionBody':
button.command,
'Text':
'<font color="{clr}"><b>{text}'
'</b></font>'.format(text=button.text, clr='#131313'),
'TextVAlign':
'middle',
'TextHAlign':
'center',
'TextOpacity':
60,
'TextSize':
'large',
'TextPaddings': [12, 8, 8, 20], # [up, left, right, bottom]
}
if hasattr(button, 'image'):
domain = Site.objects.get_current().domain
vb_btn.update({
'BgMedia': f'https://{domain}{button.image}',
'BgMediaScaleType': 'fill'
})
vb_buttons.append(vb_btn)
return {
'Type': 'keyboard',
'BgColor': '#ffffff',
'min_api_version': 6,
'Buttons': vb_buttons,
}
from viberbot import Api
from viberbot.api.bot_configuration import BotConfiguration
bot_configuration = BotConfiguration(
name='mgbot',
auth_token='46f8cbe1dee7d22a-2654d549e59d8703-5d9a149e324492c0')
viber = Api(bot_configuration)
viber.set_webhook('https://mgelios.pythonanywhere.com/viber/mgbot')
