nautilus_indicators/python/volatility/
dc.rs1use nautilus_model::data::Bar;
17use pyo3::prelude::*;
18
19use crate::{indicator::Indicator, volatility::dc::DonchianChannel};
20
21#[pymethods]
22#[pyo3_stub_gen::derive::gen_stub_pymethods]
23impl DonchianChannel {
24 #[new]
26 #[must_use]
27 pub fn py_new(period: usize) -> Self {
28 Self::new(period)
29 }
30
31 fn __repr__(&self) -> String {
32 format!("DonchianChannel({})", self.period)
33 }
34
35 #[getter]
36 #[pyo3(name = "name")]
37 fn py_name(&self) -> String {
38 self.name()
39 }
40
41 #[getter]
42 #[pyo3(name = "period")]
43 const fn py_period(&self) -> usize {
44 self.period
45 }
46
47 #[getter]
48 #[pyo3(name = "has_inputs")]
49 fn py_has_inputs(&self) -> bool {
50 self.has_inputs()
51 }
52
53 #[getter]
54 #[pyo3(name = "upper")]
55 const fn py_upper(&self) -> f64 {
56 self.upper
57 }
58
59 #[getter]
60 #[pyo3(name = "middle")]
61 const fn py_middle(&self) -> f64 {
62 self.middle
63 }
64
65 #[getter]
66 #[pyo3(name = "lower")]
67 const fn py_lower(&self) -> f64 {
68 self.lower
69 }
70
71 #[getter]
72 #[pyo3(name = "initialized")]
73 const fn py_initialized(&self) -> bool {
74 self.initialized
75 }
76
77 #[pyo3(name = "update_raw")]
78 fn py_update_raw(&mut self, high: f64, low: f64) {
79 self.update_raw(high, low);
80 }
81
82 #[pyo3(name = "handle_bar")]
83 fn py_handle_bar(&mut self, bar: &Bar) {
84 self.handle_bar(bar);
85 }
86
87 #[pyo3(name = "reset")]
88 fn py_reset(&mut self) {
89 self.reset();
90 }
91}