Move main page to a separate function

1 files changed, 17 insertions, 13 deletions

diff --git a/addon.py b/addon.py
index b360573..a7b4f3a 100644
--- a/addon.py
+++ b/addon.py
@@ -87,24 +87,28 @@ def tvshows_menu(page, folder_name):
     page.end_directory()
 
 
+def main_page(page):
+    url = page.build_url({'mode': 'folder', 'foldername': 'tv-shows'})
+    li = xbmcgui.ListItem(u'תוכניות טלוויזיה')
+    page.add_directory_item(url=url, listitem=li, isFolder=True)
+
+    url = page.build_url({'mode': 'folder', 'foldername': 'net-shows'})
+    li = xbmcgui.ListItem(u'תוכניות רשת')
+    page.add_directory_item(url=url, listitem=li, isFolder=True)
+
+    url = page.build_url({'mode': 'folder', 'foldername': 'new-items'})
+    li = xbmcgui.ListItem(u'קטעים חדשים')
+    page.add_directory_item(url=url, listitem=li, isFolder=True)
+
+    page.end_directory()
+
+
 def main():
     page = Page(sys.argv)
 
     trace("Tzafrir: {}".format(page))
     if page.mode is None:
-        url = page.build_url({'mode': 'folder', 'foldername': 'tv-shows'})
-        li = xbmcgui.ListItem(u'תוכניות טלוויזיה')
-        page.add_directory_item(url=url, listitem=li, isFolder=True)
-
-        url = page.build_url({'mode': 'folder', 'foldername': 'net-shows'})
-        li = xbmcgui.ListItem(u'תוכניות רשת')
-        page.add_directory_item(url=url, listitem=li, isFolder=True)
-
-        url = page.build_url({'mode': 'folder', 'foldername': 'new-items'})
-        li = xbmcgui.ListItem(u'קטעים חדשים')
-        page.add_directory_item(url=url, listitem=li, isFolder=True)
-
-        page.end_directory()
+        main_page(page)
 
     elif page.mode[0] == 'folder':
         foldername = page.args['foldername'][0]