72 lines
3.2 KiB
Python
72 lines
3.2 KiB
Python
import pika
|
|
|
|
import uuid
|
|
import pickle
|
|
|
|
from functions import *
|
|
from classes import *
|
|
|
|
class Researcher(object):
|
|
|
|
def __init__(self):
|
|
self.connection = pika.BlockingConnection(pika.ConnectionParameters('localhost', 5672, '/', pika.PlainCredentials("user", "password")))
|
|
self.channel = self.connection.channel()
|
|
result = self.channel.queue_declare(queue='', exclusive=True)
|
|
self.callback_queue = result.method.queue
|
|
self.channel.basic_consume(queue=self.callback_queue, on_message_callback=self.on_response, auto_ack=True)
|
|
self.response = None
|
|
self.corr_id = None
|
|
|
|
def on_response(self, ch, method, properties, body):
|
|
if self.corr_id == properties.correlation_id:
|
|
self.response = body
|
|
|
|
def call(self, proposal):
|
|
project = Project(proposal["acronym"], proposal["title"],
|
|
proposal["description"], int(proposal["amount"]))
|
|
save_project(project)
|
|
self.response = None
|
|
self.corr_id = str(uuid.uuid4())
|
|
self.channel.basic_publish(exchange='', routing_key="research_proposals",
|
|
properties=pika.BasicProperties(reply_to=self.callback_queue,
|
|
correlation_id=self.corr_id, ),
|
|
body=pickle.dumps(project))
|
|
self.connection.process_data_events(time_limit=None)
|
|
return str(self.response)
|
|
|
|
def withdraw(self, withdrawal):
|
|
self.channel.basic_publish(exchange='', routing_key='withdraw_funds',
|
|
body=f"{withdrawal['acronym']},{withdrawal['name']},{withdrawal['amount']}")
|
|
|
|
proposals = [
|
|
{"acronym": "PROJ1", "title": "Project 1", "description": "Description 1", "amount": "300000"},
|
|
{"acronym": "PROJ2", "title": "Project 2", "description": "Description 2", "amount": "400000"},
|
|
{"acronym": "PROJ3", "title": "Project 3", "description": "Description 3", "amount": "250000"},
|
|
{"acronym": "PROJ4", "title": "Project 4", "description": "Description 4", "amount": "100000"},
|
|
{"acronym": "PROJ5", "title": "Project 5", "description": "Description 5", "amount": "900000"},
|
|
{"acronym": "PROJ6", "title": "Project 6", "description": "Description 6", "amount": "420000"}
|
|
]
|
|
|
|
withdrawals = [
|
|
{"acronym": "PROJ1", "name": "Johnathan", "amount": 150000},
|
|
{"acronym": "PROJ2", "name": "Joseph", "amount": 200000},
|
|
{"acronym": "PROJ3", "name": "Jotaro", "amount": 100000},
|
|
{"acronym": "PROJ4", "name": "Josuke", "amount": 100000},
|
|
{"acronym": "PROJ5", "name": "Giorno", "amount": 250000},
|
|
{"acronym": "PROJ6", "name": "Jolyne", "amount": 430000},
|
|
{"acronym": "PROJ1", "name": "Johnathan", "amount": 100000},
|
|
{"acronym": "PROJ2", "name": "Joseph", "amount": 180000},
|
|
{"acronym": "PROJ3", "name": "Jotaro", "amount": 160000},
|
|
{"acronym": "PROJ4", "name": "Josuke", "amount": 150000},
|
|
{"acronym": "PROJ5", "name": "Giorno", "amount": 450000},
|
|
{"acronym": "PROJ6", "name": "Jolyne", "amount": 100000}
|
|
]
|
|
|
|
for proposal in proposals:
|
|
proposal_client = Researcher()
|
|
response = proposal_client.call(proposal)
|
|
|
|
for withdrawal in withdrawals:
|
|
client = Researcher()
|
|
client.withdraw(withdrawal)
|