def __init__(self, itemPath = None):

self.kodidb = KodiDB()

self.save_dir = None

self.bookmarks = None

self.plot_posts = None

self.itemPath = itemPath or utils.translateItemPath(sys.listitem.getPath())

self.file_dir = utils.BookmarkUtils.get_file_dir(self.itemPath)

self.idFile = self._get_idFile(self.itemPath)

def _get_idFile(self, itemPath):

fileName = utils.BookmarkUtils.get_file_name(itemPath)

idFiles = self.kodidb.idFiles(self.file_dir, fileName)

if len(idFiles) < 1:

raise RuntimeError("Can't get idFile!")

return idFiles[-1]['idFile']

def _get_save_dir(self):

if not self.save_dir:

self.save_dir = self.file_dir

if self.save_dir == "" or "plugin://" in self.save_dir:

self.save_dir = setting.Setting.get_addon_setting("save_dir")

if not self.save_dir:

dialog = xbmcgui.Dialog()

self.save_dir = dialog.browseSingle(0, 'Select directory for save .bmk', 'video')

setting.Setting.set_addon_setting("save_dir", self.save_dir)

return self.save_dir

def _save_bmk_posts_to_file(self, data):

file = self._get_save_dir() + '/' + utils.get_export_bookmarks_file_name(sys.listitem.getLabel())

f = xbmcvfs.File (file, 'w')

f.write(str(data))

f.close()

def _save_imgs_to_folder(self, image, seconds):

file = self._get_save_dir() + '/' + utils.get_export_bookmarks_image_name(

sys.listitem.getLabel(), str(int(seconds))

)

xbmcvfs.copy(image, file)

def _get_bookmarks(self):

if not self.bookmarks:

self.bookmarks = self.kodidb.bookmarks(self.idFile)

return self.bookmarks

def _get_plot_posts(self):

if not self.plot_posts:

plot = sys.listitem.getVideoInfoTag().getPlot()

if plot:

self.plot_posts = utils.retrieve_positions(plot)

else:

self.plot_posts = []

return self.plot_posts

def import_bookmark_from_plot(self):

positions = self._get_plot_posts()

xbmcgui.Dialog().ok("positions", str(positions))

for timeInSeconds in positions:

self.add_position(timeInSeconds)

def add_position(self, timeInSeconds, image = None):

thumb = None

if image:

thumb = u'special://profile/Thumbnails/Video/Bookmarks/%s_%s.jpg' % ( utils.getHash(image), timeInSeconds )

xbmcvfs.copy(image, thumb)

if thumb:

self.kodidb.insert_bookmark_with_thumb(self.idFile, timeInSeconds, thumb)

else:

self.kodidb.insert_bookmark(self.idFile, timeInSeconds)

if image:

xbmcvfs.delete(image)

def save_thumbnails(self):

if self._get_save_dir() != "":

for bookmark in self._get_bookmarks():

image = bookmark['thumbNailImage']

if image:

xbmc.log( "context.item.savebookmarks: same thumbnail: %s" % image, xbmc.LOGNOTICE )

seconds = bookmark['timeInSeconds']

if seconds not in self._get_plot_posts():

self._save_imgs_to_folder(image, seconds)

def save_positions(self):

positions = map(lambda x: x['timeInSeconds'], self._get_bookmarks())

xbmcgui.Dialog().ok("positions", str(positions))

xbmc.log( "context.item.savebookmarks: bookmark positions: %s" % str(positions), xbmc.LOGNOTICE )

if self._get_save_dir() != "":

xbmc.log( "context.item.savebookmarks: save to: %s" % self._get_save_dir().encode('utf-8'), xbmc.LOGNOTICE )