class ConditionEditor(QWidget): # {{{
ACTION_MAP = {
'bool' : (
(_('is true'), 'is true',),
(_('is false'), 'is false'),
(_('is undefined'), 'is undefined')
),
'ondevice' : (
(_('is true'), 'is set',),
(_('is false'), 'is not set'),
),
'identifiers' : (
(_('has id'), 'has id'),
(_('does not have id'), 'does not have id'),
),
'int' : (
(_('is equal to'), 'eq'),
(_('is less than'), 'lt'),
(_('is greater than'), 'gt')
),
'datetime' : (
(_('is equal to'), 'eq'),
(_('is less than'), 'lt'),
(_('is greater than'), 'gt'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
(_('is more days ago than'), 'older count days'),
(_('is fewer days ago than'), 'count_days'),
(_('is more days from now than'), 'newer future days'),
(_('is fewer days from now than'), 'older future days')
),
'multiple' : (
(_('has'), 'has'),
(_('does not have'), 'does not have'),
(_('has pattern'), 'has pattern'),
(_('does not have pattern'), 'does not have pattern'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
),
'single' : (
(_('is'), 'is'),
(_('is not'), 'is not'),
(_('contains'), 'contains'),
(_('does not contain'), 'does not contain'),
(_('matches pattern'), 'matches pattern'),
(_('does not match pattern'), 'does not match pattern'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
),
}
for x in ('float', 'rating'):
ACTION_MAP[x] = ACTION_MAP['int']
def __init__(self, fm, parent=None):
QWidget.__init__(self, parent)
self.fm = fm
self.action_map = self.ACTION_MAP
self.l = l = QGridLayout(self)
self.setLayout(l)
texts = _('If the ___ column ___ values')
try:
one, two, three = texts.split('___')
except:
one, two, three = 'If the ', ' column ', ' value '
self.l1 = l1 = QLabel(one)
l.addWidget(l1, 0, 0)
self.column_box = QComboBox(self)
l.addWidget(self.column_box, 0, 1)
self.l2 = l2 = QLabel(two)
l.addWidget(l2, 0, 2)
self.action_box = QComboBox(self)
l.addWidget(self.action_box, 0, 3)
self.l3 = l3 = QLabel(three)
l.addWidget(l3, 0, 4)
self.value_box = QLineEdit(self)
l.addWidget(self.value_box, 0, 5)
self.column_box.addItem('', '')
for key in sorted(
conditionable_columns(fm),
key=lambda key: sort_key(fm[key]['name'])):
self.column_box.addItem(fm[key]['name'], key)
self.column_box.setCurrentIndex(0)
self.column_box.currentIndexChanged.connect(self.init_action_box)
self.action_box.currentIndexChanged.connect(self.init_value_box)
for b in (self.column_box, self.action_box):
b.setSizeAdjustPolicy(b.AdjustToMinimumContentsLengthWithIcon)
b.setMinimumContentsLength(20)
@dynamic_property
def current_col(self):
def fget(self):
idx = self.column_box.currentIndex()
return unicode_type(self.column_box.itemData(idx) or '')
def fset(self, val):
for idx in range(self.column_box.count()):
c = unicode_type(self.column_box.itemData(idx) or '')
if c == val:
self.column_box.setCurrentIndex(idx)
return
raise ValueError('Column %r not found'%val)
return property(fget=fget, fset=fset)
@dynamic_property
def current_action(self):
def fget(self):
idx = self.action_box.currentIndex()
return unicode_type(self.action_box.itemData(idx) or '')
def fset(self, val):
for idx in range(self.action_box.count()):
c = unicode_type(self.action_box.itemData(idx) or '')
if c == val:
self.action_box.setCurrentIndex(idx)
return
raise ValueError('Action %r not valid for current column'%val)
return property(fget=fget, fset=fset)
@property
def current_val(self):
ans = unicode_type(self.value_box.text()).strip()
if self.current_col == 'languages':
rmap = {lower(v):k for k, v in lang_map().iteritems()}
ans = rmap.get(lower(ans), ans)
return ans
@dynamic_property
def condition(self):
def fget(self):
c, a, v = (self.current_col, self.current_action,
self.current_val)
if not c or not a:
return None
return (c, a, v)
def fset(self, condition):
c, a, v = condition
if not v:
v = ''
v = v.strip()
self.current_col = c
self.current_action = a
self.value_box.setText(v)
return property(fget=fget, fset=fset)
def init_action_box(self):
self.action_box.blockSignals(True)
self.action_box.clear()
self.action_box.addItem('', '')
col = self.current_col
if col:
m = self.fm[col]
dt = m['datatype']
if dt in self.action_map:
actions = self.action_map[dt]
else:
if col == 'ondevice':
k = 'ondevice'
elif col == 'identifiers':
k = 'identifiers'
else:
k = 'multiple' if m['is_multiple'] else 'single'
actions = self.action_map[k]
for text, key in actions:
self.action_box.addItem(text, key)
self.action_box.setCurrentIndex(0)
self.action_box.blockSignals(False)
self.init_value_box()
def init_value_box(self):
self.value_box.setEnabled(True)
self.value_box.setText('')
self.value_box.setInputMask('')
self.value_box.setValidator(None)
col = self.current_col
if not col:
return
action = self.current_action
if not action:
return
m = self.fm[col]
dt = m['datatype']
tt = ''
if col == 'identifiers':
tt = _('Enter either an identifier type or an '
'identifier type and value of the form identifier:value')
elif col == 'languages':
tt = _('Enter a 3 letter ISO language code, like fra for French'
' or deu for German or eng for English. You can also use'
' the full language name, in which case calibre will try to'
' automatically convert it to the language code.')
elif dt in ('int', 'float', 'rating'):
tt = _('Enter a number')
v = QIntValidator if dt == 'int' else QDoubleValidator
self.value_box.setValidator(v(self.value_box))
elif dt == 'datetime':
if action == 'count_days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the maximum days old the item can be. Zero is today. '
'Dates in the future always match')
elif action == 'older count days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the minimum days old the item can be. Zero is today. '
'Dates in the future never match')
elif action == 'older future days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the maximum days in the future the item can be. '
'Zero is today. Dates in the past always match')
elif action == 'newer future days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the minimum days in the future the item can be. '
'Zero is today. Dates in the past never match')
else:
self.value_box.setInputMask('9999-99-99')
tt = _('Enter a date in the format YYYY-MM-DD')
else:
tt = _('Enter a string.')
if 'pattern' in action:
tt = _('Enter a regular expression')
elif m.get('is_multiple', False):
tt += '\n' + _('You can match multiple values by separating'
' them with %s')%m['is_multiple']['ui_to_list']
self.value_box.setToolTip(tt)
if action in ('is set', 'is not set', 'is true', 'is false',
'is undefined'):
self.value_box.setEnabled(False)
class ConditionEditor(QWidget): # {{{
ACTION_MAP = {
'bool' : (
(_('is true'), 'is true',),
(_('is false'), 'is false'),
(_('is undefined'), 'is undefined')
),
'ondevice' : (
(_('is true'), 'is set',),
(_('is false'), 'is not set'),
),
'identifiers' : (
(_('has id'), 'has id'),
(_('does not have id'), 'does not have id'),
),
'int' : (
(_('is equal to'), 'eq'),
(_('is less than'), 'lt'),
(_('is greater than'), 'gt')
),
'datetime' : (
(_('is equal to'), 'eq'),
(_('is less than'), 'lt'),
(_('is greater than'), 'gt'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
(_('is more days ago than'), 'older count days'),
(_('is fewer days ago than'), 'count_days'),
(_('is more days from now than'), 'newer future days'),
(_('is fewer days from now than'), 'older future days')
),
'multiple' : (
(_('has'), 'has'),
(_('does not have'), 'does not have'),
(_('has pattern'), 'has pattern'),
(_('does not have pattern'), 'does not have pattern'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
),
'single' : (
(_('is'), 'is'),
(_('is not'), 'is not'),
(_('matches pattern'), 'matches pattern'),
(_('does not match pattern'), 'does not match pattern'),
(_('is set'), 'is set'),
(_('is not set'), 'is not set'),
),
}
for x in ('float', 'rating'):
ACTION_MAP[x] = ACTION_MAP['int']
def __init__(self, fm, parent=None):
QWidget.__init__(self, parent)
self.fm = fm
self.action_map = self.ACTION_MAP
self.l = l = QGridLayout(self)
self.setLayout(l)
texts = _('If the ___ column ___ values')
try:
one, two, three = texts.split('___')
except:
one, two, three = 'If the ', ' column ', ' value '
self.l1 = l1 = QLabel(one)
l.addWidget(l1, 0, 0)
self.column_box = QComboBox(self)
l.addWidget(self.column_box, 0, 1)
self.l2 = l2 = QLabel(two)
l.addWidget(l2, 0, 2)
self.action_box = QComboBox(self)
l.addWidget(self.action_box, 0, 3)
self.l3 = l3 = QLabel(three)
l.addWidget(l3, 0, 4)
self.value_box = QLineEdit(self)
l.addWidget(self.value_box, 0, 5)
self.column_box.addItem('', '')
for key in sorted(
conditionable_columns(fm),
key=lambda(key): sort_key(fm[key]['name'])):
self.column_box.addItem(fm[key]['name'], key)
self.column_box.setCurrentIndex(0)
self.column_box.currentIndexChanged.connect(self.init_action_box)
self.action_box.currentIndexChanged.connect(self.init_value_box)
for b in (self.column_box, self.action_box):
b.setSizeAdjustPolicy(b.AdjustToMinimumContentsLengthWithIcon)
b.setMinimumContentsLength(20)
@dynamic_property
def current_col(self):
def fget(self):
idx = self.column_box.currentIndex()
return unicode(self.column_box.itemData(idx) or '')
def fset(self, val):
for idx in range(self.column_box.count()):
c = unicode(self.column_box.itemData(idx) or '')
if c == val:
self.column_box.setCurrentIndex(idx)
return
raise ValueError('Column %r not found'%val)
return property(fget=fget, fset=fset)
@dynamic_property
def current_action(self):
def fget(self):
idx = self.action_box.currentIndex()
return unicode(self.action_box.itemData(idx) or '')
def fset(self, val):
for idx in range(self.action_box.count()):
c = unicode(self.action_box.itemData(idx) or '')
if c == val:
self.action_box.setCurrentIndex(idx)
return
raise ValueError('Action %r not valid for current column'%val)
return property(fget=fget, fset=fset)
@property
def current_val(self):
ans = unicode(self.value_box.text()).strip()
if self.current_col == 'languages':
rmap = {lower(v):k for k, v in lang_map().iteritems()}
ans = rmap.get(lower(ans), ans)
return ans
@dynamic_property
def condition(self):
def fget(self):
c, a, v = (self.current_col, self.current_action,
self.current_val)
if not c or not a:
return None
return (c, a, v)
def fset(self, condition):
c, a, v = condition
if not v:
v = ''
v = v.strip()
self.current_col = c
self.current_action = a
self.value_box.setText(v)
return property(fget=fget, fset=fset)
def init_action_box(self):
self.action_box.blockSignals(True)
self.action_box.clear()
self.action_box.addItem('', '')
col = self.current_col
if col:
m = self.fm[col]
dt = m['datatype']
if dt in self.action_map:
actions = self.action_map[dt]
else:
if col == 'ondevice':
k = 'ondevice'
elif col == 'identifiers':
k = 'identifiers'
else:
k = 'multiple' if m['is_multiple'] else 'single'
actions = self.action_map[k]
for text, key in actions:
self.action_box.addItem(text, key)
self.action_box.setCurrentIndex(0)
self.action_box.blockSignals(False)
self.init_value_box()
def init_value_box(self):
self.value_box.setEnabled(True)
self.value_box.setText('')
self.value_box.setInputMask('')
self.value_box.setValidator(None)
col = self.current_col
if not col:
return
action = self.current_action
if not action:
return
m = self.fm[col]
dt = m['datatype']
tt = ''
if col == 'identifiers':
tt = _('Enter either an identifier type or an '
'identifier type and value of the form identifier:value')
elif col == 'languages':
tt = _('Enter a 3 letter ISO language code, like fra for French'
' or deu for German or eng for English. You can also use'
' the full language name, in which case calibre will try to'
' automatically convert it to the language code.')
elif dt in ('int', 'float', 'rating'):
tt = _('Enter a number')
v = QIntValidator if dt == 'int' else QDoubleValidator
self.value_box.setValidator(v(self.value_box))
elif dt == 'datetime':
if action == 'count_days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the maximum days old the item can be. Zero is today. '
'Dates in the future always match')
elif action == 'older count days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the minimum days old the item can be. Zero is today. '
'Dates in the future never match')
elif action == 'older future days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the maximum days in the future the item can be. '
'Zero is today. Dates in the past always match')
elif action == 'newer future days':
self.value_box.setValidator(QIntValidator(self.value_box))
tt = _('Enter the minimum days in the future the item can be. '
'Zero is today. Dates in the past never match')
else:
self.value_box.setInputMask('9999-99-99')
tt = _('Enter a date in the format YYYY-MM-DD')
else:
tt = _('Enter a string.')
if 'pattern' in action:
tt = _('Enter a regular expression')
elif m.get('is_multiple', False):
tt += '\n' + _('You can match multiple values by separating'
' them with %s')%m['is_multiple']['ui_to_list']
self.value_box.setToolTip(tt)
if action in ('is set', 'is not set', 'is true', 'is false',
'is undefined'):
self.value_box.setEnabled(False)
class RuleEditor(QDialog): # {{{
@property
def doing_multiple(self):
return hasattr(self, 'multiple_icon_cb') and self.multiple_icon_cb.isChecked()
def __init__(self, fm, pref_name, parent=None):
QDialog.__init__(self, parent)
self.fm = fm
if pref_name == 'column_color_rules':
self.rule_kind = 'color'
rule_text = _('column coloring')
elif pref_name == 'column_icon_rules':
self.rule_kind = 'icon'
rule_text = _('column icon')
elif pref_name == 'cover_grid_icon_rules':
self.rule_kind = 'emblem'
rule_text = _('Cover grid emblem')
self.setWindowIcon(QIcon(I('format-fill-color.png')))
self.setWindowTitle(_('Create/edit a {0} rule').format(rule_text))
self.l = l = QGridLayout(self)
self.setLayout(l)
self.l1 = l1 = QLabel(_('Create a {0} rule by'
' filling in the boxes below').format(rule_text))
l.addWidget(l1, 0, 0, 1, 8)
self.f1 = QFrame(self)
self.f1.setFrameShape(QFrame.HLine)
l.addWidget(self.f1, 1, 0, 1, 8)
self.l2 = l2 = QLabel(_('Add the emblem:') if self.rule_kind == 'emblem' else _('Set the'))
l.addWidget(l2, 2, 0)
if self.rule_kind == 'color':
l.addWidget(QLabel(_('color')))
elif self.rule_kind == 'icon':
self.kind_box = QComboBox(self)
for tt, t in icon_rule_kinds:
self.kind_box.addItem(tt, t)
l.addWidget(self.kind_box, 2, 1)
self.kind_box.setToolTip(textwrap.fill(_(
'If you choose composed icons and multiple rules match, then all the'
' matching icons will be combined, otherwise the icon from the'
' first rule to match will be used.')))
else:
pass
self.l3 = l3 = QLabel(_('of the column:'))
l.addWidget(l3, 2, 2)
self.column_box = QComboBox(self)
l.addWidget(self.column_box, 2, 3)
self.l4 = l4 = QLabel(_('to'))
l.addWidget(l4, 2, 4)
if self.rule_kind == 'emblem':
l3.setVisible(False), self.column_box.setVisible(False), l4.setVisible(False)
def create_filename_box():
self.filename_box = f = QComboBox()
self.filenamebox_view = v = QListView()
v.setIconSize(QSize(32, 32))
self.filename_box.setView(v)
self.orig_filenamebox_view = f.view()
f.setMinimumContentsLength(20), f.setSizeAdjustPolicy(f.AdjustToMinimumContentsLengthWithIcon)
self.populate_icon_filenames()
if self.rule_kind == 'color':
self.color_box = ColorButton(parent=self)
self.color_label = QLabel('Sample text Sample text')
self.color_label.setTextFormat(Qt.RichText)
l.addWidget(self.color_box, 2, 5)
l.addWidget(self.color_label, 2, 6)
l.addItem(QSpacerItem(10, 10, QSizePolicy.Expanding), 2, 7)
elif self.rule_kind == 'emblem':
create_filename_box()
self.update_filename_box()
self.filename_button = QPushButton(QIcon(I('document_open.png')),
_('&Add new image'))
l.addWidget(self.filename_box)
l.addWidget(self.filename_button, 2, 6)
l.addWidget(QLabel(_('(Images should be square-ish)')), 2, 7)
l.setColumnStretch(7, 10)
else:
create_filename_box()
vb = QVBoxLayout()
self.multiple_icon_cb = QCheckBox(_('Choose &more than one icon'))
vb.addWidget(self.multiple_icon_cb)
self.update_filename_box()
self.multiple_icon_cb.clicked.connect(self.multiple_box_clicked)
vb.addWidget(self.filename_box)
l.addLayout(vb, 2, 5)
self.filename_button = QPushButton(QIcon(I('document_open.png')),
_('&Add icon'))
l.addWidget(self.filename_button, 2, 6)
l.addWidget(QLabel(_('Icons should be square or landscape')), 2, 7)
l.setColumnStretch(7, 10)
self.l5 = l5 = QLabel(
_('Only if the following conditions are all satisfied:'))
l.addWidget(l5, 3, 0, 1, 7)
self.scroll_area = sa = QScrollArea(self)
sa.setMinimumHeight(300)
sa.setMinimumWidth(950)
sa.setWidgetResizable(True)
l.addWidget(sa, 4, 0, 1, 8)
self.add_button = b = QPushButton(QIcon(I('plus.png')),
_('Add &another condition'))
l.addWidget(b, 5, 0, 1, 8)
b.clicked.connect(self.add_blank_condition)
self.l6 = l6 = QLabel(_('You can disable a condition by'
' blanking all of its boxes'))
l.addWidget(l6, 6, 0, 1, 8)
self.bb = bb = QDialogButtonBox(
QDialogButtonBox.Ok|QDialogButtonBox.Cancel)
bb.accepted.connect(self.accept)
bb.rejected.connect(self.reject)
l.addWidget(bb, 7, 0, 1, 8)
if self.rule_kind != 'color':
self.remove_button = b = bb.addButton(_('&Remove icon'), bb.ActionRole)
b.setIcon(QIcon(I('minus.png')))
b.setMenu(QMenu())
b.setToolTip('<p>' + _('Remove a previously added icon. Note that doing so will cause rules that use it to stop working.'))
self.update_remove_button()
self.conditions_widget = QWidget(self)
sa.setWidget(self.conditions_widget)
self.conditions_widget.setLayout(QVBoxLayout())
self.conditions_widget.layout().setAlignment(Qt.AlignTop)
self.conditions = []
if self.rule_kind == 'color':
for b in (self.column_box, ):
b.setSizeAdjustPolicy(b.AdjustToMinimumContentsLengthWithIcon)
b.setMinimumContentsLength(15)
for key in sorted(displayable_columns(fm),
key=lambda k: sort_key(fm[k]['name']) if k != color_row_key else 0):
if key == color_row_key and self.rule_kind != 'color':
continue
name = all_columns_string if key == color_row_key else fm[key]['name']
if name:
self.column_box.addItem(name, key)
self.column_box.setCurrentIndex(0)
if self.rule_kind == 'color':
self.color_box.color = '#000'
self.update_color_label()
self.color_box.color_changed.connect(self.update_color_label)
else:
self.rule_icon_files = []
self.filename_button.clicked.connect(self.filename_button_clicked)
self.resize(self.sizeHint())
def multiple_box_clicked(self):
self.update_filename_box()
self.update_icon_filenames_in_box()
@property
def icon_folder(self):
return os.path.join(config_dir, 'cc_icons')
def populate_icon_filenames(self):
d = self.icon_folder
self.icon_file_names = []
if os.path.exists(d):
for icon_file in os.listdir(d):
icon_file = lower(icon_file)
if os.path.exists(os.path.join(d, icon_file)) and icon_file.endswith('.png'):
self.icon_file_names.append(icon_file)
self.icon_file_names.sort(key=sort_key)
def update_filename_box(self):
doing_multiple = self.doing_multiple
model = QStandardItemModel()
self.filename_box.setModel(model)
self.icon_file_names.sort(key=sort_key)
if doing_multiple:
item = QStandardItem(_('Open to see checkboxes'))
item.setIcon(QIcon(I('blank.png')))
else:
item = QStandardItem('')
item.setFlags(Qt.ItemFlag(0))
model.appendRow(item)
for i,filename in enumerate(self.icon_file_names):
item = QStandardItem(filename)
if doing_multiple:
item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
item.setData(Qt.Unchecked, Qt.CheckStateRole)
else:
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
icon = QIcon(os.path.join(self.icon_folder, filename))
item.setIcon(icon)
model.appendRow(item)
def update_color_label(self):
pal = QApplication.palette()
bg1 = unicode_type(pal.color(pal.Base).name())
bg2 = unicode_type(pal.color(pal.AlternateBase).name())
c = self.color_box.color
self.color_label.setText('''
<span style="color: {c}; background-color: {bg1}"> {st} </span>
<span style="color: {c}; background-color: {bg2}"> {st} </span>
'''.format(c=c, bg1=bg1, bg2=bg2, st=_('Sample text')))
def sanitize_icon_file_name(self, icon_path):
n = lower(sanitize_file_name_unicode(
os.path.splitext(
os.path.basename(icon_path))[0]+'.png'))
return n.replace("'", '_')
def filename_button_clicked(self):
try:
path = choose_files(self, 'choose_category_icon',
_('Select Icon'), filters=[
(_('Images'), ['png', 'gif', 'jpg', 'jpeg'])],
all_files=False, select_only_single_file=True)
if path:
icon_path = path[0]
icon_name = self.sanitize_icon_file_name(icon_path)
if icon_name not in self.icon_file_names:
self.icon_file_names.append(icon_name)
try:
p = QIcon(icon_path).pixmap(QSize(128, 128))
d = self.icon_folder
if not os.path.exists(os.path.join(d, icon_name)):
if not os.path.exists(d):
os.makedirs(d)
with open(os.path.join(d, icon_name), 'wb') as f:
f.write(pixmap_to_data(p, format='PNG'))
except:
import traceback
traceback.print_exc()
self.update_filename_box()
self.update_remove_button()
if self.doing_multiple:
if icon_name not in self.rule_icon_files:
self.rule_icon_files.append(icon_name)
self.update_icon_filenames_in_box()
else:
self.filename_box.setCurrentIndex(self.filename_box.findText(icon_name))
self.filename_box.adjustSize()
except:
import traceback
traceback.print_exc()
return
def get_filenames_from_box(self):
if self.doing_multiple:
model = self.filename_box.model()
fnames = []
for i in range(1, model.rowCount()):
item = model.item(i, 0)
if item.checkState() == Qt.Checked:
fnames.append(lower(unicode_type(item.text())))
fname = ' : '.join(fnames)
else:
fname = lower(unicode_type(self.filename_box.currentText()))
return fname
def update_icon_filenames_in_box(self):
if self.rule_icon_files:
if not self.doing_multiple:
idx = self.filename_box.findText(self.rule_icon_files[0])
if idx >= 0:
self.filename_box.setCurrentIndex(idx)
else:
self.filename_box.setCurrentIndex(0)
else:
model = self.filename_box.model()
for icon in self.rule_icon_files:
idx = self.filename_box.findText(icon)
if idx >= 0:
item = model.item(idx)
item.setCheckState(Qt.Checked)
def update_remove_button(self):
m = self.remove_button.menu()
m.clear()
for name in self.icon_file_names:
ac = m.addAction(QIcon(os.path.join(self.icon_folder, name)), name)
connect_lambda(ac.triggered, self, lambda self: self.remove_image(self.sender().text()))
def remove_image(self, name):
try:
os.remove(os.path.join(self.icon_folder, name))
except EnvironmentError:
pass
else:
self.populate_icon_filenames()
self.update_remove_button()
self.update_filename_box()
self.update_icon_filenames_in_box()
def add_blank_condition(self):
c = ConditionEditor(self.fm, parent=self.conditions_widget)
self.conditions.append(c)
self.conditions_widget.layout().addWidget(c)
def apply_rule(self, kind, col, rule):
if kind == 'color':
if rule.color:
self.color_box.color = rule.color
else:
if self.rule_kind == 'icon':
for i, tup in enumerate(icon_rule_kinds):
if kind == tup[1]:
self.kind_box.setCurrentIndex(i)
break
self.rule_icon_files = [ic.strip() for ic in rule.color.split(':')]
if len(self.rule_icon_files) > 1:
self.multiple_icon_cb.setChecked(True)
self.update_filename_box()
self.update_icon_filenames_in_box()
for i in range(self.column_box.count()):
c = unicode_type(self.column_box.itemData(i) or '')
if col == c:
self.column_box.setCurrentIndex(i)
break
for c in rule.conditions:
ce = ConditionEditor(self.fm, parent=self.conditions_widget)
self.conditions.append(ce)
self.conditions_widget.layout().addWidget(ce)
try:
ce.condition = c
except:
import traceback
traceback.print_exc()
def accept(self):
if self.rule_kind != 'color':
fname = self.get_filenames_from_box()
if not fname:
error_dialog(self, _('No icon selected'),
_('You must choose an icon for this rule'), show=True)
return
if self.validate():
QDialog.accept(self)
def validate(self):
r = Rule(self.fm)
for c in self.conditions:
condition = c.condition
if condition is not None:
try:
r.add_condition(*condition)
except Exception as e:
import traceback
error_dialog(self, _('Invalid condition'),
_('One of the conditions for this rule is'
' invalid: <b>%s</b>')%e,
det_msg=traceback.format_exc(), show=True)
return False
if len(r.conditions) < 1:
error_dialog(self, _('No conditions'),
_('You must specify at least one non-empty condition'
' for this rule'), show=True)
return False
return True
@property
def rule(self):
r = Rule(self.fm)
if self.rule_kind != 'color':
r.color = self.get_filenames_from_box()
else:
r.color = self.color_box.color
idx = self.column_box.currentIndex()
col = unicode_type(self.column_box.itemData(idx) or '')
for c in self.conditions:
condition = c.condition
if condition is not None:
r.add_condition(*condition)
if self.rule_kind == 'icon':
kind = unicode_type(self.kind_box.itemData(
self.kind_box.currentIndex()) or '')
else:
kind = self.rule_kind
return kind, col, r
class ConfigWidget(QWidget):
# GUI definition
def __init__(self):
QWidget.__init__(self)
self.l = QVBoxLayout()
self.setLayout(self.l)
self.ll = QHBoxLayout()
self.l.addLayout(self.ll)
self.label_exe = QLabel(_('&Prince executable:'))
self.ll.addWidget(self.label_exe)
self.exe = QLineEdit(self)
self.exe.setText(prefs['prince_exe'])
self.exe.setToolTip(_('<qt>Executable for the Prince program (command-line interface)</qt>'))
self.ll.addWidget(self.exe)
self.label_exe.setBuddy(self.exe)
self.browse = QPushButton(_('&Browse') + '...', self)
self.browse.setToolTip(_('<qt>Search the Prince executable in your computer</qt>'))
self.browse.clicked.connect(self.select_exe)
self.ll.addWidget(self.browse)
self.lll = QHBoxLayout()
self.l.addLayout(self.lll)
self.label_fmts = QLabel(_('Preferred &formats:'))
self.lll.addWidget(self.label_fmts)
self.fmts = QLineEdit(self)
self.fmts.setText(','.join(prefs['formats']))
self.fmts.setToolTip(_('<qt>Comma-separated list of preferred formats to use as source, the first that matches will be used</qt>'))
self.lll.addWidget(self.fmts)
self.label_fmts.setBuddy(self.fmts)
self.add_book = QCheckBox(_('&Add PDF to the book record'))
self.add_book.setToolTip(_('<qt>Add the converted PDF to the selected book record</qt>'))
self.add_book.setChecked(prefs['add_book'])
self.l.addWidget(self.add_book)
self.show_css = QCheckBox(_('&Show CSS in the Convert dialog'))
self.show_css.setToolTip(_('<qt>Show by default the stylesheets in the Convert dialog</qt>'))
self.show_css.setChecked(prefs['show_CSS'])
self.l.addWidget(self.show_css)
self.css_layout = QVBoxLayout()
self.llll = QHBoxLayout()
self.css_layout.addLayout(self.llll)
self.css_list = QComboBox()
self.css_list.setToolTip(_('<qt>List of custom stylesheets defined. Select one to edit</qt>'))
self.css_list.setSizeAdjustPolicy(QComboBox.AdjustToContents)
self.CSS_list = prefs['custom_CSS_list'].copy()
self.default_CSS = prefs['default_CSS']
if 'custom_CSS' in prefs:
self.CSS_list[_('old')] = prefs['custom_CSS']
self.default_CSS = _('old')
if self.default_CSS not in self.CSS_list:
self.default_CSS = sorted(self.CSS_list, key=lambda x: x.lower())[0]
for key in sorted(self.CSS_list, key=lambda x: x.lower()):
self.css_list.addItem(key, key)
self.css_list.setCurrentIndex(self.css_list.findText(self.default_CSS))
self.css_list.currentIndexChanged.connect(self.set_css)
self.llll.addWidget(self.css_list)
self.css_rename = QPushButton(_('Re&name'))
self.css_rename.setToolTip(_('<qt>Rename the current stylesheet to the name on the right</qt>'))
self.css_rename.clicked.connect(self.rename_css)
self.css_rename.setEnabled(False)
self.llll.addWidget(self.css_rename)
self.css_name = QLineEdit(self)
self.css_name.setToolTip(_('<qt>Name for the new or renamed stylesheet</qt>'))
self.css_name.setText(self.css_list.currentText())
self.css_name.textChanged.connect(self.check_names)
self.llll.addWidget(self.css_name)
self.css_add = QPushButton(_('A&dd'))
self.css_add.setToolTip(_('<qt>Add a new empty stylesheet with the name on the left</qt>'))
self.css_add.clicked.connect(self.add_css)
self.css_add.setEnabled(False)
self.llll.addWidget(self.css_add)
self.css_remove = QPushButton(_('Re&move'))
self.css_remove.setToolTip(_('<qt>Remove the current stylesheet</qt>'))
self.css_remove.clicked.connect(self.remove_css)
self.llll.addWidget(self.css_remove)
self.css = TextEditWithTooltip()
self.css.setLineWrapMode(TextEditWithTooltip.NoWrap)
self.css.load_text(self.CSS_list[unicode(self.css_list.currentText())],'css')
self.css.setToolTip(_('<qt>Custom stylesheet that will be applied, if selected, to all Prince PDF conversions</qt>'))
self.css_layout.addWidget(self.css)
self.css_templates = QLabel(_('Book metadata can be used in the stylesheet. Anything between %(s1)s and %(s2)s will be processed as a calibre template. For instance, %(s3)s in the stylesheet will be replaced with the book title in the conversion.') % \
{'s1':'<span style="font-family:monospace ; font-weight:bold">@{@</span>', \
's2':'<span style="font-family:monospace ; font-weight:bold">@}@</span>', \
's3':'<span style="font-family:monospace ; font-weight:bold">@{@{title}@}@</span>'})
self.css_templates.setWordWrap(True)
self.css_layout.addWidget(self.css_templates)
self.css_box = QGroupBox(_('&Custom CSS:'))
self.css_box.setLayout(self.css_layout)
self.l.addWidget(self.css_box)
self.lllll = QHBoxLayout()
self.lllll.setAlignment(Qt.AlignLeft)
self.l.addLayout(self.lllll)
self.defaults = QPushButton(_('&Restore defaults'))
self.defaults.setToolTip(_('<qt>Restore the default settings</qt>'))
self.defaults.clicked.connect(self.restore_defaults)
self.lllll.addWidget(self.defaults, alignment=Qt.AlignLeft)
self.warning = QLabel(_('<b>Warning</b>: Deletes modified stylesheets'))
self.lllll.addWidget(self.warning)
self.adjustSize()
def select_exe(self):
'''
Create a dialog to select the Prince executable
'''
dialog = QFileDialog()
dialog.setFileMode(QFileDialog.ExistingFile)
filename = dialog.getOpenFileName(self, _('Select Prince executable'), '', '')
if filename:
try:
self.exe.setText(filename)
except(TypeError):
self.exe.setText(filename[0])
def restore_defaults(self):
'''
Restore the default settings
'''
self.exe.setText(prefs.defaults['prince_exe'])
self.fmts.setText(','.join(prefs.defaults['formats']).lower())
self.show_css.setChecked(prefs.defaults['show_CSS'])
self.add_book.setChecked(prefs.defaults['add_book'])
self.css_list.currentIndexChanged.disconnect()
self.css_list.clear()
self.CSS_list = prefs.defaults['custom_CSS_list'].copy()
self.default_CSS = prefs.defaults['default_CSS']
for key in sorted(self.CSS_list, key=lambda x: x.lower()):
self.css_list.addItem(key, key)
self.css_list.setCurrentIndex(self.css_list.findText(self.default_CSS))
self.css_name.setText(self.default_CSS)
self.css.load_text(self.CSS_list[unicode(self.css_list.currentText())],'css')
self.css_list.currentIndexChanged.connect(self.set_css)
def save_settings(self):
'''
Save the current settings
'''
prefs['prince_exe'] = unicode(self.exe.text())
prefs['formats'] = unicode(self.fmts.text().lower()).split(',')
prefs['show_CSS'] = self.show_css.isChecked()
prefs['add_book'] = self.add_book.isChecked()
self.set_css()
prefs['default_CSS'] = self.default_CSS
prefs['custom_CSS_list'] = self.CSS_list.copy()
if 'custom_CSS' in prefs:
del prefs['custom_CSS']
def set_css(self):
'''
Fill the CSS text box with the selected stylesheet
'''
self.CSS_list[self.default_CSS] = unicode(self.css.toPlainText())
self.default_CSS = unicode(self.css_list.currentText())
self.css.load_text(self.CSS_list[self.default_CSS],'css')
self.css_name.setText(self.css_list.currentText())
def add_css(self):
'''
Add a new stylesheet
'''
from calibre.gui2 import error_dialog
name = unicode(self.css_name.text())
if name in self.CSS_list:
error_dialog(self, _('Cannot add stylesheet'), _('A stylesheet with the name "%s" is already defined, use a different name.') % name, show=True)
else:
self.CSS_list[name] = ''
self.css_list.addItem(name, name)
self.css_list.setCurrentIndex(self.css_list.findText(name))
self.css_add.setEnabled(False)
self.css_rename.setEnabled(False)
def remove_css(self):
'''
Remove an existing stylesheet
'''
from calibre.gui2 import error_dialog
if (self.css_list.count() > 1):
self.css_list.currentIndexChanged.disconnect()
self.css_list.removeItem(self.css_list.currentIndex())
del self.CSS_list[self.default_CSS]
self.default_CSS = unicode(self.css_list.currentText())
self.css.load_text(self.CSS_list[self.default_CSS],'css')
self.css_list.currentIndexChanged.connect(self.set_css)
self.css_name.setText(self.css_list.currentText())
else:
error_dialog(self, _('Cannot delete the last stylesheet'), _('The last stylesheet cannot be removed. You can rename it and/or remove its contents.'), show=True)
def rename_css(self):
'''
Rename a stylesheet
'''
from calibre.gui2 import error_dialog
name = unicode(self.css_name.text())
if name in self.CSS_list:
error_dialog(self, _('Cannot rename stylesheet'), _('A stylesheet with the name "%s" is already defined, use a different name.') % name, show=True)
else:
self.CSS_list[name] = self.CSS_list.pop(self.default_CSS)
self.css_list.setItemText(self.css_list.currentIndex(),name)
self.default_CSS = name
def check_names(self, text):
name = unicode(text)
if name in self.CSS_list:
self.css_add.setEnabled(False)
self.css_rename.setEnabled(False)
else:
self.css_add.setEnabled(True)
self.css_rename.setEnabled(True)
class RuleEditor(QDialog): # {{{
@property
def doing_multiple(self):
return hasattr(self, 'multiple_icon_cb') and self.multiple_icon_cb.isChecked()
def __init__(self, fm, pref_name, parent=None):
QDialog.__init__(self, parent)
self.fm = fm
if pref_name == 'column_color_rules':
self.rule_kind = 'color'
rule_text = _('column coloring')
elif pref_name == 'column_icon_rules':
self.rule_kind = 'icon'
rule_text = _('column icon')
elif pref_name == 'cover_grid_icon_rules':
self.rule_kind = 'emblem'
rule_text = _('Cover grid emblem')
self.setWindowIcon(QIcon(I('format-fill-color.png')))
self.setWindowTitle(_('Create/edit a {0} rule').format(rule_text))
self.l = l = QGridLayout(self)
self.setLayout(l)
self.l1 = l1 = QLabel(_('Create a {0} rule by'
' filling in the boxes below').format(rule_text))
l.addWidget(l1, 0, 0, 1, 8)
self.f1 = QFrame(self)
self.f1.setFrameShape(QFrame.HLine)
l.addWidget(self.f1, 1, 0, 1, 8)
self.l2 = l2 = QLabel(_('Add the emblem:') if self.rule_kind == 'emblem' else _('Set the'))
l.addWidget(l2, 2, 0)
if self.rule_kind == 'color':
l.addWidget(QLabel(_('color')))
elif self.rule_kind == 'icon':
self.kind_box = QComboBox(self)
for tt, t in icon_rule_kinds:
self.kind_box.addItem(tt, t)
l.addWidget(self.kind_box, 2, 1)
self.kind_box.setToolTip(textwrap.fill(_(
'If you choose composed icons and multiple rules match, then all the'
' matching icons will be combined, otherwise the icon from the'
' first rule to match will be used.')))
else:
pass
self.l3 = l3 = QLabel(_('of the column:'))
l.addWidget(l3, 2, 2)
self.column_box = QComboBox(self)
l.addWidget(self.column_box, 2, 3)
self.l4 = l4 = QLabel(_('to'))
l.addWidget(l4, 2, 4)
if self.rule_kind == 'emblem':
l3.setVisible(False), self.column_box.setVisible(False), l4.setVisible(False)
def create_filename_box():
self.filename_box = f = QComboBox()
self.filenamebox_view = v = QListView()
v.setIconSize(QSize(32, 32))
self.filename_box.setView(v)
self.orig_filenamebox_view = f.view()
f.setMinimumContentsLength(20), f.setSizeAdjustPolicy(f.AdjustToMinimumContentsLengthWithIcon)
self.populate_icon_filenames()
if self.rule_kind == 'color':
self.color_box = ColorButton(parent=self)
self.color_label = QLabel('Sample text Sample text')
self.color_label.setTextFormat(Qt.RichText)
l.addWidget(self.color_box, 2, 5)
l.addWidget(self.color_label, 2, 6)
l.addItem(QSpacerItem(10, 10, QSizePolicy.Expanding), 2, 7)
elif self.rule_kind == 'emblem':
create_filename_box()
self.update_filename_box()
self.filename_button = QPushButton(QIcon(I('document_open.png')),
_('&Add new image'))
l.addWidget(self.filename_box)
l.addWidget(self.filename_button, 2, 6)
l.addWidget(QLabel(_('(Images should be square-ish)')), 2, 7)
l.setColumnStretch(7, 10)
else:
create_filename_box()
vb = QVBoxLayout()
self.multiple_icon_cb = QCheckBox(_('Choose &more than one icon'))
vb.addWidget(self.multiple_icon_cb)
self.update_filename_box()
self.multiple_icon_cb.clicked.connect(self.multiple_box_clicked)
vb.addWidget(self.filename_box)
l.addLayout(vb, 2, 5)
self.filename_button = QPushButton(QIcon(I('document_open.png')),
_('&Add icon'))
l.addWidget(self.filename_button, 2, 6)
l.addWidget(QLabel(_('Icons should be square or landscape')), 2, 7)
l.setColumnStretch(7, 10)
self.l5 = l5 = QLabel(
_('Only if the following conditions are all satisfied:'))
l.addWidget(l5, 3, 0, 1, 7)
self.scroll_area = sa = QScrollArea(self)
sa.setMinimumHeight(300)
sa.setMinimumWidth(950)
sa.setWidgetResizable(True)
l.addWidget(sa, 4, 0, 1, 8)
self.add_button = b = QPushButton(QIcon(I('plus.png')),
_('Add &another condition'))
l.addWidget(b, 5, 0, 1, 8)
b.clicked.connect(self.add_blank_condition)
self.l6 = l6 = QLabel(_('You can disable a condition by'
' blanking all of its boxes'))
l.addWidget(l6, 6, 0, 1, 8)
self.bb = bb = QDialogButtonBox(
QDialogButtonBox.Ok|QDialogButtonBox.Cancel)
bb.accepted.connect(self.accept)
bb.rejected.connect(self.reject)
l.addWidget(bb, 7, 0, 1, 8)
if self.rule_kind != 'color':
self.remove_button = b = bb.addButton(_('&Remove icon'), bb.ActionRole)
b.setIcon(QIcon(I('minus.png')))
b.setMenu(QMenu())
b.setToolTip('<p>' + _('Remove a previously added icon. Note that doing so will cause rules that use it to stop working.'))
self.update_remove_button()
self.conditions_widget = QWidget(self)
sa.setWidget(self.conditions_widget)
self.conditions_widget.setLayout(QVBoxLayout())
self.conditions_widget.layout().setAlignment(Qt.AlignTop)
self.conditions = []
if self.rule_kind == 'color':
for b in (self.column_box, ):
b.setSizeAdjustPolicy(b.AdjustToMinimumContentsLengthWithIcon)
b.setMinimumContentsLength(15)
for key in sorted(displayable_columns(fm),
key=lambda(k): sort_key(fm[k]['name']) if k != color_row_key else 0):
if key == color_row_key and self.rule_kind != 'color':
continue
name = all_columns_string if key == color_row_key else fm[key]['name']
if name:
self.column_box.addItem(name, key)
self.column_box.setCurrentIndex(0)
if self.rule_kind == 'color':
self.color_box.color = '#000'
self.update_color_label()
self.color_box.color_changed.connect(self.update_color_label)
else:
self.rule_icon_files = []
self.filename_button.clicked.connect(self.filename_button_clicked)
self.resize(self.sizeHint())
def multiple_box_clicked(self):
self.update_filename_box()
self.update_icon_filenames_in_box()
@property
def icon_folder(self):
return os.path.join(config_dir, 'cc_icons')
def populate_icon_filenames(self):
d = self.icon_folder
self.icon_file_names = []
if os.path.exists(d):
for icon_file in os.listdir(d):
icon_file = lower(icon_file)
if os.path.exists(os.path.join(d, icon_file)) and icon_file.endswith('.png'):
self.icon_file_names.append(icon_file)
self.icon_file_names.sort(key=sort_key)
def update_filename_box(self):
doing_multiple = self.doing_multiple
model = QStandardItemModel()
self.filename_box.setModel(model)
self.icon_file_names.sort(key=sort_key)
if doing_multiple:
item = QStandardItem(_('Open to see checkboxes'))
item.setIcon(QIcon(I('blank.png')))
else:
item = QStandardItem('')
item.setFlags(Qt.ItemFlag(0))
model.appendRow(item)
for i,filename in enumerate(self.icon_file_names):
item = QStandardItem(filename)
if doing_multiple:
item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
item.setData(Qt.Unchecked, Qt.CheckStateRole)
else:
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
icon = QIcon(os.path.join(self.icon_folder, filename))
item.setIcon(icon)
model.appendRow(item)
def update_color_label(self):
pal = QApplication.palette()
bg1 = unicode(pal.color(pal.Base).name())
bg2 = unicode(pal.color(pal.AlternateBase).name())
c = self.color_box.color
self.color_label.setText('''
<span style="color: {c}; background-color: {bg1}"> {st} </span>
<span style="color: {c}; background-color: {bg2}"> {st} </span>
'''.format(c=c, bg1=bg1, bg2=bg2, st=_('Sample text')))
def sanitize_icon_file_name(self, icon_path):
n = lower(sanitize_file_name_unicode(
os.path.splitext(
os.path.basename(icon_path))[0]+'.png'))
return n.replace("'", '_')
def filename_button_clicked(self):
try:
path = choose_files(self, 'choose_category_icon',
_('Select Icon'), filters=[
(_('Images'), ['png', 'gif', 'jpg', 'jpeg'])],
all_files=False, select_only_single_file=True)
if path:
icon_path = path[0]
icon_name = self.sanitize_icon_file_name(icon_path)
if icon_name not in self.icon_file_names:
self.icon_file_names.append(icon_name)
try:
p = QIcon(icon_path).pixmap(QSize(128, 128))
d = self.icon_folder
if not os.path.exists(os.path.join(d, icon_name)):
if not os.path.exists(d):
os.makedirs(d)
with open(os.path.join(d, icon_name), 'wb') as f:
f.write(pixmap_to_data(p, format='PNG'))
except:
import traceback
traceback.print_exc()
self.update_filename_box()
self.update_remove_button()
if self.doing_multiple:
if icon_name not in self.rule_icon_files:
self.rule_icon_files.append(icon_name)
self.update_icon_filenames_in_box()
else:
self.filename_box.setCurrentIndex(self.filename_box.findText(icon_name))
self.filename_box.adjustSize()
except:
import traceback
traceback.print_exc()
return
def get_filenames_from_box(self):
if self.doing_multiple:
model = self.filename_box.model()
fnames = []
for i in range(1, model.rowCount()):
item = model.item(i, 0)
if item.checkState() == Qt.Checked:
fnames.append(lower(unicode(item.text())))
fname = ' : '.join(fnames)
else:
fname = lower(unicode(self.filename_box.currentText()))
return fname
def update_icon_filenames_in_box(self):
if self.rule_icon_files:
if not self.doing_multiple:
idx = self.filename_box.findText(self.rule_icon_files[0])
if idx >= 0:
self.filename_box.setCurrentIndex(idx)
else:
self.filename_box.setCurrentIndex(0)
else:
model = self.filename_box.model()
for icon in self.rule_icon_files:
idx = self.filename_box.findText(icon)
if idx >= 0:
item = model.item(idx)
item.setCheckState(Qt.Checked)
def update_remove_button(self):
m = self.remove_button.menu()
m.clear()
for name in self.icon_file_names:
m.addAction(QIcon(os.path.join(self.icon_folder, name)), name).triggered.connect(partial(
self.remove_image, name))
def remove_image(self, name):
try:
os.remove(os.path.join(self.icon_folder, name))
except EnvironmentError:
pass
else:
self.populate_icon_filenames()
self.update_remove_button()
self.update_filename_box()
self.update_icon_filenames_in_box()
def add_blank_condition(self):
c = ConditionEditor(self.fm, parent=self.conditions_widget)
self.conditions.append(c)
self.conditions_widget.layout().addWidget(c)
def apply_rule(self, kind, col, rule):
if kind == 'color':
if rule.color:
self.color_box.color = rule.color
else:
if self.rule_kind == 'icon':
for i, tup in enumerate(icon_rule_kinds):
if kind == tup[1]:
self.kind_box.setCurrentIndex(i)
break
self.rule_icon_files = [ic.strip() for ic in rule.color.split(':')]
if len(self.rule_icon_files) > 1:
self.multiple_icon_cb.setChecked(True)
self.update_filename_box()
self.update_icon_filenames_in_box()
for i in range(self.column_box.count()):
c = unicode(self.column_box.itemData(i) or '')
if col == c:
self.column_box.setCurrentIndex(i)
break
for c in rule.conditions:
ce = ConditionEditor(self.fm, parent=self.conditions_widget)
self.conditions.append(ce)
self.conditions_widget.layout().addWidget(ce)
try:
ce.condition = c
except:
import traceback
traceback.print_exc()
def accept(self):
if self.rule_kind != 'color':
fname = self.get_filenames_from_box()
if not fname:
error_dialog(self, _('No icon selected'),
_('You must choose an icon for this rule'), show=True)
return
if self.validate():
QDialog.accept(self)
def validate(self):
r = Rule(self.fm)
for c in self.conditions:
condition = c.condition
if condition is not None:
try:
r.add_condition(*condition)
except Exception as e:
import traceback
error_dialog(self, _('Invalid condition'),
_('One of the conditions for this rule is'
' invalid: <b>%s</b>')%e,
det_msg=traceback.format_exc(), show=True)
return False
if len(r.conditions) < 1:
error_dialog(self, _('No conditions'),
_('You must specify at least one non-empty condition'
' for this rule'), show=True)
return False
return True
@property
def rule(self):
r = Rule(self.fm)
if self.rule_kind != 'color':
r.color = self.get_filenames_from_box()
else:
r.color = self.color_box.color
idx = self.column_box.currentIndex()
col = unicode(self.column_box.itemData(idx) or '')
for c in self.conditions:
condition = c.condition
if condition is not None:
r.add_condition(*condition)
if self.rule_kind == 'icon':
kind = unicode(self.kind_box.itemData(
self.kind_box.currentIndex()) or '')
else:
kind = self.rule_kind
return kind, col, r
class ValueTypeEditor(QWidget):
ValueTypes = (bool, int, float, complex, str)
def __init__(self, *args):
QWidget.__init__(self, *args)
lo = QHBoxLayout(self)
lo.setContentsMargins(0, 0, 0, 0)
lo.setSpacing(5)
# type selector
self.wtypesel = QComboBox(self)
for i, tp in enumerate(self.ValueTypes):
self.wtypesel.addItem(tp.__name__)
self.wtypesel.activated[int].connect(self._selectTypeNum)
typesel_lab = QLabel("&Type:", self)
typesel_lab.setBuddy(self.wtypesel)
lo.addWidget(typesel_lab, 0)
lo.addWidget(self.wtypesel, 0)
self.wvalue = QLineEdit(self)
self.wvalue_lab = QLabel("&Value:", self)
self.wvalue_lab.setBuddy(self.wvalue)
self.wbool = QComboBox(self)
self.wbool.addItems(["false", "true"])
self.wbool.setCurrentIndex(1)
lo.addWidget(self.wvalue_lab, 0)
lo.addWidget(self.wvalue, 1)
lo.addWidget(self.wbool, 1)
self.wvalue.hide()
# make input validators
self._validators = {int: QIntValidator(self), float: QDoubleValidator(self)}
# select bool type initially
self._selectTypeNum(0)
def _selectTypeNum(self, index):
tp = self.ValueTypes[index]
self.wbool.setVisible(tp is bool)
self.wvalue.setVisible(tp is not bool)
self.wvalue_lab.setBuddy(self.wbool if tp is bool else self.wvalue)
self.wvalue.setValidator(self._validators.get(tp, None))
def setValue(self, value):
"""Sets current value"""
for i, tp in enumerate(self.ValueTypes):
if isinstance(value, tp):
self.wtypesel.setCurrentIndex(i)
self._selectTypeNum(i)
if tp is bool:
self.wbool.setCurrentIndex(1 if value else 0)
else:
self.wvalue.setText(str(value))
return
# unknown value: set bool
self.setValue(True)
def getValue(self):
"""Returns current value, or None if no legal value is set"""
tp = self.ValueTypes[self.wtypesel.currentIndex()]
if tp is bool:
return bool(self.wbool.currentIndex())
else:
try:
return tp(self.wvalue.text())
except:
print("Error converting input to type ", tp.__name__)
traceback.print_exc()
return None
class Ui(QApplication):
def __init__(self):
super().__init__(argv)
self.w = QMainWindow()
self.w.setMinimumWidth(300)
self.w.setMinimumHeight(300)
# self.w.setWindowTitle(tr("Отправка запросов на ВС АУГО"))
self.cw = QScrollArea()
# self.__create_ui()
self.__showed = False
self.__wgts = {}
self.cb = QComboBox()
self.individuals = []
self.entities = []
self.documents = []
self.doc_files = []
self.__setupUi(self.w)
self.w.showMaximized()
def report_error(self, msg=None):
if not msg:
etype, value, tb = exc_info()
trace = ''.join(format_exception(etype, value, tb))
delim_len = 40
msg = ("*" * delim_len + "\n%s\n" + "*" * delim_len) % trace
error(msg)
mb = QMessageBox(QMessageBox.Critical, tr('Ошибка'),
str(exc_info()[1]))
mb.setDetailedText(msg)
mb.exec()
def __create_ui(self):
self.l = QGridLayout()
self.t = QTableWidget(0, 3)
self.t.setHorizontalHeaderLabels(
('№ дела (обращения)', 'Дата приёма', 'Дата отправки в СМЭВ'))
self.t.resizeColumnsToContents()
self.l.addWidget(self.t, 0, 0, 1, 2)
w = QWidget()
hl = QHBoxLayout(w)
hl.setDirection(QHBoxLayout.LeftToRight)
hl.addWidget(QWidget())
ok_b = QPushButton(tr('Добавить запрос'))
ok_b.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Minimum)
ok_b.clicked.connect(self.__show_form)
hl.addWidget(ok_b)
w.setLayout(hl)
self.l.addWidget(w, 1, 0, 1, 2)
w = QWidget()
w.setLayout(self.l)
self.cw.setWidget(w)
# self.cw.setLayout(self.l)
# self.w.setCentralWidget(self.cw)
w = QWidget()
l = QVBoxLayout()
l.addWidget(self.cw)
w.setLayout(l)
self.w.setCentralWidget(w)
def __setupUi(self, mainWindow):
mainWindow.setObjectName("MainWindow")
self.centralwidget = QWidget(mainWindow)
self.centralwidget.setObjectName("centralwidget")
self.horizontalLayout = QHBoxLayout(self.centralwidget)
self.horizontalLayout.setObjectName("horizontalLayout")
self.scrollArea = QScrollArea(self.centralwidget)
self.scrollArea.setWidgetResizable(True)
self.scrollArea.setObjectName("scrollArea")
self.scrollAreaWidgetContents = QWidget()
self.scrollAreaWidgetContents.setObjectName("scrollAreaWidgetContents")
self.gridLayout = QGridLayout(self.scrollAreaWidgetContents)
self.gridLayout.setObjectName("gridLayout")
self.__show_form()
self.scrollArea.setWidget(self.scrollAreaWidgetContents)
self.horizontalLayout.addWidget(self.scrollArea)
mainWindow.setCentralWidget(self.centralwidget)
self.menubar = QMenuBar(mainWindow)
self.menubar.setObjectName("menubar")
self.menu = QMenu(self.menubar)
self.menu.setObjectName("menu")
mainWindow.setMenuBar(self.menubar)
self.statusbar = QStatusBar(mainWindow)
self.statusbar.setObjectName("statusbar")
mainWindow.setStatusBar(self.statusbar)
self.action_1 = QAction(mainWindow)
self.action_1.setObjectName("action")
self.action_1.triggered.connect(self.send)
self.action = QAction(mainWindow)
self.action.setObjectName("action")
self.action.triggered.connect(self.on_action_triggered)
self.action_2 = QAction(mainWindow)
self.action_2.setObjectName("action_2")
self.action_3 = QAction(mainWindow)
self.action_3.setObjectName("action_3")
self.action_3.triggered.connect(self.get_response)
self.menu.addAction(self.action)
self.menubar.addAction(self.action_1)
self.menubar.addAction(self.action_2)
self.menubar.addAction(self.action_3)
self.menubar.addAction(self.menu.menuAction())
self.__retranslateUi(mainWindow)
QMetaObject.connectSlotsByName(mainWindow)
@pyqtSlot(bool)
def get_response(self):
try:
from dmsic import Integration
i = Integration(self)
res = i.get_response()[0]
if res:
QMessageBox.information(self.w, tr("Получен ответ"), str(res))
except:
self.report_error()
@pyqtSlot(bool)
def send(self):
try:
from dmsic import Integration
i = Integration(self)
declar = {}
for k, v in self.__wgts.items():
if k in ('object_address', 'AppliedDocument', 'legal_entity',
'person'):
a = {}
for key, val in v.items():
if val.metaObject().className() == 'QDateEdit':
a[key] = datetime.strptime(val.text(), '%d.%m.%Y')
else:
a[key] = val.text()
declar[k] = a
else:
if v.metaObject().className() == 'QDateEdit':
declar[k] = datetime.strptime(v.text(), '%d.%m.%Y')
else:
declar[k] = v.text()
a = declar[
'AppliedDocument'] if 'AppliedDocument' in declar else []
for v in self.documents:
d = {}
for key, val in v.items():
if val.metaObject().className() == 'QDateEdit':
d[key] = datetime.strptime(val.text(), '%d.%m.%Y')
else:
d[key] = val.text()
if not self.doc_files:
raise Exception('Добавте файл документа')
d['file_name'] = self.doc_files[self.documents.index(v)]
a.append(d)
declar['AppliedDocument'] = a
a = declar['person'] if 'person' in declar else []
for v in self.individuals:
ind = {}
for key, val in v.items():
if key in ('address', 'fact_address'):
adr = {}
for k, vl in val.items():
adr[k] = vl.text()
ind[key] = adr
else:
if val.metaObject().className() == 'QDateEdit':
ind[key] = datetime.strptime(
val.text(), '%d.%m.%Y')
else:
ind[key] = val.text()
a.append(ind)
declar['person'] = a
a = declar['legal_entity'] if 'legal_entity' in declar else []
for v in self.entities:
ent = {}
for key, val in v.items():
if key == 'address':
adr = {}
for k, vl in val.items():
adr[k] = vl.text()
ent[key] = adr
else:
if val.metaObject().className() == 'QDateEdit':
ent[key] = datetime.strptime(
val.text(), '%d.%m.%Y')
else:
ent[key] = val.text()
a.append(ent)
declar['legal_entity'] = a
i.send(declar)
mb = QMessageBox(self.w)
mb.information(self.w, tr('Готово'), tr('Запрос отправлен'))
except:
self.report_error()
@pyqtSlot(bool)
def on_action_triggered(self):
a = Ui_Dialog()
d = QDialog()
a.setupUi(d)
d.exec()
def __retranslateUi(self, MainWindow):
_translate = QCoreApplication.translate
MainWindow.setWindowTitle(
_translate("MainWindow", "Отправка запросов на ВС АУГО"))
# self.pushButton.setText(_translate("MainWindow", "Добавить"))
self.menu.setTitle(_translate("MainWindow", "Справка"))
self.action_1.setText(_translate("MainWindow", "Отправить"))
self.action_2.setText(_translate("MainWindow", "Настройка"))
self.action.setText(_translate("MainWindow", "О программе"))
self.action_3.setText(_translate("MainWindow", "Получить ответ"))
@pyqtSlot(bool)
def __show_form(self):
if self.__showed:
return
self.gridLayout.addWidget(
QLabel(tr('№ дела (обращения) <em style="color: red">*</em>')))
w = QLineEdit()
self.gridLayout.addWidget(w)
w.setFocus()
self.__wgts['declar_number'] = w
self.gridLayout.addWidget(
QLabel(
tr('Услуга (код или номер, или наименование)<em style="color: red">*</em>'
)))
w = QLineEdit()
self.gridLayout.addWidget(w)
self.__wgts['service'] = w
self.gridLayout.addWidget(
QLabel(
tr('Дата регистрации запроса <em style="color: red">*</em>')))
de = QDateEdit(QDate().currentDate())
de.setCalendarPopup(True)
self.gridLayout.addWidget(de)
self.__wgts['register_date'] = de
self.gridLayout.addWidget(
QLabel(
tr('Плановый срок предоставления услуги <em style="color: red">*</em>'
)))
de = QDateEdit()
self.__wgts['register_date'].dateChanged.connect(de.setMinimumDate)
de.setCalendarPopup(True)
de.setMinimumDate(self.__wgts['register_date'].date())
self.gridLayout.addWidget(de)
self.__wgts['end_date'] = de
gb = QGroupBox(tr('Место нахождения объекта услуги'))
gb_l = QGridLayout()
self.__wgts['object_address'] = self.__add_address(gb_l)
gb.setLayout(gb_l)
self.gridLayout.addWidget(gb, self.gridLayout.rowCount() + 1, 0, 1, 2)
doc = {}
gb = QGroupBox(tr('Приложенный документ *'))
gb_l = QGridLayout()
gb_l.addWidget(
QLabel(tr('Наименование документа <em style="color: red">*</em>')))
w = QLineEdit()
w.setMaxLength(1024)
gb_l.addWidget(w, 0, 1, 1, 1)
doc['title'] = w
gb_l.addWidget(
QLabel(tr('Номер документа <em style="color: red">*</em>')))
w = QLineEdit()
w.setMaxLength(50)
gb_l.addWidget(w)
doc['number'] = w
gb_l.addWidget(
QLabel(tr('Дата документа <em style="color: red">*</em>')))
w = QDateEdit()
w.setCalendarPopup(True)
gb_l.addWidget(w)
doc['date'] = w
gb_l.addWidget(
QLabel(
tr('Прямая ссылка на файл. Поддерживаются только пртоколы '
'HTTP, FTP <em style="color: red">*</em>')))
w = QLineEdit()
gb_l.addWidget(w)
doc['url'] = w
gb.setLayout(gb_l)
self.gridLayout.addWidget(gb, self.gridLayout.rowCount() + 1, 0, 1, 2)
self.documents.append(doc)
gb = QGroupBox(tr('Заявители *'))
self.dec_layout = QGridLayout()
self.cb = QComboBox()
self.cb.addItems(('Физическое лицо',
'Юридическое лицо/Индивидуальный предприниматель'))
self.dec_layout.addWidget(self.cb)
b = QPushButton(tr('Добавить'))
b.clicked.connect(self.add_declarant)
self.dec_layout.addWidget(b, 0, 1, 1, 1)
gb.setLayout(self.dec_layout)
self.gridLayout.addWidget(gb, self.gridLayout.rowCount() + 1, 0, 1, 2)
b = QPushButton(tr('Добавить файл документа'))
b.clicked.connect(self.__add_doc_file)
self.gridLayout.addWidget(b)
self.file_label = QLabel()
self.gridLayout.addWidget(self.file_label)
self.warn_label = QLabel(tr("Не удаляйте файл до отправки запроса"))
self.warn_label.setStyleSheet('color: red')
self.warn_label.setVisible(False)
self.gridLayout.addWidget(self.warn_label,
self.gridLayout.rowCount() + 1, 0, 1, 2)
self.__showed = True
@pyqtSlot(bool)
def __add_doc_file(self):
file_name = QFileDialog.getOpenFileName(
caption=tr('Выбурите файл'),
filter=tr('Файлы pdf (*.pdf);;Все файлы (*.*)'))[0]
self.file_label.setText(file_name)
if self.doc_files:
self.doc_files = []
self.doc_files.append(file_name)
self.warn_label.setVisible(True)
def __add_address(self, gb_l):
wgts = {}
gb_l.addWidget(QLabel(tr('Почтовый индекс')))
w = QLineEdit()
gb_l.addWidget(w, 0, 1, 1, 1)
wgts['Postal_Code'] = w
gb_l.addWidget(QLabel(tr('Регион')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Region'] = w
gb_l.addWidget(QLabel(tr('Район')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['District'] = w
gb_l.addWidget(QLabel(tr('Муниципальное образование')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['City'] = w
gb_l.addWidget(QLabel(tr('Городской район')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Urban_District'] = w
gb_l.addWidget(QLabel(tr('Сельсовет')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Soviet_Village'] = w
gb_l.addWidget(
QLabel(tr('Населенный пункт <em style="color: red">*</em>')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Locality'] = w
cb = QComboBox()
cb.addItems(('Вариант 1', 'Вариант 2'))
gb_l.addWidget(cb)
st = QStackedWidget()
p1 = QWidget()
l = QGridLayout()
l.setSpacing(3)
l.addWidget(QLabel(tr('Улица <em style="color: red">*</em>')))
w = QLineEdit()
l.addWidget(w, 0, 1, 1, 1)
wgts["Street"] = w
l.addWidget(QLabel(tr('Дом <em style="color: red">*</em>')))
w = QLineEdit()
l.addWidget(w)
wgts["House"] = w
p1.setLayout(l)
st.addWidget(p1)
p2 = QWidget()
l = QGridLayout()
l.setSpacing(3)
l.addWidget(QLabel(tr('Ориентир')))
w = QLineEdit()
l.addWidget(w, 0, 1, 1, 1)
wgts["Reference_point"] = w
p2.setLayout(l)
st.addWidget(p2)
gb_l.addWidget(st, 9, 0, 1, 2)
cb.activated.connect(st.setCurrentIndex)
gb_l.addWidget(QLabel(tr('Корпус')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Housing'] = w
gb_l.addWidget(QLabel(tr('Строение')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Building'] = w
gb_l.addWidget(QLabel(tr('Квартира')))
w = QLineEdit()
gb_l.addWidget(w)
wgts['Apartment'] = w
return wgts
@pyqtSlot(bool)
def add_declarant(self, var=True, gl=None):
if not gl:
gl = self.dec_layout
dc = {}
gb_l = QGridLayout()
if self.cb.currentIndex() == 0 or gl != self.dec_layout:
# Add Individual
gb = QGroupBox(tr('Физическое лицо *'))
gb_l.addWidget(QLabel(tr('Фамилия <em style="color: red">*</em>')))
w = QLineEdit()
gb_l.addWidget(w, 0, 1, 1, 1)
dc['surname'] = w
gb_l.addWidget(QLabel(tr('Имя <em style="color: red">*</em>')))
w = QLineEdit()
gb_l.addWidget(w)
dc['first_name'] = w
gb_l.addWidget(QLabel(tr('Отчество')))
w = QLineEdit()
gb_l.addWidget(w)
dc['patronymic'] = w
adr = QGroupBox(tr('Адрес регистрации *'))
adr_l = QGridLayout()
dc['address'] = self.__add_address(adr_l)
adr.setLayout(adr_l)
gb_l.addWidget(adr, gb_l.rowCount() + 1, 0, 1, 2)
gb.setLayout(gb_l)
gl.addWidget(gb, gl.rowCount() + 1, 0, 1, 2)
self.individuals.append(dc)
else:
# Add LegalEntity
gb = QGroupBox(
tr('Юридическое лицо/Индивидуальный предприниматель *'))
gb_l.addWidget(
QLabel(
tr('Краткое наименование ЮЛ <em style="color: red">*</em>')
))
w = QLineEdit()
gb_l.addWidget(w, 0, 1, 1, 1)
dc['name'] = w
adr = QGroupBox(tr('Юридический адрес *'))
adr_l = QGridLayout()
dc['address'] = self.__add_address(adr_l)
adr.setLayout(adr_l)
gb_l.addWidget(adr, gb_l.rowCount() + 1, 0, 1, 2)
gb.setLayout(gb_l)
gl.addWidget(gb, gl.rowCount() + 1, 0, 1, 2)
self.entities.append(dc)
class ConversionDialog(Dialog):
def __init__(self, parent, force_entire_book=False):
self.prefs = self.prefsPrep()
self.parent = parent
self.force_entire_book = force_entire_book
self.criteria = None
Dialog.__init__(self, _('Chinese Conversion'),
'chinese_conversion_dialog', parent)
def setup_ui(self):
self.quote_for_trad_target = _("Update quotes: ＂＂,＇＇ -> 「」,『』")
self.quote_for_simp_target = _("Update quotes: 「」,『』 -> ＂＂,＇＇")
# Create layout for entire dialog
layout = QVBoxLayout(self)
self.setLayout(layout)
#Create a scroll area for the top part of the dialog
self.scrollArea = QScrollArea(self)
self.scrollArea.setWidgetResizable(True)
# Create widget for all the contents of the dialog except the OK and Cancel buttons
self.scrollContentWidget = QWidget(self.scrollArea)
self.scrollArea.setWidget(self.scrollContentWidget)
widgetLayout = QVBoxLayout(self.scrollContentWidget)
# Add scrollArea to dialog
layout.addWidget(self.scrollArea)
self.operation_group_box = QGroupBox(_('Conversion Direction'))
widgetLayout.addWidget(self.operation_group_box)
operation_group_box_layout = QVBoxLayout()
self.operation_group_box.setLayout(operation_group_box_layout)
operation_group = QButtonGroup(self)
self.no_conversion_button = QRadioButton(_('No Conversion'))
operation_group.addButton(self.no_conversion_button)
self.trad_to_simp_button = QRadioButton(_('Traditional to Simplified'))
operation_group.addButton(self.trad_to_simp_button)
self.simp_to_trad_button = QRadioButton(_('Simplified to Traditional'))
operation_group.addButton(self.simp_to_trad_button)
self.trad_to_trad_button = QRadioButton(
_('Traditional to Traditional'))
operation_group.addButton(self.trad_to_trad_button)
operation_group_box_layout.addWidget(self.no_conversion_button)
operation_group_box_layout.addWidget(self.trad_to_simp_button)
operation_group_box_layout.addWidget(self.simp_to_trad_button)
operation_group_box_layout.addWidget(self.trad_to_trad_button)
self.no_conversion_button.toggled.connect(self.update_gui)
self.trad_to_simp_button.toggled.connect(self.update_gui)
self.simp_to_trad_button.toggled.connect(self.update_gui)
self.trad_to_trad_button.toggled.connect(self.update_gui)
self.style_group_box = QGroupBox(_('Language Styles'))
widgetLayout.addWidget(self.style_group_box)
style_group_box_layout = QVBoxLayout()
self.style_group_box.setLayout(style_group_box_layout)
input_layout = QHBoxLayout()
style_group_box_layout.addLayout(input_layout)
self.input_region_label = QLabel(_('Input:'))
input_layout.addWidget(self.input_region_label)
self.input_combo = QComboBox()
input_layout.addWidget(self.input_combo)
self.input_combo.addItems([_('Mainland'), _('Hong Kong'), _('Taiwan')])
self.input_combo.setToolTip(_('Select the origin region of the input'))
self.input_combo.currentIndexChanged.connect(self.update_gui)
output_layout = QHBoxLayout()
style_group_box_layout.addLayout(output_layout)
self.output_region_label = QLabel(_('Output:'))
output_layout.addWidget(self.output_region_label)
self.output_combo = QComboBox()
output_layout.addWidget(self.output_combo)
self.output_combo.addItems(
[_('Mainland'), _('Hong Kong'),
_('Taiwan')])
self.output_combo.setToolTip(
_('Select the desired region of the output'))
self.output_combo.currentIndexChanged.connect(self.update_gui)
self.use_target_phrases = QCheckBox(
_('Use output target phrases if possible'))
self.use_target_phrases.setToolTip(
_('Check to allow region specific word replacements if available'))
style_group_box_layout.addWidget(self.use_target_phrases)
self.use_target_phrases.stateChanged.connect(self.update_gui)
self.quotation_group_box = QGroupBox(_('Quotation Marks'))
widgetLayout.addWidget(self.quotation_group_box)
quotation_group_box_layout = QVBoxLayout()
self.quotation_group_box.setLayout(quotation_group_box_layout)
quotation_group = QButtonGroup(self)
self.quotation_no_conversion_button = QRadioButton(_('No Conversion'))
quotation_group.addButton(self.quotation_no_conversion_button)
self.quotation_trad_to_simp_button = QRadioButton(
self.quote_for_simp_target)
quotation_group.addButton(self.quotation_trad_to_simp_button)
self.quotation_simp_to_trad_button = QRadioButton(
self.quote_for_trad_target)
quotation_group.addButton(self.quotation_simp_to_trad_button)
quotation_group_box_layout.addWidget(
self.quotation_no_conversion_button)
quotation_group_box_layout.addWidget(
self.quotation_simp_to_trad_button)
quotation_group_box_layout.addWidget(
self.quotation_trad_to_simp_button)
self.quotation_no_conversion_button.toggled.connect(self.update_gui)
self.quotation_trad_to_simp_button.toggled.connect(self.update_gui)
self.quotation_simp_to_trad_button.toggled.connect(self.update_gui)
self.use_smart_quotes = QCheckBox(
"""Use curved 'Smart" quotes if applicable""")
self.use_smart_quotes.setToolTip(
_('Use smart curved half-width quotes rather than straight full-width quotes'
))
quotation_group_box_layout.addWidget(self.use_smart_quotes)
self.use_smart_quotes.stateChanged.connect(self.update_gui)
self.other_group_box = QGroupBox(_('Other Changes'))
widgetLayout.addWidget(self.other_group_box)
other_group_box_layout = QVBoxLayout()
self.other_group_box.setLayout(other_group_box_layout)
text_dir_layout = QHBoxLayout()
other_group_box_layout.addLayout(text_dir_layout)
direction_label = QLabel(_('Text Direction:'))
text_dir_layout.addWidget(direction_label)
self.text_dir_combo = QComboBox()
text_dir_layout.addWidget(self.text_dir_combo)
self.text_dir_combo.addItems(
[_('No Conversion'),
_('Horizontal'),
_('Vertical')])
self.text_dir_combo.setToolTip(
_('Select the desired text orientation'))
self.text_dir_combo.currentIndexChanged.connect(self.update_gui)
self.optimization_group_box = QGroupBox(
_('Reader Device Optimization'))
other_group_box_layout.addWidget(self.optimization_group_box)
optimization_group_box_layout = QVBoxLayout()
self.optimization_group_box.setLayout(optimization_group_box_layout)
punc_group = QButtonGroup(self)
self.text_dir_punc_none_button = QRadioButton(
"""No presentation optimization""")
optimization_group_box_layout.addWidget(self.text_dir_punc_none_button)
self.text_dir_punc_button = QRadioButton(
"""Optimize presentation for Readium reader""")
self.text_dir_punc_button.setToolTip(
_('Use vert/horiz punctuation presentation forms for Chrome Readium Epub3 reader'
))
optimization_group_box_layout.addWidget(self.text_dir_punc_button)
self.text_dir_punc_kindle_button = QRadioButton(
"""Optimize presentation for Kindle reader""")
self.text_dir_punc_kindle_button.setToolTip(
_('Use vert/horiz puncuation presentation forms for Kindle reader')
)
optimization_group_box_layout.addWidget(
self.text_dir_punc_kindle_button)
self.text_dir_punc_none_button.toggled.connect(self.update_gui)
self.text_dir_punc_button.toggled.connect(self.update_gui)
self.text_dir_punc_kindle_button.toggled.connect(self.update_gui)
source_group = QButtonGroup(self)
self.file_source_button = QRadioButton(_('Selected File Only'))
self.book_source_button = QRadioButton(_('Entire eBook'))
source_group.addButton(self.file_source_button)
source_group.addButton(self.book_source_button)
self.source_group_box = QGroupBox(_('Source'))
if not self.force_entire_book:
widgetLayout.addWidget(self.source_group_box)
source_group_box_layout = QVBoxLayout()
self.source_group_box.setLayout(source_group_box_layout)
source_group_box_layout.addWidget(self.file_source_button)
source_group_box_layout.addWidget(self.book_source_button)
layout.addSpacing(10)
self.button_box = QDialogButtonBox(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
self.button_box.accepted.connect(self._ok_clicked)
self.button_box.rejected.connect(self.reject)
layout.addWidget(self.button_box)
self.input_combo.setCurrentIndex(self.prefs['input_format'])
self.output_combo.setCurrentIndex(self.prefs['output_format'])
self.no_conversion_button.setChecked(self.prefs['no_conversion'])
self.trad_to_simp_button.setChecked(self.prefs['trad_to_simp'])
self.simp_to_trad_button.setChecked(self.prefs['simp_to_trad'])
self.trad_to_trad_button.setChecked(self.prefs['trad_to_trad'])
if not self.force_entire_book:
self.file_source_button.setChecked(self.prefs['use_html_file'])
self.book_source_button.setChecked(self.prefs['use_entire_book'])
else:
self.file_source_button.setChecked(False)
self.book_source_button.setChecked(True)
self.quotation_no_conversion_button.setChecked(
self.prefs['quote_no_conversion'])
self.quotation_trad_to_simp_button.setChecked(
self.prefs['quote_trad_to_simp'])
self.quotation_simp_to_trad_button.setChecked(
self.prefs['quote_simp_to_trad'])
self.use_smart_quotes.setChecked(self.prefs['use_smart_quotes'])
self.text_dir_combo.setCurrentIndex(self.prefs['orientation'])
self.text_dir_punc_none_button.setChecked(
self.prefs['no_optimization'])
self.text_dir_punc_button.setChecked(
self.prefs['readium_optimization'])
self.text_dir_punc_kindle_button.setChecked(
self.prefs['kindle_optimization'])
self.update_gui()
def update_gui(self):
if (self.quotation_trad_to_simp_button.isChecked()):
self.use_smart_quotes.setEnabled(True)
else:
self.use_smart_quotes.setEnabled(False)
if self.text_dir_combo.currentIndex() == 0:
self.optimization_group_box.setEnabled(False)
self.text_dir_punc_none_button.setEnabled(False)
self.text_dir_punc_button.setEnabled(False)
self.text_dir_punc_kindle_button.setEnabled(False)
else:
self.optimization_group_box.setEnabled(True)
self.text_dir_punc_none_button.setEnabled(True)
self.text_dir_punc_button.setEnabled(True)
self.text_dir_punc_kindle_button.setEnabled(True)
if self.no_conversion_button.isChecked():
self.input_combo.setEnabled(False)
self.output_combo.setEnabled(False)
self.use_target_phrases.setEnabled(False)
self.output_region_label.setEnabled(False)
self.input_region_label.setEnabled(False)
self.style_group_box.setEnabled(False)
elif self.trad_to_simp_button.isChecked():
self.input_combo.setEnabled(True)
#only mainland output locale for simplified output
self.output_combo.setCurrentIndex(0)
self.output_combo.setEnabled(False)
self.use_target_phrases.setEnabled(True)
self.output_region_label.setEnabled(False)
self.input_region_label.setEnabled(True)
self.style_group_box.setEnabled(True)
elif self.simp_to_trad_button.isChecked():
#only mainland input locale for simplified input
self.input_combo.setCurrentIndex(0)
self.input_combo.setEnabled(False)
self.output_combo.setEnabled(True)
self.use_target_phrases.setEnabled(True)
self.output_region_label.setEnabled(True)
self.input_region_label.setEnabled(False)
self.style_group_box.setEnabled(True)
elif self.trad_to_trad_button.isChecked():
#Trad->Trad
#currently only mainland input locale for Trad->Trad
self.input_combo.setCurrentIndex(0)
self.input_combo.setEnabled(False)
self.output_combo.setEnabled(True)
self.use_target_phrases.setEnabled(True)
self.output_region_label.setEnabled(True)
self.input_region_label.setEnabled(False)
self.style_group_box.setEnabled(True)
else:
self.input_combo.setEnabled(True)
self.output_combo.setEnabled(True)
self.use_target_phrases.setEnabled(True)
self.style_group_box.setEnabled(True)
self.output_region_label.setEnabled(True)
self.input_region_label.setEnabled(True)
def _ok_clicked(self):
output_mode = 0
if self.trad_to_simp_button.isChecked():
output_mode = 1 #trad -> simp
if self.simp_to_trad_button.isChecked():
output_mode = 2 #simp -> trad
elif self.trad_to_trad_button.isChecked():
output_mode = 3 #trad -> trad
quote_mode = 0
if self.quotation_trad_to_simp_button.isChecked():
quote_mode = 1 #trad -> simp
if self.quotation_simp_to_trad_button.isChecked():
quote_mode = 2 #simp -> trad
optimization_mode = 0
if self.text_dir_punc_button.isChecked():
optimization_mode = 1 #Readium
if self.text_dir_punc_kindle_button.isChecked():
optimization_mode = 2 #Kindle
self.criteria = (self.file_source_button.isChecked(), output_mode,
self.input_combo.currentIndex(),
self.output_combo.currentIndex(),
self.use_target_phrases.isChecked(), quote_mode,
self.use_smart_quotes.isChecked(),
self.text_dir_combo.currentIndex(), optimization_mode)
self.savePrefs()
self.accept()
def getCriteria(self):
return self.criteria
def prefsPrep(self):
from calibre.utils.config import JSONConfig
plugin_prefs = JSONConfig(
'plugins/{0}_ChineseConversion_settings'.format(PLUGIN_SAFE_NAME))
plugin_prefs.defaults['input_format'] = 0
plugin_prefs.defaults['output_format'] = 0
plugin_prefs.defaults['no_conversion'] = True
plugin_prefs.defaults['trad_to_simp'] = False
plugin_prefs.defaults['use_html_file'] = True
plugin_prefs.defaults['simp_to_trad'] = False
plugin_prefs.defaults['trad_to_trad'] = False
plugin_prefs.defaults['use_entire_book'] = True
plugin_prefs.defaults['use_target_phrases'] = True
plugin_prefs.defaults['quote_no_conversion'] = True
plugin_prefs.defaults['quote_trad_to_simp'] = False
plugin_prefs.defaults['quote_simp_to_trad'] = False
plugin_prefs.defaults['use_smart_quotes'] = False
plugin_prefs.defaults['orientation'] = 0
plugin_prefs.defaults['no_optimization'] = True
plugin_prefs.defaults['readium_optimization'] = False
plugin_prefs.defaults['kindle_optimization'] = False
return plugin_prefs
def savePrefs(self):
self.prefs['input_format'] = self.input_combo.currentIndex()
self.prefs['output_format'] = self.output_combo.currentIndex()
self.prefs['no_conversion'] = self.no_conversion_button.isChecked()
self.prefs['trad_to_simp'] = self.trad_to_simp_button.isChecked()
self.prefs['use_html_file'] = self.file_source_button.isChecked()
self.prefs['simp_to_trad'] = self.simp_to_trad_button.isChecked()
self.prefs['trad_to_trad'] = self.trad_to_trad_button.isChecked()
self.prefs['use_entire_book'] = self.book_source_button.isChecked()
self.prefs['use_target_phrases'] = self.use_target_phrases.isChecked()
self.prefs[
'quote_no_conversion'] = self.quotation_no_conversion_button.isChecked(
)
self.prefs[
'quote_trad_to_simp'] = self.quotation_trad_to_simp_button.isChecked(
)
self.prefs[
'quote_simp_to_trad'] = self.quotation_simp_to_trad_button.isChecked(
)
self.prefs['use_smart_quotes'] = self.use_smart_quotes.isChecked()
self.prefs['orientation'] = self.text_dir_combo.currentIndex()
self.prefs[
'no_optimization'] = self.text_dir_punc_none_button.isChecked()
self.prefs[
'readium_optimization'] = self.text_dir_punc_button.isChecked()
self.prefs[
'kindle_optimization'] = self.text_dir_punc_kindle_button.isChecked(
)
class ConversionDialog(Dialog):
def __init__(self, parent, force_entire_book=False):
self.prefs = self.prefsPrep()
self.parent = parent
self.force_entire_book = force_entire_book
self.criteria = None
Dialog.__init__(self, _('Chinese Conversion'), 'chinese_conversion_dialog', parent)
def setup_ui(self):
# Create layout for entire dialog
layout = QVBoxLayout(self)
self.setLayout(layout)
#Create a scroll area for the top part of the dialog
self.scrollArea = QScrollArea(self)
self.scrollArea.setWidgetResizable(True)
# Create widget for all the contents of the dialog except the OK and Cancel buttons
self.scrollContentWidget = QWidget(self.scrollArea)
self.scrollArea.setWidget(self.scrollContentWidget)
widgetLayout = QVBoxLayout(self.scrollContentWidget)
# Add scrollArea to dialog
layout.addWidget(self.scrollArea)
self.other_group_box = QGroupBox(_('Other Changes'))
widgetLayout.addWidget(self.other_group_box)
other_group_box_layout = QVBoxLayout()
self.other_group_box.setLayout(other_group_box_layout)
text_dir_layout = QHBoxLayout()
other_group_box_layout.addLayout(text_dir_layout)
direction_label = QLabel(_('Text Direction:'))
text_dir_layout.addWidget(direction_label)
self.text_dir_combo = QComboBox()
text_dir_layout.addWidget(self.text_dir_combo)
self.text_dir_combo.addItems([_('No Conversion'), _('Horizontal'), _('Vertical')])
self.text_dir_combo.setToolTip(_('Select the desired text orientation'))
self.text_dir_combo.currentIndexChanged.connect(self.update_gui)
self.optimization_group_box = QGroupBox(_('Reader Device Optimization'))
other_group_box_layout.addWidget(self.optimization_group_box)
optimization_group_box_layout = QVBoxLayout()
self.optimization_group_box.setLayout(optimization_group_box_layout)
punc_group=QButtonGroup(self)
self.text_dir_punc_none_button = QRadioButton("""No presentation optimization""")
optimization_group_box_layout.addWidget(self.text_dir_punc_none_button)
self.text_dir_punc_button = QRadioButton("""Optimize presentation for Readium reader""")
self.text_dir_punc_button.setToolTip(_('Use vert/horiz punctuation presentation forms for Chrome Readium Epub3 reader'))
optimization_group_box_layout.addWidget(self.text_dir_punc_button)
self.text_dir_punc_kindle_button = QRadioButton("""Optimize presentation for Kindle reader""")
self.text_dir_punc_kindle_button.setToolTip(_('Use vert/horiz puncuation presentation forms for Kindle reader'))
optimization_group_box_layout.addWidget(self.text_dir_punc_kindle_button)
self.text_dir_punc_none_button.toggled.connect(self.update_gui)
self.text_dir_punc_button.toggled.connect(self.update_gui)
self.text_dir_punc_kindle_button.toggled.connect(self.update_gui)
source_group=QButtonGroup(self)
self.file_source_button = QRadioButton(_('Selected File Only'))
self.book_source_button = QRadioButton(_('Entire eBook'))
source_group.addButton(self.file_source_button)
source_group.addButton(self.book_source_button)
self.source_group_box = QGroupBox(_('Source'))
if not self.force_entire_book:
widgetLayout.addWidget(self.source_group_box)
source_group_box_layout = QVBoxLayout()
self.source_group_box.setLayout(source_group_box_layout)
source_group_box_layout.addWidget(self.file_source_button)
source_group_box_layout.addWidget(self.book_source_button)
layout.addSpacing(10)
self.button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
self.button_box.accepted.connect(self._ok_clicked)
self.button_box.rejected.connect(self.reject)
layout.addWidget(self.button_box)
if not self.force_entire_book:
self.file_source_button.setChecked(self.prefs['use_html_file'])
self.book_source_button.setChecked(self.prefs['use_entire_book'])
else:
self.file_source_button.setChecked(False)
self.book_source_button.setChecked(True)
self.text_dir_combo.setCurrentIndex(self.prefs['orientation'])
self.text_dir_punc_none_button.setChecked(self.prefs['no_optimization'])
self.text_dir_punc_button.setChecked(self.prefs['readium_optimization'])
self.text_dir_punc_kindle_button.setChecked(self.prefs['kindle_optimization'])
self.update_gui()
def update_gui(self):
if self.text_dir_combo.currentIndex() == 0:
self.optimization_group_box.setEnabled(False)
self.text_dir_punc_none_button.setEnabled(False)
self.text_dir_punc_button.setEnabled(False)
self.text_dir_punc_kindle_button.setEnabled(False)
else:
self.optimization_group_box.setEnabled(True)
self.text_dir_punc_none_button.setEnabled(True)
self.text_dir_punc_button.setEnabled(True)
self.text_dir_punc_kindle_button.setEnabled(True)
def _ok_clicked(self):
optimization_mode = 0
if self.text_dir_punc_button.isChecked():
optimization_mode = 1 #Readium
if self.text_dir_punc_kindle_button.isChecked():
optimization_mode = 2 #Kindle
self.criteria = (self.text_dir_combo.currentIndex(), optimization_mode)
self.savePrefs()
self.accept()
def getCriteria(self):
return self.criteria
def prefsPrep(self):
from calibre.utils.config import JSONConfig
plugin_prefs = JSONConfig('plugins/{0}_ChineseConversion_settings'.format(PLUGIN_SAFE_NAME))
plugin_prefs.defaults['use_html_file'] = True
plugin_prefs.defaults['use_entire_book'] = True
plugin_prefs.defaults['orientation'] = 0
plugin_prefs.defaults['no_optimization'] = True
plugin_prefs.defaults['readium_optimization'] = False
plugin_prefs.defaults['kindle_optimization'] = False
return plugin_prefs
def savePrefs(self):
self.prefs['use_html_file'] = self.file_source_button.isChecked()
self.prefs['use_entire_book'] = self.book_source_button.isChecked()
self.prefs['orientation'] = self.text_dir_combo.currentIndex()
self.prefs['no_optimization'] = self.text_dir_punc_none_button.isChecked()
self.prefs['readium_optimization'] = self.text_dir_punc_button.isChecked()
self.prefs['kindle_optimization'] = self.text_dir_punc_kindle_button.isChecked()
class RemoveDialog(Dialog):
def __init__(self, parent):
from calibre_plugins.diaps_toolbag.span_div_config import plugin_prefs as prefs
self.criteria = None
self.prefs = prefs
self.parent = parent
self.help_file_name = '{0}_span_div_help.html'.format(PLUGIN_SAFE_NAME)
self.taglist = TAGLIST
Dialog.__init__(self, _('Edit Spans & Divs'), 'toolbag_spans_divs_dialog', parent)
def setup_ui(self):
DELETE_STR = _('Delete')
MODIFY_STR = _('Modify')
NO_ATTRIB_STR = _('No attributes ("naked" tag)')
self.NO_CHANGE_STR = _('No change')
layout = QVBoxLayout(self)
self.setLayout(layout)
help_layout = QHBoxLayout()
layout.addLayout(help_layout)
# Add hyperlink to a help file at the right. We will replace the correct name when it is clicked.
help_label = QLabel('<a href="http://www.foo.com/">Plugin Help</a>', self)
help_label.setTextInteractionFlags(Qt.LinksAccessibleByMouse | Qt.LinksAccessibleByKeyboard)
help_label.setAlignment(Qt.AlignRight)
help_label.linkActivated.connect(self.help_link_activated)
help_layout.addWidget(help_label)
action_layout = QHBoxLayout()
layout.addLayout(action_layout)
label = QLabel(_('Action type:'), self)
action_layout.addWidget(label)
self.action_combo = QComboBox()
action_layout.addWidget(self.action_combo)
self.action_combo.addItems([DELETE_STR, MODIFY_STR])
self.action_combo.currentIndexChanged.connect(self.update_gui)
tag_layout = QHBoxLayout()
layout.addLayout(tag_layout)
label = QLabel(_('Tag name:'), self)
tag_layout.addWidget(label)
self.tag_combo = QComboBox()
tag_layout.addWidget(self.tag_combo)
self.tag_combo.addItems(self.taglist)
self.tag_combo.currentIndexChanged.connect(self.update_gui)
attr_layout = QHBoxLayout()
layout.addLayout(attr_layout)
label = QLabel(_('Having the attribute:'), self)
attr_layout.addWidget(label)
self.attr_combo = QComboBox()
attr_layout.addWidget(self.attr_combo)
self.attr_combo.addItems(self.prefs['attrs'])
self.attr_combo.addItem(NO_ATTRIB_STR)
self.attr_combo.currentIndexChanged.connect(self.update_gui)
srch_layout = QHBoxLayout()
layout.addLayout(srch_layout)
label = QLabel(_("Whose value is (no quotes):"), self)
srch_layout.addWidget(label)
self.srch_txt = QLineEdit('', self)
srch_layout.addWidget(self.srch_txt)
self.srch_method = QCheckBox(_('Regex'), self)
srch_layout.addWidget(self.srch_method)
newtag_layout = QHBoxLayout()
layout.addLayout(newtag_layout)
label = QLabel(_('Change tag to:'), self)
newtag_layout.addWidget(label)
self.newtag_combo = QComboBox()
newtag_layout.addWidget(self.newtag_combo)
self.newtag_combo.addItem(self.NO_CHANGE_STR)
self.newtag_combo.addItems(self.prefs['{}_changes'.format(unicode(self.tag_combo.currentText()))])
if self.action_combo.currentIndex() == 0:
self.newtag_combo.setDisabled(True)
newattr_layout = QVBoxLayout()
layout.addLayout(newattr_layout)
label = QLabel(_('New attribute string to insert (entire):'), self)
newattr_layout.addWidget(label)
self.newattr_txt = QLineEdit('', self)
newattr_layout.addWidget(self.newattr_txt)
self.copy_attr = QCheckBox(_('Copy existing attribute string'), self)
self.copy_attr.stateChanged.connect(self.update_txt_box)
newattr_layout.addWidget(self.copy_attr)
if self.action_combo.currentIndex() == 0:
self.copy_attr.setDisabled(True)
self.newattr_txt.setDisabled(True)
layout.addSpacing(10)
button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
button_box.accepted.connect(self._ok_clicked)
button_box.rejected.connect(self.reject)
layout.addWidget(button_box)
def update_gui(self):
if self.attr_combo.currentIndex() == self.attr_combo.count()-1:
self.srch_txt.clear()
self.srch_txt.setDisabled(True)
self.srch_method.setChecked(False)
self.srch_method.setDisabled(True)
else:
self.srch_txt.setDisabled(False)
self.srch_method.setDisabled(False)
self.newtag_combo.clear()
self.newtag_combo.addItem(self.NO_CHANGE_STR)
self.newtag_combo.addItems(self.prefs['{}_changes'.format(unicode(self.tag_combo.currentText()))])
if self.action_combo.currentIndex() == 0:
self.newtag_combo.setCurrentIndex(0)
self.newtag_combo.setDisabled(True)
self.newattr_txt.clear()
self.newattr_txt.setDisabled(True)
self.copy_attr.setChecked(False)
self.copy_attr.setDisabled(True)
else:
self.newtag_combo.setDisabled(False)
self.newattr_txt.setDisabled(False)
self.copy_attr.setDisabled(False)
def update_txt_box(self):
if self.copy_attr.isChecked():
self.newattr_txt.clear()
self.newattr_txt.setDisabled(True)
else:
self.newattr_txt.setDisabled(False)
def _ok_clicked(self):
if self.action_combo.currentIndex() == 0:
action = 'delete'
else:
action = 'modify'
if self.attr_combo.currentIndex() == self.attr_combo.count()-1:
attribute = None
else:
attribute = unicode(self.attr_combo.currentText())
srch_str = unicode(self.srch_txt.displayText())
if not len(srch_str):
srch_str = None
if srch_str is None and attribute is not None:
return error_dialog(self.parent, _('Error'), '<p>{0}'.format(
_('Must enter a value for the attribute selected')), det_msg='', show=True)
srch_method = 'normal'
if self.srch_method.isChecked():
srch_method = 'regex'
if self.newtag_combo.currentIndex() == 0:
newtag = None
else:
newtag = unicode(self.newtag_combo.currentText())
if action == 'modify' and newtag is None and self.copy_attr.isChecked():
return error_dialog(self.parent, _('Error'), '<p>{0}'.format(
_('What--exactly--would that achieve?')), det_msg='', show=True)
new_str = unicode(self.newattr_txt.displayText())
copy_attr = False
if self.copy_attr.isChecked():
copy_attr = True
if not len(new_str):
new_str = ''
self.criteria = (srch_str, srch_method, unicode(self.tag_combo.currentText()), attribute, action, newtag, new_str, copy_attr)
self.accept()
def getCriteria(self):
return self.criteria
def help_link_activated(self, url):
def get_help_file_resource():
# Copy the HTML helpfile to the plugin directory each time the
# link is clicked in case the helpfile is updated in newer plugins.
file_path = os.path.join(config_dir, 'plugins', self.help_file_name)
with open(file_path,'w') as f:
f.write(load_resource('resources/{}'.format(self.help_file_name)))
return file_path
url = 'file:///' + get_help_file_resource()
open_url(QUrl(url))
class UArmUI(object):
def __init__(self, ui, layout):
self.main_ui = ui
self.layout = layout
super(UArmUI, self).__init__()
self.handler = UArmHandler(self)
self.lang = self.main_ui.lang
self.status = 0
self.set_ui()
self.set_disable(True)
def set_ui(self):
self._set_common_top_ui()
self._set_tab()
self._set_common_down_ui()
self.connect_slot()
def _set_common_top_ui(self):
top_frame = QFrame()
top_frame.setMaximumHeight(60)
top_layout = QVBoxLayout(top_frame)
self.layout.addWidget(top_frame)
common_top_frame = QFrame()
common_top_frame.setMinimumHeight(50)
common_top_frame.setMaximumHeight(50)
common_top_layout = QHBoxLayout(common_top_frame)
top_layout.addWidget(common_top_frame)
common_top_layout.addStretch(0)
label = QLabel(i18n[self.lang]['Type'] + ':')
self.label_type = QLabel('')
self.label_type.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_type)
label = QLabel(i18n[self.lang]['Mode'] + ':')
self.label_mode = QLabel('')
self.label_mode.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_mode)
label = QLabel(i18n[self.lang]['HardwareVersion'] + ':')
self.label_hard_version = QLabel('')
self.label_hard_version.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_hard_version)
label = QLabel(i18n[self.lang]['FirmwareVersion'] + ':')
self.label_firm_version = QLabel('')
self.label_firm_version.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_firm_version)
label_1 = QLabel(i18n[self.lang]['Connected'] + ':')
self.label_connected = QLabel()
img = QImage()
self.label_connected.setMaximumHeight(20)
self.label_connected.setMaximumWidth(20)
self.label_connected.setScaledContents(True)
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.lnt_addr = QLineEdit('COM12')
self.lnt_addr.setMaximumWidth(50)
self.lnt_addr.setMinimumWidth(30)
self.btn_connect = QPushButton(i18n[self.lang]['Connect'])
# self.btn_connect.setMaximumWidth(50)
# common_top_layout.addStretch(0)
common_top_layout.setSpacing(10)
common_top_layout.addWidget(label_1)
common_top_layout.addWidget(self.label_connected)
common_top_layout.addWidget(self.lnt_addr)
common_top_layout.addWidget(self.btn_connect)
def _set_common_down_ui(self):
slider_frame = QFrame()
slider_layout = QGridLayout(slider_frame)
self.layout.addWidget(slider_frame)
label = QLabel(i18n[self.lang]['Speed'] + ':')
self.slider_speed = QSlider(Qt.Horizontal)
self.spinbox_speed = QSpinBox()
self.slider_speed.setMinimum(1)
self.slider_speed.setMaximum(100000)
self.slider_speed.setValue(10000)
self.spinbox_speed.setSingleStep(1)
self.spinbox_speed.setMinimum(1)
self.spinbox_speed.setMaximum(100000)
self.spinbox_speed.setValue(10000)
slider_layout.addWidget(label, 0, 0)
slider_layout.addWidget(self.slider_speed, 0, 1)
slider_layout.addWidget(self.spinbox_speed, 0, 2)
# label = QLabel(i18n[self.lang]['Acc'] + ':')
# self.slider_acc = QSlider(Qt.Horizontal)
# self.spinbox_acc = QSpinBox()
# self.slider_acc.setMinimum(1)
# self.slider_acc.setMaximum(100000)
# self.slider_acc.setValue(5000)
# self.spinbox_acc.setSingleStep(1)
# self.spinbox_acc.setMinimum(1)
# self.spinbox_acc.setMaximum(100000)
# self.spinbox_acc.setValue(5000)
# slider_layout.addWidget(label, 0, 3)
# slider_layout.addWidget(self.slider_acc, 0, 4)
# slider_layout.addWidget(self.spinbox_acc, 0, 5)
btn_frame = QFrame()
btn_layout = QGridLayout(btn_frame)
self.layout.addWidget(btn_frame)
self.btn_get_device_info = QPushButton(i18n[self.lang]['GetDeviceInfo'])
self.btn_get_position = QPushButton(i18n[self.lang]['GetPosition'])
self.btn_get_polar = QPushButton(i18n[self.lang]['GetPolar'])
self.btn_get_servo_angle = QPushButton(i18n[self.lang]['GetServoAngle'])
self.btn_get_mode = QPushButton(i18n[self.lang]['GetMode'])
btn_layout.addWidget(self.btn_get_device_info, 0, 0)
btn_layout.addWidget(self.btn_get_position, 0, 1)
btn_layout.addWidget(self.btn_get_polar, 0, 2)
btn_layout.addWidget(self.btn_get_servo_angle, 0, 3)
btn_layout.addWidget(self.btn_get_mode, 0, 4)
self.combobox_servo = QComboBox()
self.combobox_servo.setStyleSheet('''color: blue;''')
for item in ['axis-1', 'axis-2', 'axis-3', 'axis-all']:
self.combobox_servo.addItem(item)
self.combobox_servo.setCurrentIndex(0)
btn_layout.addWidget(self.combobox_servo, 0, 0)
self.btn_servo_attach = QPushButton(i18n[self.lang]['ServoAttach'])
self.btn_servo_detach = QPushButton(i18n[self.lang]['ServoDetach'])
self.combobox_mode = QComboBox()
self.combobox_mode.setStyleSheet('''color: blue;''')
for item in ['normal', 'laser', '3D', 'pen']:
self.combobox_mode.addItem(item)
self.combobox_mode.setCurrentIndex(0)
self.btn_set_mode = QPushButton(i18n[self.lang]['SetMode'])
btn_layout.addWidget(self.combobox_servo, 1, 0)
btn_layout.addWidget(self.btn_servo_attach, 1, 1)
btn_layout.addWidget(self.btn_servo_detach, 1, 2)
btn_layout.addWidget(self.combobox_mode, 1, 3)
btn_layout.addWidget(self.btn_set_mode, 1, 4)
self.btn_reset = QPushButton(i18n[self.lang]['Reset'])
self.btn_pump_on = QPushButton(i18n[self.lang]['PumpOn'])
self.btn_pump_off = QPushButton(i18n[self.lang]['PumpOff'])
self.btn_gripper_catch = QPushButton(i18n[self.lang]['GripperCatch'])
self.btn_gripper_release = QPushButton(i18n[self.lang]['GripperRelease'])
btn_layout.addWidget(self.btn_reset, 2, 0)
btn_layout.addWidget(self.btn_pump_on, 2, 1)
btn_layout.addWidget(self.btn_pump_off, 2, 2)
btn_layout.addWidget(self.btn_gripper_catch, 2, 3)
btn_layout.addWidget(self.btn_gripper_release, 2, 4)
@staticmethod
def slider_spinbox_related(value, master=None, slave=None, scale=1):
try:
slave.setValue(value * scale)
except Exception as e:
print(e)
def connect_slot(self):
self.btn_connect.clicked.connect(self.connect)
self.slider_speed.valueChanged.connect(
functools.partial(self.slider_spinbox_related, slave=self.spinbox_speed, scale=1))
self.spinbox_speed.valueChanged.connect(
functools.partial(self.slider_spinbox_related, slave=self.slider_speed, scale=1))
# self.slider_acc.valueChanged.connect(
# functools.partial(self.slider_spinbox_related, slave=self.spinbox_acc, scale=1))
# self.spinbox_acc.valueChanged.connect(
# functools.partial(self.slider_spinbox_related, slave=self.slider_acc, scale=1))
self.btn_get_device_info.clicked.connect(self.get_device_info)
self.btn_get_position.clicked.connect(self.get_position)
self.btn_get_polar.clicked.connect(self.get_polar)
self.btn_get_servo_angle.clicked.connect(self.get_servo_angle)
self.btn_get_mode.clicked.connect(self.get_mode)
self.btn_servo_attach.clicked.connect(self.set_servo_attach)
self.btn_servo_detach.clicked.connect(self.set_servo_detach)
self.btn_set_mode.clicked.connect(self.set_mode)
self.btn_reset.clicked.connect(self.reset)
self.btn_pump_on.clicked.connect(self.pump_on)
self.btn_pump_off.clicked.connect(self.pump_off)
self.btn_gripper_catch.clicked.connect(self.gripper_catch)
self.btn_gripper_release.clicked.connect(self.gripper_release)
def reset_flag(self):
self.cartesian_ui.reset_flag()
self.axis_ui.reset_flag()
def update_connect_status(self, item):
try:
img = QImage()
if item and self.status != 1:
self.status = 1
logger.info('connect to {} success'.format(self.handler.port))
if img.load(connect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Disconnect'])
self.btn_connect.setStyleSheet('''color: red;font:bold;''')
self.set_disable(False)
elif not item and self.status != 0:
self.status = 0
logger.info('disconnect from {0} or failed connect {0}'.format(self.handler.port))
self.handler.cmd_que.queue.clear()
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Connect'])
self.btn_connect.setStyleSheet('''color: green;font:bold;''')
self.set_disable(True)
except Exception as e:
print(e)
def connect(self):
try:
if str(self.btn_connect.text()) == i18n[self.lang]['Connect']:
addr = self.lnt_addr.text().strip()
if addr == 'auto':
addr = None
self.btn_connect.setText('Connecting')
self.status = 2
self.btn_connect.setStyleSheet('''color: orange;font:bold;''')
self.handler.connect(addr)
# if self.window.connect(addr, report_type=report_type):
# self.btn_connect.setText(self.disconnect_label)
# self.btn_connect.setStyleSheet('''color: red;font:bold;''')
elif str(self.btn_connect.text()) == i18n[self.lang]['Disconnect']:
self.handler.disconnect()
self.btn_connect.setText(i18n[self.lang]['Connect'])
self.btn_connect.setStyleSheet('''color: green;font:bold;''')
except Exception as e:
print(e)
def get_device_info(self, event):
try:
item = {
'cmd': 'get_device_info',
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def get_mode(self, event):
try:
item = {
'cmd': 'get_mode',
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def get_position(self, event):
try:
item = {
'cmd': 'get_position',
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def get_polar(self, event):
try:
item = {
'cmd': 'get_polar',
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def get_servo_angle(self, event):
try:
item = {
'cmd': 'get_servo_angle',
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def set_servo_attach(self, event):
try:
text = self.combobox_servo.currentText()
if text == 'axis-all':
servo_id = None
else:
servo_id = int(self.combobox_servo.currentIndex())
item = {
'cmd': 'set_servo_attach',
'kwargs': {
'servo_id': servo_id
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def set_servo_detach(self, event):
try:
text = self.combobox_servo.currentText()
if text == 'axis-all':
servo_id = None
else:
servo_id = int(self.combobox_servo.currentIndex())
item = {
'cmd': 'set_servo_detach',
'kwargs': {
'servo_id': servo_id
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def pump_on(self, event):
try:
item = {
'cmd': 'set_pump',
'kwargs': {
'on': True
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def pump_off(self, event):
try:
item = {
'cmd': 'set_pump',
'kwargs': {
'on': False
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def gripper_catch(self, event):
try:
item = {
'cmd': 'set_gripper',
'kwargs': {
'catch': True
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def gripper_release(self, event):
try:
item = {
'cmd': 'set_gripper',
'kwargs': {
'catch': False
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def set_mode(self, event):
try:
mode = self.combobox_mode.currentIndex()
item = {
'cmd': 'set_mode',
'kwargs': {
'mode': int(mode)
}
}
self.handler.put_cmd_que(item)
except Exception as e:
print(e)
def reset(self, event):
try:
self.handler.cmd_que.queue.clear()
item = {
'cmd': 'reset',
'kwargs': {
'speed': self.spinbox_speed.value(),
'wait': False,
}
}
self.handler.put_cmd_que(item)
self.reset_flag()
except Exception as e:
print(e)
@staticmethod
def slider_spinbox_related(value, master=None, slave=None, scale=1):
try:
slave.setValue(value * scale)
except Exception as e:
print(e)
def _set_tab(self):
tab_widget = QTabWidget()
tab_widget.setMaximumHeight(self.main_ui.window.geometry().height() // 2)
self.layout.addWidget(tab_widget)
toolBox1 = QToolBox()
toolBox2 = QToolBox()
groupBox1 = QGroupBox()
groupBox2 = QGroupBox()
toolBox1.addItem(groupBox1, "")
toolBox2.addItem(groupBox2, "")
tab_widget.addTab(toolBox1, i18n[self.lang]['Angle'])
tab_widget.addTab(toolBox2, i18n[self.lang]['Coordinate'])
joint_layout = QVBoxLayout(groupBox1)
cartesian_layout = QVBoxLayout(groupBox2)
self.cartesian_ui = CoordinateUI(self, cartesian_layout)
self.axis_ui = AngleUI(self, joint_layout)
def update_device_info(self, device_info):
if device_info is not None:
self.label_type.setText(str(device_info.get('device_type', None)))
self.label_hard_version.setText(str(device_info.get('hardware_version', None)))
self.label_firm_version.setText(str(device_info.get('firmware_version', None)))
def update_mode(self, mode):
if mode is not None:
if mode == 0:
mode_str = 'Normal'
elif mode == 1:
mode_str = 'Laser'
elif mode == 2:
mode_str = '3D'
elif mode == 3:
mode_str = 'Pen'
else:
mode_str = self.label_mode.text()
self.label_mode.setText(mode_str)
def set_disable(self, disable):
try:
self.btn_get_device_info.setDisabled(disable)
self.btn_get_position.setDisabled(disable)
self.btn_get_polar.setDisabled(disable)
self.btn_get_servo_angle.setDisabled(disable)
self.btn_get_mode.setDisabled(disable)
self.combobox_servo.setDisabled(disable)
self.combobox_mode.setDisabled(disable)
self.btn_servo_attach.setDisabled(disable)
self.btn_servo_detach.setDisabled(disable)
self.btn_reset.setDisabled(disable)
self.btn_pump_on.setDisabled(disable)
self.btn_pump_off.setDisabled(disable)
self.btn_gripper_catch.setDisabled(disable)
self.btn_gripper_release.setDisabled(disable)
self.slider_speed.setDisabled(disable)
self.spinbox_speed.setDisabled(disable)
# self.slider_acc.setDisabled(disable)
# self.spinbox_acc.setDisabled(disable)
self.axis_ui.set_disable(disable)
self.cartesian_ui.set_disable(disable)
except Exception as e:
print(e)
class PunctDialog(Dialog):
def __init__(self, parent):
self.prefs = self.prefsPrep()
self.criteria = None
self.parent = parent
self.help_file_name = '{0}_smarten_help.html'.format(PLUGIN_SAFE_NAME)
Dialog.__init__(self, _('Smarten Punctuation (the sequel)'), 'toolbag_smarter_dialog', parent)
def setup_ui(self,):
layout = QVBoxLayout(self)
self.setLayout(layout)
help_layout = QHBoxLayout()
layout.addLayout(help_layout)
# Add hyperlink to a help file at the right. We will replace the correct name when it is clicked.
help_label = QLabel('<a href="http://www.foo.com/">Plugin Help</a>', self)
help_label.setTextInteractionFlags(Qt.LinksAccessibleByMouse | Qt.LinksAccessibleByKeyboard)
help_label.setAlignment(Qt.AlignRight)
help_label.linkActivated.connect(self.help_link_activated)
help_layout.addWidget(help_label)
self.edu_quotes = QCheckBox(_('Smarten Quotation marks'), self)
layout.addWidget(self.edu_quotes)
self.edu_quotes.setChecked(self.prefs['edu_quotes'])
self.edu_quotes.stateChanged.connect(self.quotes_gui_changes)
exceptions_group_box = QGroupBox('', self)
layout.addWidget(exceptions_group_box)
exceptions_group_box_layout = QVBoxLayout()
exceptions_group_box.setLayout(exceptions_group_box_layout)
self.use_file = QCheckBox(_('Use custom apostrophe exceptions file'), self)
exceptions_group_box_layout.addWidget(self.use_file)
if not self.edu_quotes.isChecked():
self.use_file.setDisabled(True)
else:
self.use_file.setChecked(self.prefs['use_file'])
self.use_file.stateChanged.connect(self.use_file_gui_changes)
path_layout = QHBoxLayout()
exceptions_group_box_layout.addLayout(path_layout)
self.file_path = QLineEdit('', self)
if not self.edu_quotes.isChecked() and not self.use_file.isChecked():
self.file_path.setReadOnly(True)
else:
self.file_path.setText(self.prefs['file_path'])
self.file_path.setReadOnly(True)
path_layout.addWidget(self.file_path)
self.file_button = QPushButton('...', self)
self.file_button.clicked.connect(self.getFile)
path_layout.addWidget(self.file_button)
if not self.edu_quotes.isChecked() and not self.use_file.isChecked():
self.file_button.setDisabled(True)
combo_layout = QVBoxLayout()
layout.addLayout(combo_layout)
label = QLabel(_('(em|en)-dash settings'), self)
combo_layout.addWidget(label)
self.dashes_combo = QComboBox()
combo_layout.addWidget(self.dashes_combo)
values = [_('Do not educate dashes'), _('-- = emdash (no endash support)'),
_('-- = emdash | --- = endash'), _('--- = emdash | -- = endash')]
self.dashes_combo.addItems(values)
self.dashes_combo.setCurrentIndex(self.prefs['dashes'])
# self.dashes_combo.currentIndexChanged.connect(self.update_gui)
self.ellipses = QCheckBox(_('Smarten ellipses'), self)
layout.addWidget(self.ellipses)
self.ellipses.setChecked(self.prefs['ellipses'])
self.unicode = QCheckBox(_('Educate with unicode characters (instead of entities)'), self)
layout.addWidget(self.unicode)
self.unicode.setChecked(self.prefs['unicode'])
layout.addSpacing(10)
button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
button_box.accepted.connect(self._ok_clicked)
button_box.rejected.connect(self.reject)
layout.addWidget(button_box)
def getFile(self):
unique_dlg_name = '{0}plugin:smarter_choose_dialog'.format(PLUGIN_SAFE_NAME)
caption = _('Select custom apostrophe exceptions file')
filters = [('Text files', ['txt'])]
c = choose_files(self, unique_dlg_name, caption, filters, all_files=True)
if c:
self.file_path.setReadOnly(False)
self.file_path.setText(c[0])
self.file_path.setReadOnly(True)
def _ok_clicked(self):
quotes_setting = 'q' if self.edu_quotes.isChecked() else ''
if self.dashes_combo.currentIndex() == 0:
dash_setting = ''
elif self.dashes_combo.currentIndex() == 1:
dash_setting = 'd'
elif self.dashes_combo.currentIndex() == 2:
dash_setting = 'i'
elif self.dashes_combo.currentIndex() == 3:
dash_setting = 'D'
else:
dash_setting = ''
ellipses_setting = 'e' if self.ellipses.isChecked() else ''
smarty_attr = quotes_setting + dash_setting + ellipses_setting
if smarty_attr == '':
smarty_attr = '0'
self.file_path.setReadOnly(False)
if self.use_file.isChecked() and not len(self.file_path.displayText()):
self.file_path.setReadOnly(True)
return error_dialog(self.parent, _('Error'), '<p>' +
_('Must select a custom exception file'), det_msg='', show=True)
if self.use_file.isChecked():
apos_exception_file = unicode(self.file_path.displayText())
if not os.path.exists(apos_exception_file):
apos_exception_file = None
else:
apos_exception_file = None
self.file_path.setReadOnly(True)
apos_words_list = []
if apos_exception_file is not None:
apos_words_list = self.parseExceptionsFile(os.path.normpath(apos_exception_file))
self.criteria = (smarty_attr, self.unicode.isChecked(), apos_words_list)
self.savePrefs()
self.accept()
def getCriteria(self):
return self.criteria
def quotes_gui_changes(self):
if self.edu_quotes.isChecked():
self.use_file.setDisabled(False)
if self.use_file.isChecked():
self.file_button.setDisabled(False)
else:
self.use_file.setChecked(False)
self.file_path.setReadOnly(False)
self.file_path.clear()
self.file_path.setReadOnly(True)
self.use_file.setDisabled(True)
self.file_button.setDisabled(True)
def use_file_gui_changes(self):
if self.use_file.isChecked():
self.file_button.setDisabled(False)
else:
self.file_path.setReadOnly(False)
self.file_path.clear()
self.file_path.setReadOnly(True)
self.file_button.setDisabled(True)
def prefsPrep(self):
from calibre.utils.config import JSONConfig
plugin_prefs = JSONConfig('plugins/{0}_SmarterPunct_settings'.format(PLUGIN_SAFE_NAME))
plugin_prefs.defaults['edu_quotes'] = True
plugin_prefs.defaults['use_file'] = False
plugin_prefs.defaults['file_path'] = ''
plugin_prefs.defaults['dashes'] = 1
plugin_prefs.defaults['ellipses'] = True
plugin_prefs.defaults['unicode'] = True
return plugin_prefs
def savePrefs(self):
self.prefs['edu_quotes'] = self.edu_quotes.isChecked()
self.prefs['use_file'] = self.use_file.isChecked()
self.prefs['file_path'] = unicode(self.file_path.displayText()) if len(self.file_path.displayText()) else ''
self.prefs['dashes'] = self.dashes_combo.currentIndex()
self.prefs['ellipses'] = self.ellipses.isChecked()
self.prefs['unicode'] = self.unicode.isChecked()
def help_link_activated(self, url):
def get_help_file_resource():
# Copy the HTML helpfile to the plugin directory each time the
# link is clicked in case the helpfile is updated in newer plugins.
file_path = os.path.join(config_dir, 'plugins', self.help_file_name)
with open(file_path,'w') as f:
f.write(load_resource('resources/{}'.format(self.help_file_name)))
return file_path
url = 'file:///' + get_help_file_resource()
open_url(QUrl(url))
def parseExceptionsFile(self, filename):
import os, codecs, chardet
words_list = []
bytes = min(32, os.path.getsize(filename))
raw = open(filename, 'rb').read(bytes)
if raw.startswith(codecs.BOM_UTF8):
enc = 'utf-8-sig'
else:
result = chardet.detect(raw)
enc = result['encoding']
try:
with codecs.open(filename, encoding=enc, mode='r') as fd:
words_list = [line.rstrip() for line in fd]
words_list = filter(None, words_list)
print('Exceptions list:', words_list)
except:
pass
return words_list
class ParameterMeshGroupView(ParameterView):
"""Top-level table editor item."""
meshFileChanged = pyqtSignal(str, str, float, bool)
meshGroupCheck = pyqtSignal(str, str, str)
meshGroupUnCheck = pyqtSignal(str, str, str)
meshChanged = pyqtSignal()
"""Signal: emitted when mesh is changed in the combo box."""
def __init__(self, panel, **kwargs):
"""
Create view.
Arguments:
**kwargs: Arbitrary keyword arguments.
"""
super(ParameterMeshGroupView, self).__init__(panel, **kwargs)
self.setStretchable(True)
self._mesh = QComboBox(self)
self._mesh.setObjectName("MESH")
self._msg = QLabel(self)
self._list = QTreeWidget(self)
self._list.setAllColumnsShowFocus(True)
self._list.setSelectionMode(QTreeWidget.SingleSelection)
self._list.setColumnCount(2)
titles = []
titles.append(translate("AsterStudy", "Name"))
titles.append(translate("AsterStudy", "Size"))
self._list.setHeaderLabels(titles)
self._list.header().setSectionResizeMode(QHeaderView.ResizeToContents)
self._list.header().setStretchLastSection(True)
manlabel = QLabel(translate("ParameterPanel", "Manual selection"),
self)
manlabel.setToolTip(
translate(
"ParameterPanel", "Enter manually the wanted groups if "
"not present in the list"))
self._manual = QLineEdit(self)
self._manual.setObjectName("MANUAL_INPUT")
base = self.grid()
base.addWidget(self._mesh, 0, 0, 1, -1)
base.addWidget(self._msg, 1, 0, 1, -1)
base.addWidget(self._list, 2, 0, 1, -1)
base.addWidget(manlabel, 3, 0, 1, -1)
base.addWidget(self._manual, 4, 0, 1, -1)
self._mesh.activated[int].connect(self._meshActivated)
self._updateMeshList()
self.meshFileChanged.connect(self.meshview().displayMEDFileName)
self.meshGroupCheck.connect(self.meshview().displayMeshGroup)
self.meshGroupUnCheck.connect(self.meshview().undisplayMeshGroup)
self._list.itemChanged.connect(self.meshGroupToChange)
def meshList(self):
"""
Gets the mesh commands list
Returns:
list (Command): List of commands with meshes.
"""
mlist = []
for i in xrange(self._mesh.count()):
mlist.append(self._mesh.itemData(i).name)
return mlist
def setMeshList(self, meshlist):
"""
Sets the mesh commands list
Arguments:
meshlist: List of commands with meshes.
"""
self._mesh.clear()
show_title = behavior().show_catalogue_name_in_selectors
title_mask = '{n} ({t})' if show_title else '{n}'
for meshcmd in meshlist:
title = title_mask.format(n=meshcmd.name, t=meshcmd.title)
self._mesh.addItem(title, meshcmd)
def mesh(self):
"""
Gets the currently selected mesh command object or None in error case.
Returns:
Command: Current mesh command object.
"""
idx = self._mesh.currentIndex()
return self._mesh.itemData(idx) if idx >= 0 else None
def setMesh(self, mesh):
"""
Sets the current mesh command object if it exists in the list.
Arguments:
mesh: Current mesh command object.
"""
self._mesh.setCurrentIndex(self._mesh.findData(mesh))
def message(self):
"""
Gets the info message text.
Returns:
str: info message text.
"""
return self._msg.text()
def setMessage(self, msg):
"""
Sets the info message text.
Arguments:
msg (str): info message text.
"""
self._msg.setText(msg)
self._msg.setVisible(len(msg) > 0)
def setMeshGroups(self, groups):
"""
Sets the mesh group list
Arguments:
groups (dict[int, list[tuple[str, int]]]): Mesh groups info.
"""
self._list.clear()
grp_types = sorted(groups.keys())
for typ in grp_types:
names = groups[typ]
if not names:
continue
title = MeshElemType.value2str(typ)
item = QTreeWidgetItem(self._list, [title])
for name, size in names:
sub_item = QTreeWidgetItem(item, [name, str(size)])
sub_item.setFlags(Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
sub_item.setCheckState(0, Qt.Unchecked)
sub_item.setTextAlignment(1, Qt.AlignRight)
self._list.expandAll()
def inputMeshGroups(self):
"""
Gets the mesh group names list entered manually
Returns:
list (str): List of group names.
"""
text = self._manual.text().strip()
return [i.strip() for i in text.split(",")] if len(text) > 0 else []
def setInputMeshGroups(self, groups):
"""
Sets the mesh group list entered manually
Arguments:
groups: List of mesh group names.
"""
self._manual.setText(",".join(groups))
def selectedMeshGroups(self):
"""
Gets the names of selected (checked) mesh groups.
Returns:
list (str): List of selected group names.
"""
groups = []
for i in range(self._list.topLevelItemCount()):
item = self._list.topLevelItem(i)
for j in xrange(item.childCount()):
sub_item = item.child(j)
if sub_item.checkState(0) == Qt.Checked:
groups.append(sub_item.text(0))
return list(set(groups))
def setSelectedMeshGroups(self, groups):
"""
Sets the specified group names are selected (checked)
and unchecked all other.
Arguments:
groups: List of selected mesh group names.
"""
for i in range(self._list.topLevelItemCount()):
item = self._list.topLevelItem(i)
for j in xrange(item.childCount()):
sub_item = item.child(j)
state = Qt.Checked if sub_item.text(0) in groups \
else Qt.Unchecked
sub_item.setCheckState(0, state)
@pyqtSlot(QTreeWidgetItem, int)
def meshGroupToChange(self, item, column):
"""
Emits display signal whenever the user clicks a check box
"""
meshcmd = self._meshcmd(self._mesh.currentIndex())
if meshcmd is not None:
file_name, nom_med = get_cmd_mesh(meshcmd)
if file_name is not None and nom_med is not None:
if item.checkState(column) == Qt.Checked:
self.meshGroupCheck.emit(file_name, nom_med, item.text(0))
else:
self.meshGroupUnCheck.emit(file_name, nom_med,
item.text(0))
def _meshcmd(self, index):
"""
Returns the *Command* instance associated with the panel
"""
meshcmd = None
if 0 <= index < self._mesh.count():
meshcmd = self._mesh.itemData(index)
return meshcmd
def itemValue(self, **kwargs):
"""
Get selected values.
Returns:
tuple: List with all selected mesh groups
"""
res = tuple(self.selectedMeshGroups() + self.inputMeshGroups())
return res if len(res) > 0 else None
def setItemValue(self, values):
"""
Set values of child items.
Arguments:
values: Tuple with item values (see `childValues()`).
"""
grplist = []
if values is not None:
if isinstance(values, (tuple, list)):
grplist = list(values)
else:
grplist = [values]
self.setSelectedMeshGroups(grplist)
check = dict.fromkeys(self.selectedMeshGroups())
grplist = [grp for grp in grplist if grp not in check]
self.setInputMeshGroups(grplist)
self._cache = self.itemValue()
def filterItem(self, text):
"""
Filter out the item.
Arguments:
text (str): Regular expression.
"""
regex = QRegExp(text, Qt.CaseInsensitive)
for i in range(self._list.topLevelItemCount()):
item = self._list.topLevelItem(i)
cnt_visible = 0
for j in xrange(item.childCount()):
sub_item = item.child(j)
item_text = sub_item.text(0)
hidden = text != "" and regex.indexIn(item_text) == -1
sub_item.setHidden(hidden)
if not hidden:
cnt_visible += 1
item.setHidden(cnt_visible == 0)
def _updateMeshList(self):
"""
Updates the mesh list in the combobox
"""
meshlist = avail_meshes(parameterPanel(self).pendingStorage())
meshlist.reverse()
self.setMeshList(meshlist)
msg = ""
if len(meshlist) > 1:
msg = translate("ParameterPanel", "More than one mesh found")
elif len(meshlist) == 0:
msg = translate("ParameterPanel", "No mesh found")
self.setMessage(msg)
self._meshActivated(self._mesh.currentIndex())
def _meshActivated(self, index):
"""
Updates the mesh groups in checkable list.
Invoked after mesh changing in mesh combobox.
"""
meshcmd = None
if 0 <= index < self._mesh.count():
meshcmd = self._mesh.itemData(index)
groups = {}
if meshcmd is not None:
group_type = self._meshGroupType()
file_name, nom_med = get_cmd_mesh(meshcmd)
if is_medfile(file_name) or is_reference(file_name):
self.meshFileChanged.emit(file_name, nom_med, 0.1, False)
try:
groups = get_cmd_groups(meshcmd, group_type, with_size=True)
except TypeError:
pass
self.setMeshGroups(groups)
self.meshChanged.emit()
def _meshGroupType(self):
"""
Get the type of the mesh group
Returns:
str: Mesh group type (see `MeshGroupType`).
"""
mgtype = -1
name = self.itemName()
if name.endswith("_MA") or is_contains_word(name, "MA"):
mgtype = MeshGroupType.GElement
elif name.endswith("_NO") or is_contains_word(name, "NO"):
mgtype = MeshGroupType.GNode
return mgtype
class InPlotForm(QMainWindow):
def __init__(self, mainGui):
super().__init__()
self.mainGui = mainGui
self.graphics = InPlotImage(self)
self.setCentralWidget(self.graphics)
self.initToolBar()
#self.graphics.setContextMenuPolicy(Qt.CustomContextMenu)
#self.graphics.customContextMenuRequested.connect(self.openMenu)
#def openMenu(self, position):
# action = self.graphics.mode.popMenu.exec_(self.mapToGlobal(position))
def initToolBar(self):
tb = QToolBar(self)
self.addToolBar(Qt.LeftToolBarArea, tb)
tb.actionTriggered[QAction].connect(self.toolAction)
deleteAction = QAction(QIcon('ppf/ui/icons/delete.png'), 'delete all',
self)
deleteAction.a = lambda: self.graphics.deleteAll()
tb.addAction(deleteAction)
self.penCombo = QComboBox()
tb.addWidget(self.penCombo)
pixmap = QPixmap(20, 20)
for pen in self.graphics.PENS:
pixmap.fill(pen.color())
self.penCombo.addItem(QIcon(pixmap), '')
self.penCombo.currentIndexChanged.connect(lambda: self.penChanged())
self.linewidthCombo = QComboBox()
tb.addWidget(self.linewidthCombo)
pixmap = QPixmap(20, 20)
for w in self.graphics.LINEWIDTH:
qp = QPainter()
pixmap.fill(Qt.white)
qp.begin(pixmap)
qp.fillRect(0, 0, 20, w, Qt.black)
qp.end()
self.linewidthCombo.addItem(QIcon(pixmap), '')
self.linewidthCombo.currentIndexChanged.connect(
lambda: self.linewidthChanged())
#open = QAction(QIcon("open.bmp"),"open",self)
#tb.addAction(open)
#save = QAction(QIcon("save.bmp"),"save",self)
#tb.addAction(save)
#tb.actionTriggered[QAction].connect(self.toolbtnpressed)
#self.setLayout(layout)
#self.setWindowTitle("toolbar demo")
def penChanged(self):
self.graphics.currentPen = self.graphics.PENS[
self.penCombo.currentIndex()]
def linewidthChanged(self):
self.graphics.currentWidth = self.graphics.LINEWIDTH[
self.linewidthCombo.currentIndex()]
def toolAction(self, a):
g().outText.print('action')
a.a()
class InsertSemantics(Dialog):
def __init__(self, container, parent=None):
self.container = container
self.anchor_cache = {}
self.original_type_map = {item.get('type', ''):(container.href_to_name(item.get('href'), container.opf_name), item.get('href', '').partition('#')[-1])
for item in container.opf_xpath('//opf:guide/opf:reference[@href and @type]')}
self.final_type_map = self.original_type_map.copy()
self.create_known_type_map()
Dialog.__init__(self, _('Set Semantics'), 'insert-semantics', parent=parent)
def sizeHint(self):
return QSize(800, 600)
def create_known_type_map(self):
_ = lambda x: x
self.known_type_map = {
'title-page': _('Title Page'),
'toc': _('Table of Contents'),
'index': _('Index'),
'glossary': _('Glossary'),
'acknowledgements': _('Acknowledgements'),
'bibliography': _('Bibliography'),
'colophon': _('Colophon'),
'copyright-page': _('Copyright page'),
'dedication': _('Dedication'),
'epigraph': _('Epigraph'),
'foreword': _('Foreword'),
'loi': _('List of Illustrations'),
'lot': _('List of Tables'),
'notes': _('Notes'),
'preface': _('Preface'),
'text': _('Text'),
}
_ = __builtins__['_']
type_map_help = {
'title-page': _('Page with title, author, publisher, etc.'),
'index': _('Back-of-book style index'),
'text': _('First "real" page of content'),
}
t = _
all_types = [(k, (('%s (%s)' % (t(v), type_map_help[k])) if k in type_map_help else t(v))) for k, v in self.known_type_map.iteritems()]
all_types.sort(key=lambda x: sort_key(x[1]))
self.all_types = OrderedDict(all_types)
def setup_ui(self):
self.l = l = QVBoxLayout(self)
self.setLayout(l)
self.tl = tl = QFormLayout()
self.semantic_type = QComboBox(self)
for key, val in self.all_types.iteritems():
self.semantic_type.addItem(val, key)
tl.addRow(_('Type of &semantics:'), self.semantic_type)
self.target = t = QLineEdit(self)
t.setPlaceholderText(_('The destination (href) for the link'))
tl.addRow(_('&Target:'), t)
l.addLayout(tl)
self.hline = hl = QFrame(self)
hl.setFrameStyle(hl.HLine)
l.addWidget(hl)
self.h = h = QHBoxLayout()
l.addLayout(h)
names = [n for n, linear in self.container.spine_names]
fn, f = create_filterable_names_list(names, filter_text=_('Filter files'), parent=self)
self.file_names, self.file_names_filter = fn, f
fn.selectionModel().selectionChanged.connect(self.selected_file_changed)
self.fnl = fnl = QVBoxLayout()
self.la1 = la = QLabel(_('Choose a &file:'))
la.setBuddy(fn)
fnl.addWidget(la), fnl.addWidget(f), fnl.addWidget(fn)
h.addLayout(fnl), h.setStretch(0, 2)
fn, f = create_filterable_names_list([], filter_text=_('Filter locations'), parent=self)
self.anchor_names, self.anchor_names_filter = fn, f
fn.selectionModel().selectionChanged.connect(self.update_target)
fn.doubleClicked.connect(self.accept, type=Qt.QueuedConnection)
self.anl = fnl = QVBoxLayout()
self.la2 = la = QLabel(_('Choose a &location (anchor) in the file:'))
la.setBuddy(fn)
fnl.addWidget(la), fnl.addWidget(f), fnl.addWidget(fn)
h.addLayout(fnl), h.setStretch(1, 1)
self.bb.addButton(self.bb.Help)
self.bb.helpRequested.connect(self.help_requested)
l.addWidget(self.bb)
self.semantic_type_changed()
self.semantic_type.currentIndexChanged.connect(self.semantic_type_changed)
self.target.textChanged.connect(self.target_text_changed)
def help_requested(self):
d = info_dialog(self, _('About semantics'), _(
'Semantics refer to additional information about specific locations in the book.'
' For example, you can specify that a particular location is the dedication or the preface'
' or the table of contents and so on.\n\nFirst choose the type of semantic information, then'
' choose a file and optionally a location within the file to point to.\n\nThe'
' semantic information will be written in the <guide> section of the opf file.'))
d.resize(d.sizeHint())
d.exec_()
def semantic_type_changed(self):
item_type = unicode(self.semantic_type.itemData(self.semantic_type.currentIndex()) or '')
name, frag = self.final_type_map.get(item_type, (None, None))
self.show_type(name, frag)
def show_type(self, name, frag):
self.file_names_filter.clear(), self.anchor_names_filter.clear()
self.file_names.clearSelection(), self.anchor_names.clearSelection()
if name is not None:
row = self.file_names.model().find_name(name)
if row is not None:
sm = self.file_names.selectionModel()
sm.select(self.file_names.model().index(row), sm.ClearAndSelect)
if frag:
row = self.anchor_names.model().find_name(frag)
if row is not None:
sm = self.anchor_names.selectionModel()
sm.select(self.anchor_names.model().index(row), sm.ClearAndSelect)
self.target.blockSignals(True)
if name is not None:
self.target.setText(name + (('#' + frag) if frag else ''))
else:
self.target.setText('')
self.target.blockSignals(False)
def target_text_changed(self):
name, frag = unicode(self.target.text()).partition('#')[::2]
item_type = unicode(self.semantic_type.itemData(self.semantic_type.currentIndex()) or '')
self.final_type_map[item_type] = (name, frag or None)
def selected_file_changed(self, *args):
rows = list(self.file_names.selectionModel().selectedRows())
if not rows:
self.anchor_names.model().set_names([])
else:
name, positions = self.file_names.model().data(rows[0], Qt.UserRole)
self.populate_anchors(name)
def populate_anchors(self, name):
if name not in self.anchor_cache:
from calibre.ebooks.oeb.base import XHTML_NS
root = self.container.parsed(name)
self.anchor_cache[name] = sorted(
(set(root.xpath('//*/@id')) | set(root.xpath('//h:a/@name', namespaces={'h':XHTML_NS}))) - {''}, key=primary_sort_key)
self.anchor_names.model().set_names(self.anchor_cache[name])
self.update_target()
def update_target(self):
rows = list(self.file_names.selectionModel().selectedRows())
if not rows:
return
name = self.file_names.model().data(rows[0], Qt.UserRole)[0]
href = name
frag = ''
rows = list(self.anchor_names.selectionModel().selectedRows())
if rows:
anchor = self.anchor_names.model().data(rows[0], Qt.UserRole)[0]
if anchor:
frag = '#' + anchor
href += frag
self.target.setText(href or '#')
@property
def changed_type_map(self):
return {k:v for k, v in self.final_type_map.iteritems() if v != self.original_type_map.get(k, None)}
def apply_changes(self, container):
from calibre.ebooks.oeb.polish.opf import set_guide_item, get_book_language
from calibre.translations.dynamic import translate
lang = get_book_language(container)
for item_type, (name, frag) in self.changed_type_map.iteritems():
title = self.known_type_map[item_type]
if lang:
title = translate(lang, title)
set_guide_item(container, item_type, title, name, frag=frag)
@classmethod
def test(cls):
import sys
from calibre.ebooks.oeb.polish.container import get_container
c = get_container(sys.argv[-1], tweak_mode=True)
d = cls(c)
if d.exec_() == d.Accepted:
import pprint
pprint.pprint(d.changed_type_map)
d.apply_changes(d.container)
class StartMenu(QWidget):
def __init__(self, parent):
super(StartMenu, self).__init__(parent)
self.tophbox = QHBoxLayout()
self.hbox = QHBoxLayout()
self.vbox = QVBoxLayout()
self.label = QLabel()
self.label.setPixmap(QPixmap('img/new-game.png'))
self.label.setScaledContents(True)
self.label.setFixedSize(600, 200)
self.tophbox.addWidget(self.label)
self.startbutton = Button(self, 'Start')
self.startbutton.setEnabled(False)
self.startbutton.setFixedHeight(100)
self.tophbox.addWidget(self.startbutton)
self.playeramt_label = QLabel('Number of players:')
self.playeramt_label.setFixedWidth(125)
self.playeramount = QComboBox()
self.playeramount.setStyleSheet('color: rgb(0, 0, 0)')
self.playeramount.setFixedWidth(50)
self.playeramount.addItems([str(i) for i in range(2, 13)])
self.playeramount.setCurrentIndex(2)
self.playeramount.setMaxVisibleItems(11)
self.playeramount.currentTextChanged.connect(self.form_player_entries)
self.score_label = QLabel('Score limit:')
self.score_label.setFixedWidth(65)
self.score_limit = QLineEdit()
self.score_limit.setMaximumWidth(40)
self.score_limit.setPalette(QPalette(Qt.white))
self.score_limit.setText('16')
self.mode_label = QLabel('Game Mode:')
self.mode_label.setFixedWidth(85)
self.mode_select = QComboBox()
self.mode_select.addItems(['Deal-1', 'Deal-4'])
self.mode_select.setPalette(QPalette(Qt.white))
self.mode_select.setFixedWidth(100)
self.mode_select.currentTextChanged.connect(self.update_playeramount)
self.autofill_button = Button(self, 'Auto Fill')
self.autofill_button.clicked.connect(self.auto_fill)
self.clear_button = Button(self, 'Clear All')
self.clear_button.clicked.connect(self.clear_all)
self.player_entries = QVBoxLayout()
self.spacer = QSpacerItem(20, 40, QSizePolicy.Minimum, QSizePolicy.Expanding)
self.hbox.addWidget(self.playeramt_label)
self.hbox.addWidget(self.playeramount)
self.hbox.addWidget(self.score_label)
self.hbox.addWidget(self.score_limit)
self.hbox.addWidget(self.mode_label)
self.hbox.addWidget(self.mode_select)
self.hbox.addWidget(self.autofill_button)
self.hbox.addWidget(self.clear_button)
self.vbox.addLayout(self.tophbox)
self.vbox.addLayout(self.hbox)
self.vbox.addLayout(self.player_entries)
self.vbox.addItem(self.spacer)
self.setLayout(self.vbox)
self.form_player_entries()
def form_player_entries(self):
amt = self.playeramount.currentIndex() + 2 #desired amount of player entries
curr = self.player_entries.count() #current amount of player entries
dif = amt - self.player_entries.count() #difference between desired and current entries
if dif < 0: #if too many entries currently
for i in range(curr-1, amt-1, -1): #remove starting from last entry
rm = self.player_entries.itemAt(i).widget()
self.player_entries.removeWidget(rm)
rm.setParent(None)
else: #if too few entries, add until desired amount reached
for i in range(dif):
new_entry = PlayerInfoField(self, self.player_entries.count())
new_entry.name_field.textChanged.connect(self.check_filled)
self.player_entries.addWidget(new_entry)
self.check_filled()
def check_filled(self): #Enables start button when player fields are correctly filled
ready = True
for i in range(self.player_entries.count()):
entry = self.player_entries.itemAt(i).widget()
if entry.name_field.text() == '':
ready = False
break
if ready:
self.startbutton.setEnabled(True)
else:
self.startbutton.setEnabled(False)
def auto_fill(self): #Generates fills the rest of the form automatically
for i in range(self.player_entries.count()):
entry = self.player_entries.itemAt(i).widget()
if entry.name_field.text() == '':
entry.generate_name()
entry.AItoggle.setChecked(True)
def clear_all(self):
for i in range(self.player_entries.count()):
entry = self.player_entries.itemAt(i).widget()
entry.name_field.clear()
entry.AItoggle.setChecked(False)
def update_playeramount(self, mode):
ind = self.playeramount.currentIndex()
if mode == 'Deal-1': #Limit max players to 12
self.playeramount.clear()
self.playeramount.addItems([str(i) for i in range(2, 13)])
self.playeramount.setCurrentIndex(ind)
if mode == 'Deal-4': #Limit max players to 4
self.playeramount.clear()
self.playeramount.addItems([str(i) for i in range(2, 5)])
self.playeramount.setCurrentIndex(min(2, ind))
self.check_filled()
def extract_info_and_init_game(self): #Creates a game object based on the info
pointlimit = int(self.score_limit.text())
dealmode = self.mode_select.currentIndex()
game = Game(pointlimit, dealmode)
for i in range(self.player_entries.count()):
entry = self.player_entries.itemAt(i).widget()
name = entry.name_field.text()
if entry.AItoggle.isChecked():
difficulty = entry.AIdifficulty.currentIndex()
game.add_player(name, difficulty)
else:
game.add_player(name, 5)
return game
class Bool(Base):
def setup_ui(self, parent):
self.widgets = [QLabel('&'+self.col_metadata['name']+':', parent)]
w = QWidget(parent)
self.widgets.append(w)
l = QHBoxLayout()
l.setContentsMargins(0, 0, 0, 0)
w.setLayout(l)
self.combobox = QComboBox(parent)
l.addWidget(self.combobox)
t = _('Yes')
c = QPushButton(t, parent)
width = c.fontMetrics().boundingRect(t).width() + 7
c.setMaximumWidth(width)
l.addWidget(c)
c.clicked.connect(self.set_to_yes)
t = _('No')
c = QPushButton(t, parent)
width = c.fontMetrics().boundingRect(t).width() + 7
c.setMaximumWidth(width)
l.addWidget(c)
c.clicked.connect(self.set_to_no)
t = _('Clear')
c = QPushButton(t, parent)
width = c.fontMetrics().boundingRect(t).width() + 7
c.setMaximumWidth(width)
l.addWidget(c)
c.clicked.connect(self.set_to_cleared)
c = QLabel('', parent)
c.setMaximumWidth(1)
l.addWidget(c, 1)
w = self.combobox
items = [_('Yes'), _('No'), _('Undefined')]
icons = [I('ok.png'), I('list_remove.png'), I('blank.png')]
if not self.db.prefs.get('bools_are_tristate'):
items = items[:-1]
icons = icons[:-1]
for icon, text in zip(icons, items):
w.addItem(QIcon(icon), text)
def setter(self, val):
val = {None: 2, False: 1, True: 0}[val]
if not self.db.prefs.get('bools_are_tristate') and val == 2:
val = 1
self.combobox.setCurrentIndex(val)
def getter(self):
val = self.combobox.currentIndex()
return {2: None, 1: False, 0: True}[val]
def set_to_yes(self):
self.combobox.setCurrentIndex(0)
def set_to_no(self):
self.combobox.setCurrentIndex(1)
def set_to_cleared(self):
self.combobox.setCurrentIndex(2)
class ConfigWidget(QWidget):
# GUI definition
def __init__(self):
QWidget.__init__(self)
self.l = QVBoxLayout()
self.setLayout(self.l)
self.ll = QHBoxLayout()
self.l.addLayout(self.ll)
self.label_exe = QLabel(_('&Prince executable:'))
self.ll.addWidget(self.label_exe)
self.exe = QLineEdit(self)
self.exe.setText(prefs['prince_exe'])
self.exe.setToolTip(
_('<qt>Executable for the Prince program (command-line interface)</qt>'
))
self.ll.addWidget(self.exe)
self.label_exe.setBuddy(self.exe)
self.browse = QPushButton(_('&Browse') + '...', self)
self.browse.setToolTip(
_('<qt>Search the Prince executable in your computer</qt>'))
self.browse.clicked.connect(self.select_exe)
self.ll.addWidget(self.browse)
self.lll = QHBoxLayout()
self.l.addLayout(self.lll)
self.label_fmts = QLabel(_('Preferred &formats:'))
self.lll.addWidget(self.label_fmts)
self.fmts = QLineEdit(self)
self.fmts.setText(','.join(prefs['formats']))
self.fmts.setToolTip(
_('<qt>Comma-separated list of preferred formats to use as source, the first that matches will be used</qt>'
))
self.lll.addWidget(self.fmts)
self.label_fmts.setBuddy(self.fmts)
self.add_book = QCheckBox(_('&Add PDF to the book record'))
self.add_book.setToolTip(
_('<qt>Add the converted PDF to the selected book record</qt>'))
self.add_book.setChecked(prefs['add_book'])
self.l.addWidget(self.add_book)
self.show_css = QCheckBox(_('&Show CSS in the Convert dialog'))
self.show_css.setToolTip(
_('<qt>Show by default the styles in the Convert dialog</qt>'))
self.show_css.setChecked(prefs['show_CSS'])
self.l.addWidget(self.show_css)
self.css_layout = QVBoxLayout()
self.llll = QHBoxLayout()
self.css_layout.addLayout(self.llll)
self.css_list = QComboBox()
self.css_list.setToolTip(
_('<qt>List of custom styles defined. Select one to edit</qt>'))
self.css_list.setSizeAdjustPolicy(QComboBox.AdjustToContents)
self.CSS_list = prefs['custom_CSS_list'].copy()
self.default_CSS = prefs['default_CSS']
if 'custom_CSS' in prefs:
self.CSS_list[_('old')] = prefs['custom_CSS']
self.default_CSS = _('old')
if self.default_CSS not in self.CSS_list:
self.default_CSS = sorted(self.CSS_list,
key=lambda x: x.lower())[0]
for key in sorted(self.CSS_list, key=lambda x: x.lower()):
self.css_list.addItem(key, key)
self.css_list.setCurrentIndex(self.css_list.findText(self.default_CSS))
self.css_list.currentIndexChanged.connect(self.set_css)
self.llll.addWidget(self.css_list)
self.css_rename = QPushButton(_('Re&name'))
self.css_rename.setToolTip(
_('<qt>Rename the current style to the name on the right</qt>'))
self.css_rename.clicked.connect(self.rename_css)
self.css_rename.setEnabled(False)
self.llll.addWidget(self.css_rename)
self.css_name = QLineEdit(self)
self.css_name.setToolTip(
_('<qt>Name for the new or renamed style</qt>'))
self.css_name.setText(self.css_list.currentText())
self.css_name.textChanged.connect(self.check_names)
self.llll.addWidget(self.css_name)
self.css_add = QPushButton(_('A&dd'))
self.css_add.setToolTip(
_('<qt>Add a new empty style with the name on the left</qt>'))
self.css_add.clicked.connect(self.add_css)
self.css_add.setEnabled(False)
self.llll.addWidget(self.css_add)
self.css_remove = QPushButton(_('Re&move'))
self.css_remove.setToolTip(_('<qt>Remove the current style</qt>'))
self.css_remove.clicked.connect(self.remove_css)
self.llll.addWidget(self.css_remove)
self.llll_ = QHBoxLayout()
self.css_layout.addLayout(self.llll_)
self.label_args = QLabel(_('Addi&tional command-line arguments:'))
self.llll_.addWidget(self.label_args)
self.args = QLineEdit(self)
# Make sure custom_CSS_list and custom_args_list have the same keys
if 'custom_args_list' in prefs:
self.args_list = prefs['custom_args_list'].copy()
else:
self.args_list = {}
for key in self.CSS_list:
if not key in self.args_list:
self.args_list[key] = ''
for key in self.args_list:
if not key in self.CSS_list:
del self.args_list[key]
self.args.setText(self.args_list[unicode(self.css_list.currentText())])
self.args.setToolTip(
_('<qt>Additional command-line arguments used in conversions with this style</qt>'
))
self.llll_.addWidget(self.args)
self.label_args.setBuddy(self.args)
self.css = TextEditWithTooltip(self, expected_geometry=(80, 20))
self.css.setLineWrapMode(TextEditWithTooltip.NoWrap)
self.css.load_text(self.CSS_list[unicode(self.css_list.currentText())],
'css')
self.css.setToolTip(
_('<qt>Custom stylesheet that will be applied, if selected, to all Prince PDF conversions</qt>'
))
self.css_layout.addWidget(self.css)
self.css_templates = QLabel(_('Book metadata can be used in the stylesheet. Anything between %(s1)s and %(s2)s will be processed as a calibre template. For instance, %(s3)s in the stylesheet will be replaced with the book title in the conversion.') % \
{'s1':'<span style="font-family:monospace ; font-weight:bold">@{@</span>', \
's2':'<span style="font-family:monospace ; font-weight:bold">@}@</span>', \
's3':'<span style="font-family:monospace ; font-weight:bold">@{@{title}@}@</span>'})
self.css_templates.setWordWrap(True)
self.css_layout.addWidget(self.css_templates)
self.css_box = QGroupBox(_('&Custom styles:'))
self.css_box.setLayout(self.css_layout)
self.l.addWidget(self.css_box)
self.lllll = QHBoxLayout()
self.lllll.setAlignment(Qt.AlignLeft)
self.l.addLayout(self.lllll)
self.defaults = QPushButton(_('&Restore defaults'))
self.defaults.setToolTip(_('<qt>Restore the default settings</qt>'))
self.defaults.clicked.connect(self.restore_defaults)
self.lllll.addWidget(self.defaults, alignment=Qt.AlignLeft)
self.warning = QLabel(_('<b>Warning</b>: Deletes modified styles'))
self.lllll.addWidget(self.warning)
self.adjustSize()
def select_exe(self):
'''
Create a dialog to select the Prince executable
'''
dialog = QFileDialog()
dialog.setFileMode(QFileDialog.ExistingFile)
filename = dialog.getOpenFileName(self, _('Select Prince executable'),
'', '')
if filename:
try:
self.exe.setText(filename)
except (TypeError):
self.exe.setText(filename[0])
def restore_defaults(self):
'''
Restore the default settings
'''
self.exe.setText(prefs.defaults['prince_exe'])
self.fmts.setText(','.join(prefs.defaults['formats']).lower())
self.show_css.setChecked(prefs.defaults['show_CSS'])
self.add_book.setChecked(prefs.defaults['add_book'])
self.css_list.currentIndexChanged.disconnect()
self.css_list.clear()
self.CSS_list = prefs.defaults['custom_CSS_list'].copy()
self.args_list = prefs.defaults['custom_args_list'].copy()
self.default_CSS = prefs.defaults['default_CSS']
for key in sorted(self.CSS_list, key=lambda x: x.lower()):
self.css_list.addItem(key, key)
self.css_list.setCurrentIndex(self.css_list.findText(self.default_CSS))
self.css_name.setText(self.default_CSS)
self.css.load_text(self.CSS_list[unicode(self.css_list.currentText())],
'css')
self.args.setText(self.args_list[unicode(self.css_list.currentText())])
self.css_list.currentIndexChanged.connect(self.set_css)
def save_settings(self):
'''
Save the current settings
'''
prefs['prince_exe'] = unicode(self.exe.text())
prefs['formats'] = unicode(self.fmts.text().lower()).split(',')
prefs['show_CSS'] = self.show_css.isChecked()
prefs['add_book'] = self.add_book.isChecked()
self.set_css()
prefs['default_CSS'] = self.default_CSS
prefs['custom_CSS_list'] = self.CSS_list.copy()
prefs['custom_args_list'] = self.args_list.copy()
if 'custom_CSS' in prefs:
del prefs['custom_CSS']
def set_css(self):
'''
Fill the CSS text box with the selected stylesheet, and similarly for command-line arguments
'''
self.CSS_list[self.default_CSS] = unicode(self.css.toPlainText())
self.args_list[self.default_CSS] = unicode(self.args.text())
self.default_CSS = unicode(self.css_list.currentText())
self.css.load_text(self.CSS_list[self.default_CSS], 'css')
self.args.setText(self.args_list[self.default_CSS])
self.css_name.setText(self.css_list.currentText())
def add_css(self):
'''
Add a new style
'''
from calibre.gui2 import error_dialog
name = unicode(self.css_name.text())
if name in self.CSS_list:
error_dialog(
self,
_('Cannot add style'),
_('A style with the name "%s" is already defined, use a different name.'
) % name,
show=True)
else:
self.CSS_list[name] = ''
self.args_list[name] = ''
self.css_list.addItem(name, name)
self.css_list.setCurrentIndex(self.css_list.findText(name))
self.css_add.setEnabled(False)
self.css_rename.setEnabled(False)
def remove_css(self):
'''
Remove an existing style
'''
from calibre.gui2 import error_dialog
if (self.css_list.count() > 1):
self.css_list.currentIndexChanged.disconnect()
self.css_list.removeItem(self.css_list.currentIndex())
del self.CSS_list[self.default_CSS]
del self.args_list[self.default_CSS]
self.default_CSS = unicode(self.css_list.currentText())
self.css.load_text(self.CSS_list[self.default_CSS], 'css')
self.args.setText(self.args_list[self.default_CSS])
self.css_list.currentIndexChanged.connect(self.set_css)
self.css_name.setText(self.css_list.currentText())
else:
error_dialog(
self,
_('Cannot delete the last style'),
_('The last style cannot be removed. You can rename it and/or remove its contents.'
),
show=True)
def rename_css(self):
'''
Rename a style
'''
from calibre.gui2 import error_dialog
name = unicode(self.css_name.text())
if name in self.CSS_list:
error_dialog(
self,
_('Cannot rename style'),
_('A style with the name "%s" is already defined, use a different name.'
) % name,
show=True)
else:
self.CSS_list[name] = self.CSS_list.pop(self.default_CSS)
self.args_list[name] = self.args_list.pop(self.default_CSS)
self.css_list.setItemText(self.css_list.currentIndex(), name)
self.default_CSS = name
def check_names(self, text):
name = unicode(text)
if name in self.CSS_list:
self.css_add.setEnabled(False)
self.css_rename.setEnabled(False)
else:
self.css_add.setEnabled(True)
self.css_rename.setEnabled(True)