class MainWindow(QtGui.QMainWindow, Ui_MainWindow):
currentUser = None
currentGroup = None
def __init__(self, parent=None ):
QtGui.QMainWindow.__init__(self, parent)
self.setupUi(self)
#self.connection_string="sqlite:///%s/../../davstorage/db/devel.db" % (os.getcwd())
self.connection_string="postgres://*****:*****@localhost/davstorage"
self.dbhandler = DBHandler(self.connection_string)
widget = AuthWindow(self.dbhandler,self)
widget.setModal(True)
widget.exec_()
if self.dbhandler.getCurrentUser() == None:
self.close()
self.update_data()
def event(self, event):
if event == QtCore.QEvent.Close:
self.update_data()
return super(MainWindow, self).event(event)
def add_user(self):
widget = UserWindow(self.dbhandler,parent=self)
widget.show()
def delete_user(self):
user = self.dbhandler.session.query(User).filter_by(login=self.currentUser.__str__()).first()
user.is_deleted=True
self.dbhandler.session.add(user)
self.dbhandler.session.commit()
self.update_data()
def edit_user(self):
widget = UserWindow(self.dbhandler,self.currentUser,parent=self)
widget.show()
pass
def add_group(self):
widget = GroupWindow(self.dbhandler,None,pgroup =self.currentGroup,parent=self)
widget.show()
pass
def edit_group(self):
widget = GroupWindow(self.dbhandler,self.currentGroup,pgroup =self.currentGroup, parent=self)
widget.show()
pass
def delete_group(self):
user = self.dbhandler.session.query(Group).filter_by(name=self.currentGroup.__str__()).first()
user.is_deleted=True
self.dbhandler.session.add(user)
self.dbhandler.session.commit()
self.update_data()
def tab_changed(self, widget):
self.update_data()
pass
def restr_group(self):
widget = RestrictionsWindow(self.dbhandler,self.currentGroup,parent=self)
widget.show()
pass
def update_data(self):
users = self.dbhandler.getUsers()
self.lstUsers.clear()
for u in users:
self.lstUsers.addItem(QtGui.QListWidgetItem(u.login))
self.treeGroups.clear()
usr = self.dbhandler.getCurrentUser();
for group in self.dbhandler.getGroups():
rs = self.dbhandler.session.query(ActionRestrict).filter_by(actor_id=usr.id, actor_type=1, object_type=2, object_id = group.id).first()
if rs != None and rs.action & actions["ADMIN"] != 0 :
item=QtGui.QTreeWidgetItem([group.name])
item.addChildren(self.get_groupWidgetTree(group, usr))
#item = self.get_groupWidgetTree(group)
self.treeGroups.addTopLevelItem(item)
else:
subs = self.get_groupWidgetTree(group, usr)
for i in subs:
self.treeGroups.addTopLevelItem(i)
def get_groupWidgetTree(self, group, usr):
lst = []
for g in group.subgroups:
rs = self.dbhandler.session.query(ActionRestrict).filter_by(actor_id=usr.id, actor_type=1, object_type=2, object_id = g.id).first()
subs = self.get_groupWidgetTree(g, usr)
if rs != None and rs.action & actions["ADMIN"] != 0 :
gi = QtGui.QTreeWidgetItem([g.name])
gi.addChildren(subs)
gi.setData(0,32, rs.action)
lst.append(gi)
else:
for i in subs:
d = i.data(0,32)
if d.toUInt()[0] & actions["ADMIN"] != 0:
lst.append(i)
return lst
# def get_groupWidgetTree(self, group):
# lst = []
# for g in group.subgroups:
# gi = QtGui.QTreeWidgetItem([g.name])
# gi.addChildren(self.get_groupWidgetTree(g))
# lst.append(gi)
#
# return lst
def groupItemChanged(self, item):
if item == None:
self.currentGroup = None
else:
self.currentGroup = item.text(0)
pass
def userItemChanged(self, item):
self.currentUser = item.text()
pass
