FileResource.py

from flask_restful import reqparse, Resource
from flask import request
from models import File
from serializers import file_schema


class FileResource(Resource):

  def __init__(self):

    self.parser = reqparse.RequestParser()

    self.parser.add_argument(
      'document',
      required=True,
      location='json',
      help='Cade o arquivo?'
    )

    self.parser.add_argument(
      'user_id',
      required=True, 
      location='json',
      help='Id do usuario responsavel do documento nao pode ser vazio'
    )

    self.parser.add_argument(
      'extension',
      required=True, 
      location='json',
      help='Extensao nao pode ser vazio'
    )

  def post(self):
    data = self.parser.parse_args()

    new_file = File(
      document=data['document'],
      user_id=data['user_id'],
      extension=data['extension']
    )

    try:
      new_file.save()
      return {'message': 'Arquivo salvo com sucesso'}, 200
    except Exception as err:
      print(err)
      return {'message': 'Algo esta incorreto'}, 500

  def get(self):
    if 'id' in requests.args:
      doc = File.find_file_by_id(request.args['id'])

      if doc:
        return file_schema.jsonify(doc)
      else: {'message': 'Arquivo nao encontrado'}, 404
    else:
      return File.retrieve_all_files()

  def put(self):
    data = self.parser.parse_args()

    doc = File.get_file_by_id(data['id'])

    if doc:
      doc.document = data['document']
      doc.user_id = data['user_id']
      doc.extension = data['extension']

      try:
        doc.save()
        return {'message': 'Arquivo atualizado com sucesso'}, 200

      except Exception as err:
        print(err)
        return {'message': 'Algo esta incorreto'}, 500
    else:
      return {'message': 'Arquivo nao encontrado'}, 404

  def delete(self):
    if 'id' in requests.args:
      try:
        File.remove(requests.args['id'])
        return {'message': 'Arquivo deletado com sucesso'}, 200
      except Exception as err:
        print(err)
        return {'message': 'Algo esta incorreto'}, 500
    else:
      return {'message': 'BAD REQUEST'}, 400