Source code for camelot.view.controls.editors.datetimeeditor

#  ============================================================================
#
#  Copyright (C) 2007-2011 Conceptive Engineering bvba. All rights reserved.
#  www.conceptive.be / project-camelot@conceptive.be
#
#  This file is part of the Camelot Library.
#
#  This file may be used under the terms of the GNU General Public
#  License version 2.0 as published by the Free Software Foundation
#  and appearing in the file license.txt included in the packaging of
#  this file.  Please review this information to ensure GNU
#  General Public Licensing requirements will be met.
#
#  If you are unsure which license is appropriate for your use, please
#  visit www.python-camelot.com or contact project-camelot@conceptive.be
#
#  This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
#  WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
#
#  For use of this library in commercial applications, please contact
#  project-camelot@conceptive.be
#
#  ============================================================================

import datetime

from PyQt4 import QtGui, QtCore
from PyQt4.QtCore import Qt

from customeditor import CustomEditor, set_background_color_palette
from dateeditor import DateEditor
from camelot.view.proxy import ValueLoading

[docs]class TimeValidator(QtGui.QValidator): def __init__(self, parent, nullable): QtGui.QValidator.__init__(self, parent) self._nullable = nullable
[docs] def validate(self, input, pos): parts = str(input).split(':') if len(parts)!=2: return (QtGui.QValidator.Invalid, pos) if str(input)=='--:--' and self._nullable: return (QtGui.QValidator.Acceptable, pos) for part in parts: if not part.isdigit(): return (QtGui.QValidator.Invalid, pos) if len(part) not in (1,2): return (QtGui.QValidator.Intermediate, pos) if not int(parts[0]) in range(0,24): return (QtGui.QValidator.Invalid, pos) if not int(parts[1]) in range(0,60): return (QtGui.QValidator.Invalid, pos) return (QtGui.QValidator.Acceptable, pos)
[docs]class DateTimeEditor(CustomEditor): """Widget for editing date and time separated and with popups""" def __init__(self, parent, editable=True, nullable=True, field_name = 'datetime', **kwargs): CustomEditor.__init__(self, parent) self.setObjectName( field_name ) import itertools self.nullable = nullable layout = QtGui.QHBoxLayout() self.dateedit = DateEditor(self, editable=editable, nullable=nullable, **kwargs) self.dateedit.editingFinished.connect( self.editing_finished ) layout.addWidget(self.dateedit, 1) self.timeedit = QtGui.QComboBox(self) self.timeedit.setEditable(True) if not editable: self.timeedit.setEnabled(False) time_entries = [entry for entry in itertools.chain(*(('%02i:00'%i, '%02i:30'%i) for i in range(0,24)))] self.timeedit.addItems(time_entries) self.timeedit.setValidator(TimeValidator(self, nullable)) self.timeedit.currentIndexChanged.connect( self.editing_finished ) self.timeedit.editTextChanged.connect( self.editing_finished ) self.timeedit.setFocusPolicy( Qt.StrongFocus ) layout.addWidget(self.timeedit, 1) # focus proxy is needed to activate the editor with a single click self.setFocusProxy(self.dateedit) self.setLayout(layout) layout.setContentsMargins( 0, 0, 0, 0) layout.setSpacing(0) @QtCore.pyqtSlot(QtCore.QString) @QtCore.pyqtSlot(int) @QtCore.pyqtSlot()
[docs] def editing_finished(self, _arg=None): if self.time() and self.date(): self.editingFinished.emit()
[docs] def get_value(self): time_value = self.time() date_value = self.date() if time_value not in (None, ValueLoading) and date_value not in (None, ValueLoading): value = datetime.datetime(hour=time_value.hour(), minute=time_value.minute(), second=time_value.second(), year=date_value.year, month=date_value.month, day=date_value.day) else: value = None return CustomEditor.get_value(self) or value
[docs] def set_value(self, value): value = CustomEditor.set_value(self, value) if value: self.dateedit.set_value(value.date()) self.timeedit.lineEdit().setText('%02i:%02i'%(value.hour, value.minute)) else: self.dateedit.set_value(None) self.timeedit.lineEdit().setText('--:--')
[docs] def date(self): return self.dateedit.get_value()
[docs] def time(self): text = str(self.timeedit.currentText()) if text=='--:--': return None parts = text.split(':') return QtCore.QTime(int(parts[0]), int(parts[1]))
[docs] def set_enabled(self, editable=True): self.timeedit.setEnabled(editable) self.dateedit.setEnabled(editable)
[docs] def set_field_attributes(self, editable = True, background_color = None, tooltip = None, **kwargs): self.set_enabled(editable) self.set_background_color(background_color) self.dateedit.line_edit.setToolTip(unicode(tooltip or ''))
[docs] def set_background_color(self, background_color): self.dateedit.set_background_color( background_color ) set_background_color_palette( self.timeedit.lineEdit(), background_color )

Comments
blog comments powered by Disqus