class TestDatabase(unittest.TestCase):
def setUp(self):
'''
Stardard init method: runs before each test_* method
Initializes a PluginManager
'''
self.profile_path = tempfile.mkdtemp()
profile = Profile(self.profile_path, 'testing')
dirname = os.path.dirname(__file__)
self._csvfile = os.path.join(dirname, 'sample_talks.csv')
db_file = os.path.join(self.profile_path, 'presentations.db')
self.db = QtDBConnector(db_file, PluginManager(profile))
def tearDown(self):
'''
Generic unittest.TestCase.tearDown()
'''
shutil.rmtree(self.profile_path)
def test_get_talks(self):
"""Simply test that a query is returned"""
self.assertIsInstance(self.db.get_talks(), QtSql.QSqlQuery)
def test_get_events(self):
"""Simply test that a query is returned"""
self.assertIsInstance(self.db.get_events(), QtSql.QSqlQuery)
def test_get_talk_ids(self):
"""Simply test that a query is returned"""
self.assertIsInstance(self.db.get_talk_ids(), QtSql.QSqlQuery)
def test_get_talks_by_event(self):
"""Simply test that a query is returned"""
self.assertIsInstance(self.db.get_talks_by_event("SC2011"), QtSql.QSqlQuery)
def test_get_talks_by_room(self):
"""Simply test that a query is returned"""
self.assertIsInstance(self.db.get_talks_by_room("T105"), QtSql.QSqlQuery)
def test_get_presentation(self):
"""Simply test that a presentation is returned"""
self.assertIsInstance(self.db.get_presentation(1), Presentation)
def test_get_presentations_model(self):
"""Simply test that a model is returned"""
self.assertIsInstance(self.db.get_presentations_model(), QtSql.QSqlTableModel)
def test_get_events_model(self):
"""Simply test that a model is returned"""
self.assertIsInstance(self.db.get_events_model(), QtSql.QSqlQueryModel)
def test_get_rooms_model(self):
"""Simply test that a model is returned"""
self.assertIsInstance(self.db.get_rooms_model("SC2011"), QtSql.QSqlQueryModel)
def test_get_talks_model(self):
"""Simply test that a model is returned"""
self.assertIsInstance(self.db.get_talks_model("SC2011", "T105"), QtSql.QSqlQueryModel)
def test_add_talks_from_rss(self):
"""Test that talks are retrieved from the RSS feed"""
feed1 = "http://fosslc.org/drupal/presentations_rss/summercamp2010"
feed2 = "http://fosslc.org/drupal/presentations_rss/sc2011"
presentation1 = Presentation("Managing map data in a database", "Andrew Ross")
presentation2 = Presentation("Building NetBSD", "David Maxwell")
self.db.add_talks_from_rss(feed1)
self.assertTrue(self.db.presentation_exists(presentation1))
self.db.add_talks_from_rss(feed2)
self.assertTrue(self.db.presentation_exists(presentation2))
def test_add_talks_from_csv(self):
"""Test that talks are retrieved from the CSV file"""
fname = self._csvfile
presentation = Presentation("Building NetBSD", "David Maxwell")
self.db.add_talks_from_csv(fname)
self.assertTrue(self.db.presentation_exists(presentation))
class TalkEditorApp(FreeseerApp):
"""
Freeseer talk database editor main gui class
"""
def __init__(self, core=None):
FreeseerApp.__init__(self)
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8(":/freeseer/logo.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.setWindowIcon(icon)
self.resize(960, 400)
self.mainWidget = QtGui.QWidget()
self.mainLayout = QtGui.QHBoxLayout()
self.mainWidget.setLayout(self.mainLayout)
self.setCentralWidget(self.mainWidget)
self.editorWidget = EditorWidget()
self.editorWidget.editor.setColumnHidden(5, True)
self.addTalkWidget = AddTalkWidget()
self.mainLayout.addWidget(self.editorWidget)
self.mainLayout.addWidget(self.addTalkWidget)
# Initialize geometry, to be used for restoring window positioning.
self.geometry = None
self.config = Config(settings.configdir)
self.db = QtDBConnector(settings.configdir)
#
# Setup Menubar
#
self.actionExportCsv = QtGui.QAction(self)
self.actionExportCsv.setObjectName(_fromUtf8("actionExportCsv"))
# Actions
self.menuFile.insertAction(self.actionExit, self.actionExportCsv)
# --- End Menubar
#
# Talk Editor Connections
#
# Add Talk Widget
self.connect(self.addTalkWidget.addButton, QtCore.SIGNAL("clicked()"), self.add_talk)
self.connect(self.addTalkWidget.cancelButton, QtCore.SIGNAL("clicked()"), self.hide_add_talk_widget)
self.addTalkWidget.setHidden(True)
# Editor Widget
self.connect(
self.editorWidget.rssLineEdit, QtCore.SIGNAL("returnPressed()"), self.editorWidget.rssPushButton.click
)
self.connect(self.editorWidget.rssPushButton, QtCore.SIGNAL("clicked()"), self.add_talks_from_rss)
self.connect(self.editorWidget.addButton, QtCore.SIGNAL("clicked()"), self.show_add_talk_widget)
self.connect(self.editorWidget.removeButton, QtCore.SIGNAL("clicked()"), self.remove_talk)
self.connect(self.editorWidget.clearButton, QtCore.SIGNAL("clicked()"), self.confirm_reset)
self.connect(self.editorWidget.closeButton, QtCore.SIGNAL("clicked()"), self.close)
# CSV Widget
self.connect(self.editorWidget.csvFileSelectButton, QtCore.SIGNAL("clicked()"), self.csv_file_select)
self.connect(self.editorWidget.csvPushButton, QtCore.SIGNAL("clicked()"), self.add_talks_from_csv)
self.connect(self.actionExportCsv, QtCore.SIGNAL("triggered()"), self.export_talks_to_csv)
# Load default language
actions = self.menuLanguage.actions()
for action in actions:
if action.data().toString() == self.config.default_language:
action.setChecked(True)
self.translate(action)
break
self.load_presentations_model()
###
### Translation
###
def retranslate(self):
self.setWindowTitle(self.uiTranslator.translate("TalkEditorApp", "Freeseer Talk Editor"))
#
# Reusable Strings
#
self.confirmDBClearTitleString = self.uiTranslator.translate("TalkEditorApp", "Clear Database")
self.confirmDBClearQuestionString = self.uiTranslator.translate(
"TalkEditorApp", "Are you sure you want to clear the DB?"
)
# --- End Reusable Strings
#
# Menubar
#
self.actionExportCsv.setText(self.uiTranslator.translate("TalkEditorApp", "&Export to CSV"))
# --- End Menubar
#
# AddTalkWidget
#
self.addTalkWidget.addTalkGroupBox.setTitle(self.uiTranslator.translate("TalkEditorApp", "Add Talk"))
self.addTalkWidget.titleLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Title"))
self.addTalkWidget.presenterLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Presenter"))
self.addTalkWidget.eventLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Event"))
self.addTalkWidget.roomLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Room"))
self.addTalkWidget.dateLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Date"))
self.addTalkWidget.timeLabel.setText(self.uiTranslator.translate("TalkEditorApp", "Time"))
self.addTalkWidget.addButton.setText(self.uiTranslator.translate("TalkEditorApp", "Add"))
self.addTalkWidget.cancelButton.setText(self.uiTranslator.translate("TalkEditorApp", "Cancel"))
# --- End AddTalkWidget
#
# EditorWidget
#
self.editorWidget.rssLabel.setText(self.uiTranslator.translate("TalkEditorApp", "URL"))
self.editorWidget.rssPushButton.setText(self.uiTranslator.translate("TalkEditorApp", "Load talks from RSS"))
self.editorWidget.csvLabel.setText(self.uiTranslator.translate("TalkEditorApp", "File"))
self.editorWidget.csvPushButton.setText(self.uiTranslator.translate("TalkEditorApp", "Load talks from CSV"))
self.editorWidget.addButton.setText(self.uiTranslator.translate("TalkEditorApp", "Add"))
self.editorWidget.removeButton.setText(self.uiTranslator.translate("TalkEditorApp", "Remove"))
self.editorWidget.clearButton.setText(self.uiTranslator.translate("TalkEditorApp", "Clear"))
self.editorWidget.closeButton.setText(self.uiTranslator.translate("TalkEditorApp", "Close"))
# --- End EditorWidget
def load_presentations_model(self):
# Load Presentation Model
self.presentationModel = self.db.get_presentations_model()
self.editorWidget.editor.setModel(self.presentationModel)
def show_add_talk_widget(self):
self.editorWidget.setHidden(True)
self.addTalkWidget.setHidden(False)
def hide_add_talk_widget(self):
self.editorWidget.setHidden(False)
self.addTalkWidget.setHidden(True)
def add_talk(self):
date = self.addTalkWidget.dateEdit.date()
time = self.addTalkWidget.timeEdit.time()
datetime = QtCore.QDateTime(date, time)
presentation = Presentation(
unicode(self.addTalkWidget.titleLineEdit.text()),
unicode(self.addTalkWidget.presenterLineEdit.text()),
"", # description
"", # level
unicode(self.addTalkWidget.eventLineEdit.text()),
unicode(self.addTalkWidget.roomLineEdit.text()),
unicode(datetime.toString()),
)
# Do not add talks if they are empty strings
if len(presentation.title) == 0:
return
self.db.insert_presentation(presentation)
# cleanup
self.addTalkWidget.titleLineEdit.clear()
self.addTalkWidget.presenterLineEdit.clear()
self.presentationModel.select()
self.hide_add_talk_widget()
def remove_talk(self):
try:
row_clicked = self.editorWidget.editor.currentIndex().row()
except:
return
self.presentationModel.removeRow(row_clicked)
self.presentationModel.select()
def reset(self):
self.db.clear_database()
self.presentationModel.select()
def confirm_reset(self):
"""
Presents a confirmation dialog to ask the user if they are sure they
wish to remove the talk database.
If Yes call the reset() function.
"""
confirm = QtGui.QMessageBox.question(
self,
self.confirmDBClearTitleString,
self.confirmDBClearQuestionString,
QtGui.QMessageBox.Yes | QtGui.QMessageBox.No,
QtGui.QMessageBox.No,
)
if confirm == QtGui.QMessageBox.Yes:
self.reset()
def add_talks_from_rss(self):
rss_url = unicode(self.editorWidget.rssLineEdit.text())
self.db.add_talks_from_rss(rss_url)
self.presentationModel.select()
def closeEvent(self, event):
logging.info("Exiting talk database editor...")
self.geometry = self.saveGeometry()
event.accept()
def csv_file_select(self):
dirpath = str(self.editorWidget.csvLineEdit.text())
fname = QtGui.QFileDialog.getOpenFileName(self, "Select file", "", "*.csv")
if fname:
self.editorWidget.csvLineEdit.setText(fname)
def add_talks_from_csv(self):
fname = self.editorWidget.csvLineEdit.text()
if fname:
self.db.add_talks_from_csv(fname)
self.presentationModel.select()
def export_talks_to_csv(self):
dirpath = str(self.editorWidget.csvLineEdit.text())
fname = QtGui.QFileDialog.getSaveFileName(self, "Select file", "", "*.csv")
if fname:
self.db.export_talks_to_csv(fname)
class TalkEditorApp(FreeseerApp):
'''
Freeseer talk database editor main gui class
'''
def __init__(self, recordapp=None):
FreeseerApp.__init__(self)
self.recordapp = recordapp
icon = QIcon()
icon.addPixmap(QPixmap(_fromUtf8(":/freeseer/logo.png")), QIcon.Normal,
QIcon.Off)
self.setWindowIcon(icon)
self.resize(960, 400)
self.mainWidget = QWidget()
self.mainLayout = QHBoxLayout()
self.mainWidget.setLayout(self.mainLayout)
self.setCentralWidget(self.mainWidget)
self.editorWidget = EditorWidget()
self.editorWidget.editor.setColumnHidden(5, True)
self.addTalkWidget = AddTalkWidget()
self.mainLayout.addWidget(self.editorWidget)
self.mainLayout.addWidget(self.addTalkWidget)
# Initialize geometry, to be used for restoring window positioning.
self.geometry = None
self.config = Config(settings.configdir)
self.db = QtDBConnector(settings.configdir)
#
# Setup Menubar
#
self.actionExportCsv = QAction(self)
self.actionExportCsv.setObjectName(_fromUtf8("actionExportCsv"))
# Actions
self.menuFile.insertAction(self.actionExit, self.actionExportCsv)
# --- End Menubar
#
# Talk Editor Connections
#
# Add Talk Widget
self.connect(self.addTalkWidget.addButton, SIGNAL('clicked()'),
self.add_talk)
self.connect(self.addTalkWidget.cancelButton, SIGNAL('clicked()'),
self.hide_add_talk_widget)
self.addTalkWidget.setHidden(True)
# Editor Widget
self.connect(self.editorWidget.rssLineEdit, SIGNAL('returnPressed()'),
self.editorWidget.rssPushButton.click)
self.connect(self.editorWidget.rssPushButton, SIGNAL('clicked()'),
self.add_talks_from_rss)
self.connect(self.editorWidget.addButton, SIGNAL('clicked()'),
self.show_add_talk_widget)
self.connect(self.editorWidget.removeButton, SIGNAL('clicked()'),
self.remove_talk)
self.connect(self.editorWidget.clearButton, SIGNAL('clicked()'),
self.confirm_reset)
self.connect(self.editorWidget.closeButton, SIGNAL('clicked()'),
self.close)
# CSV Widget
self.connect(self.editorWidget.csvLineEdit, SIGNAL('returnPressed()'),
self.editorWidget.csvPushButton.click)
self.connect(self.editorWidget.csvFileSelectButton,
SIGNAL('clicked()'), self.csv_file_select)
self.connect(self.editorWidget.csvPushButton, SIGNAL('clicked()'),
self.add_talks_from_csv)
self.connect(self.actionExportCsv, SIGNAL('triggered()'),
self.export_talks_to_csv)
# Load default language
actions = self.menuLanguage.actions()
for action in actions:
if action.data().toString() == self.config.default_language:
action.setChecked(True)
self.translate(action)
break
self.load_presentations_model()
###
### Translation
###
def retranslate(self):
self.setWindowTitle(
self.app.translate("TalkEditorApp", "Freeseer Talk Editor"))
#
# Reusable Strings
#
self.confirmDBClearTitleString = self.app.translate(
"TalkEditorApp", "Clear Database")
self.confirmDBClearQuestionString = self.app.translate(
"TalkEditorApp", "Are you sure you want to clear the DB?")
# --- End Reusable Strings
#
# Menubar
#
self.actionExportCsv.setText(
self.app.translate("TalkEditorApp", "&Export to CSV"))
# --- End Menubar
#
# AddTalkWidget
#
self.addTalkWidget.addTalkGroupBox.setTitle(
self.app.translate("TalkEditorApp", "Add Talk"))
self.addTalkWidget.titleLabel.setText(
self.app.translate("TalkEditorApp", "Title"))
self.addTalkWidget.presenterLabel.setText(
self.app.translate("TalkEditorApp", "Presenter"))
self.addTalkWidget.eventLabel.setText(
self.app.translate("TalkEditorApp", "Event"))
self.addTalkWidget.roomLabel.setText(
self.app.translate("TalkEditorApp", "Room"))
self.addTalkWidget.dateLabel.setText(
self.app.translate("TalkEditorApp", "Date"))
self.addTalkWidget.timeLabel.setText(
self.app.translate("TalkEditorApp", "Time"))
self.addTalkWidget.addButton.setText(
self.app.translate("TalkEditorApp", "Add"))
self.addTalkWidget.cancelButton.setText(
self.app.translate("TalkEditorApp", "Cancel"))
# --- End AddTalkWidget
#
# EditorWidget
#
self.editorWidget.rssLabel.setText(
self.app.translate("TalkEditorApp", "URL"))
self.editorWidget.rssPushButton.setText(
self.app.translate("TalkEditorApp", "Load talks from RSS"))
self.editorWidget.csvLabel.setText(
self.app.translate("TalkEditorApp", "File"))
self.editorWidget.csvPushButton.setText(
self.app.translate("TalkEditorApp", "Load talks from CSV"))
self.editorWidget.addButton.setText(
self.app.translate("TalkEditorApp", "Add"))
self.editorWidget.removeButton.setText(
self.app.translate("TalkEditorApp", "Remove"))
self.editorWidget.clearButton.setText(
self.app.translate("TalkEditorApp", "Clear"))
self.editorWidget.closeButton.setText(
self.app.translate("TalkEditorApp", "Close"))
# --- End EditorWidget
def load_presentations_model(self):
# Load Presentation Model
self.presentationModel = self.db.get_presentations_model()
self.editorWidget.editor.setModel(self.presentationModel)
def show_add_talk_widget(self):
self.editorWidget.setHidden(True)
self.addTalkWidget.setHidden(False)
def hide_add_talk_widget(self):
self.editorWidget.setHidden(False)
self.addTalkWidget.setHidden(True)
def add_talk(self):
date = self.addTalkWidget.dateEdit.date()
time = self.addTalkWidget.timeEdit.time()
datetime = QDateTime(date, time)
presentation = Presentation(
unicode(self.addTalkWidget.titleLineEdit.text()),
unicode(self.addTalkWidget.presenterLineEdit.text()),
"", # description
"", # level
unicode(self.addTalkWidget.eventLineEdit.text()),
unicode(self.addTalkWidget.roomLineEdit.text()),
unicode(datetime.toString(Qt.ISODate)))
# Do not add talks if they are empty strings
if (len(presentation.title) == 0):
return
self.db.insert_presentation(presentation)
# cleanup
self.addTalkWidget.titleLineEdit.clear()
self.addTalkWidget.presenterLineEdit.clear()
self.presentationModel.select()
self.hide_add_talk_widget()
# If this is launched from the recording app
# refresh the talk list
if self.recordapp:
self.recordapp.load_event_list()
def remove_talk(self):
try:
row_clicked = self.editorWidget.editor.currentIndex().row()
except:
return
self.presentationModel.removeRow(row_clicked)
self.presentationModel.select()
def reset(self):
self.db.clear_database()
self.presentationModel.select()
def confirm_reset(self):
"""
Presents a confirmation dialog to ask the user if they are sure they
wish to remove the talk database.
If Yes call the reset() function.
"""
confirm = QMessageBox.question(self, self.confirmDBClearTitleString,
self.confirmDBClearQuestionString,
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if confirm == QMessageBox.Yes:
self.reset()
def add_talks_from_rss(self):
rss_url = unicode(self.editorWidget.rssLineEdit.text())
self.db.add_talks_from_rss(rss_url)
self.presentationModel.select()
def closeEvent(self, event):
log.info('Exiting talk database editor...')
self.geometry = self.saveGeometry()
event.accept()
def csv_file_select(self):
dirpath = str(self.editorWidget.csvLineEdit.text())
fname = QFileDialog.getOpenFileName(self, 'Select file', "", "*.csv")
if fname:
self.editorWidget.csvLineEdit.setText(fname)
def add_talks_from_csv(self):
fname = self.editorWidget.csvLineEdit.text()
if fname:
self.db.add_talks_from_csv(fname)
self.presentationModel.select()
def export_talks_to_csv(self):
dirpath = str(self.editorWidget.csvLineEdit.text())
fname = QFileDialog.getSaveFileName(self, 'Select file', "", "*.csv")
if fname:
self.db.export_talks_to_csv(fname)
