232 lines
8.8 KiB
Python
232 lines
8.8 KiB
Python
#!/bin/env python
|
|
import os, sys, urllib.parse, re, magic, shutil
|
|
from appdirs import AppDirs
|
|
from PyQt5 import QtCore, QtGui, QtWidgets, QtWebEngineWidgets
|
|
from PyQt5.QtCore import QSettings, QCoreApplication, QUrl, QTimer
|
|
from PyQt5.QtWidgets import QApplication, QWidget, qApp, QMainWindow, QStyle, QAction, QTabWidget, QInputDialog, QFileDialog, QLineEdit
|
|
from PyQt5.QtGui import QIcon
|
|
from PyQt5.QtWebEngineWidgets import QWebEngineView, QWebEngineDownloadItem
|
|
from PyQt5.QtWebChannel import QWebChannel
|
|
|
|
DATA_DIR = AppDirs("TabletopLibrary", "knotteye").user_data_dir
|
|
LIB_DIR = '/lib'
|
|
|
|
def _mkdir(_dir):
|
|
if os.path.isdir(_dir): pass
|
|
elif os.path.isfile(_dir):
|
|
raise OSError("%s exists as a regular file." % _dir)
|
|
else:
|
|
parent, directory = os.path.split(_dir)
|
|
if parent and not os.path.isdir(parent): _mkdir(parent)
|
|
if directory: os.mkdir(_dir)
|
|
|
|
|
|
class TabBar(QtWidgets.QTabBar):
|
|
def tabSizeHint(self, index):
|
|
s = QtWidgets.QTabBar.tabSizeHint(self, index)
|
|
s.transpose()
|
|
return s
|
|
|
|
def paintEvent(self, event):
|
|
painter = QtWidgets.QStylePainter(self)
|
|
opt = QtWidgets.QStyleOptionTab()
|
|
|
|
for i in range(self.count()):
|
|
self.initStyleOption(opt, i)
|
|
painter.drawControl(QtWidgets.QStyle.CE_TabBarTabShape, opt)
|
|
painter.save()
|
|
|
|
s = opt.rect.size()
|
|
s.transpose()
|
|
r = QtCore.QRect(QtCore.QPoint(), s)
|
|
r.moveCenter(opt.rect.center())
|
|
opt.rect = r
|
|
|
|
c = self.tabRect(i).center()
|
|
painter.translate(c)
|
|
painter.rotate(90)
|
|
painter.translate(-c)
|
|
painter.drawControl(QtWidgets.QStyle.CE_TabBarTabLabel, opt);
|
|
painter.restore()
|
|
|
|
|
|
class TabWidget(QtWidgets.QTabWidget):
|
|
def __init__(self, *args, **kwargs):
|
|
QtWidgets.QTabWidget.__init__(self, *args, **kwargs)
|
|
self.setTabBar(TabBar(self))
|
|
self.setTabPosition(QtWidgets.QTabWidget.West)
|
|
|
|
|
|
class App(QMainWindow):
|
|
settings = QSettings()
|
|
currentDownloads = []
|
|
def __init__(self):
|
|
self.vertTabDict = {}
|
|
super().__init__()
|
|
self.title = 'Tabletop Library'
|
|
self.initUI()
|
|
|
|
def initUI(self):
|
|
self.setWindowTitle(self.title)
|
|
self.setGeometry(self.settings.value('left', type=int) or 10, self.settings.value('top', type=int) or 10, self.settings.value('width', type=int) or 640, self.settings.value('height', type=int) or 480)
|
|
|
|
exitAction = QAction('&Exit', self)
|
|
exitAction.setShortcut('Ctrl+Q')
|
|
exitAction.setStatusTip('Exit application')
|
|
exitAction.triggered.connect(self.close)
|
|
|
|
saveAction = QAction('Save', self, triggered=self.saveCurrent)
|
|
saveAction.setShortcut('Ctrl+S')
|
|
saveAction.setStatusTip('Save current tab')
|
|
|
|
saveAllAction = QAction('Save All', self, triggered=self.saveAll)
|
|
saveAllAction.setShortcut('Ctrl+Shift+S')
|
|
saveAllAction.setStatusTip('Save all tabs')
|
|
|
|
newCharAction = QAction("New Collection", self, triggered=self.newCharDialog)
|
|
newCharAction.setShortcut('Ctrl+Shift+N')
|
|
|
|
newFileAction = QAction("New File", self, triggered=self.newFileDialog)
|
|
newFileAction.setShortcut('Ctrl+N')
|
|
|
|
menubar = self.menuBar()
|
|
|
|
filemenu = menubar.addMenu("File")
|
|
filemenu.addAction(newFileAction)
|
|
filemenu.addAction(newCharAction)
|
|
filemenu.insertSeparator(QAction())
|
|
filemenu.addAction(saveAction)
|
|
filemenu.addAction(saveAllAction)
|
|
filemenu.insertSeparator(QAction())
|
|
filemenu.addAction(exitAction)
|
|
|
|
self.saveOnExitPref = QAction('Save On Exit', self, checkable=True)
|
|
self.saveOnExitPref.setChecked(self.settings.value('saveOnExit', type=bool))
|
|
self.saveOnExitPref.setToolTip('May cause the app to be sluggish when closing')
|
|
|
|
editmenu = menubar.addMenu("Edit")
|
|
editmenu.setToolTipsVisible(True)
|
|
editmenu.addAction(self.saveOnExitPref)
|
|
|
|
self.vtabs = TabWidget()
|
|
|
|
self.getChars()
|
|
|
|
self.setCentralWidget(self.vtabs)
|
|
self.show()
|
|
|
|
QtWebEngineWidgets.QWebEngineProfile.defaultProfile().downloadRequested.connect( self.on_downloadRequested )
|
|
|
|
def closeEvent(self, event):
|
|
if self.saveOnExitPref.isChecked():
|
|
self.saveAll()
|
|
if not self.currentDownloads and self.saveOnExitPref.isChecked():
|
|
QTimer.singleShot(200, self.close)
|
|
event.ignore()
|
|
return
|
|
self.settings.setValue("left", self.x())
|
|
self.settings.setValue("top", self.y())
|
|
self.settings.setValue("width", self.width())
|
|
self.settings.setValue("height", self.height())
|
|
self.settings.setValue('saveOnExit', self.saveOnExitPref.isChecked())
|
|
event.accept()
|
|
|
|
def getChars(self):
|
|
_mkdir(DATA_DIR+"/collections")
|
|
for _, dirs, _ in os.walk(DATA_DIR+"/collections"):
|
|
for direc in dirs:
|
|
self.addChar(direc)
|
|
break
|
|
|
|
def addChar(self, name):
|
|
_mkdir(DATA_DIR+"/collections/"+name)
|
|
self.vertTabDict[name] = CTab(QTabWidget(), name)
|
|
self.vtabs.addTab(self.vertTabDict[name].qobject, name)
|
|
for _, _, files in os.walk(DATA_DIR+"/collections/"+name):
|
|
for file in files:
|
|
mime = magic.from_file(DATA_DIR+"/collections/"+name+"/"+file, mime=True)
|
|
if(mime == "application/pdf"):
|
|
self.vertTabDict[name].makeChild(file)
|
|
return
|
|
break
|
|
|
|
def saveCurrent(self):
|
|
self.vtabs.currentWidget().currentWidget().page().runJavaScript('document.getElementById("download").click()')
|
|
|
|
@QtCore.pyqtSlot("QWebEngineDownloadItem*")
|
|
def on_downloadRequested(self, download):
|
|
regex = re.compile('^file://[\\/\w:]+viewer\.html\?file=')
|
|
#strip out the file://whatever/viewer.html?file= and we're left with the file path
|
|
download.setPath(regex.sub('', download.page().url().toString()))
|
|
download.accept()
|
|
self.currentDownloads.append(download)
|
|
download.finished.connect(self.finishDownload)
|
|
|
|
def finishDownload(self):
|
|
for d in self.currentDownloads:
|
|
if d.isFinished():
|
|
del d
|
|
return
|
|
|
|
def saveAll(self):
|
|
for vt in self.vertTabDict:
|
|
for t in self.vertTabDict[vt].children:
|
|
self.vertTabDict[vt].children[t].page().runJavaScript('document.getElementById("download").click()')
|
|
|
|
def newCharDialog(self):
|
|
text, ok = QInputDialog.getText(self, "New Collection", "Name:", QLineEdit.Normal, "")
|
|
if ok and text != '':
|
|
self.addChar(text)
|
|
|
|
def newFileDialog(self):
|
|
qfd = QFileDialog(self)
|
|
qfd.setFileMode(QFileDialog.ExistingFile)
|
|
qfd.setNameFilters(["TabletopLibrary Documents (*.pdf)", "All Files (*)"])
|
|
item, ok = QInputDialog.getItem(self, "Add to which collection?","Collection:", self.vertTabDict.keys(), self.vtabs.currentIndex(), False)
|
|
if not ok:
|
|
return
|
|
qfd.filesSelected.connect(self.vertTabDict[item].addFile)
|
|
qfd.show()
|
|
|
|
|
|
class CTab(object):
|
|
def __init__(self, qobject, name, children={}):
|
|
self.qobject = qobject
|
|
self.children = children
|
|
self.name = name
|
|
|
|
def makeChild(self, filename):
|
|
print('makechild')
|
|
self.children[filename] = QWebEngineView()
|
|
self.qobject.addTab(self.children[filename], filename)
|
|
try:
|
|
viewerfile = sys.argv[1]
|
|
except:
|
|
if os.path.isfile(DATA_DIR+'/minified/web/viewer.html'):
|
|
viewerfile = DATA_DIR+'/minified/web/viewer.html'
|
|
elif os.path.isfile(LIB_DIR+'/tabletoplibrary/minified/web/viewer.html'):
|
|
viewerfile = LIB_DIR+'/tabletoplibrary/minified/web/viewer.html'
|
|
else:
|
|
print("No viewer file")
|
|
exit(1)
|
|
self.children[filename].load(QUrl('file://'+viewerfile+'?file='+urllib.parse.quote(DATA_DIR+'/collections/'+self.name+'/'+filename)))
|
|
self.children[filename].show()
|
|
|
|
def addFile(self, pathlist):
|
|
# Check for duplicates because of this bug
|
|
# https://forum.qt.io/topic/41356/qfiledialog-signal-fileselected-is-triggered-twice-for-one-user-choice-changed-in-recent-qt/
|
|
for path in pathlist:
|
|
_mkdir(DATA_DIR+"/collections/"+self.name)
|
|
if os.path.isfile(DATA_DIR+'/collections/'+self.name+'/'+os.path.basename(path)):
|
|
return
|
|
shutil.copyfile(path, DATA_DIR+'/collections/'+self.name+'/'+os.path.basename(path))
|
|
self.makeChild(os.path.basename(path))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app = QApplication(sys.argv)
|
|
app.setOrganizationName("knotteye")
|
|
app.setApplicationName("Tabletop Library")
|
|
ex = App()
|
|
sys.exit(app.exec_())
|