File: rpcitylist.py

package info (click to toggle)
routeplanner 0.17
  • links: PTS
  • area: main
  • in suites: sarge
  • size: 1,836 kB
  • ctags: 264
  • sloc: python: 3,282; makefile: 58; sh: 40
file content (146 lines) | stat: -rw-r--r-- 4,944 bytes parent folder | download | duplicates (3)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
# Code for the GUI interface to RoutePlanner
#
# Copyright (C) 1996-2001 Chris Lawrence
# This file may be freely distributed under the terms of the RoutePlanner
# license.  A copy should appear as 'LICENSE' in the archive that this
# file was included in.

import string
import gtk
import gobject
#from gnome.ui import *

class CityList:
    def __init__(self, wtree):
        self.wtree = wtree
        self.citywin = wtree.get_widget('choosecity')
        self.cityview = wtree.get_widget('citychooser')
        self.cityentry = wtree.get_widget('cityentry')
        self.showallbutton = wtree.get_widget('showall')
        self.visible = []

        self.citystore = gtk.ListStore(gobject.TYPE_PYOBJECT,
                                       gobject.TYPE_STRING)
        renderer = gtk.CellRendererText()
        col = gtk.TreeViewColumn('City', renderer, text=1)
        self.cityview.append_column(col)
        self.cityview.set_model(self.citystore)

        sel = self.cityview.get_selection()
        sel.set_mode(gtk.SELECTION_SINGLE)
        sel.connect("changed", self.on_citychooser_select_row)

        dict = {}
        for key in dir(self.__class__):
            dict[key] = getattr(self, key)
        self.wtree.signal_autoconnect(dict)

    def startup(self, city='', db=None, setfun=None, setdata=None):
        self.db = db
        #self.cities = map(str, db.cities)
        self.cities = db.cities

        # Update the CList
        self.update_citylist()
        if city and city in self.cities:
            cnum = self.cities.index(city)
            sel = self.cityview.get_selection()
            sel.select_path(cnum)
            self.selected = city
            self.selrow = cnum
        else:
            self.selected = self.selrow = None
            self.cityentry.set_text('')
        
        self.citywin.connect("delete_event", self.on_city_cancel_clicked)
        
        if setfun:
            self.callwhendone = (setfun, setdata)
        else:
            self.callwhendone = None

        self.cityentry.grab_focus()
        self.citywin.show()

    def on_citychooser_select_row(self, *args):
        sel = args[0]
        if isinstance(sel, gtk.TreeView):
            sel = sel.get_selection()

        model, it = sel.get_selected()
        if it:
            self.selected = model.get_value(it, 0)
            cityname = model.get_value(it, 1)
            self.cityentry.set_text(cityname)
            path = model.get_path(it)
            self.selrow = path[0]
        else:
            self.selected = self.selrow = None
            self.cityentry.set_text('')

    def on_citychooser_button_press_event(self, button, event, *args):
        if event.type == gtk.gdk._2BUTTON_PRESS and event.button == 1 and \
               self.cityview.get_selection().get_selected():
            self.on_city_ok_clicked()

    def on_cityentry_activate(self, *args):
        text = self.cityentry.get_text().decode('UTF-8')
        matching = self.db.CitiesMatchIndex(text, self.visible)
        if not matching:
            dialog = gtk.MessageDialog(self.citywin, type=gtk.MESSAGE_ERROR,
                                       message_format='No matching cities.',
                                       buttons=gtk.BUTTONS_CLOSE)
            dialog.run()
            dialog.destroy()
            return

        if len(matching) == 1:
            match, lmatch = matching[0], self.visible.index(matching[0])
            sel = self.cityview.get_selection()
            sel.select_path(lmatch)
            self.cityview.scroll_to_cell((lmatch,), None, False, 0.5, 0)
            if text.lower() == unicode(self.cities[match]).lower():
                self.on_city_ok_clicked()
            return

        self.update_citylist(tuple(matching))

    def on_city_ok_clicked(self, *args):
        if not self.selected:
            self.on_cityentry_activate()
            return

        self.citywin.hide()
        if self.callwhendone:
            #print self.selected
            setfun, setdata = self.callwhendone
            setfun(self.selected, self.selrow, setdata)
        self.citystore.clear()

    def on_city_cancel_clicked(self, *args):
        self.citywin.hide()
        self.citystore.clear()
        return True

    def on_city_showall_clicked(self, *args):
        self.update_citylist()

    def update_citylist(self, selected=()):
        if not selected:
            selected = range(len(self.cities))

        self.citystore.clear()

        for i in range(len(selected)):
            sel = selected[i]
            it = self.citystore.append()
            try:
                cityname = unicode(self.cities[sel])
            except:
                print self.cities[sel].city
            
            self.citystore.set(it, 0, self.cities[sel], 1, cityname)
            
        self.visible = list(selected)

        self.showallbutton.set_sensitive(len(selected) != len(self.cities))