R/analyzer_orientacoes.R

In parlametria/perfil-parlamentarR: Perfil Parlamentar

library(tidyverse)
library(here)

#' @title Processa orientação de proposições votadas em plenário para um determinado ano
#' @description Recupera informação das orientações dos partidos para um determinado ano
#' @param ano Ano para ocorrência das votações em plenário
#' @param url Link para dados das proposições selecionadas para captura das votações em plenário
#' Se url é diferente de NULL, então considerará lista de proposições presentes nos dados disponíveis através da URL
#' @return Dataframe com informações das orientações
#' @examples
#' orientacao <- process_orientacao_por_ano_camara(2019)
#' @export
process_orientacao_por_ano_camara <- function(ano = 2019) {
  proposicoes_votadas <- fetch_proposicoes_votadas_camara(ano)
  
  proposicoes <- proposicoes_votadas %>%
    filter(lubridate::year(data_votacao) == ano) %>% 
    distinct(id)
  
  orientacao <- tibble(id_proposicao = proposicoes$id) %>%
    mutate(dados = map(id_proposicao,
                       fetch_orientacoes_proposicao_camara)) %>%
    unnest(dados) %>%
    distinct()
  
  return(orientacao)
}

#' @title Processa orientação de proposições (selecionadas pela equipe VA) votadas em plenário para um conjunto de anos
#' @description Recupera informação das orientações dos partidos para um conjunto de anos para as
#' proposicoes selecionadas pela equipe VA.
#' @param anos Vector com lista de anos
#' @return Lista contendo orientações
#' @examples
#' orientacao <- process_orientacao_anos_camara(2019)
#' @export
process_orientacao_anos_camara <-
  function(anos = c(2019, 2020, 2021)) {
    library(tidyverse)
    library(here)
    
    orientacao <- tibble(ano = anos) %>%
      mutate(dados = map(ano,
                         process_orientacao_por_ano_camara)) %>%
      unnest(dados) %>%
      distinct() %>%
      mutate(casa = "camara")
    
    return(orientacao)
}


#' @title Processa orientação dos votos de plenário para um conjunto de votações
#' @description Recupera informação das orientações dos partidos para um conjunto de votações
#' @param votos_datapath Caminho para o dataframe contendo dados de votações
#' @return Lista contendo orientações
#' @examples
#' votos <- processa_votos_senado()
#' orientacao <- process_orientacao_votos_senado()
#' @export
process_orientacao_votos_senado <- function(votos) {
    library(tidyverse)
  
    orientacoes_governo <- define_orientacao_governo(votos)
    
    partidos <- votos %>%
      select(partido) %>%
      distinct() %>%
      filter(!partido %in% c("SPARTIDO", "S/Partido"))
    
    orientacoes_partido <-
      purrr::map_df(partidos$partido,
                    ~ calcula_voto_maioria_absoluta(votos, .x)) %>% 
      rename(partido_bloco = partido)
    
    orientacoes <- orientacoes_governo %>%
      bind_rows(orientacoes_partido) %>%
      distinct(id_votacao, partido_bloco, .keep_all = TRUE) %>%
      select(id_votacao, partido_bloco, voto)
    
    return(orientacoes)
    
  }

#' @title Define a orientação de um partido para um conjunto de votos
#' @description Recupera informação das orientações dos partidos para um conjunto de votos utilizando a maioria absoluta.
#' No caso de empate entre os votos, o voto será o do líder do partido.
#' @param votos Dataframe contendo dados de votos
#' @param sigla_partido Sigla do partido a ter orientações recuperadas
#' @return Dataframe contendo informações de orientações de um partido
get_voto_lider <- function(lideres, votos, id_votacao) {
  library(tidyverse)
  
  if (nrow(lideres) > 0) {
    voto <- (lideres$id %>%
               purrr::map(function(x) {
                 data <- votos %>%
                   filter(id_parlamentar == x &
                            id_votacao_votos %in% id_votacao) %>%
                   pull(voto)
               }))[[1]]
  } else {
    voto <- (
      votos %>%
        filter(id_votacao_votos %in% id_votacao) %>%
        group_by(voto) %>%
        mutate(maioria = n(),
               voto_final = which.max(maioria)) %>%
        arrange(desc(voto_final)) %>%
        pull(voto)
    )[[1]]
  }
  
  if (length(voto) == 0) {
    return(as.numeric(NA))
  } else{
    return(as.numeric(voto))
  }
}

#' @title Define a orientação de um partido para um conjunto de votos
#' @description Recupera informação das orientações dos partidos para um conjunto de votos utilizando a maioria absoluta.
#' No caso de empate entre os votos, o voto será o do líder do partido.
#' @param votos Dataframe contendo dados de votos
#' @param sigla_partido Sigla do partido a ter orientações recuperadas
#' @return Dataframe contendo informações de orientações de um partido
#' @examples
#' orientacao <- calcula_voto_maioria_absoluta(votos, "PSL")
calcula_voto_maioria_absoluta <- function(votos, sigla_partido) {
  library(tidyverse)
  
  print(paste0("Calculando orientação do partido ", sigla_partido))
  
  orientacoes <- votos %>%
    group_by(id_votacao, partido, voto) %>%
    filter(partido == sigla_partido) %>%
    count() %>%
    ungroup() %>%
    group_by(id_votacao) %>%
    filter(n == max(n)) %>%
    mutate(empate = if_else(n() > 1, 1, 0)) %>%
    distinct() %>%
    select(-n)
  
  lideres <- .fetch_liderancas_senado() %>%
    filter(bloco_partido == sigla_partido)
  
  votos <- votos %>%
    rename(id_votacao_votos = id_votacao)
  
  if (nrow(orientacoes %>% filter(empate == 1)) > 0) {
    orientacoes <- orientacoes %>%
      mutate(voto = if_else(
        empate == 1,
        get_voto_lider(lideres, votos, id_votacao),
        as.numeric(voto)
      )) %>%
      unique()
  }
  
  orientacoes <- orientacoes %>%
    select(-empate) %>%
    ungroup()
  
  return(orientacoes)
}

#' @title Define a orientação do Governo
#' @description Recupera informação das orientações do Governo com base nos votos do Líder
#' @param votos Dataframe contendo dados de votos
#' @return Dataframe contendo informações de orientações do Governo
#' @examples
#' orientacao <- define_orientacao_governo(votos)
define_orientacao_governo <- function(votos) {
  library(tidyverse)
  
  lideres <- .fetch_liderancas_senado() %>%
    filter(bloco_partido == "Governo", stringr::str_detect(cargo, "^Líder"))
  
  orientacoes <- votos %>%
    filter(id_parlamentar %in% lideres$id) %>%
    mutate(partido_bloco = "Governo") %>%
    select(id_votacao,
           partido_bloco,
           voto)
  
  return(orientacoes)
  
}