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: DKNG

Browse source
This commit is contained in:
Elijah McMorris 2025-05-24 03:29:29 -07:00
parent b0088a98a4
commit 1ad252cd00
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

@ -10,6 +10,7 @@ use crate::util::df::DF;
pub enum Ticker {
ARKW,
CRWV,
DKNG,
ETOR,
MKFG,
LUNR,
@ -30,6 +31,7 @@ impl Ticker {
match self {
Self::ARKW => Self::arkw(df),
Self::CRWV => Self::crwv(df),
Self::DKNG => Self::dkng(df),
Self::ETOR => Self::etor(df),
Self::MKFG => Self::mkfg(df),
Self::LUNR => Self::lunr(df),
@ -107,6 +109,24 @@ impl Ticker {
Ok(df.into())
}
fn dkng(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("AFTKINGS")))
.then(lit("DRAFTKINGS"))
.otherwise(col("company"))
.alias("company")])
.collect()
{
df = x;
}
Ok(df.into())
}
fn etor(df: DF) -> Result<DF, Error> {
let mut df = df.collect()?;
@ -278,6 +298,17 @@ mod tests {
&[Some("ETORO GROUP"), Some("ETORO GROUP")]
)?,
)]
#[case::dkng(
Ticker::DKNG,
defualt_df(
&[Some("DKNG")],
&[Some("AFTKINGS")],
)?,
defualt_df(
&[Some("DKNG")],
&[Some("DRAFTKINGS")]
)?,
)]
#[case::mkfg(
Ticker::MKFG,
defualt_df(