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
use std::fmt::Display;
use crate::constants::{BT_WALLET_HOTKEY, BT_WALLET_NAME, BT_WALLET_PATH};
#[derive(Clone)]
pub struct WalletConfig {
pub name: String,
pub path: String,
pub hotkey: String,
}
impl WalletConfig {
/// Creates a new WalletConfig instance.
///
/// Arguments:
/// name (Option<String>): Optional wallet name. Defaults to "default" if not provided.
/// hotkey (Option<String>): Optional hotkey name. Defaults to "default" if not provided.
/// path (Option<String>): Optional wallet path. Defaults to "~/.bittensor/wallets/" if not provided.
/// Returns:
/// wallet_config (WalletConfig): A new WalletConfig instance.
pub fn new(name: Option<String>, hotkey: Option<String>, path: Option<String>) -> Self {
WalletConfig {
name: name.unwrap_or_else(|| BT_WALLET_NAME.to_string()),
hotkey: hotkey.unwrap_or_else(|| BT_WALLET_HOTKEY.to_string()),
path: path.unwrap_or_else(|| BT_WALLET_PATH.to_string()),
}
}
}
#[derive(Clone)]
pub struct Config {
pub wallet: WalletConfig,
}
impl Display for Config {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
write!(
f,
"Config(name: '{}', path: '{}', hotkey: '{}'",
self.wallet.name, self.wallet.path, self.wallet.hotkey
)
}
}
impl Config {
/// Creates a new Config instance.
///
/// Arguments:
/// name (Option<String>): Optional wallet name. Defaults to "default" if not provided.
/// hotkey (Option<String>): Optional hotkey name. Defaults to "default" if not provided.
/// path (Option<String>): Optional wallet path. Defaults to "~/.bittensor/wallets/" if not provided.
/// Returns:
/// config (Config): A new Config instance.
pub fn new(name: Option<String>, hotkey: Option<String>, path: Option<String>) -> Config {
Config {
wallet: WalletConfig::new(name, hotkey, path),
}
}
/// Returns the wallet name.
pub fn name(&self) -> String {
self.wallet.name.clone()
}
/// Returns the wallet path.
pub fn path(&self) -> String {
self.wallet.path.clone()
}
/// Returns the hotkey name.
pub fn hotkey(&self) -> String {
self.wallet.hotkey.clone()
}
}