Как отловить исключение во Flask из sqlalchemy @validate decorator - PullRequest
0 голосов
/ 30 января 2019

У меня интересная проблема.По сути, у меня есть это API-приложение Flask restful, где у меня есть HTML-страница, на которой вошедшие в систему пользователи могут создать пользователя API.Это обрабатывается Apiusers класса модели SQLAlchemy, в котором я использую декоратор validates, чтобы убедиться, что имя пользователя и адрес электронной почты соответствуют правилам.Все это прекрасно работает, когда пользователь вводит неверную информацию, он выдает исключение, но я не могу поймать его во Flask, несмотря ни на что.Я хочу поймать его и отправить сообщение о нем обратно пользователю, но вместо этого он выдает встроенную ошибку. AssertionError, которая в результате выдает страницу с 500 внутренними ошибками сервера, когда не находится в режиме отладки.

Код очень похож наthis: https://nunie123.github.io/sqlalchemy-validation.html

Пробовал с использованием различных типов исключений - ValueError, определяя пользовательский класс исключений и, наконец, перехватывая все ошибки с помощью: кроме: но исключение всегда заканчивается необработанным.Я также столкнулся с несколько похожей проблемой, как отмечалось здесь: https://github.com/marshmallow-code/webargs/issues/122, где есть упоминание о том, что флеш-отдых перехватывает обработку всех исключений HTTP, но это с 2016 года. В любом случае мне удавалось поймать все исключения уникальных ограничений SQLAlchemy.хорошо в том же месте, прежде чем использовать validates, так что я не думаю, что это проблема.

models.py:

from sqlalchemy.orm import validates    

class Apiusers(db.Model):
    __tablename__ = 'apiusers'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(64), index=True, unique=True)
    email = db.Column(db.String(120), index=True)
    password_hash = db.Column(db.String(128))


    def __repr__(self):
        return self.username

    @validates('username')
    def validate_username(self, key, username):
      if not username:
        raise AssertError('No username provided')
      if Apiusers.query.filter(Apiusers.username == username).first():
        raise AssertError('Username is already in use')

      if len(username) < 5 or len(username) > 20:
        raise AssertError('Username must be between 5 and 20 characters')

      return username

    @validates('email')
    def validate_email(self, key, email):
      if not email:
        raise AssertError('No email provided')
      if not re.match("[^@]+@[^@]+\.[^@]+", email):
        raise AssertError('Provided email is not an email address')
      return email


views.py:

@app.route('/register', methods=['GET', 'POST'])
def register():
  if request.method == 'POST':
    _user = request.form['username']
    _email = request.form['emails']
    record = Apiusers(username=_user, email=_email)
    try:
      db.session.add(record)
      db.session.commit()
    except AssertError as e:
      flash('{}'.format(str(e)), 'danger')
      return render_template("register.html")
    return render_template("regresult.html")

Вот также трассировка стека при попытке добавить пустое имя пользователя:

Traceback (most recent call last):
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 2309, in 
__call__
return self.wsgi_app(environ, start_response)
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 2295, in 
wsgi_app
response = self.handle_exception(e)
File "/usr/local/lib/python3.6/site-packages/flask_restful/__init__.py", line 269, in error_router
return original_handler(e)
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 1741, in handle_exception
reraise(exc_type, exc_value, tb)
File "/usr/local/lib/python3.6/site-packages/flask/_compat.py", line 35, in reraise
raise value
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 2292, in wsgi_app
response = self.full_dispatch_request()
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 1815, in full_dispatch_request
rv = self.handle_user_exception(e)
File "/usr/local/lib/python3.6/site-packages/flask_restful/__init__.py", line 269, in error_router
return original_handler(e)
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 1718, in handle_user_exception
reraise(exc_type, exc_value, tb)
File "/usr/local/lib/python3.6/site-packages/flask/_compat.py", line 35, in reraise
raise value
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 1813, in full_dispatch_request
rv = self.dispatch_request()
File "/usr/local/lib/python3.6/site-packages/flask/app.py", line 1799, in dispatch_request
return self.view_functions[rule.endpoint](**req.view_args)
File "/opt/app/api2/app.py", line 189, in regresult
record = Apiusers(username=_user, email=_email)
File "<string>", line 4, in __init__

File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/state.py", line 428, in _initialize_instance
manager.dispatch.init_failure(self, args, kwargs)
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/util/langhelpers.py", line 67, in __exit__
compat.reraise(exc_type, exc_value, exc_tb)
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/util/compat.py", line 277, in reraise
raise value
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/state.py", line 425, in _initialize_instance
return manager.original_init(*mixed[1:], **kwargs)
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/ext/declarative/base.py", line 801, in _declarative_constructor
setattr(self, k, kwargs[k])
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/attributes.py", line 263, in __set__
instance_state(instance), instance_dict(instance), value, None
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/attributes.py", line 810, in set
state, dict_, value, old, initiator
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/attributes.py", line 818, in fire_replace_event
state, value, previous, initiator or self._replace_token
File "/usr/local/lib/python3.6/site-packages/sqlalchemy/orm/util.py", line 168, in set_
return validator(state.obj(), key, value)
File "/opt/app/api2/app.py", line 95, in validate_name
assert value != '', "Invalid username"
AssertionError: Invalid username

Ответы [ 2 ]

0 голосов
/ 30 января 2019

На самом деле просто разобрался с ответом.После перемещения строки кода, которая вызывает модель Apiusers в блоке try - кроме всех, все исключения перехватываются и обрабатываются все нормально :-) Ранее они заканчивались тем, что не обрабатывались, так как эта строка находилась вне блока.Вот что происходит, когда ты пропускаешь свой послеобеденный кофе.

try:
  record = Apiusers(username=_user, email=_email)
  db.session.add(record)
  db.session.commit()
0 голосов
/ 30 января 2019

Необработанное утверждение приведет к завершению программы.Смотрите здесь .Попробуйте вместо этого что-то вроде:

if not email:
    flash('Your error message')
    return redirect(request.url)

if not re.match("[^@]+@[^@]+\.[^@]+", email):
    flash('Your error message')
    return redirect(request.url)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...