def add_url(self, url, group_name, index): """ Adds url to gorup in view, needs index to read from database Args: url (string): url address of rss group_name (string): name of the group index (int): index of url in database """ url_row = QTreeWidgetItem([url]) url_row.rss_type = "url" url_row.url_index = index self.urls[f"{group_name}_{url}"] = url_row self.groups[group_name].addChild(url_row)
def add_group(self, group_name, urls, indexes): """ Adds groups and all of its urls to the view Args: group_name (string): name of group urls (list): list of urls indexes (list): list of url indexes in database """ group_tree = QTreeWidgetItem([group_name]) group_tree.rss_type = "group" self.groups[group_name] = group_tree for url, idx in zip(urls, indexes): self.add_url(url, group_name, idx) self.addTopLevelItem(group_tree)