def autoDownloadMap(self, map_name, parent=None):
grabber = LiquipediaGrabber()
try:
QApplication.setOverrideCursor(Qt.WaitCursor)
sc2map = grabber.get_map(map_name)
except MapNotFound:
QMessageBox.critical(
self, _("Map not found"),
_('"{}" was not found on Liquipedia.').format(map_name))
return
finally:
QApplication.restoreOverrideCursor()
map_name = sc2map.get_name()
try:
QApplication.setOverrideCursor(Qt.WaitCursor)
images = grabber.get_images(sc2map.get_map_images())
image = ""
for size in sorted(images):
if not image or size <= 2500 * 2500:
image = images[size]
url = grabber._base_url + image
if parent is None:
parent = self.view
downloader = MapDownloader(parent, map_name, url)
downloader.download()
if map_name not in scctool.settings.maps:
scctool.settings.maps.append(map_name)
except Exception:
raise
finally:
QApplication.restoreOverrideCursor()
def __init__(self, manager):
super().__init__()
self.__manager = manager
self.__grabber = LiquipediaGrabber()
self.setTimeout(30)
self.addTask('refresh_data', self.__refresh_data)
self.addTask('refresh_stats', self.__refresh_stats)
self.addTask('refresh_mappool', self.__refresh_mappool)
def test_liquipedia():
grabber = LiquipediaGrabber()
mappool = list(grabber.get_ladder_mappool())
assert len(mappool) == 7
stats = list(grabber.get_map_stats(mappool))
assert len(stats) == len(mappool)
for map_stats in stats:
assert map_stats['map'] in mappool
def downloadLadderMaps(self):
players_per_team, ok = QInputDialog.getItem(
self,
_('Select the type of ladder maps to download'),
_('Please select a map type') + ':',
['1vs1', '2vs2', '3vs3', '4vs4'],
editable=False)
players_per_team = int(players_per_team[0])
found_a_map = False
for sc2map in LiquipediaGrabber().get_ladder_mappool(players_per_team):
if not sc2map in scctool.settings.maps:
found_a_map = True
self.controller.autoDownloadMap(sc2map, self)
scctool.settings.maps.append(sc2map)
items = self.maplist.findItems(sc2map, Qt.MatchExactly)
if len(items) == 0:
item = QListWidgetItem(sc2map)
self.maplist.addItem(item)
self.maplist.setCurrentItem(item)
else:
self.maplist.setCurrentItem(items[0])
self.changePreview()
if not found_a_map:
QMessageBox.information(
self, _("No missing map"),
_('All of the current ladder maps are already present.'))
def createWindow(self, mainWindow, placeholder, team):
"""Create readme sub window."""
super().__init__(None)
self.mainWindow = mainWindow
self.controller = mainWindow.controller
self.team = team
self.liquipediaGrabber = LiquipediaGrabber()
self.setWindowIcon(QIcon(
scctool.settings.getResFile("liquipedia.png")))
self.setWindowModality(Qt.ApplicationModal)
mainLayout = QGridLayout()
self.qle_search = QLineEdit(placeholder)
self.qle_search.setAlignment(Qt.AlignCenter)
self.qle_search.returnPressed.connect(self.search)
completer = QCompleter(
scctool.settings.config.getMyTeams() +
self.controller.historyManager.getTeamList(), self.qle_search)
completer.setCaseSensitivity(Qt.CaseInsensitive)
completer.setCompletionMode(QCompleter.UnfilteredPopupCompletion)
completer.setWrapAround(True)
self.qle_search.setCompleter(completer)
mainLayout.addWidget(self.qle_search, 0, 0, 1, 2)
searchButton = QPushButton(_("Search"))
searchButton.clicked.connect(self.search)
mainLayout.addWidget(searchButton, 0, 2)
self.box = QGroupBox(_("Results"))
layout = QHBoxLayout()
self.result_list = QListWidget()
self.result_list.setViewMode(QListWidget.IconMode)
self.result_list.itemDoubleClicked.connect(self.doubleClicked)
self.result_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.result_list.customContextMenuRequested.connect(
self.listItemRightClicked)
self.result_list.setIconSize(QSize(75, 75))
# list.setWrapping(False)
# list.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.result_list.setAcceptDrops(False)
self.result_list.setDragEnabled(False)
layout.addWidget(self.result_list)
self.box.setLayout(layout)
mainLayout.addWidget(self.box, 1, 0, 1, 3)
selectButton = QPushButton(" " + _("Use Selected Logo") + " ")
selectButton.clicked.connect(self.applyLogo)
closeButton = QPushButton(_("Cancel"))
closeButton.clicked.connect(self.close)
mainLayout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Minimum), 2,
0)
mainLayout.addWidget(closeButton, 2, 1)
mainLayout.addWidget(selectButton, 2, 2)
self.setLayout(mainLayout)
self.setWindowTitle(_("Liquipedia Image Search"))
self.resize(
QSize(mainWindow.size().width() * 0.9,
self.sizeHint().height()))
relativeChange = QPoint(mainWindow.size().width() / 2,
mainWindow.size().height() / 3)\
- QPoint(self.size().width() / 2,
self.size().height() / 3)
self.move(mainWindow.pos() + relativeChange)
class SubwindowLiquipediaSearch(QWidget):
"""Show readme sub window."""
nams = dict()
results = dict()
data = dict()
def createWindow(self, mainWindow, placeholder, team):
"""Create readme sub window."""
super().__init__(None)
self.mainWindow = mainWindow
self.controller = mainWindow.controller
self.team = team
self.liquipediaGrabber = LiquipediaGrabber()
self.setWindowIcon(QIcon(
scctool.settings.getResFile("liquipedia.png")))
self.setWindowModality(Qt.ApplicationModal)
mainLayout = QGridLayout()
self.qle_search = QLineEdit(placeholder)
self.qle_search.setAlignment(Qt.AlignCenter)
self.qle_search.returnPressed.connect(self.search)
completer = QCompleter(
scctool.settings.config.getMyTeams() +
self.controller.historyManager.getTeamList(), self.qle_search)
completer.setCaseSensitivity(Qt.CaseInsensitive)
completer.setCompletionMode(QCompleter.UnfilteredPopupCompletion)
completer.setWrapAround(True)
self.qle_search.setCompleter(completer)
mainLayout.addWidget(self.qle_search, 0, 0, 1, 2)
searchButton = QPushButton(_("Search"))
searchButton.clicked.connect(self.search)
mainLayout.addWidget(searchButton, 0, 2)
self.box = QGroupBox(_("Results"))
layout = QHBoxLayout()
self.result_list = QListWidget()
self.result_list.setViewMode(QListWidget.IconMode)
self.result_list.itemDoubleClicked.connect(self.doubleClicked)
self.result_list.setContextMenuPolicy(Qt.CustomContextMenu)
self.result_list.customContextMenuRequested.connect(
self.listItemRightClicked)
self.result_list.setIconSize(QSize(75, 75))
# list.setWrapping(False)
# list.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.result_list.setAcceptDrops(False)
self.result_list.setDragEnabled(False)
layout.addWidget(self.result_list)
self.box.setLayout(layout)
mainLayout.addWidget(self.box, 1, 0, 1, 3)
selectButton = QPushButton(" " + _("Use Selected Logo") + " ")
selectButton.clicked.connect(self.applyLogo)
closeButton = QPushButton(_("Cancel"))
closeButton.clicked.connect(self.close)
mainLayout.addItem(
QSpacerItem(0, 0, QSizePolicy.Expanding, QSizePolicy.Minimum), 2,
0)
mainLayout.addWidget(closeButton, 2, 1)
mainLayout.addWidget(selectButton, 2, 2)
self.setLayout(mainLayout)
self.setWindowTitle(_("Liquipedia Image Search"))
self.resize(
QSize(mainWindow.size().width() * 0.9,
self.sizeHint().height()))
relativeChange = QPoint(mainWindow.size().width() / 2,
mainWindow.size().height() / 3)\
- QPoint(self.size().width() / 2,
self.size().height() / 3)
self.move(mainWindow.pos() + relativeChange)
def clean(self):
self.nams.clear()
self.results.clear()
self.data.clear()
def search(self):
QApplication.setOverrideCursor(Qt.WaitCursor)
self.clean()
loading_map = QPixmap(scctool.settings.getResFile("loading.png"))
try:
self.result_list.clear()
idx = 0
search_str = self.qle_search.text()
for name, thumb in self.liquipediaGrabber.image_search(search_str):
self.data[idx] = name
name = name.replace('/commons/File:', '')
self.results[idx] = QListWidgetItem(QIcon(loading_map), name)
self.results[idx].setSizeHint(QSize(80, 90))
url = base_url + thumb
self.nams[idx] = QNetworkAccessManager()
self.nams[idx].finished.connect(
lambda reply, i=idx: self.finishRequest(reply, i))
self.nams[idx].get(QNetworkRequest(QUrl(url)))
self.result_list.addItem(self.results[idx])
if idx == 0:
self.result_list.setCurrentItem(self.results[idx])
idx += 1
self.box.setTitle(
_("Results for '{}': {}").format(search_str, idx))
except Exception as e:
module_logger.exception("message")
finally:
QApplication.restoreOverrideCursor()
def finishRequest(self, reply, idx):
img = QImage()
img.loadFromData(reply.readAll())
map = QPixmap(img).scaled(75, 75, Qt.KeepAspectRatio)
self.results[idx].setIcon(QIcon(map))
def applyLogo(self, skip=False):
QApplication.setOverrideCursor(Qt.WaitCursor)
item = self.result_list.currentItem()
if item is not None and (skip or item.isSelected()):
for idx, iteritem in self.results.items():
if item is iteritem:
images = self.liquipediaGrabber.get_images(self.data[idx])
image = ""
for size in sorted(images):
if not image or size <= 600 * 600:
image = images[size]
self.downloadLogo(base_url + image)
break
QApplication.restoreOverrideCursor()
self.close()
def doubleClicked(self, item):
QApplication.setOverrideCursor(Qt.WaitCursor)
for idx, iteritem in self.results.items():
if item is iteritem:
images = self.liquipediaGrabber.get_images(self.data[idx])
image = ""
for size in sorted(images):
if not image or size <= 600 * 600:
image = images[size]
self.downloadLogo(base_url + image)
break
QApplication.restoreOverrideCursor()
self.close()
def listItemRightClicked(self, QPos):
self.listMenu = QMenu()
menu_item = self.listMenu.addAction(_("Open on Liquipedia"))
menu_item.triggered.connect(self.openLiquipedia)
menu_item = self.listMenu.addAction(_("Use as Team Logo"))
menu_item.triggered.connect(lambda: self.applyLogo(True))
parentPosition = self.result_list.mapToGlobal(QPoint(0, 0))
self.listMenu.move(parentPosition + QPos)
self.listMenu.show()
def openLiquipedia(self):
item = self.result_list.currentItem()
for idx, iteritem in self.results.items():
if item is iteritem:
url = base_url + self.data[idx]
self.controller.openURL(url)
break
def downloadLogo(self, url):
logo = LogoDownloader(self.controller, self, url).download()
logo.refreshData()
map = logo.provideQPixmap()
if self.team == 1:
self.controller.logoManager.setTeam1Logo(logo)
self.mainWindow.team1_icon.setPixmap(map)
self.mainWindow.refreshLastUsed()
elif self.team == 2:
self.controller.logoManager.setTeam2Logo(logo)
self.mainWindow.team2_icon.setPixmap(map)
self.mainWindow.refreshLastUsed()
def closeEvent(self, event):
"""Handle close event."""
try:
self.clean()
event.accept()
except Exception as e:
module_logger.exception("message")
class MapStatsThread(TasksThread):
newMapData = pyqtSignal(str, object)
newMapPool = pyqtSignal(object)
def __init__(self, manager):
super().__init__()
self.__manager = manager
self.__grabber = LiquipediaGrabber()
self.setTimeout(30)
self.addTask('refresh_data', self.__refresh_data)
self.addTask('refresh_stats', self.__refresh_stats)
self.addTask('refresh_mappool', self.__refresh_mappool)
def setMaps(self, maps, full=False):
if full:
self.__fullmaps = maps
else:
self.__maps = maps
def __refresh_data(self):
try:
map = self.__fullmaps.pop()
try:
liquipediaMap = self.__grabber.get_map(map)
stats = liquipediaMap.get_stats()
info = liquipediaMap.get_info()
data = dict()
data['tvz'] = stats['tvz']
data['zvp'] = stats['zvp']
data['pvt'] = stats['pvt']
data['creator'] = info['creator']
data['size'] = info['size']
data['spawn-positions'] = info['spawn-positions']
data['refreshed'] = int(time.time())
self.newMapData.emit(map, data)
module_logger.info('Map {} found.'.format(map))
except MapNotFound:
module_logger.info('Map {} not found.'.format(map))
except ConnectionError:
module_logger.info('Connection Error for map {}.'.format(map))
except Exception as e:
module_logger.exception("message")
except IndexError:
self.deactivateTask('refresh_data')
def __refresh_stats(self):
try:
for stats in self.__grabber.get_map_stats(self.__maps):
map = stats['map']
data = dict()
data['tvz'] = stats['tvz']
data['zvp'] = stats['zvp']
data['pvt'] = stats['pvt']
data['refreshed'] = int(time.time())
module_logger.info('Map {} found.'.format(map))
self.newMapData.emit(map, data)
finally:
self.deactivateTask('refresh_stats')
def __refresh_mappool(self):
try:
mappool = list(self.__grabber.get_ladder_mappool())
self.newMapPool.emit(mappool)
module_logger.info('Current map pool found.')
finally:
self.deactivateTask('refresh_mappool')
def addFromLquipedia(self):
"""Add a map from Liquipedia."""
grabber = LiquipediaGrabber()
search_str = ''
while True:
search_str, ok = QInputDialog.getText(self,
_('Map Name'),
_('Map Name') + ':',
text=search_str)
search_str.strip()
try:
if ok and search_str:
if search_str.lower() == 'tbd':
QMessageBox.critical(
self, _("Error"),
_('"{}" is not a valid map name.').format(
search_str))
continue
try:
QApplication.setOverrideCursor(Qt.WaitCursor)
sc2map = grabber.get_map(search_str)
except MapNotFound:
QMessageBox.critical(
self, _("Map not found"),
_('"{}" was not found on Liquipedia.').format(
search_str))
continue
finally:
QApplication.restoreOverrideCursor()
map_name = sc2map.get_name()
if (map_name in scctool.settings.maps):
buttonReply = QMessageBox.warning(
self, _("Duplicate Entry"),
_("Map {} is already in list! Overwrite?".format(
map_name)), QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if buttonReply == QMessageBox.No:
break
else:
self.controller.deleteMap(map_name)
try:
QApplication.setOverrideCursor(Qt.WaitCursor)
images = grabber.get_images(sc2map.get_map_images())
image = ""
for size in sorted(images):
if not image or size <= 2500 * 2500:
image = images[size]
url = grabber._base_url + image
downloader = MapDownloader(self, map_name, url)
downloader.download()
if map_name not in scctool.settings.maps:
scctool.settings.maps.append(map_name)
items = self.maplist.findItems(map_name,
Qt.MatchExactly)
if len(items) == 0:
item = QListWidgetItem(map_name)
self.maplist.addItem(item)
self.maplist.setCurrentItem(item)
else:
self.maplist.setCurrentItem(items[0])
self.changePreview()
except Exception:
raise
finally:
QApplication.restoreOverrideCursor()
except Exception as e:
module_logger.exception("message")
QMessageBox.critical(self, _("Error"), str(e))
break