prb_1/API/main.py

import datetime
import random

import jwt

from data.connect import init_db, connect, User, Document, DocumentCategory, Comment
from flask import Flask, Response, request, jsonify

app = Flask(__name__)
app.config['SECRET_KEY'] = 'jyeraghueykgaeyugheaughkawefy'


@app.route('/api/v1/SignIn', methods=['POST'])
def login():
    try:
        data = request.json

        with connect() as session:
            user = session.query(User).filter(User.email == data['name'], User.password == data['password']).first()

        if user is None:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1304
            }, 403

        token = jwt.encode({
            'sub': user.email,
            'exp': datetime.datetime.utcnow() + datetime.timedelta(minutes=180)
        }, app.config['SECRET_KEY'], algorithm='HS256')
        return jsonify({'token': token})
    except Exception:
        return {
            'timestamp': int(datetime.datetime.now().timestamp()),
            'message': 'Неправильно сформированный запрос',
            'errorCode': 1400
        }, 400


@app.route('/api/v1/Documents', methods=['GET'])
def get_documents():
    try:
        token = request.headers.get('Authorization')

        if not token:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1304
            }, 403

        try:
            data = jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
        except jwt.ExpiredSignatureError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 401
        except jwt.InvalidTokenError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 403

        resp = []

        with connect() as session:
            documents = session.query(Document).all()

            for document in documents:
                # category = session.query(DocumentCategory).filter(DocumentCategory.id == document.category_id).first()
                # comments = session.query(Comment).filter(Comment.document_id == document.id).all()

                category = document.category
                comments = document.comments

                resp.append({
                    'id': document.id,
                    'title': document.title,
                    'date_created': document.date_created,
                    'date_updated': document.date_updated,
                    'category': document.category.title,
                    'has_comments': True if len(document.comments) else False
                })

        return resp

    except Exception:
        return {
            'timestamp': int(datetime.datetime.now().timestamp()),
            'message': 'Неправильно сформированный запрос',
            'errorCode': 1400
        }, 400


@app.route('/api/v1/Document/<int:documentId>/Comments', methods=['GET'])
def get_comments(documentId):
    try:
        token = request.headers.get('Authorization')

        if not token:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1304
            }, 403

        try:
            data = jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
        except jwt.ExpiredSignatureError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 401
        except jwt.InvalidTokenError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 403

        resp = []

        with connect() as session:
            comments = session.query(Comment).filter(Comment.document_id == documentId).all()

            if not comments:
                return {
                    'timestamp': int(datetime.datetime.now().timestamp()),
                    'message': 'Документ не найден',
                    'errorCode': 1404
                }, 404

            for comment in comments:
                resp.append({
                    'id': comment.id,
                    'document_id': comment.document_id,
                    'text': comment.text,
                    'date_created': comment.date_created,
                    'date_updated': comment.date_updated,
                    'author': {
                        'name': f'{comment.user.last_name} {comment.user.first_name}',
                        'position': comment.user.post.title,
                    }
                })
        return resp

    except Exception:
        return {
            'timestamp': int(datetime.datetime.now().timestamp()),
            'message': 'Неправильно сформированный запрос',
            'errorCode': 1400
        }, 400


@app.route('/api/v1/Document/<int:documentId>/Comments', methods=['POST'])
def create_comment(documentId):
    try:
        token = request.headers.get('Authorization')

        if not token:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1304
            }, 403

        try:
            data = jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
        except jwt.ExpiredSignatureError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 401
        except jwt.InvalidTokenError:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильные авторизационные даныне',
                'errorCode': 1401
            }, 403

        resp = []

        comment_data = request.json

        if not comment_data:
            return {
                'timestamp': int(datetime.datetime.now().timestamp()),
                'message': 'Неправильно сформированный запрос',
                'errorCode': 1400
            }, 400

        with connect() as session:
            document = session.query(Document).filter(Document.id == documentId).first()

            if document is None:
                return {
                    'timestamp': int(datetime.datetime.now().timestamp()),
                    'message': 'Документ не найден',
                    'errorCode': 1404
                }, 404

            user = session.query(User).filter(User.id == comment_data['user_id']).first()

            if user is None:
                return {
                    'timestamp': int(datetime.datetime.now().timestamp()),
                    'message': 'Пользователь не найден',
                    'errorCode': 1404
                }, 404

            comment = Comment(
                text=comment_data['text'],
                document_id=document.id,
                date_created=datetime.datetime.now(),
                date_updated=datetime.datetime.now(),
                user_id=user.id,
            )
            session.add(comment)
            session.commit()

        return Response(status=200)

    except Exception:
        return {
            'timestamp': int(datetime.datetime.now().timestamp()),
            'message': 'Неправильно сформированный запрос',
            'errorCode': 1400
        }, 400


@app.route('/protected', methods=['GET'])
def protected():
    token = request.headers.get('Authorization')
    if not token:
        return jsonify({'message': 'Token is missing!'}), 403

    try:
        data = jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
        return jsonify({'message': f'Welcome {data["sub"]}!'})
    except jwt.ExpiredSignatureError:
        return jsonify({'message': 'Token has expired!'}), 401
    except jwt.InvalidTokenError:
        return jsonify({'message': 'Invalid token!'}), 401


def main():
    init_db()
    app.run()


if __name__ == '__main__':
    main()