示例#1
0
 def _find_local(self, photo, unknowns):
     granularity = int(photo['datetakengranularity'])
     min_taken_date = datetime.strptime(
         photo['datetaken'], '%Y-%m-%d %H:%M:%S')
     if granularity <= 0:
         max_taken_date = min_taken_date + timedelta(seconds=1)
     elif granularity <= 4:
         max_taken_date = min_taken_date + timedelta(days=31)
     else:
         max_taken_date = min_taken_date + timedelta(days=366)
     candidates = []
     for candidate in unknowns:
         if not candidate.metadata.date_taken:
             continue
         date_taken = candidate.metadata.date_taken['datetime']
         if date_taken < min_taken_date or date_taken > max_taken_date:
             continue
         candidates.append(candidate)
     if not candidates:
         return None
     rsp = requests.get(photo['url_t'])
     if rsp.status_code == 200:
         flickr_icon = rsp.content
     else:
         logger.error('HTTP error %d (%s)', rsp.status_code, photo['url_t'])
         return None
     # get user to choose matching image file
     dialog = QtWidgets.QDialog(parent=self)
     dialog.setWindowTitle(translate('FlickrTab', 'Select an image'))
     dialog.setLayout(QtWidgets.QFormLayout())
     dialog.layout().setFieldGrowthPolicy(
         QtWidgets.QFormLayout.AllNonFixedFieldsGrow)
     pixmap = QtGui.QPixmap()
     pixmap.loadFromData(flickr_icon)
     label = QtWidgets.QLabel()
     label.setPixmap(pixmap)
     dialog.layout().addRow(label, QtWidgets.QLabel(translate(
         'FlickrTab', 'Which image file matches\nthis picture on Flickr?')))
     divider = QtWidgets.QFrame()
     divider.setFrameStyle(QtWidgets.QFrame.HLine)
     dialog.layout().addRow(divider)
     buttons = {}
     for candidate in candidates:
         label = QtWidgets.QLabel()
         pixmap = candidate.image.pixmap()
         if pixmap:
             label.setPixmap(pixmap)
         else:
             label.setText(candidate.image.text())
         button = QtWidgets.QPushButton(
             os.path.basename(candidate.path))
         button.setToolTip(candidate.path)
         button.setCheckable(True)
         button.clicked.connect(dialog.accept)
         dialog.layout().addRow(label, button)
         buttons[button] = candidate
     button = QtWidgets.QPushButton(translate('FlickrTab', 'No match'))
     button.setDefault(True)
     button.clicked.connect(dialog.reject)
     dialog.layout().addRow('', button)
     if dialog.exec_() == QtWidgets.QDialog.Accepted:
         for button, candidate in buttons.items():
             if button.isChecked():
                 return candidate
     return None
示例#2
0
 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)