NexVeridian/ark-invest-api-rust-data
1
0
Fork 0
mirror of https://github.com/NexVeridian/ark-invest-api-rust-data.git synced 2025-09-02 01:49:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: CRCL

Browse source
This commit is contained in:
Elijah McMorris 2025-06-10 16:06:27 -07:00
parent e297be713e
commit 86edeab90e
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
1 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/util/format/ticker.rs
View file

@ -9,6 +9,7 @@ use crate::util::df::DF;
#[derive(Debug, strum_macros::Display, EnumIter, Clone, Copy, PartialEq, Eq)]
pub enum Ticker {
ARKW,
CRLC,
CRWV,
DKNG,
ETOR,
@ -31,6 +32,7 @@ impl Ticker {
pub fn format(&self, df: DF) -> Result<DF, Error> {
match self {
Self::ARKW => Self::arkw(df),
Self::CRLC => Self::crlc(df),
Self::CRWV => Self::crwv(df),
Self::DKNG => Self::dkng(df),
Self::ETOR => Self::etor(df),
@ -93,6 +95,24 @@ impl Ticker {
Ok(df.into())
}
fn crlc(df: DF) -> Result<DF, Error> {
let mut df = df.collect()?;
if let Ok(x) = df
.clone()
.lazy()
.with_columns(vec![when(col("company").eq(lit("CIRCLE INTERNET GROUP")))
.then(lit("CRLC"))
.otherwise(col("ticker"))
.alias("ticker")])
.collect()
{
df = x;
}
Ok(df.into())
}
fn crwv(df: DF) -> Result<DF, Error> {
let mut df = df.collect()?;
@ -302,6 +322,17 @@ mod tests {
&[Some("ARKB"), Some("ARKB"), Some("ARKB"), Some("ARKB")],
)?,
)]
#[case::crlc(
Ticker::CRLC,
defualt_df(
&[Some("CRLC"), None::<&str>],
&[Some("CIRCLE INTERNET GROUP"), Some("CIRCLE INTERNET GROUP")],
)?,
defualt_df(
&[Some("CRLC"), Some("CRLC")],
&[Some("CIRCLE INTERNET GROUP"), Some("CIRCLE INTERNET GROUP")]
)?,
)]
#[case::crwv(
Ticker::CRWV,
defualt_df(