otr-seriesassistant/inputDialog.py at master · dahoo/otr-seriesassistant · 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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@author: Daniel Hoffmann
"""

import helpers
import gtk, requests, os.path
from xml.dom.minidom import parseString

class ChooseSeriesDialog(object):
    def __init__(self):
        self.builder = gtk.Builder()
        self.builder.add_from_file(os.path.join(os.path.split(os.path.realpath(__file__))[0],
                                        "seriesChooserDialog.ui"))
        self.builder.connect_signals(self)

        helpers.create_treeview_column(self.obj('tv_results'), 'Name', 0,
                               gtk.CellRendererSpin())

        helpers.create_treeview_column(self.obj('tv_results'), 'ID',  1,
                               gtk.CellRendererText())

        window = self.builder.get_object("dialog1")
        window.show_all()

    def obj(self, name):
        """
        returns glade object 'name'
        """
        return self.builder.get_object(name)

    def on_bt_search_clicked(self, action, *args):
        name = self.obj('en_name').get_text()
        language = ''
        if self.obj('rb_german').get_active():
            language = '&language=de'
        elif self.obj('rb_all').get_active():
            language = '&language=all'
        response = requests.get('http://www.thetvdb.com/api/GetSeries.php?seriesname=' + name + language)

        if response.status_code == 200:
            xmlFile = parseString(response.content)
            for element in xmlFile.getElementsByTagName('Series'):
                self.obj('ls_series').append((helpers.getNodeText(element, 'SeriesName'),
                         helpers.getNodeText(element, 'seriesid')))
            self.obj('tv_results').set_cursor(0)

    def on_bt_ok_clicked(self, action, *args):
        path = self.obj('tv_results').get_cursor()[0]
        if path is not None:
            row = self.obj('ls_series')[path]
            self.response = (row[0], row[1])
            self.quit()

    def on_bt_cancel_clicked(self, action, *args):
        self.response = None
        self.quit()

    def run(self):
        try:
            gtk.main()
        except KeyboardInterrupt:
            pass

    def quit(self):
        gtk.main_quit()

def getSeries():
    dialog = ChooseSeriesDialog()
    dialog.run()
    response = dialog.response
    dialog.builder.get_object("dialog1").destroy()
    return response

if __name__ == '__main__':
    series = getSeries()
    if series:
        print "Selected: %s" % getSeries()[0]
    gtk.main()