import os

from t_tech.invest import Client
from t_tech.invest.schemas import (
    CreateFavoriteGroupRequest,
    DeleteFavoriteGroupRequest,
    EditFavoritesActionType as At,
    EditFavoritesRequestInstrument,
    GetFavoriteGroupsRequest,
)

TOKEN = os.environ["INVEST_TOKEN"]


def main():
    with Client(TOKEN) as client:
        r = client.instruments.get_favorites()

        print("Список избранных инструментов:")
        for i in r.favorite_instruments:
            print(f"{i.uid} - {i.name}")

        request = CreateFavoriteGroupRequest()
        request.group_name = "My test favorite group"
        request.group_color = "aa0000"  # red color
        r = client.instruments.create_favorite_group(request=request)
        group_id = r.group_id
        print(f"Создана новая группа избранного с ИД: {group_id}")

        client.instruments.edit_favorites(
            instruments=[EditFavoritesRequestInstrument(instrument_id="BBG001M2SC01")],
            action_type=At.EDIT_FAVORITES_ACTION_TYPE_ADD,
            group_id=group_id,
        )

        request = GetFavoriteGroupsRequest()
        request.instrument_id = ["BBG001M2SC01"]
        r = client.instruments.get_favorite_groups(request=request)
        print(f"Список групп избранного:")
        for i in r.groups:
            print(
                f"{i.group_id} - {i.group_name}. Количество элементов: {i.size}. "
                f"Содержит выбранный инструмент {request.instrument_id[0]}: "
                f"{i.contains_instrument} "
            )

        request = DeleteFavoriteGroupRequest()
        request.group_id = group_id
        client.instruments.delete_favorite_group(request=request)
        print(f"Удалена группа избранного с ИД: {group_id}")


if __name__ == "__main__":
    main()