Minor changes required for running tests

[ipypbx] / src / ipypbx / main.py

diff --git a/src/ipypbx/main.py b/src/ipypbx/main.py
index d1d7bb5..f0422d6 100644 (file)
--- a/src/ipypbx/main.py
+++ b/src/ipypbx/main.py
@@ -17,12 +17,11 @@
 
 import os
 import sys
-from ipypbx import controllers, sql, ui
+from ipypbx import controllers, ui
 from PyQt4 import QtCore, QtGui, QtSql
 
 
 # Working directory path.
-# NOTE: ~/.ipypbx gives an error - for some reason QT doesn't like the dot?
 PREFIX = os.path.expanduser('~/.ipypbx')
 
 # Database file name.
@@ -48,21 +47,68 @@ def setupDb(prefix=PREFIX, dbname=DB_NAME):
     
     if db.open():
         if created:
-            for query in sql.creation_queries:
-                QtSql.QSqlQuery().exec_(query)
+            # Load script from local file.
+            sql_script = open(os.path.join(os.path.dirname(__file__), 'create.sql')).read()
+
+            # Split into individual queries.
+            sql_queries = sql_script.split(';')
+
+            # Execute all queries except last that is empty.
+            query = QtSql.QSqlQuery()
+            for query_string in sql_queries[:-1]:
+                query.exec_(query_string)
+
+        return db, created
     else:
+        # Something went horribly wrong.
         QtGui.QMessageBox.warning(
             None, "Fatal Error", "Database Error: %s" % db.lastError().text())
         sys.exit(1)
 
-    
+
 if __name__ == '__main__':
     app = QtGui.QApplication(sys.argv)
-    setupDb()
-    views = ui.MainWindow()
-    views.show()
 
-    connectionController = controllers.ConnectionController(parent=views)
+    # Localize UI.
+    locale = QtCore.QLocale.system().name()
+    translator = QtCore.QTranslator()
+    
+    if translator.load("ipypbx_%s" % locale.toLower(), "ipypbx/locale"):
+        QtGui.QApplication.installTranslator(translator)
+
+    # Initialize main window.
+    db, created = setupDb()
+    main = QtGui.QMainWindow()
+    views = ui.Ui_MainWindow()
+    views.setupUi(main)
+    main.show()
+
+    # Setup controllers.
+    main.controllers = {}
+    for basename in ('Connection', 'SipProfile', 'Domain', 'Gateway',
+                     'Endpoint', 'Extension'):
+        main.controllers[basename.lower()] = getattr(
+            controllers, basename + 'Controller'
+            )(parent=main, views=views)
+
+    # Select first row.
+    connection_model = main.controllers['connection'].model
+    connection_index = connection_model.index(0, 0)
+
+    # Get first row index.
+    first_row_id, ok = connection_model.data(
+        connection_index.sibling(0, 0)).toInt()
+    
+    if ok:
+        # We have to explicitly emit row selection signal since previous state
+        # was unselected.
+        QtCore.QObject.emit(
+            main.controllers['connection'].view_list.selectionModel(),
+            QtCore.SIGNAL('currentRowChanged(QModelIndex, QModelIndex)'),
+            connection_index, connection_index)
+
+    # TODO: not needed anymore?
+    if created:
+        main.controllers['connection'].add()
+    sys.exit(app.exec_())
 
-    app.exec_()
-#    sys.exit()