mirror of
https://github.com/nikdoof/smsbot.git
synced 2025-12-17 19:59:28 +00:00
Fix linting issues
This commit is contained in:
@@ -27,9 +27,9 @@ console_scripts =
|
|||||||
|
|
||||||
[flake8]
|
[flake8]
|
||||||
format = wemake
|
format = wemake
|
||||||
ignore = E501,D,WPS226,WPS110, WPS210,WPS231,WPS202
|
ignore = D,E501,WPS432,WPS323,WPS2,S104,WPS412
|
||||||
max-line-length = 120
|
max-line-length = 120
|
||||||
exclude = setup.py
|
exclude = setup.py,env,build
|
||||||
|
|
||||||
[darglint]
|
[darglint]
|
||||||
docstring_style=sphinx
|
docstring_style=sphinx
|
||||||
|
|||||||
@@ -3,23 +3,22 @@ import os
|
|||||||
import sys
|
import sys
|
||||||
from functools import wraps
|
from functools import wraps
|
||||||
|
|
||||||
|
import pkg_resources
|
||||||
from flask import Flask, abort, current_app, request
|
from flask import Flask, abort, current_app, request
|
||||||
from telegram.ext import CommandHandler, Updater
|
from telegram.ext import CommandHandler, Updater
|
||||||
from twilio.request_validator import RequestValidator
|
from twilio.request_validator import RequestValidator
|
||||||
from waitress import serve
|
from waitress import serve
|
||||||
import pkg_resources
|
|
||||||
|
|
||||||
pkg_version = pkg_resources.require("smsbot")[0].version
|
pkg_version = pkg_resources.require('smsbot')[0].version
|
||||||
|
|
||||||
|
|
||||||
class TelegramSmsBot(object):
|
class TelegramSmsBot(object):
|
||||||
|
|
||||||
owner_id = None
|
def __init__(self, token, owner=None, subscribers=None):
|
||||||
subscriber_ids = []
|
|
||||||
|
|
||||||
def __init__(self, token):
|
|
||||||
self.logger = logging.getLogger(self.__class__.__name__)
|
self.logger = logging.getLogger(self.__class__.__name__)
|
||||||
self.bot_token = token
|
self.bot_token = token
|
||||||
|
self.owner_id = owner
|
||||||
|
self.subscriber_ids = subscribers or []
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
self.logger.info('Starting bot...')
|
self.logger.info('Starting bot...')
|
||||||
@@ -44,7 +43,7 @@ class TelegramSmsBot(object):
|
|||||||
|
|
||||||
def subscribe_handler(self, update, context):
|
def subscribe_handler(self, update, context):
|
||||||
self.logger.info('/subscribe command received')
|
self.logger.info('/subscribe command received')
|
||||||
if not update.message.chat['id'] in self.subscriber_ids:
|
if update.message.chat['id'] not in self.subscriber_ids:
|
||||||
self.logger.info('{0} subscribed'.format(update.message.chat['username']))
|
self.logger.info('{0} subscribed'.format(update.message.chat['username']))
|
||||||
self.subscriber_ids.append(update.message.chat['id'])
|
self.subscriber_ids.append(update.message.chat['id'])
|
||||||
self.send_owner('{0} has subscribed'.format(update.message.chat['username']))
|
self.send_owner('{0} has subscribed'.format(update.message.chat['username']))
|
||||||
@@ -85,16 +84,16 @@ class TelegramSmsBot(object):
|
|||||||
self.subscriber_ids.append(chat_id)
|
self.subscriber_ids.append(chat_id)
|
||||||
|
|
||||||
|
|
||||||
def validate_twilio_request(f):
|
def validate_twilio_request(func):
|
||||||
"""Validates that incoming requests genuinely originated from Twilio"""
|
"""Validates that incoming requests genuinely originated from Twilio"""
|
||||||
@wraps(f)
|
@wraps(func)
|
||||||
def decorated_function(*args, **kwargs):
|
def decorated_function(*args, **kwargs): # noqa: WPS430
|
||||||
# Create an instance of the RequestValidator class
|
# Create an instance of the RequestValidator class
|
||||||
twilio_token = os.environ.get('SMSBOT_TWILIO_AUTH_TOKEN')
|
twilio_token = os.environ.get('SMSBOT_TWILIO_AUTH_TOKEN')
|
||||||
|
|
||||||
if not twilio_token:
|
if not twilio_token:
|
||||||
current_app.logger.warning('Twilio request validation skipped due to SMSBOT_TWILIO_AUTH_TOKEN missing')
|
current_app.logger.warning('Twilio request validation skipped due to SMSBOT_TWILIO_AUTH_TOKEN missing')
|
||||||
return f(*args, **kwargs)
|
return func(*args, **kwargs)
|
||||||
|
|
||||||
validator = RequestValidator(twilio_token)
|
validator = RequestValidator(twilio_token)
|
||||||
|
|
||||||
@@ -103,14 +102,14 @@ def validate_twilio_request(f):
|
|||||||
request_valid = validator.validate(
|
request_valid = validator.validate(
|
||||||
request.url,
|
request.url,
|
||||||
request.form,
|
request.form,
|
||||||
request.headers.get('X-TWILIO-SIGNATURE', ''))
|
request.headers.get('X-TWILIO-SIGNATURE', ''),
|
||||||
|
)
|
||||||
|
|
||||||
# Continue processing the request if it's valid, return a 403 error if
|
# Continue processing the request if it's valid, return a 403 error if
|
||||||
# it's not
|
# it's not
|
||||||
if request_valid or current_app.debug:
|
if request_valid or current_app.debug:
|
||||||
return f(*args, **kwargs)
|
return func(*args, **kwargs)
|
||||||
else:
|
return abort(403)
|
||||||
return abort(403)
|
|
||||||
return decorated_function
|
return decorated_function
|
||||||
|
|
||||||
|
|
||||||
@@ -123,7 +122,7 @@ class TwilioWebhookHandler(object):
|
|||||||
self.app.add_url_rule('/message', 'message', self.message, methods=['POST'])
|
self.app.add_url_rule('/message', 'message', self.message, methods=['POST'])
|
||||||
self.app.add_url_rule('/call', 'call', self.call, methods=['POST'])
|
self.app.add_url_rule('/call', 'call', self.call, methods=['POST'])
|
||||||
|
|
||||||
def set_bot(self, bot):
|
def set_bot(self, bot): # noqa: WPS615
|
||||||
self.bot = bot
|
self.bot = bot
|
||||||
|
|
||||||
def index(self):
|
def index(self):
|
||||||
@@ -135,7 +134,7 @@ class TwilioWebhookHandler(object):
|
|||||||
@validate_twilio_request
|
@validate_twilio_request
|
||||||
def message(self):
|
def message(self):
|
||||||
|
|
||||||
message = "From: {From}\n\n{Body}".format(**request.values.to_dict())
|
message = 'From: {From}\n\n{Body}'.format(**request.values.to_dict())
|
||||||
|
|
||||||
current_app.logger.info('Received SMS from {From}: {Body}'.format(**request.values.to_dict()))
|
current_app.logger.info('Received SMS from {From}: {Body}'.format(**request.values.to_dict()))
|
||||||
self.bot.send_subscribers(message)
|
self.bot.send_subscribers(message)
|
||||||
@@ -182,4 +181,4 @@ def main():
|
|||||||
# Start webhooks
|
# Start webhooks
|
||||||
webhooks = TwilioWebhookHandler()
|
webhooks = TwilioWebhookHandler()
|
||||||
webhooks.set_bot(telegram_bot)
|
webhooks.set_bot(telegram_bot)
|
||||||
webhooks.serve(host=listen_host, port=listen_port)
|
webhooks.serve(host=listen_host, port=listen_port)
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
from . import main
|
from smsbot import main
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
main()
|
main()
|
||||||
|
|||||||
Reference in New Issue
Block a user