class TestInfo(unittest.TestCase):
def setUp(self):
self.registry = CollectorRegistry()
self.info = Info('i', 'help', registry=self.registry)
self.labels = Info('il', 'help', ['l'], registry=self.registry)
def test_info(self):
self.assertEqual(1, self.registry.get_sample_value('i_info', {}))
self.info.info({'a': 'b', 'c': 'd'})
self.assertEqual(None, self.registry.get_sample_value('i_info', {}))
self.assertEqual(
1, self.registry.get_sample_value('i_info', {
'a': 'b',
'c': 'd'
}))
def test_labels(self):
self.assertRaises(ValueError, self.labels.labels('a').info, {'l': ''})
self.labels.labels('a').info({'foo': 'bar'})
self.assertEqual(
1,
self.registry.get_sample_value('il_info', {
'l': 'a',
'foo': 'bar'
}))
def test_duplicate_metrics_raises(self):
registry = CollectorRegistry()
Counter('c_total', 'help', registry=registry)
self.assertRaises(ValueError, Counter, 'c_total', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'c_total', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'c_created', 'help', registry=registry)
Gauge('g_created', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'g_created', 'help', registry=registry)
self.assertRaises(ValueError, Counter, 'g', 'help', registry=registry)
Summary('s', 'help', registry=registry)
self.assertRaises(ValueError, Summary, 's', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 's_created', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 's_sum', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 's_count', 'help', registry=registry)
# We don't currently expose quantiles, but let's prevent future
# clashes anyway.
self.assertRaises(ValueError, Gauge, 's', 'help', registry=registry)
Histogram('h', 'help', registry=registry)
self.assertRaises(ValueError, Histogram, 'h', 'help', registry=registry)
# Clashes aggaint various suffixes.
self.assertRaises(ValueError, Summary, 'h', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'h_count', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'h_sum', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'h_bucket', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'h_created', 'help', registry=registry)
# The name of the histogram itself is also taken.
self.assertRaises(ValueError, Gauge, 'h', 'help', registry=registry)
Info('i', 'help', registry=registry)
self.assertRaises(ValueError, Gauge, 'i_info', 'help', registry=registry)
def test_target_info_duplicate_detected(self):
registry = CollectorRegistry(target_info={'foo': 'bar'})
self.assertRaises(ValueError, Info, 'target', 'help', registry=registry)
registry.set_target_info({})
i = Info('target', 'help', registry=registry)
registry.set_target_info({})
self.assertRaises(ValueError, Info, 'target', 'help', registry=registry)
self.assertRaises(ValueError, registry.set_target_info, {'foo': 'bar'})
registry.unregister(i)
registry.set_target_info({'foo': 'bar'})
class TinvestCollector(object):
account_id: str
client: SyncClient
registry: CollectorRegistry = CollectorRegistry()
position_average_price: Gauge = Gauge("position_average_price",
"Average position price",
labelnames=["name", "currency"],
registry=registry)
position_average_price_no_nkd: Gauge = Gauge(
"position_average_price_no_nkd",
"Average position price",
labelnames=["name", "currency"],
registry=registry)
position_last_price: Gauge = Gauge("position_last_price",
"Position last price",
labelnames=["name"],
registry=registry)
position_close_price: Gauge = Gauge("position_close_price",
"Position close price",
labelnames=["name"],
registry=registry)
position_expected_yield: Gauge = Gauge("position_expected_yield",
"Expected yield",
labelnames=["name", "currency"],
registry=registry)
position_balance: Gauge = Gauge("position_balance",
"Balance",
labelnames=["name"],
registry=registry)
position_info: Info = Info("position",
"Position information",
labelnames=["name"],
registry=registry)
position_lots: Gauge = Gauge("position_lots",
"Lots count",
labelnames=["name"],
registry=registry)
currency_balance: Gauge = Gauge("currency_balance",
"Currency balance",
labelnames=["name"],
registry=registry)
etf_info: Info = Info("etf",
"ETF information",
labelnames=["name"],
registry=registry)
etf_last_price: Gauge = Gauge("etf_last_price",
"ETF last price",
labelnames=["name"],
registry=registry)
operation_commission: Gauge = Gauge(
"operation_commission",
"Operations commission",
labelnames=["figi", "currency", "instrument_type", "operation_type"],
registry=registry)
operation_broker_commission: Gauge = Gauge(
"operation_broker_commission",
"Broker commission",
labelnames=["figi", "currency", "instrument_type"],
registry=registry)
operation_coupon: Gauge = Gauge(
"operation_coupon",
"Coupon payments sum",
labelnames=["figi", "currency", "instrument_type"],
registry=registry)
operation_dividend: Gauge = Gauge(
"operation_dividend",
"Dividend payments sum",
labelnames=["figi", "currency", "instrument_type"],
registry=registry)
def __init__(self, token: str, account_id: str):
self.client = SyncClient(token, use_sandbox=False)
self.account_id = account_id
def collect(self):
self.__generate_positions_metrics()
self.__generate_currencies_metrics()
self.__generate_etfs_metrics()
self.__generate_operations_metrics()
for metric in self.registry.collect():
yield metric
def __generate_positions_metrics(self):
for position in self.__get_positions():
if position.average_position_price_no_nkd is not None:
self.position_average_price_no_nkd.labels(position.name,
position.average_position_price_no_nkd.currency.name). \
set(position.average_position_price_no_nkd.value)
currency = position.average_position_price_no_nkd.currency.name
else:
self.position_average_price.labels(
position.name, position.average_position_price.currency.name). \
set(position.average_position_price.value)
currency = position.average_position_price.currency.name
self.position_expected_yield.labels(position.name, position.expected_yield.currency.name). \
set(position.expected_yield.value)
self.position_balance.labels(position.name).set(position.balance)
orders = self.client.get_market_orderbook(position.figi, 0)
self.position_last_price.labels(position.name).set(
orders.payload.last_price)
self.position_close_price.labels(position.name).set(
orders.payload.close_price)
self.position_info.labels(position.name).info({
"currency":
currency,
"blocked":
position.blocked or "0",
"figi":
position.figi,
"instrument_type":
position.instrument_type,
"isin":
position.isin or "",
"ticker":
position.ticker,
})
self.position_lots.labels(position.name).set(position.lots)
def __generate_currencies_metrics(self):
for currency in self.__get_currencies():
self.currency_balance.labels(currency.currency.name).set(
currency.balance)
def __generate_etfs_metrics(self):
for etf in self.__get_etfs():
orders = self.client.get_market_orderbook(etf.figi, 0)
self.etf_last_price.labels(etf.name).set(orders.payload.last_price)
self.etf_info.labels(etf.name).info({
"currency":
etf.currency.name,
"figi":
etf.figi,
"isin":
etf.isin,
"ticker":
etf.ticker,
"trade_status":
orders.payload.trade_status.name,
})
def __generate_operations_metrics(self):
for operation in self.__get_operations():
if operation.commission is not None:
self.operation_commission.labels(operation.figi, operation.commission.currency.name,
operation.instrument_type.name, operation.operation_type.name). \
inc(float(operation.commission.value))
if operation.operation_type.name == 'broker_commission':
self.operation_broker_commission.labels(operation.figi, operation.currency.name,
operation.instrument_type.name). \
inc(float(operation.payment))
elif operation.operation_type.name == 'coupon':
self.operation_coupon.labels(operation.figi, operation.currency.name, operation.instrument_type.name). \
inc(float(operation.payment))
elif operation.operation_type.name == 'dividend':
self.operation_dividend.labels(operation.figi, operation.currency.name,
operation.instrument_type.name). \
inc(float(operation.payment))
def __get_positions(self):
portfolio = self.client.get_portfolio(self.account_id)
return portfolio.payload.positions
def __get_currencies(self):
currencies = self.client.get_portfolio_currencies(self.account_id)
return currencies.payload.currencies
def __get_etfs(self):
etfs = self.client.get_market_etfs()
return etfs.payload.instruments
def __get_operations(self):
operations = self.client.get_operations(
datetime.now() - timedelta(minutes=1),
datetime.now(),
broker_account_id=self.account_id)
return operations.payload.operations
def setUp(self):
self.registry = CollectorRegistry()
self.info = Info('i', 'help', registry=self.registry)
self.labels = Info('il', 'help', ['l'], registry=self.registry)
