-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathapp.py
79 lines (63 loc) · 2.33 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from flask import Flask, request, jsonify
from flask_sqlalchemy import SQLAlchemy
import pika
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///tasks.db'
db = SQLAlchemy(app)
class Task(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(80), nullable=False)
description = db.Column(db.String(120))
status = db.Column(db.String(20), nullable=False)
def to_dict(self):
return {
'id': self.id,
'title': self.title,
'description': self.description,
'status': self.status
}
def send_message_to_queue(task):
connection = pika.BlockingConnection(pika.ConnectionParameters(host='localhost'))
channel = connection.channel()
channel.queue_declare(queue='task_queue')
if isinstance(task, Task):
body = str(task.to_dict())
else:
body = str(task)
channel.basic_publish(exchange='', routing_key='task_queue', body=body)
connection.close()
@app.route('/tasks', methods=['POST'])
def create_task():
data = request.json
task = Task(title=data['title'], description=data['description'], status=data['status'])
db.session.add(task)
db.session.commit()
send_message_to_queue(task)
return jsonify(task.to_dict()), 201
@app.route('/tasks', methods=['GET'])
def get_tasks():
tasks = Task.query.all()
return jsonify([task.to_dict() for task in tasks])
@app.route('/tasks/<int:task_id>', methods=['PUT'])
def update_task(task_id):
task = db.session.get(Task, task_id)
if task is None:
return jsonify({'message': 'Tarefa não encontrada'}), 404
data = request.json
task.title = data.get('title', task.title)
task.description = data.get('description', task.description)
task.status = data.get('status', task.status)
db.session.commit()
send_message_to_queue(task)
return jsonify(task.to_dict()), 200
@app.route('/tasks/<int:task_id>', methods=['DELETE'])
def delete_task(task_id):
task = Task.query.get(task_id)
if task is None:
return jsonify({'message': 'Tarefa não encontrada'}), 404
db.session.delete(task)
db.session.commit()
send_message_to_queue({'id': task_id, 'action': 'delete'})
return jsonify({'message': 'Tarefa deletada com sucesso'}), 200
if __name__ == '__main__':
app.run(debug=True)