Fix linting issues

This commit is contained in:
2022-06-08 19:56:16 +01:00
parent 8f5218c2bb
commit a2692fecc9
3 changed files with 21 additions and 22 deletions

View File

@@ -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

View File

@@ -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)

View File

@@ -1,4 +1,4 @@
from . import main from smsbot import main
if __name__ == '__main__': if __name__ == '__main__':
main() main()