class ManagedTVMenu(object):
''' Provide windows for displaying managed shows and episodes,
and tools for manipulating the objects and managed file '''
def __init__(self):
self.dbh = DatabaseHandler()
@staticmethod
@utils.logged_function
def move_episodes_to_staged(items):
''' Remove all managed episodes in specified show from library, and add them to staged '''
STR_MOVING_ALL_x_EPISODES_BACK_TO_STAGED = utils.getlocalizedstring(
32034)
STR_ALL_x_EPISODES_MOVED_TO_STAGED = utils.getlocalizedstring(32035)
show_title = items[0].show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(
utils.ADDON_NAME,
STR_MOVING_ALL_x_EPISODES_BACK_TO_STAGED % show_title)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.remove_from_library()
item.set_as_staged()
progress_dialog.close()
utils.notification(STR_ALL_x_EPISODES_MOVED_TO_STAGED % show_title)
@utils.logged_function
def move_all_seasons_to_staged(self, show_title):
''' Remove all managed episodes in specified show from library, and add them to staged '''
STR_MOVING_ALL_x_SEASONS_BACK_TO_STAGED = 'Movendo temporadas de %s para staged'
STR_ALL_x_SEASONS_MOVED_TO_STAGED = 'Todas as temporadas de %s movidas para staged'
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(
utils.ADDON_NAME,
STR_MOVING_ALL_x_SEASONS_BACK_TO_STAGED % show_title)
items = self.dbh.get_content_items(status='managed',
mediatype='tvshow',
show_title=show_title,
order='Show_Title')
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.remove_from_library()
item.set_as_staged()
progress_dialog.close()
utils.notification(STR_ALL_x_SEASONS_MOVED_TO_STAGED % show_title)
@utils.logged_function
def move_all_to_staged(self):
''' Remove all managed tvshow items from library, and add them to staged '''
STR_MOVING_ALL_TV_SHOWS_BACK_TO_STAGED = utils.getlocalizedstring(
32026)
STR_ALL_TV_SHOWS_MOVED_TO_STAGED = utils.getlocalizedstring(32027)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_MOVING_ALL_TV_SHOWS_BACK_TO_STAGED)
managed_tv_items = self.dbh.get_content_items(status='managed',
mediatype='tvshow',
order='Show_Title')
for index, item in enumerate(managed_tv_items):
percent = 100 * index / len(managed_tv_items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.remove_from_library()
item.set_as_staged()
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOWS_MOVED_TO_STAGED)
@staticmethod
@utils.logged_function
def remove_episodes(items):
''' Remove all episodes in specified show from library '''
STR_REMOVING_ALL_x_EPISODES = utils.getlocalizedstring(32032)
STR_ALL_x_EPISODES_REMOVED = utils.getlocalizedstring(32033)
show_title = items[0].show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_REMOVING_ALL_x_EPISODES % show_title)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
item.remove_from_library()
item.delete()
progress_dialog.close()
utils.notification(STR_ALL_x_EPISODES_REMOVED % show_title)
@utils.logged_function
def remove_seasons(self, items, show_title):
''' Remove all seasons in specified show from library '''
STR_REMOVING_ALL_X_SEASONS = 'Removendo temporadas de: %s'
STR_ALL_X_SEASONS_REMOVED = 'Todas as temporadas de %s, foram removidas'
seasons = items[0]
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_REMOVING_ALL_X_SEASONS % show_title)
for season_number in seasons:
percent = 100 * season_number / len(seasons)
progress_dialog.update(percent,
line2=show_title,
line3=str("Season: %s" % season_number))
self.dbh.remove_from(mediatype='tvshow',
show_title=show_title,
season=season_number)
xbmc.sleep(300)
progress_dialog.close()
utils.notification(STR_ALL_X_SEASONS_REMOVED % show_title)
@utils.logged_function
def remove_all(self):
''' Remove all managed tvshow items from library '''
STR_REMOVING_ALL_TV_SHOWS = utils.getlocalizedstring(32024)
STR_ALL_TV_SHOWS_REMOVED = utils.getlocalizedstring(32025)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_REMOVING_ALL_TV_SHOWS)
managed_tv_items = self.dbh.get_content_items(status='managed',
mediatype='tvshow',
order='Show_Title')
for index, item in enumerate(managed_tv_items):
percent = 100 * index / len(managed_tv_items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
item.remove_from_library()
item.delete()
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOWS_REMOVED)
@utils.logged_function
def episode_options(self, item, season_number):
''' Provide options for a single managed episode in a dialog window '''
STR_REMOVE = utils.getlocalizedstring(32017)
STR_MOVE_BACK_TO_STAGED = utils.getlocalizedstring(32018)
STR_BACK = utils.getlocalizedstring(32011)
STR_MANAGED_EPISODE_OPTIONS = utils.getlocalizedstring(32036)
lines = [STR_REMOVE, STR_MOVE_BACK_TO_STAGED, STR_BACK]
ret = xbmcgui.Dialog().select(
'{0} - {1} - {2}'.format(utils.ADDON_NAME,
STR_MANAGED_EPISODE_OPTIONS,
item.episode_title_with_id), lines)
if ret >= 0:
if lines[ret] == STR_REMOVE:
item.remove_from_library()
item.delete()
return self.view_episodes(item.show_title, season_number)
elif lines[ret] == STR_MOVE_BACK_TO_STAGED:
item.remove_from_library()
item.set_as_staged()
return self.view_episodes(item.show_title, season_number)
elif lines[ret] == STR_BACK:
return self.view_episodes(item.show_title, season_number)
return self.view_episodes(item.show_title, season_number)
@utils.logged_function
def view_episodes(self, show_title, season_number):
''' Displays all managed episodes in the specified show,
which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_MANAGED_x_EPISODES = utils.getlocalizedstring(
32028) % show_title
STR_REMOVE_ALL_EPISODES = utils.getlocalizedstring(32029)
STR_MOVE_ALL_EPISODES_BACK_TO_STAGED = utils.getlocalizedstring(32030)
STR_BACK = utils.getlocalizedstring(32011)
STR_MANAGED_x_EPISODES = utils.getlocalizedstring(32031) % show_title
managed_episodes = self.dbh.get_content_items(
status='managed',
mediatype='tvshow',
order='Show_Title',
show_title=show_title,
season_number=season_number)
if not managed_episodes:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_MANAGED_x_EPISODES)
return self.view_shows()
lines = [str(x) for x in managed_episodes]
lines += [
STR_REMOVE_ALL_EPISODES, STR_MOVE_ALL_EPISODES_BACK_TO_STAGED,
STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_MANAGED_x_EPISODES),
lines)
if ret >= 0:
if ret < len(managed_episodes): # managed item
for i, item in enumerate(managed_episodes):
if ret == i:
return self.episode_options(item, season_number)
elif lines[ret] == STR_REMOVE_ALL_EPISODES:
self.remove_episodes(managed_episodes)
return self.view_shows()
elif lines[ret] == STR_MOVE_ALL_EPISODES_BACK_TO_STAGED:
self.move_episodes_to_staged(managed_episodes)
return self.view_shows()
elif lines[ret] == STR_BACK:
return self.view_seasons(show_title)
return self.view_seasons(show_title)
@utils.logged_function
def view_seasons(self, show_title):
''' Displays all managed seasons in the specified show,
which are selectable and lead to options.
Also provides additional options at bottom of menu '''
# TODO: functions to remove all or add all if necessary
STR_NO_MANAGED_X_SEASONS = str('No managed %s seasons') % show_title
STR_REMOVE_ALL_SEASONS = 'Remove all seasons'
STR_MOVE_ALL_SEASONS_BACK_TO_STAGED = 'Move all seasons back to staged'
STR_BACK = utils.getlocalizedstring(32011)
STR_MANAGED_X_SEASONS = str('Managed %s Seasons') % show_title
managed_seasons = self.dbh.get_content_items(status='managed',
mediatype='tvshow',
order='Season',
show_title=show_title)
if not managed_seasons:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_MANAGED_X_SEASONS)
return self.view_shows()
lines = [str('[B]Season %s[/B]' % x) for x in managed_seasons]
lines += [
STR_REMOVE_ALL_SEASONS, STR_MOVE_ALL_SEASONS_BACK_TO_STAGED,
STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_MANAGED_X_SEASONS), lines)
selection = lines[ret]
if ret >= 0:
if selection == STR_REMOVE_ALL_SEASONS:
# TODO: remove by title only
self.remove_seasons(managed_seasons, show_title)
return self.view_shows()
elif selection == STR_MOVE_ALL_SEASONS_BACK_TO_STAGED:
self.move_all_seasons_to_staged(show_title)
return self.view_shows()
elif selection == STR_BACK:
self.view_shows()
return self.view_shows()
else: # managed item
return self.view_episodes(show_title=show_title,
season_number=''.join(
filter(str.isdigit, selection)))
return self.view_shows()
@utils.logged_function
def view_shows(self):
''' Display all managed tvshows, which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_MANAGED_TV_SHOWS = utils.getlocalizedstring(32020)
STR_REMOVE_ALL_TV_SHOWS = utils.getlocalizedstring(32021)
STR_MOVE_ALL_TV_SHOWS_BACK_TO_STAGED = utils.getlocalizedstring(32022)
STR_BACK = utils.getlocalizedstring(32011)
STR_MANAGED_TV_SHOWS = utils.getlocalizedstring(32023)
managed_tvshows = self.dbh.get_all_shows('managed')
if not managed_tvshows:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_MANAGED_TV_SHOWS)
return
lines = ['[B]{}[/B]'.format(x) for x in managed_tvshows]
lines += [
STR_REMOVE_ALL_TV_SHOWS, STR_MOVE_ALL_TV_SHOWS_BACK_TO_STAGED,
STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_MANAGED_TV_SHOWS), lines)
if ret >= 0:
if ret < len(managed_tvshows):
for show_title in managed_tvshows:
if managed_tvshows[ret] == show_title:
self.view_seasons(show_title)
break
elif lines[ret] == STR_REMOVE_ALL_TV_SHOWS:
self.remove_all()
elif lines[ret] == STR_MOVE_ALL_TV_SHOWS_BACK_TO_STAGED:
self.move_all_to_staged()
elif lines[ret] == STR_BACK:
return
class StagedTVMenu(object):
''' Provide windows for displaying staged tvshows and episodes,
and tools for managing the items '''
def __init__(self):
self.dbh = DatabaseHandler()
@staticmethod
@utils.logged_function
def add_all_episodes(items):
''' Add all episodes from specified show to library '''
STR_ADDING_ALL_x_EPISODES = utils.getlocalizedstring(32071)
STR_ALL_x_EPISODES_ADDED = utils.getlocalizedstring(32072)
show_title = items[0].show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_ADDING_ALL_x_EPISODES % show_title)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.add_to_library()
progress_dialog.close()
utils.notification(STR_ALL_x_EPISODES_ADDED % show_title)
@utils.logged_function
def add_all_seasons(self, show_title):
''' Add all episodes from specified show to library '''
STR_ADDING_ALL_x_SEASONS = 'Adding all %s seasons...'
STR_ALL_x_SEASONS_ADDED = 'All %s seasons added'
staged_seasons = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title',
show_title=show_title)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_ADDING_ALL_x_SEASONS % show_title)
for index, item in enumerate(staged_seasons):
percent = 100 * index / len(staged_seasons)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(100)
item.add_to_library()
progress_dialog.close()
utils.notification(STR_ALL_x_SEASONS_ADDED % show_title)
@utils.logged_function
def add_all_seasons_with_metadata(self, show_title):
''' Add all seasons in the specified show with metadata to the library '''
STR_ADDING_ALL_x_SEASONS_WITH_METADATA = 'Adding all %s seasons with metadata...'
STR_ALL_x_SEASONS_WITH_METADATA_ADDED = 'All %s seasons with metadata added'
staged_seasons = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title',
show_title=show_title)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(
utils.ADDON_NAME,
STR_ADDING_ALL_x_SEASONS_WITH_METADATA % show_title)
for index, item in enumerate(staged_seasons):
percent = 100 * index / len(staged_seasons)
if os.path.exists(item.episode_nfo[0]):
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.add_to_library()
progress_dialog.update(percent, line2=' ', line3=' ')
progress_dialog.close()
utils.notification(STR_ALL_x_SEASONS_WITH_METADATA_ADDED % show_title)
@staticmethod
@utils.logged_function
def add_all_episodes_with_metadata(staged_episodes, show_title):
''' Add all episodes in the specified show with metadata to the library '''
STR_ADDING_ALL_x_EPISODES_WITH_METADATA = utils.getlocalizedstring(
32073)
STR_ALL_x_EPISODES_WITH_METADATA_ADDED = utils.getlocalizedstring(
32074)
# show_title = items[0]
# progress_dialog = xbmcgui.DialogProgress()
# progress_dialog.create(
# utils.ADDON_NAME, STR_ADDING_ALL_x_EPISODES_WITH_METADATA % show_title
# )
# for index, item in enumerate(items):
# percent = 100 * index / len(items)
# # nfo_path = os.path.join(metadata_dir, item.clean_title + '.nfo')
# if os.path.exists(item.episode_nfo[0]):
# progress_dialog.update(
# percent,
# line2=item.show_title,
# line3=item.episode_title_with_id
# )
# xbmc.sleep(200)
# item.add_to_library()
# progress_dialog.update(percent, line2=' ', line3=' ')
# progress_dialog.close()
# utils.notification(STR_ALL_x_EPISODES_WITH_METADATA_ADDED % show_title)
@staticmethod
@utils.logged_function
def generate_all_episodes_metadata(items):
''' Generate metadata items for all episodes in show '''
STR_GENERATING_ALL_x_METADATA = utils.getlocalizedstring(32077)
STR_ALL_x_METADATA_CREATED = utils.getlocalizedstring(32078)
show_title = items[0].show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_GENERATING_ALL_x_METADATA % show_title)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.create_metadata_item()
progress_dialog.close()
utils.notification(STR_ALL_x_METADATA_CREATED % show_title)
@utils.logged_function
def generate_all_seasons_metadata(self, show_title):
''' Generate metadata items for all seasons in show '''
STR_GENERATING_ALL_x_METADATA = 'Generating all %s metadata...'
STR_ALL_x_METADATA_CREATED = 'All %s metadata created'
staged_seasons = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title',
show_title=show_title)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_GENERATING_ALL_x_METADATA % show_title)
for index, item in enumerate(staged_seasons):
percent = 100 * index / len(staged_seasons)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.create_metadata_item()
progress_dialog.close()
utils.notification(STR_ALL_x_METADATA_CREATED % show_title)
@staticmethod
def rename_dialog(item):
''' Prompt input for new name, and rename if non-empty string '''
input_ret = xbmcgui.Dialog().input("Title", defaultt=item.show_title)
if input_ret:
item.rename(input_ret)
@staticmethod
@utils.logged_function
def rename_episodes_using_metadata(items):
''' Rename all episodes in show using nfo files '''
STR_RENAMING_x_EPISODES_USING_METADATA = utils.getlocalizedstring(
32075)
STR_x_EPISODES_RENAMED_USING_METADATA = utils.getlocalizedstring(32076)
show_title = items[0].show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(
utils.ADDON_NAME,
STR_RENAMING_x_EPISODES_USING_METADATA % show_title)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
# TODO: fix rename_using_metadata() current is not used
# item.rename_using_metadata()
progress_dialog.close()
utils.notification(STR_x_EPISODES_RENAMED_USING_METADATA % show_title)
@utils.logged_function
def add_all_shows(self):
''' Add all tvshow items to library '''
STR_ADDING_ALL_TV_SHOWS = utils.getlocalizedstring(32059)
STR_ALL_TV_SHOWS_ADDED = utils.getlocalizedstring(32060)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_ADDING_ALL_TV_SHOWS)
staged_tv_items = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title')
for index, item in enumerate(staged_tv_items):
percent = 100 * index / len(staged_tv_items)
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.add_to_library()
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOWS_ADDED)
@utils.logged_function
def add_all_with_metadata(self):
''' Add all tvshow items with nfo file to library'''
STR_ADDING_ALL_TV_SHOW_ITEMS_WITH_METADATA = utils.getlocalizedstring(
32061)
STR_ALL_TV_SHOW_ITEMS_WITH_METADATA_ADDED = utils.getlocalizedstring(
32062)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_ADDING_ALL_TV_SHOW_ITEMS_WITH_METADATA)
staged_tv_items = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title')
# content menaget precisa retornar episode_nfo[0]
for index, item in enumerate(staged_tv_items):
percent = 100 * index / len(staged_tv_items)
if os.path.exists(item.episode_nfo[0]):
progress_dialog.update(percent,
line2=item.show_title,
line3=item.episode_title_with_id)
xbmc.sleep(200)
item.add_to_library()
progress_dialog.update(percent, line2=' ', line3=' ')
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOW_ITEMS_WITH_METADATA_ADDED)
@utils.logged_function
def generate_all_metadata(self):
''' Create metadata for all staged tvshow items '''
STR_GENERATING_ALL_TV_SHOW_METADATA = utils.getlocalizedstring(32063)
STR_ALL_TV_SHOW_METADATA_CREATED = utils.getlocalizedstring(32064)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_GENERATING_ALL_TV_SHOW_METADATA)
staged_tv_items = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title')
for index, item in enumerate(staged_tv_items):
percent = 100 * index / len(staged_tv_items)
progress_dialog.update(percent, line2=item.show_title)
xbmc.sleep(200)
item.create_metadata_item()
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOW_METADATA_CREATED)
@utils.logged_function
def read_all_metadata(self):
''' Read metadata for all staged tvshow items '''
STR_READING_ALL_TV_SHOW_METADATA = utils.getlocalizedstring(32145)
STR_ALL_TV_SHOW_METADATA_READ = utils.getlocalizedstring(32146)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_READING_ALL_TV_SHOW_METADATA)
staged_tv_items = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Show_Title')
for index, item in enumerate(staged_tv_items):
percent = 100 * index / len(staged_tv_items)
progress_dialog.update(percent, line2=item.show_title)
xbmc.sleep(200)
# item.read_metadata_item()
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOW_METADATA_READ)
@utils.logged_function
def remove_all(self):
''' Remove all staged tvshow items '''
STR_REMOVING_ALL_TV_SHOWS = utils.getlocalizedstring(32024)
STR_ALL_TV_SHOW_REMOVED = utils.getlocalizedstring(32025)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_REMOVING_ALL_TV_SHOWS)
self.dbh.remove_from(status='staged',
mediatype='tvshow',
show_title=None,
directory=None)
progress_dialog.close()
utils.notification(STR_ALL_TV_SHOW_REMOVED)
@utils.logged_function
def remove_all_seasons(self, show_title):
''' Remove all seasons from the specified show '''
STR_REMOVING_ALL_x_SEASONS = utils.getlocalizedstring(
32032) % show_title
STR_ALL_x_SEASONS_REMOVED = utils.getlocalizedstring(
32033) % show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_REMOVING_ALL_x_SEASONS)
self.dbh.remove_from(status='staged',
mediatype='tvshow',
show_title=show_title,
directory=None)
progress_dialog.close()
utils.notification(STR_ALL_x_SEASONS_REMOVED)
@utils.logged_function
def remove_all_episodes(self, show_title):
''' Remove all episodes from the specified show '''
STR_REMOVING_ALL_x_EPISODES = utils.getlocalizedstring(
32032) % show_title
STR_ALL_x_EPISODES_REMOVED = utils.getlocalizedstring(
32033) % show_title
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_REMOVING_ALL_x_EPISODES)
self.dbh.remove_from(status='staged',
mediatype='tvshow',
show_title=show_title,
directory=None)
progress_dialog.close()
utils.notification(STR_ALL_x_EPISODES_REMOVED)
@utils.logged_function
def remove_and_block_show(self, show_title):
''' Remove all seasons from specified show from the library,
delete metadata, and add to blocked list '''
# Remove from staged
self.remove_all_seasons(show_title)
# Delete metadata folder
clean_show_title = utils.clean_name(show_title)
metadata_dir = os.path.join(utils.METADATA_FOLDER, 'TV',
clean_show_title)
utils.fs.remove_dir(metadata_dir)
# Add show title to blocked
self.dbh.add_blocked_item(show_title, 'tvshow')
@utils.logged_function
def episode_options(self, item, season_number):
''' Provide options for a single staged episode in a dialog window '''
#TODO: rename associated metadata when renaming
#TODO: rename show title
#TODO: remove item (including metadata)
STR_ADD = utils.getlocalizedstring(32048)
STR_REMOVE = utils.getlocalizedstring(32017)
STR_REMOVE_AND_BLOCK_EPISODE = utils.getlocalizedstring(32079)
STR_RENAME = utils.getlocalizedstring(32050)
STR_AUTOMATICALLY_RENAME_USING_METADTA = utils.getlocalizedstring(
32051)
STR_GENERATE_METADATA_ITEM = utils.getlocalizedstring(32052)
STR_BACK = utils.getlocalizedstring(32011)
STR_STAGED_EPISODE_OPTIONS = utils.getlocalizedstring(32080)
lines = [
STR_ADD, STR_REMOVE, STR_REMOVE_AND_BLOCK_EPISODE, STR_RENAME,
STR_AUTOMATICALLY_RENAME_USING_METADTA, STR_GENERATE_METADATA_ITEM,
STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1} - {2}'.format(utils.ADDON_NAME,
STR_STAGED_EPISODE_OPTIONS,
item.show_title), lines)
if ret >= 0:
if lines[ret] == STR_ADD:
item.add_to_library()
self.view_episodes(item.show_title, season_number)
elif lines[ret] == STR_REMOVE:
item.delete()
self.view_episodes(item.show_title, season_number)
elif lines[ret] == STR_REMOVE_AND_BLOCK_EPISODE:
item.remove_and_block()
self.view_episodes(item.show_title, season_number)
elif lines[ret] == STR_RENAME:
self.rename_dialog(item)
self.episode_options(item, season_number)
elif lines[ret] == STR_GENERATE_METADATA_ITEM:
item.create_metadata_item()
self.episode_options(item, season_number)
elif lines[ret] == STR_AUTOMATICALLY_RENAME_USING_METADTA:
item.rename_using_metadata()
self.episode_options(item, season_number)
elif lines[ret] == STR_BACK:
self.view_episodes(item.show_title, season_number)
else:
self.view_episodes(item.show_title, season_number)
@utils.logged_function
def view_episodes(self, show_title, season_number):
''' Display all staged episodes in the specified show,
which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_STAGED_x_EPISODES = utils.getlocalizedstring(32065) % show_title
STR_ADD_ALL_EPISODES = utils.getlocalizedstring(32066)
STR_ADD_ALL_EPISODES_WITH_METADATA = utils.getlocalizedstring(32067)
STR_REMOVE_ALL_EPISODES = utils.getlocalizedstring(32029)
STR_REMOVE_AND_BLOCK_TV_SHOW = utils.getlocalizedstring(32068)
STR_AUTOMATICALLY_RENAME_ALL_EPISODES_USING_METADATA = utils.getlocalizedstring(
32069)
STR_GENERATE_ALL_METADATA_ITEMS = utils.getlocalizedstring(32040)
STR_BACK = utils.getlocalizedstring(32011)
STR_STAGED_x_EPISODES = utils.getlocalizedstring(32070) % show_title
staged_episodes = self.dbh.get_content_items(
status='staged',
mediatype='tvshow',
order='Show_Title',
show_title=show_title,
season_number=season_number)
if not staged_episodes:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_STAGED_x_EPISODES)
self.view_shows()
return
lines = [str(x) for x in staged_episodes]
lines += [
STR_ADD_ALL_EPISODES, STR_ADD_ALL_EPISODES_WITH_METADATA,
STR_REMOVE_ALL_EPISODES, STR_REMOVE_AND_BLOCK_TV_SHOW,
STR_AUTOMATICALLY_RENAME_ALL_EPISODES_USING_METADATA,
STR_GENERATE_ALL_METADATA_ITEMS, STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_STAGED_x_EPISODES), lines)
if ret >= 0:
if ret < len(staged_episodes): # staged item
for i, item in enumerate(staged_episodes):
if ret == i:
self.episode_options(item, season_number)
break
elif lines[ret] == STR_ADD_ALL_EPISODES:
self.add_all_episodes(staged_episodes)
self.view_shows()
elif lines[ret] == STR_ADD_ALL_EPISODES_WITH_METADATA:
self.add_all_episodes_with_metadata(staged_episodes,
show_title)
self.view_episodes(show_title, season_number)
elif lines[ret] == STR_REMOVE_ALL_EPISODES:
self.remove_all_episodes(show_title)
self.view_shows()
elif lines[ret] == STR_REMOVE_AND_BLOCK_TV_SHOW:
self.remove_and_block_show(show_title)
self.view_shows()
elif lines[
ret] == STR_AUTOMATICALLY_RENAME_ALL_EPISODES_USING_METADATA:
self.rename_episodes_using_metadata(staged_episodes)
self.view_episodes(show_title, season_number)
elif lines[ret] == STR_GENERATE_ALL_METADATA_ITEMS:
self.generate_all_episodes_metadata(staged_episodes)
self.view_episodes(show_title, season_number)
elif lines[ret] == STR_BACK:
self.view_seasons(show_title)
else:
self.view_seasons(show_title)
@utils.logged_function
def view_seasons(self, show_title):
''' Display all staged seasons in the specified show,
which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_STAGED_x_SEASONS = 'No staged %s seasons' % show_title
STR_ADD_ALL_SEASONS = 'Add all seasons'
STR_ADD_ALL_SEASONS_WITH_METADATA = 'Add all seasons with metadata'
STR_REMOVE_ALL_SEASONS = 'Remove all seasons'
STR_REMOVE_AND_BLOCK_TV_SHOW = utils.getlocalizedstring(32068)
STR_AUTOMATICALLY_RENAME_ALL_SEASONS_USING_METADATA = utils.getlocalizedstring(
32069)
STR_GENERATE_ALL_METADATA_ITEMS = utils.getlocalizedstring(32040)
STR_BACK = utils.getlocalizedstring(32011)
STR_STAGED_x_SEASONS = 'Staged %s seasons' % show_title
staged_seasons = self.dbh.get_content_items(status='staged',
mediatype='tvshow',
order='Season',
show_title=show_title)
if not staged_seasons:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_STAGED_x_SEASONS)
self.view_shows()
return
lines = [str('[B]Season %s[/B]' % x) for x in staged_seasons]
lines += [
STR_ADD_ALL_SEASONS, STR_ADD_ALL_SEASONS_WITH_METADATA,
STR_REMOVE_ALL_SEASONS, STR_REMOVE_AND_BLOCK_TV_SHOW,
STR_AUTOMATICALLY_RENAME_ALL_SEASONS_USING_METADATA,
STR_GENERATE_ALL_METADATA_ITEMS, STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_STAGED_x_SEASONS), lines)
selection = lines[ret]
if ret >= 0:
if selection == STR_ADD_ALL_SEASONS:
self.add_all_seasons(show_title)
self.view_shows()
elif selection == STR_ADD_ALL_SEASONS_WITH_METADATA:
self.add_all_seasons_with_metadata(show_title)
self.view_seasons(show_title)
elif selection == STR_REMOVE_ALL_SEASONS:
self.remove_all_seasons(show_title)
self.view_shows()
elif selection == STR_REMOVE_AND_BLOCK_TV_SHOW:
self.remove_and_block_show(show_title)
self.view_shows()
elif (selection ==
STR_AUTOMATICALLY_RENAME_ALL_SEASONS_USING_METADATA):
self.rename_seasons_using_metadata(staged_seasons)
self.view_seasons(show_title)
elif selection == STR_GENERATE_ALL_METADATA_ITEMS:
self.generate_all_seasons_metadata(show_title)
self.view_seasons(show_title)
elif selection == STR_BACK:
self.view_shows()
else: # staged item
self.view_episodes(show_title,
season_number=''.join(
filter(str.isdigit, selection)))
else:
self.view_shows()
@utils.logged_function
def view_shows(self):
''' Display all managed tvshows, which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_STAGED_TV_SHOWS = utils.getlocalizedstring(32054)
STR_ADD_ALL_TV_SHOWS = utils.getlocalizedstring(32055)
STR_ADD_ALL_ITEMS_WITH_METADTA = utils.getlocalizedstring(32056)
STR_REMOVE_ALL_TV_SHOWS = utils.getlocalizedstring(32057)
STR_GENERATE_ALL_METADATA_ITEMS = utils.getlocalizedstring(32040)
STR_READ_ALL_METADATA_ITEMS = utils.getlocalizedstring(32147)
STR_BACK = utils.getlocalizedstring(32011)
STR_STAGED_TV_SHOWS = utils.getlocalizedstring(32058)
staged_tvshows = self.dbh.get_all_shows('staged')
if not staged_tvshows:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_STAGED_TV_SHOWS)
return
lines = ['[B]{}[/B]'.format(x) for x in staged_tvshows]
lines += [
STR_ADD_ALL_TV_SHOWS, STR_ADD_ALL_ITEMS_WITH_METADTA,
STR_REMOVE_ALL_TV_SHOWS, STR_GENERATE_ALL_METADATA_ITEMS,
STR_READ_ALL_METADATA_ITEMS, STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_STAGED_TV_SHOWS), lines)
if ret >= 0:
if ret < len(staged_tvshows): # staged item
for show_title in staged_tvshows:
if staged_tvshows[ret] == show_title:
self.view_seasons(show_title)
break
elif lines[ret] == STR_ADD_ALL_TV_SHOWS:
self.add_all_shows()
elif lines[ret] == STR_ADD_ALL_ITEMS_WITH_METADTA:
self.add_all_with_metadata()
self.view_shows()
elif lines[ret] == STR_REMOVE_ALL_TV_SHOWS:
self.remove_all()
elif lines[ret] == STR_GENERATE_ALL_METADATA_ITEMS:
self.generate_all_metadata()
self.view_shows()
elif lines[ret] == STR_READ_ALL_METADATA_ITEMS:
self.read_all_metadata()
self.view_shows()
elif lines[ret] == STR_BACK:
return
class StagedMoviesMenu(object):
''' Provide windows for displaying staged movies,
and tools for managing the items '''
#TODO: don't commit sql changes for "... all" until end
#TODO: decorator for "...all" commands
#TODO: load staged movies on init, use as instance variable, refresh as needed
def __init__(self):
self.dbh = DatabaseHandler()
@staticmethod
@utils.logged_function
def add_all(items):
''' Add all staged movies to library '''
STR_ADDING_ALL_MOVIES = utils.getlocalizedstring(32042)
STR_ALL_MOVIES_ADDED = utils.getlocalizedstring(32043)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_ADDING_ALL_MOVIES)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(int(percent), item.movie_title)
item.add_to_library()
progress_dialog.close()
utils.notification(STR_ALL_MOVIES_ADDED)
@staticmethod
@utils.logged_function
def add_all_with_metadata(items):
''' Add all movies with nfo files to the library '''
# TODO: Remove code duplication with MovieItem.add_to_library_if_metadata
STR_ADDING_ALL_MOVIES_WITH_METADATA = utils.getlocalizedstring(32044)
STR_ALL_MOVIES_WITH_METADTA_ADDED = utils.getlocalizedstring(32045)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_ADDING_ALL_MOVIES_WITH_METADATA)
for index, item in enumerate(items):
percent = 100 * index / len(items)
if os.path.exists(item.movie_nfo[0]):
progress_dialog.update(int(percent), item.movie_title)
item.add_to_library()
progress_dialog.close()
utils.notification(STR_ALL_MOVIES_WITH_METADTA_ADDED)
@staticmethod
@utils.logged_function
def clean_up_metadata():
''' Remove all unused metadata '''
STR_MOVIE_METADATA_CLEANED = utils.getlocalizedstring(32136)
metadata_dir = os.path.join(utils.METADATA_FOLDER, 'Movies')
for folder in os.listdir(metadata_dir):
full_path = os.path.join(metadata_dir, folder)
if os.path.isdir(full_path):
folder_contents = os.listdir(full_path)
if len(folder_contents) == 1 and fnmatch(
folder_contents[0], '*.strm'):
utils.log_msg(
'Removing metadata folder {}'.format(full_path),
loglevel=xbmc.LOGNOTICE)
shutil.rmtree(full_path)
utils.notification(STR_MOVIE_METADATA_CLEANED)
@staticmethod
@utils.logged_function
def generate_all_metadata(items):
''' Generate metadata items for all staged movies '''
STR_GENERATING_ALL_MOVIE_METADATA = utils.getlocalizedstring(32046)
STR_ALL_MOVIE_METADTA_CREATED = utils.getlocalizedstring(32047)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME,
STR_GENERATING_ALL_MOVIE_METADATA)
for index, item in enumerate(items):
percent = 100 * index / len(items)
progress_dialog.update(int(percent), item.movie_title)
item.create_metadata_item()
progress_dialog.close()
utils.notification(STR_ALL_MOVIE_METADTA_CREATED)
@staticmethod
def rename_dialog(item):
''' Prompt input for new name, and rename if non-empty string '''
#TODO: move to utils or parent class so it's not duplicated
input_ret = xbmcgui.Dialog().input("Title", defaultt=item.movie_title)
if input_ret:
item.rename(input_ret)
@utils.logged_function
def options(self, item):
''' Provide options for a single staged movie in a dialog window '''
#TODO: add a back button
STR_ADD = utils.getlocalizedstring(32048)
STR_REMOVE = utils.getlocalizedstring(32017)
STR_REMOVE_AND_BLOCK = utils.getlocalizedstring(32049)
STR_RENAME = utils.getlocalizedstring(32050)
STR_AUTOMATICALLY_RENAME_USING_METADTA = utils.getlocalizedstring(
32051)
STR_GENERATE_METADATA_ITEM = utils.getlocalizedstring(32052)
STR_STAGED_MOVIE_OPTIONS = utils.getlocalizedstring(32053)
lines = [
STR_ADD,
STR_REMOVE,
STR_REMOVE_AND_BLOCK,
#STR_RENAME, STR_AUTOMATICALLY_RENAME_USING_METADTA,
STR_GENERATE_METADATA_ITEM
]
ret = xbmcgui.Dialog().select(
'{0} - {1} - {2}'.format(utils.ADDON_NAME,
STR_STAGED_MOVIE_OPTIONS,
item.movie_title), lines)
if ret >= 0:
if lines[ret] == STR_ADD:
item.add_to_library()
self.view_all()
elif lines[ret] == STR_REMOVE:
item.delete()
self.view_all()
elif lines[ret] == STR_REMOVE_AND_BLOCK:
item.remove_and_block()
self.view_all()
elif lines[ret] == STR_RENAME:
self.rename_dialog(item)
self.options(item)
elif lines[ret] == STR_GENERATE_METADATA_ITEM:
item.create_metadata_item()
self.options(item)
elif lines[ret] == STR_AUTOMATICALLY_RENAME_USING_METADTA:
item.rename_using_metadata()
self.options(item)
else:
self.view_all()
@utils.logged_function
def remove_all(self):
''' Remove all staged movies '''
STR_REMOVING_ALL_MOVIES = utils.getlocalizedstring(32013)
STR_ALL_MOVIES_REMOVED = utils.getlocalizedstring(32014)
progress_dialog = xbmcgui.DialogProgress()
progress_dialog.create(utils.ADDON_NAME, STR_REMOVING_ALL_MOVIES)
self.dbh.remove_from(status='staged',
mediatype='movie',
show_title=None,
directory=None)
progress_dialog.close()
utils.notification(STR_ALL_MOVIES_REMOVED)
@utils.logged_function
def view_all(self):
''' Display all staged movies, which are selectable and lead to options.
Also provides additional options at bottom of menu '''
STR_NO_STAGED_MOVIES = utils.getlocalizedstring(32037)
STR_ADD_ALL_MOVIES = utils.getlocalizedstring(32038)
STR_ADD_ALL_MOVIES_WITH_METADATA = utils.getlocalizedstring(32039)
STR_REMOVE_ALL_MOVIES = utils.getlocalizedstring(32009)
STR_GENERATE_ALL_METADATA_ITEMS = utils.getlocalizedstring(32040)
STR_BACK = utils.getlocalizedstring(32011)
STR_STAGED_MOVIES = utils.getlocalizedstring(32041)
STR_CLEAN_UP_METADATA = utils.getlocalizedstring(32135)
staged_movies = self.dbh.get_content_items(status='staged',
mediatype='movie',
order='Title')
if not staged_movies:
xbmcgui.Dialog().ok(utils.ADDON_NAME, STR_NO_STAGED_MOVIES)
return
lines = [str(x) for x in staged_movies]
lines += [
STR_ADD_ALL_MOVIES, STR_ADD_ALL_MOVIES_WITH_METADATA,
STR_REMOVE_ALL_MOVIES, STR_GENERATE_ALL_METADATA_ITEMS,
STR_CLEAN_UP_METADATA, STR_BACK
]
ret = xbmcgui.Dialog().select(
'{0} - {1}'.format(utils.ADDON_NAME, STR_STAGED_MOVIES), lines)
if ret >= 0:
if ret < len(staged_movies): # staged item
for i, item in enumerate(staged_movies):
if ret == i:
self.options(item)
break
elif lines[ret] == STR_ADD_ALL_MOVIES:
self.add_all(staged_movies)
elif lines[ret] == STR_ADD_ALL_MOVIES_WITH_METADATA:
self.add_all_with_metadata(staged_movies)
self.view_all()
elif lines[ret] == STR_REMOVE_ALL_MOVIES:
self.remove_all()
elif lines[ret] == STR_GENERATE_ALL_METADATA_ITEMS:
self.generate_all_metadata(staged_movies)
self.view_all()
elif lines[ret] == STR_CLEAN_UP_METADATA:
self.clean_up_metadata()
self.view_all()
elif lines[ret] == STR_BACK:
return