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)