diff --git a/main.py b/main.py index 369702d..3b12c63 100644 --- a/main.py +++ b/main.py @@ -123,8 +123,8 @@ def home_menu(): # Battle Leaderboard add_dir( get_string(30052), BASE_URL + '/battle-leaderboard/recorded', 3, MEDIA_DIR + 'leader.png', '', '', 'top' ) - # TODO: replace with getting catergories from https://rumble.com/browse - add_dir( get_string(29916), BASE_URL + '/browse', 3, MEDIA_DIR + 'viral.png', '', '', 'cat_list' ) + # Getting categories + add_dir( 'Categories', BASE_URL + '/browse', 3, MEDIA_DIR + 'viral.png', '', '', 'cat_list' ) # TODO: add livestream section from https://rumble.com/browse/live @@ -244,6 +244,8 @@ def list_rumble( url, cat ): amount = dir_list_create( data, cat, 'channel_video', False, 2 ) elif cat == 'following': amount = dir_list_create( data, cat, 'following', False, 2 ) + elif cat == 'cat_list': + amount = dir_list_create( data, cat, cat, False ) return amount @@ -322,7 +324,7 @@ def dir_list_create( data, cat, video_type='video', search = False, play=0 ): cat = 'channel_video' #open get url and open player - add_dir( clean_text( title ), BASE_URL + link.strip(), 3, img, img, '', cat ) + add_dir( clean_text( title ), BASE_URL + link.strip() + '/videos', 3, img, img, '', cat ) elif video_type == 'following': following = re.compile(r'\s]+)(?:\s*data-js=\"main_menu_live_channel\")?\s*>\s*
\s*\s*\s*([^<]+)', re.MULTILINE|re.DOTALL|re.IGNORECASE).findall(data)