class PhotiniMap(QtWidgets.QWidget): def __init__(self, image_list, parent=None): super(PhotiniMap, self).__init__(parent) self.app = QtWidgets.QApplication.instance() self.image_list = image_list name = self.__module__.split('.')[-1] self.script_dir = pkg_resources.resource_filename( 'photini', 'data/' + name + '/') self.drag_icon = QtGui.QPixmap( os.path.join(self.script_dir, '../map_pin_grey.png')) self.drag_hotspot = 11, 35 self.search_string = None self.map_loaded = False self.marker_info = {} self.map_status = {} self.dropped_images = [] self.geocoder = self.get_geocoder() self.setLayout(QtWidgets.QHBoxLayout()) ## left side left_side = QtWidgets.QGridLayout() # latitude & longitude self.coords = LatLongDisplay(self.image_list) left_side.addWidget(self.coords.label, 0, 0) self.coords.changed.connect(self.new_coords) left_side.addWidget(self.coords, 0, 1) # altitude label = QtWidgets.QLabel(translate('MapTabsAll', 'Altitude')) label.setAlignment(Qt.AlignRight) left_side.addWidget(label, 1, 0) self.altitude = DoubleSpinBox() self.altitude.setSuffix(' m') self.altitude.new_value.connect(self.new_altitude) left_side.addWidget(self.altitude, 1, 1) if hasattr(self.geocoder, 'get_altitude'): self.altitude_button = QtWidgets.QPushButton( translate('MapTabsAll', 'Get altitude from map')) self.altitude_button.clicked.connect(self.get_altitude) left_side.addWidget(self.altitude_button, 2, 1) else: self.altitude_button = None # search label = QtWidgets.QLabel(translate('MapTabsAll', 'Search')) label.setAlignment(Qt.AlignRight) left_side.addWidget(label, 3, 0) self.edit_box = ComboBox() self.edit_box.setEditable(True) self.edit_box.setInsertPolicy(QtWidgets.QComboBox.NoInsert) self.edit_box.lineEdit().setPlaceholderText( translate('MapTabsAll', '<new search>')) self.edit_box.lineEdit().returnPressed.connect(self.search) self.edit_box.activated.connect(self.goto_search_result) self.clear_search() self.edit_box.setEnabled(False) left_side.addWidget(self.edit_box, 3, 1) # search terms and conditions for n, widget in enumerate(self.geocoder.search_terms()): left_side.addWidget(widget, n + 4, 0, 1, 2) left_side.setColumnStretch(1, 1) left_side.setRowStretch(7, 1) self.layout().addLayout(left_side) # map # create handler for calls from JavaScript self.call_handler = CallHandler(parent=self) self.map = MapWebView(self.call_handler) self.map.drop_text.connect(self.drop_text) self.map.setAcceptDrops(False) self.layout().addWidget(self.map) self.layout().setStretch(1, 1) # other init self.image_list.image_list_changed.connect(self.image_list_changed) @QtSlot() @catch_all def image_list_changed(self): self.redraw_markers() self.coords.refresh() self.update_altitude() self.see_selection() @QtSlot() @catch_all def initialise(self): page = '''<!DOCTYPE html> <html> <head> <meta charset="utf-8" /> <meta name="viewport" content="initial-scale=1.0, user-scalable=no" /> <style type="text/css"> html, body {{ height: 100%; margin: 0; padding: 0 }} #mapDiv {{ position: relative; width: 100%; height: 100% }} </style> {initialize} {head} <script type="text/javascript" src="script.js"></script> </head> <body ondragstart="return false"> <div id="mapDiv"></div> </body> </html>''' lat, lng = eval( self.app.config_store.get('map', 'centre', '(51.0, 0.0)')) zoom = int(eval(self.app.config_store.get('map', 'zoom', '11'))) if QtWebEngineWidgets: initialize = ''' <script type="text/javascript" src="qrc:///qtwebchannel/qwebchannel.js"> </script> <script type="text/javascript"> var python; function initialize() {{ new QWebChannel(qt.webChannelTransport, doLoadMap); }} function doLoadMap(channel) {{ python = channel.objects.python; loadMap({lat}, {lng}, {zoom}); }} </script>''' else: initialize = ''' <script type="text/javascript"> function initialize() {{ loadMap({lat}, {lng}, {zoom}); }} </script>''' initialize = initialize.format(lat=lat, lng=lng, zoom=zoom) page = page.format(initialize=initialize, head=self.get_head()) QtWidgets.QApplication.setOverrideCursor(Qt.WaitCursor) self.map.setHtml(page, QtCore.QUrl.fromLocalFile(self.script_dir)) @catch_all def initialize_finished(self): QtWidgets.QApplication.restoreOverrideCursor() self.map_loaded = True self.edit_box.setEnabled(True) self.map.setAcceptDrops(True) self.redraw_markers() def refresh(self): self.image_list.set_drag_to_map(self.drag_icon, self.drag_hotspot) if not self.map_loaded: self.initialise() return lat, lng = eval(self.app.config_store.get('map', 'centre')) zoom = int(eval(self.app.config_store.get('map', 'zoom'))) self.JavaScript('setView({!r},{!r},{:d})'.format(lat, lng, zoom)) def do_not_close(self): return False @catch_all def new_status(self, status): self.map_status.update(status) for key in ('centre', 'zoom'): if key in status: self.app.config_store.set('map', key, repr(self.map_status[key])) @QtSlot(int, int, str) @catch_all def drop_text(self, x, y, text): self.dropped_images = eval(text) self.JavaScript('markerDrop({:d},{:d})'.format(x, y)) @catch_all def marker_drop(self, lat, lng): for path in self.dropped_images: image = self.image_list.get_image(path) image.metadata.latlong = lat, lng self.dropped_images = [] self.redraw_markers() self.coords.refresh() self.see_selection() @QtSlot() @catch_all def new_coords(self): self.redraw_markers() self.update_altitude() self.see_selection() @QtSlot(object) @catch_all def new_altitude(self, value): for image in self.image_list.get_selected_images(): image.metadata.altitude = value self.update_altitude() def update_altitude(self): images = self.image_list.get_selected_images() if not images: self.altitude.set_value(None) self.altitude.setEnabled(False) if self.altitude_button: self.altitude_button.setEnabled(False) return values = [] for image in images: value = image.metadata.altitude if value not in values: values.append(value) if len(values) > 1: self.altitude.set_multiple(choices=filter(None, values)) else: self.altitude.set_value(values[0]) self.altitude.setEnabled(True) if self.altitude_button: self.altitude_button.setEnabled(bool(self.coords.get_value())) def see_selection(self): locations = [] for image in self.image_list.get_selected_images(): latlong = image.metadata.latlong if not latlong: continue location = [latlong['lat'], latlong['lon']] if location not in locations: locations.append(location) if not locations: return self.JavaScript('fitPoints({})'.format(repr(locations))) @QtSlot(list) @catch_all def new_selection(self, selection): self.redraw_markers() self.coords.refresh() self.update_altitude() self.see_selection() def redraw_markers(self): if not self.map_loaded: return for info in self.marker_info.values(): info['images'] = [] for image in self.image_list.get_images(): latlong = image.metadata.latlong if not latlong: continue for info in self.marker_info.values(): if info['latlong'] == latlong: info['images'].append(image) break else: for i in range(len(self.marker_info) + 2): marker_id = i if marker_id not in self.marker_info: break self.marker_info[marker_id] = { 'images': [image], 'latlong': LatLon(latlong), 'selected': image.selected, } self.JavaScript('addMarker({:d},{!r},{!r},{:d})'.format( marker_id, latlong['lat'], latlong['lon'], image.selected)) for marker_id in list(self.marker_info.keys()): info = self.marker_info[marker_id] if not info['images']: self.JavaScript('delMarker({:d})'.format(marker_id)) del self.marker_info[marker_id] elif info['selected'] != any([x.selected for x in info['images']]): info['selected'] = not info['selected'] self.JavaScript('enableMarker({:d},{:d})'.format( marker_id, info['selected'])) def plot_track(self, tracks): latlngs = [] for t in tracks: latlngs.append([[x[1], x[2]] for x in t]) self.JavaScript('plotTrack({!r})'.format(latlngs)) @QtSlot() @catch_all def get_altitude(self): altitude = self.geocoder.get_altitude(self.coords.get_value()) if altitude is not None: self.new_altitude(round(altitude, 1)) @QtSlot() @catch_all def search(self, search_string=None, bounded=True): if not search_string: search_string = self.edit_box.lineEdit().text() self.edit_box.clearEditText() if not search_string: return self.search_string = search_string self.clear_search() if bounded: bounds = self.map_status['bounds'] else: bounds = None for result in self.geocoder.search(search_string, bounds=bounds): north, east, south, west, name = result self.edit_box.addItem(name, (north, east, south, west)) self.edit_box.set_dropdown_width() self.edit_box.showPopup() def clear_search(self): self.edit_box.clear() self.edit_box.addItem('') if self.search_string: self.edit_box.addItem(translate('MapTabsAll', '<widen search>'), 'widen') self.edit_box.addItem( translate('MapTabsAll', '<repeat search>', 'repeat')) @QtSlot(int) @catch_all def goto_search_result(self, idx): self.edit_box.setCurrentIndex(0) self.edit_box.clearFocus() if idx == 0: return data = self.edit_box.itemData(idx) if data is None: return if data == 'widen': # widen search self.search(search_string=self.search_string, bounded=False) elif data == 'repeat': # repeat search self.search(search_string=self.search_string) elif data[-1] is None: self.JavaScript('setView({},{},{})'.format( data[0], data[1], self.map_status['zoom'])) else: self.JavaScript('adjustBounds({},{},{},{})'.format(*data)) @catch_all def marker_click(self, marker_id): self.image_list.select_images(self.marker_info[marker_id]['images']) @catch_all def marker_drag(self, lat, lng): self.coords.set_value('{:.6f}, {:.6f}'.format(lat, lng)) @catch_all def marker_drag_end(self, lat, lng, marker_id): info = self.marker_info[marker_id] for image in info['images']: image.metadata.latlong = lat, lng info['latlong'] = LatLon((lat, lng)) self.coords.refresh() def JavaScript(self, command): if self.map_loaded: self.map.page().do_java_script(command)
class PhotiniMap(QtWidgets.QSplitter): def __init__(self, image_list, parent=None): super(PhotiniMap, self).__init__(parent) self.app = QtWidgets.QApplication.instance() self.image_list = image_list self.geocode_cache = OrderedDict() name = self.__module__.split('.')[-1] self.api_key = key_store.get(name, 'api_key') self.search_key = key_store.get('opencage', 'api_key') self.script_dir = pkg_resources.resource_filename( 'photini', 'data/' + name + '/') self.drag_icon = QtGui.QPixmap( os.path.join(self.script_dir, '../map_pin_grey.png')) self.drag_hotspot = 11, 35 self.search_string = None self.map_loaded = False self.marker_info = {} self.map_status = {} self.dropped_images = [] self.setChildrenCollapsible(False) left_side = QtWidgets.QWidget() self.addWidget(left_side) left_side.setLayout(QtWidgets.QFormLayout()) left_side.layout().setContentsMargins(0, 0, 0, 0) left_side.layout().setFieldGrowthPolicy( QtWidgets.QFormLayout.AllNonFixedFieldsGrow) # map # create handler for calls from JavaScript self.call_handler = CallHandler(parent=self) self.map = MapWebView(self.call_handler) self.map.drop_text.connect(self.drop_text) self.map.setAcceptDrops(False) self.addWidget(self.map) # search search_layout = QtWidgets.QFormLayout() search_layout.setContentsMargins(0, 0, 0, 0) search_layout.setVerticalSpacing(0) search_layout.setFieldGrowthPolicy( QtWidgets.QFormLayout.AllNonFixedFieldsGrow) self.edit_box = ComboBox() self.edit_box.setEditable(True) self.edit_box.setInsertPolicy(QtWidgets.QComboBox.NoInsert) self.edit_box.lineEdit().setPlaceholderText( translate('PhotiniMap', '<new search>')) self.edit_box.lineEdit().returnPressed.connect(self.search) self.edit_box.activated.connect(self.goto_search_result) self.clear_search() self.edit_box.setEnabled(False) search_layout.addRow(translate('PhotiniMap', 'Search'), self.edit_box) # search terms and conditions terms = self.search_terms() if terms: search_layout.addRow(*terms) left_side.layout().addRow(search_layout) if terms: divider = QtWidgets.QFrame() divider.setFrameStyle(QtWidgets.QFrame.HLine) left_side.layout().addRow(divider) left_side.layout().addItem( QtWidgets.QSpacerItem(1, 1000, vPolicy=QtWidgets.QSizePolicy.Expanding)) # latitude & longitude layout = QtWidgets.QHBoxLayout() self.coords = SingleLineEdit() self.coords.editingFinished.connect(self.new_coords) self.coords.setEnabled(False) layout.addWidget(self.coords) # convert lat/lng to location info self.auto_location = QtWidgets.QPushButton( translate('PhotiniMap', six.unichr(0x21e8) + ' address')) self.auto_location.setFixedHeight(self.coords.height()) self.auto_location.setEnabled(False) self.auto_location.clicked.connect(self.get_address) layout.addWidget(self.auto_location) left_side.layout().addRow(translate('PhotiniMap', 'Lat, long'), layout) # location info self.location_widgets = [] self.location_info = QtWidgets.QTabWidget() tab_bar = QTabBar() self.location_info.setTabBar(tab_bar) tab_bar.context_menu.connect(self.location_tab_context_menu) tab_bar.tabMoved.connect(self.location_tab_moved) self.location_info.setElideMode(Qt.ElideLeft) self.location_info.setMovable(True) self.location_info.setEnabled(False) left_side.layout().addRow(self.location_info) # address lookup (and default search) terms and conditions layout = QtWidgets.QHBoxLayout() if terms: widget = CompactButton( self.tr('Address lookup\npowered by OpenCage')) else: widget = CompactButton( self.tr('Search && lookup\npowered by OpenCage')) widget.clicked.connect(self.load_tou_opencage) layout.addWidget(widget) widget = CompactButton( self.tr('Geodata © OpenStreetMap\ncontributors')) widget.clicked.connect(self.load_tou_osm) layout.addWidget(widget) left_side.layout().addRow(layout) # other init self.image_list.image_list_changed.connect(self.image_list_changed) self.splitterMoved.connect(self.new_split) self.block_timer = QtCore.QTimer(self) self.block_timer.setInterval(5000) self.block_timer.setSingleShot(True) self.block_timer.timeout.connect(self.enable_search) def search_terms(self): return None @QtCore.pyqtSlot() @catch_all def load_tou_opencage(self): QtGui.QDesktopServices.openUrl( QtCore.QUrl('https://geocoder.opencagedata.com/')) @QtCore.pyqtSlot() @catch_all def load_tou_osm(self): QtGui.QDesktopServices.openUrl( QtCore.QUrl('http://www.openstreetmap.org/copyright')) @QtCore.pyqtSlot(int, int) @catch_all def new_split(self, pos, index): self.app.config_store.set('map', 'split', str(self.sizes())) @QtCore.pyqtSlot() @catch_all def image_list_changed(self): self.redraw_markers() self.display_coords() self.display_location() self.see_selection() @QtCore.pyqtSlot() @catch_all def initialise(self): page = ''' <!DOCTYPE html> <html> <head> <meta charset="utf-8" /> <meta name="viewport" content="initial-scale=1.0, user-scalable=no" /> <style type="text/css"> html, body {{ height: 100%; margin: 0; padding: 0 }} #mapDiv {{ position: relative; width: 100%; height: 100% }} </style> <script type="text/javascript"> var initData = {{key: "{key}", lat: {lat}, lng: {lng}, zoom: {zoom}}}; </script> {initialize} {head} <script type="text/javascript" src="script.js" async></script> </head> <body ondragstart="return false"> <div id="mapDiv"></div> </body> </html> ''' lat, lng = eval( self.app.config_store.get('map', 'centre', '(51.0, 0.0)')) zoom = int(eval(self.app.config_store.get('map', 'zoom', '11'))) if using_qtwebengine: initialize = ''' <script type="text/javascript" src="qrc:///qtwebchannel/qwebchannel.js"> </script> <script type="text/javascript"> var python; function initialize() { new QWebChannel(qt.webChannelTransport, function (channel) { python = channel.objects.python; loadMap(); }); } </script> ''' else: initialize = ''' <script type="text/javascript"> function initialize() { loadMap(); } </script> ''' page = page.format(lat=lat, lng=lng, zoom=zoom, initialize=initialize, key=self.api_key, head=self.get_head()) QtWidgets.QApplication.setOverrideCursor(Qt.WaitCursor) self.map.setHtml(page, QtCore.QUrl.fromLocalFile(self.script_dir)) @catch_all def initialize_finished(self): QtWidgets.QApplication.restoreOverrideCursor() self.map_loaded = True self.edit_box.setEnabled(True) self.map.setAcceptDrops(True) self.redraw_markers() self.display_coords() def refresh(self): self.setSizes( eval(self.app.config_store.get('map', 'split', str(self.sizes())))) self.image_list.set_drag_to_map(self.drag_icon, self.drag_hotspot) if not self.map_loaded: self.initialise() return lat, lng = eval(self.app.config_store.get('map', 'centre')) zoom = int(eval(self.app.config_store.get('map', 'zoom'))) self.JavaScript('setView({!r},{!r},{:d})'.format(lat, lng, zoom)) def do_not_close(self): return False @catch_all def new_status(self, status): self.map_status.update(status) for key in ('centre', 'zoom'): if key in status: self.app.config_store.set('map', key, repr(self.map_status[key])) @QtCore.pyqtSlot(int, int, six.text_type) @catch_all def drop_text(self, x, y, text): self.dropped_images = eval(text) self.JavaScript('markerDrop({:d},{:d})'.format(x, y)) @catch_all def marker_drop(self, lat, lng): for path in self.dropped_images: image = self.image_list.get_image(path) image.metadata.latlong = lat, lng self.dropped_images = [] self.redraw_markers() self.display_coords() self.see_selection() @QtCore.pyqtSlot() @catch_all def new_coords(self): text = self.coords.get_value().strip() if not text: for image in self.image_list.get_selected_images(): image.metadata.latlong = None self.redraw_markers() return try: lat, lng = map(float, text.split(',')) except Exception: self.display_coords() return for image in self.image_list.get_selected_images(): image.metadata.latlong = lat, lng self.redraw_markers() self.display_coords() self.see_selection() def see_selection(self): locations = [] for image in self.image_list.get_selected_images(): latlong = image.metadata.latlong if not latlong: continue location = [latlong.lat, latlong.lon] if location not in locations: locations.append(location) if not locations: return self.JavaScript('fitPoints({})'.format(repr(locations))) @QtCore.pyqtSlot(QtGui.QContextMenuEvent) @catch_all def location_tab_context_menu(self, event): idx = self.location_info.tabBar().tabAt(event.pos()) self.location_info.setCurrentIndex(idx) menu = QtWidgets.QMenu(self) menu.addAction(self.tr('Duplicate location'), self.duplicate_location) menu.addAction(self.tr('Delete location'), self.delete_location) action = menu.exec_(event.globalPos()) @QtCore.pyqtSlot() @catch_all def duplicate_location(self): idx = self.location_info.currentIndex() for image in self.image_list.get_selected_images(): # duplicate data location = Location(self._get_location(image, idx) or {}) # shuffle data up location_list = list(image.metadata.location_shown or []) location_list.insert(idx, location) image.metadata.location_shown = location_list # display data self.display_location() @QtCore.pyqtSlot() @catch_all def delete_location(self): idx = self.location_info.currentIndex() for image in self.image_list.get_selected_images(): # shuffle data down location_list = list(image.metadata.location_shown or []) if idx == 0: if location_list: location = location_list[0] else: location = None image.metadata.location_taken = location if idx <= len(location_list): del location_list[max(idx - 1, 0)] image.metadata.location_shown = location_list # display data self.display_location() @QtCore.pyqtSlot(int, int) @catch_all def location_tab_moved(self, idx_a, idx_b): self.pending_move = idx_a, idx_b # do actual swap when idle to avoid seg fault QtCore.QTimer.singleShot(0, self._location_tab_moved) @QtCore.pyqtSlot() @catch_all def _location_tab_moved(self): idx_a, idx_b = self.pending_move # swap data for image in self.image_list.get_selected_images(): temp_a = self._get_location(image, idx_a) temp_b = self._get_location(image, idx_b) self._set_location(image, idx_a, temp_b) self._set_location(image, idx_b, temp_a) # adjust tab names for idx in range(min(idx_a, idx_b), max(idx_a, idx_b) + 1): self.set_tab_text(idx) # display data self.display_location() def _get_location(self, image, idx): if idx == 0: return image.metadata.location_taken elif not image.metadata.location_shown: return None elif idx <= len(image.metadata.location_shown): return image.metadata.location_shown[idx - 1] return None def _set_location(self, image, idx, location): if idx == 0: image.metadata.location_taken = location else: location_list = list(image.metadata.location_shown or []) while len(location_list) < idx: location_list.append(None) location_list[idx - 1] = location image.metadata.location_shown = location_list @QtCore.pyqtSlot(object, dict) @catch_all def new_location(self, widget, new_value): idx = self.location_info.indexOf(widget) for image in self.image_list.get_selected_images(): temp = dict(self._get_location(image, idx) or {}) temp.update(new_value) self._set_location(image, idx, temp) # new_location can be called when changing tab, so don't delete # tabs until later QtCore.QTimer.singleShot(0, self.display_location) def display_coords(self): images = self.image_list.get_selected_images() if not images: self.coords.set_value(None) self.auto_location.setEnabled(False) return values = [] for image in images: value = image.metadata.latlong if value not in values: values.append(value) if len(values) > 1: self.coords.set_multiple(choices=filter(None, values)) self.auto_location.setEnabled(False) else: self.coords.set_value(values[0]) self.auto_location.setEnabled( bool(values[0]) and not self.block_timer.isActive()) def set_tab_text(self, idx): if idx == 0: text = self.tr('camera') else: text = self.tr('subject {}').format(idx) self.location_info.setTabText(idx, text) @QtCore.pyqtSlot() @catch_all def display_location(self): images = self.image_list.get_selected_images() # get required number of tabs count = 0 for image in images: if image.metadata.location_shown: count = max(count, len(image.metadata.location_shown)) count += 2 # add or remove tabs if self.location_info.currentIndex() >= count: self.location_info.setCurrentIndex(count - 1) idx = self.location_info.count() while idx < count: if not self.location_widgets: widget = LocationInfo() widget.new_value.connect(self.new_location) self.location_widgets.append(widget) self.location_info.addTab(self.location_widgets.pop(), '') self.set_tab_text(idx) idx += 1 while idx > count: idx -= 1 self.location_widgets.append(self.location_info.widget(idx)) self.location_info.removeTab(idx) # display data for idx in range(count): widget = self.location_info.widget(idx) if images: values = defaultdict(list) for image in images: location = self._get_location(image, idx) or {} for key in widget.members: value = None if key in location: value = location[key] if value not in values[key]: values[key].append(value) for key in widget.members: if len(values[key]) > 1: widget.members[key].set_multiple( choices=filter(None, values[key])) else: widget.members[key].set_value(values[key][0]) else: for key in widget.members: widget.members[key].set_value(None) @QtCore.pyqtSlot(list) @catch_all def new_selection(self, selection): self.coords.setEnabled(bool(selection)) self.location_info.setEnabled(bool(selection)) self.redraw_markers() self.display_coords() self.display_location() self.see_selection() def redraw_markers(self): if not self.map_loaded: return for info in self.marker_info.values(): info['images'] = [] for image in self.image_list.get_images(): latlong = image.metadata.latlong if not latlong: continue for info in self.marker_info.values(): if info['latlong'] == latlong: info['images'].append(image) break else: for i in range(len(self.marker_info) + 2): marker_id = i if marker_id not in self.marker_info: break self.marker_info[marker_id] = { 'images': [image], 'latlong': LatLon(latlong), 'selected': image.selected, } self.JavaScript('addMarker({:d},{!r},{!r},{:d})'.format( marker_id, latlong.lat, latlong.lon, image.selected)) for marker_id in list(self.marker_info.keys()): info = self.marker_info[marker_id] if not info['images']: self.JavaScript('delMarker({:d})'.format(marker_id)) del self.marker_info[marker_id] elif info['selected'] != any([x.selected for x in info['images']]): info['selected'] = not info['selected'] self.JavaScript('enableMarker({:d},{:d})'.format( marker_id, info['selected'])) def plot_track(self, tracks): latlngs = [] for t in tracks: latlngs.append([[x[1], x[2]] for x in t]) self.JavaScript('plotTrack({!r})'.format(latlngs)) @QtCore.pyqtSlot() @catch_all def enable_search(self): self.block_timer.stop() self.edit_box.lineEdit().setEnabled(self.map_loaded) if self.search_string: item = self.edit_box.model().item(1) item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) item = self.edit_box.model().item(2) item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled) self.display_coords() def disable_search(self): self.edit_box.lineEdit().setEnabled(False) if self.search_string: item = self.edit_box.model().item(1) item.setFlags(~(Qt.ItemIsSelectable | Qt.ItemIsEnabled)) item = self.edit_box.model().item(2) item.setFlags(~(Qt.ItemIsSelectable | Qt.ItemIsEnabled)) self.auto_location.setEnabled(False) self.block_timer.start() focus = QtWidgets.QApplication.focusWidget() if focus: focus.clearFocus() def do_geocode(self, params): cache_key = params['q'] if 'bounds' in params: cache_key += params['bounds'] if cache_key in self.geocode_cache: return self.geocode_cache[cache_key] self.disable_search() params['key'] = self.search_key params['abbrv'] = '1' params['no_annotations'] = '1' lang, encoding = locale.getdefaultlocale() if lang: params['language'] = lang with Busy(): try: rsp = requests.get( 'https://api.opencagedata.com/geocode/v1/json', params=params, timeout=5) except Exception as ex: logger.error(str(ex)) return [] if rsp.status_code >= 400: logger.error('Search error %d', rsp.status_code) return [] rsp = rsp.json() status = rsp['status'] if status['code'] != 200: logger.error('Search error %d: %s', status['code'], status['message']) return [] if rsp['total_results'] < 1: logger.error('No results found') return [] rate = rsp['rate'] self.block_timer.setInterval(5000 * rate['limit'] // max(rate['remaining'], 1)) self.geocode_cache[cache_key] = rsp['results'] while len(self.geocode_cache) > 20: self.geocode_cache.popitem(last=False) return rsp['results'] def geocode(self, search_string, bounds=None): params = { 'q': search_string, 'limit': '20', } if bounds: north, east, south, west = bounds w = east - west h = north - south if min(w, h) < 10.0: lat, lon = self.map_status['centre'] north = min(lat + 5.0, 90.0) south = max(lat - 5.0, -90.0) east = lon + 5.0 west = lon - 5.0 params['bounds'] = '{!r},{!r},{!r},{!r}'.format( west, south, east, north) for result in self.do_geocode(params): yield (result['bounds']['northeast']['lat'], result['bounds']['northeast']['lng'], result['bounds']['southwest']['lat'], result['bounds']['southwest']['lng'], result['formatted']) # Map OpenCage address components to IPTC address heirarchy. There # are many possible components (user generated data) so any # unrecognised ones are put in 'sublocation'. See # https://github.com/OpenCageData/address-formatting/blob/master/conf/components.yaml address_map = { 'world_region': ('continent', ), 'country_code': ('ISO_3166-1_alpha-3', 'ISO_3166-1_alpha-2', 'country_code'), 'country_name': ('country', 'country_name'), 'province_state': ('county', 'county_code', 'local_administrative_area', 'state_district', 'state', 'state_code', 'province', 'region', 'island'), 'city': ('village', 'locality', 'hamlet', 'neighbourhood', 'city_district', 'suburb', 'city', 'town', 'postcode'), 'sublocation': ('house_number', 'street_number', 'house', 'public_building', 'building', 'water', 'road', 'pedestrian', 'path', 'residential', 'street_name', 'street', 'cycleway', 'footway', 'place'), 'ignore': ('political_union', 'road_reference', 'road_reference_intl', 'road_type', '_type'), } @QtCore.pyqtSlot() @catch_all def get_address(self): results = self.do_geocode( {'q': self.coords.get_value().replace(' ', '')}) if not results: return address = dict(results[0]['components']) if 'county_code' in address and 'county' in address: del address['county_code'] if 'state_code' in address and 'state' in address: del address['state_code'] self.new_location(self.location_info.currentWidget(), Location.from_address(address, self.address_map)) @QtCore.pyqtSlot() @catch_all def search(self, search_string=None, bounded=True): if not search_string: search_string = self.edit_box.lineEdit().text() self.edit_box.clearEditText() if not search_string: return self.search_string = search_string self.clear_search() if bounded: bounds = self.map_status['bounds'] else: bounds = None for result in self.geocode(search_string, bounds=bounds): north, east, south, west, name = result self.edit_box.addItem(name, (north, east, south, west)) self.edit_box.set_dropdown_width() self.edit_box.showPopup() def clear_search(self): self.edit_box.clear() self.edit_box.addItem('') if self.search_string: self.edit_box.addItem(translate('PhotiniMap', '<widen search>')) self.edit_box.addItem(translate('PhotiniMap', '<repeat search>')) @QtCore.pyqtSlot(int) @catch_all def goto_search_result(self, idx): self.edit_box.setCurrentIndex(0) self.edit_box.clearFocus() if idx == 0: return if self.search_string and idx == 1: # widen search self.search(search_string=self.search_string, bounded=False) return if self.search_string and idx == 2: # repeat search self.search(search_string=self.search_string) return view = self.edit_box.itemData(idx) if view[-1] is None: self.JavaScript('setView({},{},{})'.format( view[0], view[1], self.map_status['zoom'])) else: self.JavaScript('adjustBounds({},{},{},{})'.format(*view)) @catch_all def marker_click(self, marker_id): self.image_list.select_images(self.marker_info[marker_id]['images']) @catch_all def marker_drag(self, lat, lng): self.coords.set_value('{:.6f}, {:.6f}'.format(lat, lng)) @catch_all def marker_drag_end(self, lat, lng, marker_id): info = self.marker_info[marker_id] for image in info['images']: image.metadata.latlong = lat, lng info['latlong'] = LatLon((lat, lng)) self.display_coords() def JavaScript(self, command): if self.map_loaded: self.map.page().do_java_script(command)