diff --git a/src/pyside6/trade_widget.py b/src/pyside6/trade_widget.py index a54a9cd..93cc66f 100644 --- a/src/pyside6/trade_widget.py +++ b/src/pyside6/trade_widget.py @@ -62,8 +62,12 @@ class TradeWidget(QWidget): class TradeCollection(QWidget): """docstring for TradeCollection.""" - def __init__(self, message: Message): + + def __init__(self, + message: Message, + parent): super(TradeCollection, self).__init__() + self.parent = parent self.main_layout = QVBoxLayout() self.setLayout(self.main_layout) trade = TradeWidget(message, self) @@ -80,4 +84,5 @@ class TradeCollection(QWidget): def del_trade(self, message: Message): del self.trades[message.trade.__hash__] if not self.trades: + self.parent.del_collection(message.trade.unique_item()) self.deleteLater()