21
21
import time
22
22
from collections import defaultdict
23
23
24
- from aqt . qt import *
24
+ from PyQt4 import QtCore , QtGui
25
25
from .lang import _
26
26
27
27
28
+ INFO_TEMPLATE = u'' .join ([
29
+ _ ('QUERIED' ) + u'<br>' + 45 * u'-' + u'<br>' ,
30
+ _ ('SUCCESS' ) + u' {} ' + _ ('WORDS' ) + u'<br>' ,
31
+ _ ('SKIPED' ) + u' {} ' + _ ('WORDS' ) + u'<br>' ,
32
+ _ ('UPDATE' ) + u' {} ' + _ ('FIELDS' ) + u'<br>' ,
33
+ _ ('FAILURE' ) + u' {} ' + _ ('WORDS' ) + u'<br>'
34
+ ])
35
+
36
+
28
37
class ProgressWindow (object ):
29
38
"""
30
39
Query progress window
31
40
"""
32
41
33
42
def __init__ (self , mw ):
34
43
self .mw = mw
35
- self .app = QApplication .instance ()
44
+ self .app = QtGui . QApplication .instance ()
36
45
self ._win = None
37
46
self ._msg_count = defaultdict (int )
38
47
@@ -45,22 +54,20 @@ def update_labels(self, data):
45
54
else :
46
55
return
47
56
48
- number_info = ''
49
- words_number , fields_number , fails_number = (
50
- self ._msg_count [ 'words_number' ] ,
51
- self ._msg_count [ 'fields_number' ] ,
52
- self ._msg_count [ 'fails_number' ]
57
+ words_number , fields_number , fails_number , skips_number = (
58
+ self . _msg_count . get ( 'words_number' , 0 ),
59
+ self ._msg_count . get ( 'fields_number' , 0 ) ,
60
+ self ._msg_count . get ( 'fails_number' , 0 ) ,
61
+ self ._msg_count . get ( 'skips_number' , 0 )
53
62
)
54
- if words_number or fields_number :
55
- number_info += _ ('QUERIED' ) + u'<br>' + 45 * u'-'
56
- number_info += u'<br>{0}: {1} {2}' .format (
57
- _ ('SUCCESS' ), words_number , _ ('WORDS' ))
58
- number_info += u'<br>{0}: {1} {2}' .format (
59
- _ ('UPDATE' ), fields_number , _ ('FIELDS' ))
60
- number_info += u'<br>{0}: {1} {2}' .format (
61
- _ ('FAILURE' ), fails_number , _ ('WORDS' ))
62
-
63
- self ._update (label = number_info , value = words_number )
63
+ number_info = INFO_TEMPLATE .format (
64
+ words_number ,
65
+ skips_number ,
66
+ fields_number ,
67
+ fails_number
68
+ )
69
+
70
+ self ._update (label = number_info , value = words_number + skips_number + fails_number )
64
71
self ._win .adjustSize ()
65
72
66
73
def update_title (self , title ):
@@ -73,8 +80,8 @@ def start(self, max=0, min=0, label=None, parent=None):
73
80
# setup window
74
81
label = label or _ ("Processing..." )
75
82
parent = parent or self .app .activeWindow () or self .mw
76
- self ._win = QProgressDialog (label , '' , min , max , parent )
77
- self ._win .setWindowModality (Qt .ApplicationModal )
83
+ self ._win = QtGui . QProgressDialog (label , '' , min , max , parent )
84
+ self ._win .setWindowModality (QtCore . Qt .ApplicationModal )
78
85
self ._win .setCancelButton (None )
79
86
self ._win .canceled .connect (self .finish )
80
87
self ._win .setWindowTitle ("Querying..." )
@@ -107,12 +114,12 @@ def _update(self, label=None, value=None, process=True, maybeShow=True):
107
114
self ._counter = value or (self ._counter + 1 )
108
115
self ._win .setValue (self ._counter )
109
116
if process and elapsed >= 0.2 :
110
- self .app .processEvents (QEventLoop .ExcludeUserInputEvents )
117
+ self .app .processEvents (QtCore . QEventLoop .ExcludeUserInputEvents )
111
118
self ._lastUpdate = time .time ()
112
119
113
120
def _setBusy (self ):
114
121
self ._disabled = True
115
- self .mw .app .setOverrideCursor (QCursor (Qt .WaitCursor ))
122
+ self .mw .app .setOverrideCursor (QtGui . QCursor (QtCore . Qt .WaitCursor ))
116
123
117
124
def _unsetBusy (self ):
118
125
self ._disabled = False
0 commit comments