diff --git a/src/Channel/__init__.py b/src/Channel/__init__.py index d257d88..0988394 100644 --- a/src/Channel/__init__.py +++ b/src/Channel/__init__.py @@ -51,3 +51,6 @@ class Channel: def set_logo(self, logo: wx.Bitmap) -> None: self.m_logo = logo + + def refresh(self) -> None: + pass diff --git a/src/main.py b/src/main.py index 624a089..56d14aa 100644 --- a/src/main.py +++ b/src/main.py @@ -110,7 +110,9 @@ class Cast(wx.Frame): import_from_newpipe(subfile) subscriptions = get_subscriptions() for channel in subscriptions: - channels.append(YouTube.YouTube(channel[0], channel[1])) + yt_chan = YouTube.YouTube(channel[0], channel[1]) + yt_chan.refresh() + channels.append(yt_chan) self.m_providers[1].set_channels(channels) self.m_providers[1].make_latest() self.show_channel_list(None,self.m_selected_provider_index) @@ -178,18 +180,40 @@ class Cast(wx.Frame): self.m_sizer = wx.BoxSizer(wx.VERTICAL) # self.m_sizer.AddSpacer(SPACER_HEIGHT * 4) channel = self.m_selected_provider.get_channel_by_index(index) + def refresh_callback(event): + if self.m_selected_provider_index == 1 and index == 0: + with wx.BusyInfo("Please wait, working..."): + for chan in self.m_selected_provider.get_channels(): + chan.refresh() + wait = 1 + while wait > 0: + wait = 0 + for chan in self.m_selected_provider.get_channels(): + if chan.wait(): + wait += 1 + time.sleep(1) + wx.GetApp().Yield() + else: + channel.refresh() + self.show_video_list(event,index ) + + refreshbtn = wx.Button(self.m_panel, + -1, + label="Refresh", + size=(WIDTH, BTN_HEIGHT)) + refreshbtn.Bind(wx.EVT_BUTTON, refresh_callback) + self.m_sizer.Add(refreshbtn) if channel.wait(): with wx.BusyInfo("Please wait, working..."): - index = 0 - while channel.wait(): time.sleep(1) wx.GetApp().Yield() - callback = lambda event: self.show_channel_list( + back_callback = lambda event: self.show_channel_list( event, self.m_selected_provider_index) - self.add_back_button(callback) + self.add_back_button(back_callback) + for item in channel.get_items(): # type: ignore inner_sizer = wx.BoxSizer(wx.VERTICAL)