veri.py

import os
import json
import logging
from logging.config import dictConfig
import stomp
import time

from flask import Flask, request, Response, session
from werkzeug.exceptions import InternalServerError


dictConfig({
    'version': 1,
    'root': {
        'level': 'DEBUG'
    }
})


class VerificationQueueListener(stomp.ConnectionListener):
    def __init__(self, connection):
        self.connection = connection


    def on_error(self, frame):
        app.logger.error('received an err: %s', frame)


    def on_message(self, frame):
        msg = json.loads(frame.body)
        app.logger.debug('received a verification request for: %s', msg)

        accepted = do_verification(msg['title']) and do_verification(msg['content'])

        result = {
            'id': msg['id'],
            'status': 'accepted' if accepted else 'rejected'
        }

        self.connection.send(body=json.dumps(result), destination='verification-result.queue')
        app.logger.debug('done')


def connect_to_artemis():
    artemis_host = os.environ.get('HNOTES_ARTEMIS_HOST', 'localhost')
    artemis_port = os.environ.get('HNTOES_ARTEMIS_PORT', 61616)
    c = stomp.Connection([(artemis_host, artemis_port)], heartbeats=(4000, 4000), auto_content_length=False)
    c.set_listener('', VerificationQueueListener(c))

    c.connect('artemis', 'artemis', wait=True)
    c.subscribe(destination='verification.queue', id=1)

    return c


def do_verification(text: str) -> bool:
    blocked_words = [
        'banana'
    ]

    accepted = True
    for b in blocked_words:
        if b in text:
            accepted = False
            break

    return accepted


def maybe_error():
    ctr = session.get('counter', 0)
    ctr += 1
    session['counter'] = ctr
    if ctr % 5 != 0:
        raise InternalServerError('Encountered an internal error, pls try again later')


connected = False
logger = logging.getLogger(__name__)
while not connected:
    try:
        connect_to_artemis()
        connected = True
        logger.info("Connected to Artemis!")
    except stomp.exception.ConnectFailedException:
        logger.error("Couldn't connect to Artemis. Will retry in 5 seconds")
        time.sleep(5)

app = Flask(__name__)


@app.route('/')
def health_check():
    return { 'status': 'ok' }


@app.route('/verificator', methods=['POST'])
def verify_input():
    maybe_error()

    txt = str(request.data)
    accepted = do_verification(txt)
    result = {
        'status': 'accepted' if accepted else 'rejected',
        'length': len(txt)
    }

    app.logger.info('Input is: %s; result is: %s', input, str(result))

    return result


@app.route('/error')
def get_error():
    return Response('there seems to be an error', 500)