toxygen/src/main.py
2016-02-20 23:50:10 +03:00

119 lines
3.4 KiB
Python

from loginscreen import LoginScreen
from settings import Settings
from mainscreen import MainWindow
from profile import Profile, tox_factory
import sys
from PySide import QtCore, QtGui
from tox import Tox
from bootstrap import node_generator
class login(object):
def __init__(self, arr):
self.arr = arr
def login_screen_close(self, t, number=-1, default=False, name=None):
""" Function which processes data from login screen
:param t: 0 - window was closed, 1 - new profile was created, 2 - profile loaded
:param number: num of choosen profile in list (-1 by default)
:param default: was or not choosen profile marked as default
:param name: name of new profile
"""
print str(t), str(number), str(default), str(name)
self.t = t
self.num = number
self.default = default
self.name = name
def get_data(self):
return self.arr[self.num]
def status(a, b, c):
print 'WOW, it works!'
print str(b)
def friend_status(a, b, c, d, e):
print 'Friend connected! Friend number: ' + str(c)
def message(a, b, c, d, e, f):
print 'Message: ', str(d)
def main():
"""
main function of app. loads loginscreen if needed and starts mainscreen
"""
app = QtGui.QApplication(sys.argv)
app.setWindowIcon(QtGui.QIcon('images/icon.png'))
settings = Settings()
if not settings['auto_profile']:
# show login screen if default profile not found
ls = LoginScreen()
ls.setWindowIconText("Toxygen")
profiles = Profile.find_profiles()
ls.update_select(map(lambda x: x[1], profiles))
_login = login(profiles)
ls.update_on_close(_login.login_screen_close)
ls.show()
app.connect(app, QtCore.SIGNAL("lastWindowClosed()"), app, QtCore.SLOT("quit()"))
app.exec_()
if not _login.t:
return
elif _login.t == 1: # create new profile
# TODO: add creation of new profile
path = Settings.get_default_path()
name = _login.name if _login.name else 'Toxygen User'
return
else: # load existing profile
path, name = _login.get_data()
if _login.default:
settings['auto_profile'] = (path, name)
settings.save()
else:
path, name = settings['auto_profile']
# loading profile
print str(path), str(name)
data = Profile.open_profile(path, name)
ms = MainWindow()
# creating tox instance
tox = tox_factory(data, settings)
# bootstrap
for data in node_generator():
tox.bootstrap(*data)
# TODO: set all callbacks
tox.callback_friend_message(message, 0)
tox.callback_self_connection_status(status, 0)
#tox.callback_friend_connection_status(friend_status, 0)
# starting thread for tox iterate
mainloop = ToxIterateThread(tox)
mainloop.start()
ms.show()
app.connect(app, QtCore.SIGNAL("lastWindowClosed()"), app, QtCore.SLOT("quit()"))
app.exec_()
mainloop.stop = True
mainloop.wait()
del tox
class ToxIterateThread(QtCore.QThread):
def __init__(self, tox):
QtCore.QThread.__init__(self)
self.tox = tox
self.stop = False
def run(self):
while not self.stop:
self.tox.iterate()
self.msleep(self.tox.iteration_interval())
if __name__ == '__main__':
main()