plugin.program.shortlist/capture.py at master · Gothicawakening/plugin.program.shortlist · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#
#       Copyright (C) 2014-
#       Sean Poyser (seanpoyser@gmail.com)
#
#  This Program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2, or (at your option)
#  any later version.
#
#  This Program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with XBMC; see the file COPYING.  If not, write to
#  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
#  http://www.gnu.org/copyleft/gpl.html
#

import sys
import xbmc
import xbmcgui
import xbmcaddon
import os
import pickle
from shortlistitem import ShortlistItem
from database import *

__addon__       = xbmcaddon.Addon(id='plugin.program.shortlist')

# if __name__ == '__main__':
def main():

    lst = listDatabases()
    # xbmc.log( "; ".join(lst), xbmc.LOGNOTICE);

    dialog = xbmcgui.Dialog()
    ret = dialog.contextmenu(lst)

    if ret != -1:
        dbNice = lst[ret]
        dbName = dbNice.lower() + ".db"
        #xbmc.log( dbName, xbmc.LOGNOTICE);

        videoInfoTag = sys.listitem.getVideoInfoTag()

        item = ShortlistItem()
        item.filename = xbmc.getInfoLabel('ListItem.FilenameAndPath')
        # item.filename = sys.listitem.getfilename()
        item.title = sys.listitem.getLabel()
        item.duration = sys.listitem.getduration()
        item.rating = videoInfoTag.getRating()
        item.year = videoInfoTag.getYear()
        item.date = videoInfoTag.getPremiered()
        item.plot = videoInfoTag.getPlot()
        item.plotoutline = videoInfoTag.getPlotOutline()
        item.thumb = sys.listitem.getArt( 'thumb' )
        item.poster = sys.listitem.getArt( 'poster' )
        item.fanart = sys.listitem.getArt( 'fanart' )

        result = addItemToDatabase( dbName, item )

        if result:
            title = "Added"
            message = "'%s' to %s" % (item.title, dbNice)
        else:
            lst = ['Keep','Remove']
            ret = dialog.contextmenu(lst)

            if ret == 0:
                # Keep
                title = "Kept"
                message = "'%s' in %s" % (item.title, dbNice)
            elif ret == 1:
                # Delete
                deleteItemFromDatabase( dbName, item.filename )
                title = "Deleted"
                message = "'%s' from %s" % (item.title, dbNice)

        xbmc.executebuiltin("Notification(\"%s\", \"%s\")" % (title, message) )

main()