py_mcai_worker_sdk 1.1.3

AMQP Worker to listen and provide trait to process message
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

mod ebu_ttml_live;
mod format_context;
mod frame;
#[cfg(test)]
mod tests;

pub use ebu_ttml_live::PyEbuTtmlLive;
pub use format_context::FormatContext;
pub use frame::Frame;
use mcai_worker_sdk::prelude::GenericFilter;
use pyo3::prelude::*;
use pyo3::types::{PyDict, PyList, PyString};
use std::collections::HashMap;

#[pyclass]
#[derive(Clone, Debug)]
pub(crate) struct PyGenericFilter {
  pub name: String,
  pub label: Option<String>,
  pub parameters: HashMap<String, String>,
}

impl From<PyGenericFilter> for GenericFilter {
  fn from(filter: PyGenericFilter) -> Self {
    GenericFilter {
      name: filter.name.clone(),
      label: filter.label.clone(),
      parameters: filter.parameters,
    }
  }
}

impl From<&PyDict> for PyGenericFilter {
  fn from(py_dict: &PyDict) -> Self {
    let name = py_dict
      .get_item("name")
      .unwrap()
      .downcast::<PyString>()
      .unwrap()
      .extract::<String>()
      .unwrap();

    let label = py_dict.get_item("label").map(|label| {
      label
        .downcast::<PyString>()
        .unwrap()
        .extract::<String>()
        .unwrap()
    });

    let parameters = py_dict
      .get_item("parameters")
      .unwrap()
      .downcast::<PyDict>()
      .unwrap()
      .extract::<HashMap<String, String>>()
      .unwrap();

    PyGenericFilter {
      name,
      label,
      parameters,
    }
  }
}

pub(crate) fn extract_generic_filters(py_list: &PyList) -> Vec<GenericFilter> {
  py_list
    .iter()
    .map(|item| item.downcast::<PyDict>().unwrap())
    .map(PyGenericFilter::from)
    .map(GenericFilter::from)
    .collect()
}