Skip to content

Commit

Permalink
add Proposal info
Browse files Browse the repository at this point in the history
  • Loading branch information
@Mr-Leshiy
Mr-Leshiy committed Nov 9, 2022
1 parent a5c855f commit 9c51c1f
Showing 1 changed file with 126 additions and 17 deletions.
143 changes: 126 additions & 17 deletions catalyst-toolbox/src/archive/vit_ss.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,14 @@ use csv::Writer;
use serde::{Deserialize, Serialize};
use std::{fs::File, path::Path};
use vit_servicing_station_lib::{
db::{self, load_db_connection_pool, queries::funds::query_fund_by_id},
utils::serde::{deserialize_unix_timestamp_from_rfc3339, serialize_unix_timestamp_as_rfc3339},
db::{
self, load_db_connection_pool,
queries::{funds::query_fund_by_id, proposals::query_all_proposals},
},
utils::serde::{
deserialize_string_as_bytes, deserialize_unix_timestamp_from_rfc3339, serialize_bin_as_str,
serialize_unix_timestamp_as_rfc3339,
},
v0::errors::HandleError,
};

Expand All @@ -23,41 +29,105 @@ pub enum Error {
}

#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
pub struct Fund {
struct Fund {
#[serde(default = "Default::default")]
pub id: i32,
id: i32,
#[serde(alias = "fundName")]
pub fund_name: String,
fund_name: String,
#[serde(alias = "fundGoal")]
pub fund_goal: String,
fund_goal: String,
#[serde(alias = "votingPowerThreshold")]
pub voting_power_threshold: i64,
voting_power_threshold: i64,
#[serde(alias = "fundStartTime")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
pub fund_start_time: i64,
fund_start_time: i64,
#[serde(alias = "fundEndTime")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
pub fund_end_time: i64,
fund_end_time: i64,
#[serde(alias = "nextFundStartTime")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
pub next_fund_start_time: i64,
next_fund_start_time: i64,
#[serde(alias = "registrationSnapshotTime")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
pub registration_snapshot_time: i64,
registration_snapshot_time: i64,
#[serde(alias = "nextRegistrationSnapshotTime")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
pub next_registration_snapshot_time: i64,
// #[serde(flatten)]
// pub stage_dates: FundStageDates,
next_registration_snapshot_time: i64,
#[serde(alias = "resultsUrl")]
pub results_url: String,
results_url: String,
#[serde(alias = "surveyUrl")]
pub survey_url: String,
survey_url: String,
}

#[derive(Serialize, Deserialize, PartialEq, Eq, Debug, Clone)]
struct Proposal {
#[serde(alias = "internalId")]
internal_id: i32,
#[serde(alias = "proposalId")]
proposal_id: String,
#[serde(alias = "categoryId", default = "Default::default")]
category_id: String,
#[serde(alias = "categoryName")]
category_name: String,
#[serde(alias = "categoryDescription", default = "Default::default")]
category_description: String,
#[serde(alias = "proposalTitle")]
proposal_title: String,
#[serde(alias = "proposalSummary")]
proposal_summary: String,
#[serde(alias = "proposalPublicKey")]
proposal_public_key: String,
#[serde(alias = "proposalFunds")]
proposal_funds: i64,
#[serde(alias = "proposalUrl")]
proposal_url: String,
#[serde(alias = "proposalFilesUrl")]
proposal_files_url: String,
#[serde(alias = "proposalImpactScore")]
proposal_impact_score: i64,
#[serde(alias = "proposerName")]
proposer_name: String,
#[serde(alias = "proposerEmail")]
proposer_email: String,
#[serde(alias = "proposerUrl")]
proposer_url: String,
#[serde(alias = "proposerRelevantExperience")]
proposer_relevant_experience: String,
#[serde(alias = "chainProposalId")]
#[serde(serialize_with = "serialize_bin_as_str")]
#[serde(deserialize_with = "deserialize_string_as_bytes")]
chain_proposal_id: Vec<u8>,
#[serde(alias = "chainProposalIndex")]
chain_proposal_index: i64,
#[serde(alias = "chainVoteplanId")]
chain_voteplan_id: String,
#[serde(alias = "chainVoteStartTime", default = "Default::default")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
chain_vote_start_time: i64,
#[serde(alias = "chainVoteEndTime", default = "Default::default")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
chain_vote_end_time: i64,
#[serde(alias = "chainCommitteeEndTime", default = "Default::default")]
#[serde(serialize_with = "serialize_unix_timestamp_as_rfc3339")]
#[serde(deserialize_with = "deserialize_unix_timestamp_from_rfc3339")]
chain_committee_end_time: i64,
#[serde(alias = "chainVoteplanPayload")]
chain_voteplan_payload: String,
#[serde(alias = "chainVoteEncryptionKey")]
chain_vote_encryption_key: String,
#[serde(alias = "fundId")]
fund_id: i32,
#[serde(alias = "challengeId")]
challenge_id: i32,
#[serde(alias = "reviewsCount")]
reviews_count: i32,
}

fn csv_writer(output_dir: &Path, name: String) -> Result<Writer<File>, Error> {
Expand Down Expand Up @@ -92,13 +162,52 @@ pub async fn generate_archive_files(
next_fund_start_time: fund.next_fund_start_time,
registration_snapshot_time: fund.registration_snapshot_time,
next_registration_snapshot_time: fund.next_registration_snapshot_time,
// stage_dates: fund.stage_dates,
results_url: fund.results_url,
survey_url: fund.survey_url,
})?;
chanllenges_writer.serialize(fund.challenges)?;
vote_plans_writer.serialize(fund.chain_vote_plans)?;
goals_writer.serialize(fund.goals)?;

// proposals info
let mut proposals_writer = csv_writer(output_dir, format!("proposals_{}.csv", fund_id))?;

let proposals = query_all_proposals(&db_pool).await?;

proposals_writer.serialize(
proposals
.into_iter()
.map(|el| Proposal {
internal_id: el.proposal.internal_id,
proposal_id: el.proposal.proposal_id,
category_id: el.proposal.proposal_category.category_id,
category_name: el.proposal.proposal_category.category_name,
category_description: el.proposal.proposal_category.category_description,
proposal_title: el.proposal.proposal_title,
proposal_summary: el.proposal.proposal_summary,
proposal_public_key: el.proposal.proposal_public_key,
proposal_funds: el.proposal.proposal_funds,
proposal_url: el.proposal.proposal_url,
proposal_files_url: el.proposal.proposal_files_url,
proposal_impact_score: el.proposal.proposal_impact_score,
proposer_name: el.proposal.proposer.proposer_name,
proposer_email: el.proposal.proposer.proposer_email,
proposer_url: el.proposal.proposer.proposer_url,
proposer_relevant_experience: el.proposal.proposer.proposer_relevant_experience,
chain_proposal_id: el.proposal.chain_proposal_id,
chain_proposal_index: el.proposal.chain_proposal_index,
chain_voteplan_id: el.proposal.chain_voteplan_id,
chain_vote_start_time: el.proposal.chain_vote_start_time,
chain_vote_end_time: el.proposal.chain_vote_end_time,
chain_committee_end_time: el.proposal.chain_committee_end_time,
chain_voteplan_payload: el.proposal.chain_voteplan_payload,
chain_vote_encryption_key: el.proposal.chain_vote_encryption_key,
fund_id: el.proposal.fund_id,
challenge_id: el.proposal.challenge_id,
reviews_count: el.proposal.reviews_count,
})
.collect::<Vec<Proposal>>(),
)?;

Ok(())
}

0 comments on commit 9c51c1f

Please sign in to comment.