/
gui.py
executable file
·143 lines (108 loc) · 4.06 KB
/
gui.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
#!/usr/bin/python
# Author: Marco Dinacci - www.intransitione.com -
from PySide.QtCore import *
from PySide.QtGui import *
import photoflow, config
from os.path import expanduser as getHomeDir
# TODO launch in a new thread the import process
# every image processed send a signal from the import thread
# and set the current filename here.
# Use a timer that every 20 ms or so refresh the image with the current one
# as there's no need to resize ans show every photo, the human eya is not that fast
class MainWindow(QMainWindow):
_source = None
def __init__(self):
super(MainWindow, self).__init__()
self.setWindowTitle("Photoflow")
self.setFixedSize(350,150)
hbox = QHBoxLayout()
# browse button and label
self.dirLabel = QLabel("Select the directory containing the photos")
self.dirBtn = QPushButton("Browse")
self.dirBtn.clicked.connect(self._showImportDialog)
hbox.addWidget(self.dirLabel)
hbox.addWidget(self.dirBtn)
vbox = QVBoxLayout()
vbox.addLayout(hbox)
"""
# preview image
self.preview = QPixmap()
self._previewLabel = QLabel()
self._previewLabel.setPixmap(self.preview)
vbox.addWidget(self._previewLabel)
"""
self.progressBar = QProgressBar()
self.progressBar.setEnabled(False)
self.progressBar.setVisible(False)
vbox.addWidget(self.progressBar)
# import button
self.importBtn = QPushButton("Import")
self.importBtn.setEnabled(False)
self.importBtn.clicked.connect(self._import)
vbox.addWidget(self.importBtn)
self.mainWidget = QWidget()
self.mainWidget.setLayout(vbox)
self.setCentralWidget(self.mainWidget)
self.resize(400, 300)
self.message("Ready.")
def update(self, filename):
self.message("Importing %s" % filename)
if self.progressBar.isEnabled():
self.progressBar.setValue(self.progressBar.value()+1)
"""
self._currentImage = filename
img = QImage(filename)
self.preview = QPixmap.fromImage(img).scaledToWidth(380)
self._previewLabel.setPixmap(self.preview)
self._previewLabel.adjustSize()
"""
def set_total(self, total):
if total > 0:
self.total = total
self.progressBar.reset()
self.progressBar.setRange(1,total)
self.progressBar.setEnabled(True)
self.progressBar.setVisible(True)
def alert(self, msg):
dialog = QDialog()
label = QLabel(msg)
closeBtn = QPushButton("Close")
closeBtn.clicked.connect(lambda:dialog.hide())
vbox = QVBoxLayout()
vbox.addWidget(label)
vbox.addWidget(closeBtn)
dialog.setLayout(vbox)
dialog.show()
def exit(self):
pass
def message(self, msg, timeout = 0):
self.statusBar().showMessage(msg, timeout)
def _import(self):
if self._source is not None:
mm = photoflow.MediaManager()
mm.set_source(self._source)
mm.set_destination(config.PHOTOS_DESTINATION)
mm.register_gui(self)
self.message("Importing photos...")
mm.import_photos()
# TODO show progress
self.message("Finished importing photos", 5000)
mm.set_destination(config.VIDEOS_DESTINATION)
self.message("Importing videos...")
mm.import_videos()
self.message("Finished importing videos", 5000)
def _showImportDialog(self):
options = QFileDialog.DontResolveSymlinks | QFileDialog.ShowDirsOnly
directory = QFileDialog.getExistingDirectory(self, self.dirLabel.text(),
getHomeDir("~"), options)
if directory:
self._source = directory
self.dirLabel.setText(directory)
self.importBtn.setEnabled(True)
if __name__ == "__main__":
import sys
app = QApplication([])
win = MainWindow()
win.show()
app.exec_()
sys.exit()