43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import pika
|
|
|
|
import pickle
|
|
import os
|
|
|
|
from classes import Project
|
|
|
|
if not os.path.exists("approved_projects"):
|
|
os.makedirs("approved_projects")
|
|
|
|
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost', 5672, '/', pika.PlainCredentials("user", "password")))
|
|
|
|
channel = connection.channel()
|
|
|
|
channel.queue_declare(queue='get_project')
|
|
channel.queue_declare(queue='save_project')
|
|
|
|
def get_project(ch, method, properties, body):
|
|
project_id = body.decode()
|
|
with open(os.path.join("approved_projects", f"{project_id}.txt"), "rb") as file:
|
|
project = file.read()
|
|
channel.basic_publish(exchange='',
|
|
routing_key=properties.reply_to,
|
|
properties=pika.BasicProperties(correlation_id=properties.correlation_id),
|
|
body=project)
|
|
print(f"got project {project_id}")
|
|
|
|
|
|
def save_project(ch, method, properties, body):
|
|
project_data = body.decode()
|
|
acronym, title, description, amount = project_data.split(',')
|
|
|
|
project = Project(acronym, title, description, int(amount))
|
|
filename = os.path.join("approved_projects", f"{acronym}.txt")
|
|
with open(filename, "wb+") as file:
|
|
pickle.dump(project, file)
|
|
|
|
|
|
channel.basic_consume(queue='get_project', on_message_callback=get_project, auto_ack=False)
|
|
channel.basic_consume(queue='save_project', on_message_callback=save_project, auto_ack=True)
|
|
|
|
channel.start_consuming()
|