Make IPyPBX work with SQL tables created by django models
[ipypbx] / src / ipypbx / main.py
1 # Copyright (c) Stas Shtin, 2010
2
3 # This file is part of IPyPBX.
4
5 # IPyPBX is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9
10 # IPyPBX is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14
15 # You should have received a copy of the GNU General Public License
16 # along with IPyPBX.  If not, see <http://www.gnu.org/licenses/>.
17
18 import os
19 import sys
20 from ipypbx import controllers, ui
21 from PyQt4 import QtCore, QtGui, QtSql
22
23
24 # Working directory path.
25 # NOTE: ~/.ipypbx gives an error - for some reason QT doesn't like the dot?
26 PREFIX = os.path.expanduser('~/.ipypbx')
27
28 # Database file name.
29 DB_NAME = 'ipypbx.db'
30
31
32 def setupDb(prefix=PREFIX, dbname=DB_NAME):
33     """
34     Setup database.
35     """
36     created = False
37     # Create it if necessary.
38     if prefix:
39         if not os.path.exists(prefix):
40             created = True
41             os.mkdir(prefix, 0700)
42         
43         os.chdir(prefix)
44     
45     # Connect to SQLite database.
46     db = QtSql.QSqlDatabase.addDatabase("QSQLITE")
47     db.setDatabaseName(dbname)
48     
49     if db.open():
50         if created:
51             # Load script from local file.
52             sql_script = open(os.path.join(os.path.dirname(__file__), 'create.sql')).read()
53
54             # Split into individual queries.
55             sql_queries = sql_script.split(';')
56
57             # Execute all queries except BEGIN/COMMIT sequences.
58             query = QtSql.QSqlQuery()
59             for query_string in sql_queries[1:-2]:
60                 query.exec_(query_string)
61     else:
62         # Something went horribly wrong.
63         QtGui.QMessageBox.warning(
64             None, "Fatal Error", "Database Error: %s" % db.lastError().text())
65         sys.exit(1)
66
67
68 def localize():
69     locale = QtCore.QLocale.system().name()
70     translator = QtCore.QTranslator()
71     
72     if translator.load("ipypbx_%s" % locale.toLower(), "ipypbx/locale"):
73         QtGui.QApplication.installTranslator(translator)
74
75     
76 if __name__ == '__main__':
77     app = QtGui.QApplication(sys.argv)
78     setupDb()
79     main = QtGui.QMainWindow()
80     views = ui.Ui_MainWindow()
81     views.setupUi(main)
82     localize()
83     main.show()
84
85     connectionController = controllers.ConnectionController(
86         parent=main, views=views)
87
88     app.exec_()
89 #    sys.exit()