46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
import pika
|
|
|
|
import pickle
|
|
|
|
from functions import log, save_project
|
|
from classes import *
|
|
import colours
|
|
|
|
remaining_funds = 1000000
|
|
|
|
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost', 5672, '/', pika.PlainCredentials("user", "password")))
|
|
|
|
channel = connection.channel()
|
|
|
|
channel.queue_declare(queue='research_proposals')
|
|
|
|
def process_proposal(ch, method, properties, body):
|
|
global remaining_funds
|
|
project = pickle.loads(body)
|
|
if 200000 <= project.amount <= 500000 and project.amount <= remaining_funds:
|
|
remaining_funds -= project.amount
|
|
channel.basic_publish(exchange='',
|
|
routing_key=properties.reply_to,
|
|
properties=pika.BasicProperties(correlation_id=properties.correlation_id),
|
|
body="approved")
|
|
project.status = 1
|
|
save_project(project)
|
|
log(f"project {colours.INFO}{project.acronym} {colours.OK}approved{colours.RESET}, {colours.INFO}{remaining_funds}{colours.RESET} remaining")
|
|
else:
|
|
channel.basic_publish(exchange='',
|
|
routing_key=properties.reply_to,
|
|
properties=pika.BasicProperties(correlation_id=properties.correlation_id),
|
|
body="rejected")
|
|
project.status = 2
|
|
save_project(project)
|
|
log(f"project {colours.INFO}{project.acronym} {colours.ERROR}rejected{colours.RESET}, {colours.INFO}{remaining_funds}{colours.RESET} remaining")
|
|
|
|
channel.basic_consume(queue='research_proposals', on_message_callback=process_proposal, auto_ack=True)
|
|
|
|
try:
|
|
channel.start_consuming()
|
|
except KeyboardInterrupt:
|
|
channel.stop_consuming()
|
|
connection.close()
|
|
|