def loadDatabase(self):
print("loading db")
#TODO: Decide if a local copy should be used
#TODO: download the file if it doesn't exist locally
#TODO: update the file if it exits locally
try:
self._card_database = CardDatabase()
except Exception as e:
self.errorMessage("Failed to load card database (%s)\r\nExiting..." % str(e), block=True)
print(e)
sys.exit(1)
class MainWindow(QMainWindow, Ui_MainWindow):
_card_database = []
_cubeData = None
_cubeFile = None
def __init__(self, *args, **kwargs):
QMainWindow.__init__(self, *args, **kwargs)
self.setupUi(self)
self.errorMessageDialog = QErrorMessage(parent=self)
self.errorMessageDialog.setWindowModality(Qt.ApplicationModal)
self.grouping_model = GroupingModel(parent=self)
self.tablemodel = CubeModel(parent=self)
self.comboBox_Grouping.setModel(self.grouping_model)
self.tableView.setModel(self.tablemodel)
self.tableView.setItemDelegate(CardsDelegate(self))
self.tableView.setEditTriggers(QAbstractItemView.DoubleClicked)
self.tableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.tableView.customContextMenuRequested.connect(self.contextMenuEvent)
QTimer.singleShot(500, lambda: self._initCont(*sys.argv))
def _initCont(self, *args):
if self.dockWidget_Breakdown.isFloating():
self.dockWidget_Breakdown.move(self.pos().x() + self.width() + 20,
self.pos().y())
self.dockWidget_Breakdown.resize(-1,
self.height())
self.loadDatabase()
if len(args) > 1:
self.errorMessage(args[1])
self._fileLoad(args[1])
# ------------------------------- Read-Only Properties --------------------------------
#placehodler
# ------------------------------- ??? --------------------------------
def errorMessage(self, msg, block=False):
self.errorMessageDialog.showMessage(str(msg))
if block:
self.errorMessageDialog.exec_()
def loadDatabase(self):
print("loading db")
#TODO: Decide if a local copy should be used
#TODO: download the file if it doesn't exist locally
#TODO: update the file if it exits locally
try:
self._card_database = CardDatabase()
except Exception as e:
self.errorMessage("Failed to load card database (%s)\r\nExiting..." % str(e), block=True)
print(e)
sys.exit(1)
# ------------------------------- FILE Actions --------------------------------
FILE_EXT_FORMATS = 'Cube Data File (*.cube);;XML File (*.xml);;All Files (*)'
def fileOpen(self):
options = QFileDialog.Options()
cubeFile = QFileDialog.getOpenFileName(self,
"Open Cube Data file",
self._cubeFile,
self.FILE_EXT_FORMATS,
options=options)
if cubeFile[0]:
self._fileLoad(cubeFile[0])
def fileSave(self):
options = QFileDialog.Options()
cubeFile = QFileDialog.getSaveFileName(self,
"Save Cube Data File",
self._cubeFile,
self.FILE_EXT_FORMATS,
options=options)
if cubeFile:
self._fileSave(cubeFile)
def _fileLoad(self, cubeFile):
print("Parsing {0}".format(cubeFile))
try:
data = etree.parse(cubeFile)
self._cubeData = data
self.grouping_model.beginResetModel()
self.grouping_model.endResetModel()
self._cubeFile = cubeFile
except Exception as e:
self.errorMessage("Failed to load file " + str(e))
print(e)
def _fileSave(self, cubeFile=None):
if cubeFile == None:
cubeFile = self._cubeFile
if not cubeFile:
self.fileSave()
return
self._cubeData.write(cubeFile,
pretty_print=True,
method="xml",
encoding="UTF-8",
xml_declaration=True)
# ------------------------------- HELP Actions --------------------------------
def helpAbout(self):
QMessageBox.about(self,
"About %s" % __appname__,
__doc__.strip())
# ------------------------------- ??? Actions --------------------------------
def resetTable(self):
self.tablemodel.beginResetModel()
self.tablemodel.endResetModel()
for i in range(self.tablemodel.columnCount()):
self.tableView.setColumnWidth(i, CARD_WIDTH)
for i in range(self.tablemodel.rowCount()):
self.tableView.setRowHeight(i, CARD_HEIGHT)
# ------------------------------- EXPORT Actions --------------------------------
def exportToGoogleSpreadSheet(self, username, password, sheetname):
# Login with your Google account
gc = gspread.login(username, password)
# Open a worksheet from spreadsheet with one shot
sh = gc.open(sheetname)
wks = sh.worksheet(self.comboBox_Grouping.currentText())
for i, vhead in enumerate(self.tablemodel.v_header):
wks.update_cell(i+2, 1, vhead)
for j, hhead in enumerate(self.tablemodel.h_header):
wks.update_cell(1, j+2, hhead)
d = self.tablemodel.index(i, j).data(Qt.DisplayRole).toPyObject()
if d:
cell_data = r'=IMAGE("%s", 4, %s, %s)' % (quote(self._card_database.findByName(d)[0].imageURL), CARD_HEIGHT, CARD_WIDTH)
else:
cell_data = r''
wks.update_cell(i+2, j+2, cell_data)
# Fetch a cell range
#cell_list = wks.range('A1:B7')
#print cell_list
print("DONE!")
# ------------------------------- Table Helpers --------------------------------
def contextMenuEvent(self, event):
if self.tableView.selectionModel().selection().indexes():
for i in self.tableView.selectionModel().selection().indexes():
row, column = i.row(), i.column()
self.menu = QMenu(self)
renameAction = QAction('Rename', self)
renameAction.triggered.connect(lambda: self.renameSlot(i))
self.menu.addAction(renameAction)
#queryAction = QAction('Query Cards', self)
#queryAction.triggered.connect(lambda: self.querySlot(i))
#self.menu.addAction(queryAction)
deleteAction = QAction('Delete Card', self)
deleteAction.triggered.connect(lambda: self.deleteSlot(i))
self.menu.addAction(deleteAction)
self.menu.popup(QCursor.pos())
def renameSlot(self, i):
prev_text = i.data(Qt.DisplayRole).toString()
text, ok = QInputDialog.getText(self,
'Input Dialog',
'Enter card name:',
text=prev_text)
if ok:
self.tablemodel.setData(i.row(), i.column(), text)
if __name__ == "__main__":
import os
import sys
import random
from lxml import etree
from PyQt5.QtWidgets import QApplication #Required due to card using QPixMap
fpath, _ = os.path.split(__file__)
sys.path.append(os.path.join(fpath, ".."))
from mtg.Database import CardDatabase
app = QApplication(sys.argv)
db = CardDatabase()
data = etree.parse(r"MyCube.xml")
cards = data.xpath(".//cards/card")
main_cubedb = db.filterInNameList([card.text for card in cards])
main_cubedb = main_cubedb.sort(key = lambda card: card.name.lower())
main_cubedb = main_cubedb.filterDuplicates()
for pack in data.xpath(".//booster/pack"):
pack = BoosterPack(pack)
print(pack)
for p in range(0, 10):