class LightDMManager:
def __init__(self):
self.scriptDir = os.path.dirname(os.path.realpath(__file__))
# Load window and widgets
self.builder = Gtk.Builder()
self.builder.add_from_file(
join(self.scriptDir,
'../../share/lightdm-manager/lightdm-manager.glade'))
# Main window objects
go = self.builder.get_object
self.window = go('ldmWindow')
self.swUsers = go('swUsers')
self.tvUsers = go('tvUsers')
self.btnSave = go('btnSave')
self.imgBackground = go('imgBackground')
self.btnUsers = go('btnUsers')
self.btnAppearance = go('btnAppearance')
self.chkHideUsers = go('chkHideUsers')
self.ebFace = go('ebFace')
self.imgFace = go('imgFace')
self.nbLightDM = go('nbLightDM')
self.cmbThemes = go('cmbThemes')
# Read from config file
self.cfg = Config('lightdm-manager.conf')
self.lightdmConf = self.cfg.getValue('CONFIG', 'lightdmConf')
self.desktopbaseDir = self.cfg.getValue('CONFIG', 'desktopbaseDir')
gktGreeterConf = self.cfg.getValue('CONFIG', 'gtkGreeterConf')
kdeGreeterConf = self.cfg.getValue('CONFIG', 'kdeGreeterConf')
if exists(gktGreeterConf):
self.greeterConf = gktGreeterConf
else:
self.greeterConf = kdeGreeterConf
# Translations
title = _("LightDM Manager")
self.window.set_title(title)
self.btnUsers.set_label("_{}".format(_("Users")))
self.btnAppearance.set_label("_{}".format(_("Appearance")))
go('lblBackground').set_label(_("Background"))
go('lblTheme').set_label(_("Theme"))
go('lblLightDmMenu').set_label(_("Menu"))
self.chkHideUsers.set_label(_("Hide users"))
go('lblUsersFace').set_label(_("User icon"))
go('lblUsersAutologin').set_label(_("Auto-login"))
# Get current background image
self.cfgGreeter = Config(self.greeterConf)
try:
self.curBgPath = self.cfgGreeter.getValue('greeter', 'background')
self.curTheme = self.cfgGreeter.getValue('greeter', 'theme-name')
except:
self.curBgPath = None
self.curTheme = None
# Get current auto-login user
self.cfgLightdm = Config(self.lightdmConf)
try:
self.curAutoUser = self.cfgLightdm.getValue(
'SeatDefaults', 'autologin-user').strip()
self.curHideUsers = False
ghu = self.cfgLightdm.getValue('SeatDefaults',
'greeter-hide-users').strip()
if 'true' in ghu:
self.curHideUsers = True
except:
self.curAutoUser = None
self.curHideUsers = False
# Init
self.usr = User()
self.newbgImg = self.curBgPath
self.newAutoUser = self.curAutoUser
self.newTheme = self.curTheme
self.themes = []
self.selectedMenuItem = None
self.debug = False
self.logPath = ''
self.prevPath = None
self.tempFace = "/tmp/face"
self.newFaces = []
self.loggedUser = functions.getUserLoginName()
self.curUser = self.loggedUser
self.selectImg = join(self.scriptDir,
'../../share/lightdm-manager/select.png')
# Handle arguments
try:
opts, args = getopt.getopt(sys.argv[1:], 'dl:', ['debug', 'log='])
except getopt.GetoptError:
print(("Arguments cannot be parsed: %s" % str(sys.argv[1:])))
sys.exit(1)
for opt, arg in opts:
if opt in ('-d', '--debug'):
self.debug = True
elif opt in ('-l', '--log'):
self.logPath = arg
# Initialize logging
if self.debug:
if not self.logPath:
self.logPath = 'lightdm-manager.log'
self.log = Logger(self.logPath, 'debug', True, None, self.window)
# Backup config files because ConfigParser does not preserve commented lines
if not exists("%s.org" % self.greeterConf):
copy(self.greeterConf, "%s.org" % self.greeterConf)
self.log.write(
"%(conf1)s copied to %(conf2)s.org" % {
"conf1": self.greeterConf,
"conf2": self.greeterConf
}, 'LightDMManager.main', 'debug')
if not exists("%s.org" % self.lightdmConf):
copy(self.lightdmConf, "%s.org" % self.lightdmConf)
self.log.write(
"%(conf1)s copied to %(conf2)s.org" % {
"conf1": self.lightdmConf,
"conf2": self.lightdmConf
}, 'LightDMManager.main', 'debug')
# Initiate the treeview handler and connect the custom toggle event with usersCheckBoxToggled
self.tvHandler = TreeViewHandler(self.tvUsers, self.log)
self.tvHandler.connect('checkbox-toggled', self.usersCheckBoxToggled)
# Get users
self.users = self.usr.getUsers()
self.fillUsers()
self.tvHandler.selectValue(self.curUser, 1)
self.setBackground(self.curBgPath)
self.cmbHandlerThemes = ComboBoxHandler(self.cmbThemes)
self.listThemes()
self.chkHideUsers.set_active(self.curHideUsers)
# Show users menu
self.on_btnUsers_clicked(None)
self.on_tvUsers_cursor_changed(None)
self.version = functions.getPackageVersion('lightdm-manager')
# Connect the signals and show the window
self.builder.connect_signals(self)
self.window.show()
# ===============================================
# Menu section functions
# ===============================================
def on_btnUsers_clicked(self, widget, event=None):
if self.selectedMenuItem != menuItems[0]:
self.selectedMenuItem = menuItems[0]
self.nbLightDM.set_current_page(0)
def on_btnAppearance_clicked(self, widget, event=None):
if self.selectedMenuItem != menuItems[1]:
self.selectedMenuItem = menuItems[1]
self.nbLightDM.set_current_page(1)
# ===============================================
# Functions
# ===============================================
def on_ebFace_enter_notify_event(self, widget, event):
self.window.get_window().set_cursor(Gdk.Cursor(Gdk.CursorType.HAND2))
def on_ebFace_leave_notify_event(self, widget, event):
self.window.get_window().set_cursor(None)
def on_ebFace_button_release_event(self, widget, event):
home = self.usr.getUserHomeDir(self.curUser)
primaryGroup = self.usr.getUserPrimaryGroupName(self.curUser)
imagePath = SelectImageDialog(_('Select user image'), home,
self.window).show()
if imagePath is not None:
tempUserImg = "%(tempFace)s.%(curUser)s" % {
"tempFace": self.tempFace,
"curUser": self.curUser
}
self.newFaces.append(
[tempUserImg,
join(home, ".face"), self.curUser, primaryGroup])
print((">>> self.newFaces = %s" % self.newFaces))
ih = ImageHandler(imagePath)
ih.makeFaceImage(tempUserImg)
if exists(tempUserImg):
self.imgFace.set_from_pixbuf(ih.pixbuf)
else:
# This should never happen
self.imgFace.set_from_file(self.selectImg)
def on_tvUsers_cursor_changed(self, widget):
self.curUser = self.tvHandler.getSelectedValue(1)
showFace = None
if self.newFaces:
for face in self.newFaces:
if face[2] == self.curUser and exists(face[0]):
showFace = GdkPixbuf.Pixbuf.new_from_file(face[0])
if showFace is None:
showFace = self.usr.getUserFacePixbuf(self.curUser)
if showFace is None:
# Still no user icon found: show select image
self.imgFace.set_from_file(self.selectImg)
else:
self.imgFace.set_from_pixbuf(showFace)
def on_btnSave_clicked(self, widget):
saved = False
saveHideUsers = False
saveAutoUser = False
saveFaces = False
saveBackground = False
saveTheme = False
if self.chkHideUsers.get_active() != self.curHideUsers:
saveHideUsers = True
if self.curAutoUser != self.newAutoUser:
saveAutoUser = True
if self.newFaces:
saveFaces = True
if self.curBgPath != self.newbgImg:
saveBackground = True
self.newTheme = self.cmbHandlerThemes.getValue()
if self.curTheme != self.newTheme:
saveTheme = True
if saveHideUsers or saveAutoUser or saveFaces or saveBackground or saveTheme:
qd = QuestionDialog(
_("LightDM settings"),
_("Settings have changed\n\nDo you want to save the new settings?"
), self.window)
answer = qd.show()
if answer:
if saveAutoUser:
if self.newAutoUser is not None:
# Save the auto-login user
self.cfgLightdm.setValue('SeatDefaults',
'autologin-user',
self.newAutoUser)
self.cfgLightdm.setValue('SeatDefaults',
'autologin-user-timeout', '0')
self.curAutoUser = self.newAutoUser
self.log.write(
"New auto-login user: %(usr)s" %
{"usr": self.curAutoUser},
'LightDMManager.saveSettings', 'debug')
else:
self.cfgLightdm.removeOption('SeatDefaults',
'autologin-user')
self.cfgLightdm.removeOption('SeatDefaults',
'autologin-user-timeout')
self.curAutoUser = None
self.log.write("Auto-login disabled",
'LightDMManager.saveSettings', 'debug')
if saveHideUsers:
hideUsers = str(self.chkHideUsers.get_active()).lower()
self.cfgLightdm.setValue('SeatDefaults',
'greeter-hide-users', hideUsers)
self.log.write(
"Hide users saved: %(users)s" % {"users": hideUsers},
'LightDMManager.saveSettings', 'debug')
if saveFaces:
for face in self.newFaces:
if exists(face[0]):
copy(face[0], face[1])
if exists(face[1]):
os.system(
"chown %(owner)s:%(group)s %(path)s" % {
"owner": face[2],
"group": face[3],
"path": face[1]
})
self.log.write("User icons saved",
'LightDMManager.saveSettings', 'debug')
if saveTheme:
self.cfgGreeter.setValue('greeter', 'theme-name',
self.newTheme)
self.curTheme = self.newTheme
self.log.write(
"Theme saved: %(theme)s" % {"theme": self.curTheme},
'LightDMManager.saveSettings', 'debug')
if saveBackground:
if os.path.exists(self.newbgImg):
self.cfgGreeter.setValue('greeter', 'background',
self.newbgImg)
self.curBgPath = self.newbgImg
self.log.write(
"Background saved: %(background)s" %
{"background": self.curBgPath},
'LightDMManager.saveSettings', 'debug')
saved = True
else:
if os.path.exists(self.curBgPath):
self.setBackground(self.curBgPath)
self.log.write(
"Current background: %(background)s" %
{"background": self.curBgPath},
'LightDMManager.saveSettings', 'debug')
else:
self.imgBackground.set_from_file(
join(self.scriptDir,
'../../share/lightdm-manager/select.png'))
self.log.write("No background set",
'LightDMManager.saveSettings', 'debug')
self.fillUsers()
if saved:
self.curHideUsers = self.chkHideUsers.get_active()
self.curAutoUser = self.newAutoUser
self.newFaces = []
self.curBgPath = self.newbgImg
self.curTheme = self.newTheme
MessageDialogSafe(_("Saved"),
_("LightDM settings saved successfully."),
Gtk.MessageType.INFO, self.window).show()
def on_ebBackground_button_release_event(self, widget, event):
self.newbgImg = SelectImageDialog(_("Choose background image"),
self.desktopbaseDir,
self.window).show()
if exists(self.newbgImg) and self.newbgImg != self.curBgPath:
self.setBackground(self.newbgImg)
self.log.write(
_("New background: %(bg)s") % {"bg": self.newbgImg},
'LightDMManager.chooseFile', 'info')
def on_ebBackground_enter_notify_event(self, widget, event):
self.window.get_window().set_cursor(Gdk.Cursor(Gdk.CursorType.HAND2))
def on_ebBackground_leave_notify_event(self, widget, event):
self.window.get_window().set_cursor(None)
# This method is fired by the TreeView.checkbox-toggled event
def usersCheckBoxToggled(self, obj, path, colNr, toggleValue):
path = int(path)
model = self.tvUsers.get_model()
itr = model.get_iter(path)
user = model[itr][1]
if self.prevPath != path or toggleValue:
# Only one toggle box can be selected (or none)
self.tvHandler.treeviewToggleAll([0], False, 1, user)
# Save current path
self.prevPath = path
# Save selected user
self.newAutoUser = user
self.log.write(
_("Auto-login user selected: %(usr)s") % {"usr": user},
'LightDMManager.usersCheckBoxToggled', 'info')
elif self.prevPath == path and not toggleValue:
self.newAutoUser = None
def listThemes(self):
themeDir = '/usr/share/themes'
themeDirLocal = '~/.local/share/themes'
dirs = functions.locate('gtk-*', themeDir, True) + functions.locate(
'gtk-*', themeDirLocal, True)
for path in dirs:
dirList = path.split('/')
for d in dirList:
if 'gtk-' in d:
break
themeName = d
if themeName not in self.themes:
self.themes.append(themeName)
if self.themes:
self.cmbHandlerThemes.fillComboBox(self.themes)
if self.curTheme in self.themes:
self.cmbHandlerThemes.selectValue(self.curTheme)
def fillUsers(self):
selUsr = False
contentList = []
i = 0
for usr in self.users:
if usr == self.curAutoUser:
selUsr = True
self.prevPath = i
else:
selUsr = False
contentList.append([selUsr, usr])
i += 1
# Fill treeview with users
#fillTreeview(contentList, columnTypesList, columnHideList=[-1], setCursor=0, setCursorWeight=400, firstItemIsColName=False, appendToExisting=False, appendToTop=False)
columnTypesList = ['bool', 'str']
self.tvHandler.fillTreeview(contentList, columnTypesList)
def setBackground(self, path):
# Set Background
if path is not None:
if exists(path):
ih = ImageHandler(path)
ih.resizeImage(height=200)
self.imgBackground.set_from_pixbuf(ih.pixbuf)
else:
self.imgBackground.set_from_file(self.selectImg)
else:
self.imgBackground.set_from_file(self.selectImg)
# ===============================================
# General functions
# ===============================================
def on_ldmWindow_destroy(self, widget, data=None):
# Close the app
self.on_btnSave_clicked(None)
for tmp in self.newFaces:
os.remove(tmp[0])
Gtk.main_quit()
class LightDMManager:
def __init__(self):
self.scriptDir = os.path.dirname(os.path.realpath(__file__))
# Load window and widgets
self.builder = Gtk.Builder()
self.builder.add_from_file(join(self.scriptDir, '../../share/lightdm-manager/lightdm-manager.glade'))
# Main window objects
go = self.builder.get_object
self.window = go('ldmWindow')
self.swUsers = go('swUsers')
self.tvUsers = go('tvUsers')
self.btnSave = go('btnSave')
self.imgBackground = go('imgBackground')
self.btnUsers = go('btnUsers')
self.btnAppearance = go('btnAppearance')
self.chkHideUsers = go('chkHideUsers')
self.ebFace = go('ebFace')
self.imgFace = go('imgFace')
self.nbLightDM = go('nbLightDM')
self.cmbThemes = go('cmbThemes')
# Read from config file
self.cfg = Config('lightdm-manager.conf')
self.lightdmConf = self.cfg.getValue('CONFIG', 'lightdmConf')
self.desktopbaseDir = self.cfg.getValue('CONFIG', 'desktopbaseDir')
gktGreeterConf = self.cfg.getValue('CONFIG', 'gtkGreeterConf')
kdeGreeterConf = self.cfg.getValue('CONFIG', 'kdeGreeterConf')
if exists(gktGreeterConf):
self.greeterConf = gktGreeterConf
else:
self.greeterConf = kdeGreeterConf
# Translations
title = _("LightDM Manager")
self.window.set_title(title)
self.btnUsers.set_label("_{}".format(_("Users")))
self.btnAppearance.set_label("_{}".format(_("Appearance")))
go('lblBackground').set_label(_("Background"))
go('lblTheme').set_label(_("Theme"))
go('lblLightDmMenu').set_label(_("Menu"))
self.chkHideUsers.set_label(_("Hide users"))
go('lblUsersFace').set_label(_("User icon"))
go('lblUsersAutologin').set_label(_("Auto-login"))
# Get current background image
self.cfgGreeter = Config(self.greeterConf)
try:
self.curBgPath = self.cfgGreeter.getValue('greeter', 'background')
self.curTheme = self.cfgGreeter.getValue('greeter', 'theme-name')
except:
self.curBgPath = None
self.curTheme = None
# Get current auto-login user
self.cfgLightdm = Config(self.lightdmConf)
try:
self.curAutoUser = self.cfgLightdm.getValue('SeatDefaults', 'autologin-user').strip()
self.curHideUsers = False
ghu = self.cfgLightdm.getValue('SeatDefaults', 'greeter-hide-users').strip()
if 'true' in ghu:
self.curHideUsers = True
except:
self.curAutoUser = None
self.curHideUsers = False
# Init
self.usr = User()
self.newbgImg = self.curBgPath
self.newAutoUser = self.curAutoUser
self.newTheme = self.curTheme
self.themes = []
self.selectedMenuItem = None
self.debug = False
self.logPath = ''
self.prevPath = None
self.tempFace = "/tmp/face"
self.newFaces = []
self.loggedUser = functions.getUserLoginName()
self.curUser = self.loggedUser
self.selectImg = join(self.scriptDir, '../../share/lightdm-manager/select.png')
# Handle arguments
try:
opts, args = getopt.getopt(sys.argv[1:], 'dl:', ['debug', 'log='])
except getopt.GetoptError:
print(("Arguments cannot be parsed: %s" % str(sys.argv[1:])))
sys.exit(1)
for opt, arg in opts:
if opt in ('-d', '--debug'):
self.debug = True
elif opt in ('-l', '--log'):
self.logPath = arg
# Initialize logging
if self.debug:
if not self.logPath:
self.logPath = 'lightdm-manager.log'
self.log = Logger(self.logPath, 'debug', True, None, self.window)
# Backup config files because ConfigParser does not preserve commented lines
if not exists("%s.org" % self.greeterConf):
copy(self.greeterConf, "%s.org" % self.greeterConf)
self.log.write("%(conf1)s copied to %(conf2)s.org" % { "conf1": self.greeterConf, "conf2": self.greeterConf }, 'LightDMManager.main', 'debug')
if not exists("%s.org" % self.lightdmConf):
copy(self.lightdmConf, "%s.org" % self.lightdmConf)
self.log.write("%(conf1)s copied to %(conf2)s.org" % { "conf1": self.lightdmConf, "conf2": self.lightdmConf }, 'LightDMManager.main', 'debug')
# Initiate the treeview handler and connect the custom toggle event with usersCheckBoxToggled
self.tvHandler = TreeViewHandler(self.tvUsers, self.log)
self.tvHandler.connect('checkbox-toggled', self.usersCheckBoxToggled)
# Get users
self.users = self.usr.getUsers()
self.fillUsers()
self.tvHandler.selectValue(self.curUser, 1)
self.setBackground(self.curBgPath)
self.cmbHandlerThemes = ComboBoxHandler(self.cmbThemes)
self.listThemes()
self.chkHideUsers.set_active(self.curHideUsers)
# Show users menu
self.on_btnUsers_clicked(None)
self.on_tvUsers_cursor_changed(None)
self.version = functions.getPackageVersion('lightdm-manager')
# Connect the signals and show the window
self.builder.connect_signals(self)
self.window.show()
# ===============================================
# Menu section functions
# ===============================================
def on_btnUsers_clicked(self, widget, event=None):
if self.selectedMenuItem != menuItems[0]:
self.selectedMenuItem = menuItems[0]
self.nbLightDM.set_current_page(0)
def on_btnAppearance_clicked(self, widget, event=None):
if self.selectedMenuItem != menuItems[1]:
self.selectedMenuItem = menuItems[1]
self.nbLightDM.set_current_page(1)
# ===============================================
# Functions
# ===============================================
def on_ebFace_enter_notify_event(self, widget, event):
self.window.get_window().set_cursor(Gdk.Cursor(Gdk.CursorType.HAND2))
def on_ebFace_leave_notify_event(self, widget, event):
self.window.get_window().set_cursor(None)
def on_ebFace_button_release_event(self, widget, event):
home = self.usr.getUserHomeDir(self.curUser)
primaryGroup = self.usr.getUserPrimaryGroupName(self.curUser)
imagePath = SelectImageDialog(_('Select user image'), home, self.window).show()
if imagePath is not None:
tempUserImg = "%(tempFace)s.%(curUser)s" % {"tempFace": self.tempFace, "curUser": self.curUser}
self.newFaces.append([tempUserImg, join(home, ".face"), self.curUser, primaryGroup])
print((">>> self.newFaces = %s" % self.newFaces))
ih = ImageHandler(imagePath)
ih.makeFaceImage(tempUserImg)
if exists(tempUserImg):
self.imgFace.set_from_pixbuf(ih.pixbuf)
else:
# This should never happen
self.imgFace.set_from_file(self.selectImg)
def on_tvUsers_cursor_changed(self, widget):
self.curUser = self.tvHandler.getSelectedValue(1)
showFace = None
if self.newFaces:
for face in self.newFaces:
if face[2] == self.curUser and exists(face[0]):
showFace = GdkPixbuf.Pixbuf.new_from_file(face[0])
if showFace is None:
showFace = self.usr.getUserFacePixbuf(self.curUser)
if showFace is None:
# Still no user icon found: show select image
self.imgFace.set_from_file(self.selectImg)
else:
self.imgFace.set_from_pixbuf(showFace)
def on_btnSave_clicked(self, widget):
saved = False
saveHideUsers = False
saveAutoUser = False
saveFaces = False
saveBackground = False
saveTheme = False
if self.chkHideUsers.get_active() != self.curHideUsers:
saveHideUsers = True
if self.curAutoUser != self.newAutoUser:
saveAutoUser = True
if self.newFaces:
saveFaces = True
if self.curBgPath != self.newbgImg:
saveBackground = True
self.newTheme = self.cmbHandlerThemes.getValue()
if self.curTheme != self.newTheme:
saveTheme = True
if saveHideUsers or saveAutoUser or saveFaces or saveBackground or saveTheme:
qd = QuestionDialog(_("LightDM settings"), _("Settings have changed\n\nDo you want to save the new settings?"), self.window)
answer = qd.show()
if answer:
if saveAutoUser:
if self.newAutoUser is not None:
# Save the auto-login user
self.cfgLightdm.setValue('SeatDefaults', 'autologin-user', self.newAutoUser)
self.cfgLightdm.setValue('SeatDefaults', 'autologin-user-timeout', '0')
self.curAutoUser = self.newAutoUser
self.log.write("New auto-login user: %(usr)s" % { "usr": self.curAutoUser }, 'LightDMManager.saveSettings', 'debug')
else:
self.cfgLightdm.removeOption('SeatDefaults', 'autologin-user')
self.cfgLightdm.removeOption('SeatDefaults', 'autologin-user-timeout')
self.curAutoUser = None
self.log.write("Auto-login disabled", 'LightDMManager.saveSettings', 'debug')
if saveHideUsers:
hideUsers = str(self.chkHideUsers.get_active()).lower()
self.cfgLightdm.setValue('SeatDefaults', 'greeter-hide-users', hideUsers)
self.log.write("Hide users saved: %(users)s" % {"users": hideUsers}, 'LightDMManager.saveSettings', 'debug')
if saveFaces:
for face in self.newFaces:
if exists(face[0]):
copy(face[0], face[1])
if exists(face[1]):
os.system("chown %(owner)s:%(group)s %(path)s" % {"owner": face[2], "group": face[3], "path": face[1]})
self.log.write("User icons saved", 'LightDMManager.saveSettings', 'debug')
if saveTheme:
self.cfgGreeter.setValue('greeter', 'theme-name', self.newTheme)
self.curTheme = self.newTheme
self.log.write("Theme saved: %(theme)s" % { "theme": self.curTheme }, 'LightDMManager.saveSettings', 'debug')
if saveBackground:
if os.path.exists(self.newbgImg):
self.cfgGreeter.setValue('greeter', 'background', self.newbgImg)
self.curBgPath = self.newbgImg
self.log.write("Background saved: %(background)s" % { "background": self.curBgPath }, 'LightDMManager.saveSettings', 'debug')
saved = True
else:
if os.path.exists(self.curBgPath):
self.setBackground(self.curBgPath)
self.log.write("Current background: %(background)s" % { "background": self.curBgPath }, 'LightDMManager.saveSettings', 'debug')
else:
self.imgBackground.set_from_file(join(self.scriptDir, '../../share/lightdm-manager/select.png'))
self.log.write("No background set", 'LightDMManager.saveSettings', 'debug')
self.fillUsers()
if saved:
self.curHideUsers = self.chkHideUsers.get_active()
self.curAutoUser = self.newAutoUser
self.newFaces = []
self.curBgPath = self.newbgImg
self.curTheme = self.newTheme
MessageDialogSafe(_("Saved"), _("LightDM settings saved successfully."), Gtk.MessageType.INFO, self.window).show()
def on_ebBackground_button_release_event(self, widget, event):
self.newbgImg = SelectImageDialog(_("Choose background image"), self.desktopbaseDir, self.window).show()
if exists(self.newbgImg) and self.newbgImg != self.curBgPath:
self.setBackground(self.newbgImg)
self.log.write(_("New background: %(bg)s") % { "bg": self.newbgImg }, 'LightDMManager.chooseFile', 'info')
def on_ebBackground_enter_notify_event(self, widget, event):
self.window.get_window().set_cursor(Gdk.Cursor(Gdk.CursorType.HAND2))
def on_ebBackground_leave_notify_event(self, widget, event):
self.window.get_window().set_cursor(None)
# This method is fired by the TreeView.checkbox-toggled event
def usersCheckBoxToggled(self, obj, path, colNr, toggleValue):
path = int(path)
model = self.tvUsers.get_model()
itr = model.get_iter(path)
user = model[itr][1]
if self.prevPath != path or toggleValue:
# Only one toggle box can be selected (or none)
self.tvHandler.treeviewToggleAll([0], False, 1, user)
# Save current path
self.prevPath = path
# Save selected user
self.newAutoUser = user
self.log.write(_("Auto-login user selected: %(usr)s") % { "usr": user }, 'LightDMManager.usersCheckBoxToggled', 'info')
elif self.prevPath == path and not toggleValue:
self.newAutoUser = None
def listThemes(self):
themeDir = '/usr/share/themes'
themeDirLocal = '~/.local/share/themes'
dirs = functions.locate('gtk-*', themeDir, True) + functions.locate('gtk-*', themeDirLocal, True)
for path in dirs:
dirList = path.split('/')
for d in dirList:
if 'gtk-' in d:
break
themeName = d
if themeName not in self.themes:
self.themes.append(themeName)
if self.themes:
self.cmbHandlerThemes.fillComboBox(self.themes)
if self.curTheme in self.themes:
self.cmbHandlerThemes.selectValue(self.curTheme)
def fillUsers(self):
selUsr = False
contentList = []
i = 0
for usr in self.users:
if usr == self.curAutoUser:
selUsr = True
self.prevPath = i
else:
selUsr = False
contentList.append([selUsr, usr])
i += 1
# Fill treeview with users
#fillTreeview(contentList, columnTypesList, columnHideList=[-1], setCursor=0, setCursorWeight=400, firstItemIsColName=False, appendToExisting=False, appendToTop=False)
columnTypesList = ['bool', 'str']
self.tvHandler.fillTreeview(contentList, columnTypesList)
def setBackground(self, path):
# Set Background
if path is not None:
if exists(path):
ih = ImageHandler(path)
ih.resizeImage(height=200)
self.imgBackground.set_from_pixbuf(ih.pixbuf)
else:
self.imgBackground.set_from_file(self.selectImg)
else:
self.imgBackground.set_from_file(self.selectImg)
# ===============================================
# General functions
# ===============================================
def on_ldmWindow_destroy(self, widget, data=None):
# Close the app
self.on_btnSave_clicked(None)
for tmp in self.newFaces:
os.remove(tmp[0])
Gtk.main_quit()
class UserManager(object):
def __init__(self):
self.scriptDir = abspath(dirname(__file__))
# Load window and widgets
self.builder = Gtk.Builder()
self.builder.add_from_file(join(self.scriptDir, '../../share/usermanager/usermanager.glade'))
# Main window objects
go = self.builder.get_object
self.window = go('usermanagerWindow')
self.tvUsersMain = go('tvUsersMain')
self.tvUserGroupsMain = go('tvUserGroupsMain')
self.chkShowSystemUsers = go('chkShowSystemUsers')
self.btnUserEdit = go('btnUserEdit')
self.btnUserRemove = go('btnUserRemove')
# User window objects
self.windowUser = go('usermanagerUserWindow')
self.nbUser = go('nbUser')
self.txtLoginName = go('txtLoginName')
self.txtRealName = go('txtRealName')
self.txtUserID = go('txtUserID')
self.txtHomeDirectory = go('txtHomeDirectory')
self.cmbShells = go('cmbShells')
self.cmbPrimaryGroup = go('cmbPrimaryGroup')
self.cmbPrimaryGroupEntry = go('cmbPrimaryGroupEntry')
self.tvUserGroups = go('tvUserGroups')
self.ebFace = go('ebFace')
self.imgFace = go('imgFace')
self.lblUserID = go('lblUserID')
self.txtPassword = go('txtPassword')
self.txtLastChanged = go('txtLastChanged')
self.cmbValidUntilMonth = go('cmbValidUntilMonth')
self.spbValidUntilDay = go('spbValidUntilDay')
self.spbValidUntilYear = go('spbValidUntilYear')
self.radEnabled = go('radEnabled')
self.radDisabled = go('radDisabled')
self.radValidUntilAlways = go('radValidUntilAlways')
self.radValidUntilDate = go('radValidUntilDate')
self.spbRequirePasswordAfter = go('spbRequirePasswordAfter')
self.spbWarnBeforePasswordExpires = go('spbWarnBeforePasswordExpires')
self.spbDisableAccountWhenPasswordExpires = go('spbDisableAccountWhenPasswordExpires')
self.spbEnforceMinimumPasswordAge = go('spbEnforceMinimumPasswordAge')
self.chkRequirePasswordAfter = go('chkRequirePasswordAfter')
self.chkEnforceMinimumPasswordAge = go('chkEnforceMinimumPasswordAge')
self.tvGroupsMain = go('tvGroupsMain')
# Group window objects
self.windowGroup = go('usermanagerGroupWindow')
self.txtGroupName = go('txtGroupName')
self.txtGroupID = go('txtGroupID')
self.tvAccounts = go('tvAccounts')
self.tvSelectedAccounts = go('tvSelectedAccounts')
self.tvGroupAccounts = go('tvGroupAccounts')
# Main window translations
self.window.set_title(_("User manager"))
self.chkShowSystemUsers.set_label(_("Show system users"))
self.btnUserEdit.set_label(_("Edit"))
self.btnUserRemove.set_label(_("Remove"))
go('btnUserAdd').set_label(_("Add"))
go('lblUsersTab').set_text(_("Users"))
go('lblGroupsTab').set_text(_("Groups"))
go('lblUsersMain').set_text(go('lblUsersTab').get_text())
go('lblUserGroupsMain').set_text(_("User groups"))
go('lblGroupsMain').set_text(go('lblGroupsTab').get_text())
go('lblGroupAccountsMain').set_text(_("Group accounts"))
# User window translations
self.windowUser.set_title(_("User settings"))
go('lblUserDetails').set_text(_("Details"))
go('lblStatus').set_text(_("Status"))
go('lblLoginName').set_text(_("Login name"))
go('lblRealName').set_text(_("Real name"))
go('lblUserID').set_text(_("User ID"))
go('lblPrimaryGroup').set_text(_("Primary group"))
go('lblHomeDirectory').set_text(_("Home directory"))
go('lblShell').set_text(_("Shell"))
self.radEnabled.set_label(_("Enabled"))
self.radDisabled.set_label(_("Disabled"))
#go('lblPrivilegesAndGroups').set_text(_("Privileges and groups"))
go('lblPrivilegesAndGroups').set_text(go('lblGroupsTab').get_text())
go('lblPrivileges').set_text(_("Privileges"))
go('lblUserGroups').set_text(go('lblGroupsTab').get_text())
go('lblPasswordSecurity').set_text(_("Password"))
go('lblPassword').set_text(go('lblPasswordSecurity').get_text())
go('lblLastChanged').set_text(_("Last changed"))
go('lblValidUntil').set_text(_("Valid until"))
self.radValidUntilAlways.set_label(_("Always"))
go('lblPasswordAging').set_text(_("Password aging"))
self.chkRequirePasswordAfter.set_label(_("Require password after"))
go('lblWarnBeforePasswordExpires').set_text(_("Warn before password expires after"))
go('lblDisableAccountWhenPasswordExpires').set_text(_("Disable account when password expires after"))
self.chkEnforceMinimumPasswordAge.set_label(_("Enforce minimum password age"))
go('lblDays1').set_text(_("days"))
go('lblDays2').set_text(go('lblDays1').get_text())
go('lblDays3').set_text(go('lblDays1').get_text())
go('lblDays4').set_text(go('lblDays1').get_text())
go('btnSaveUser').set_label(_("Save"))
go('btnCancelUser').set_label(_("Cancel"))
# Group window translations
self.windowGroup.set_title(_("Group settings"))
go('lblGroupName').set_text(_("Group name"))
go('lblGroupID').set_text(_("Group ID"))
go('lblAccounts').set_text(_("Available accounts"))
go('lblSelectedAccounts').set_text(_("Selected accounts"))
go('btnOkGroup').set_label(go('btnSaveUser').get_label())
go('btnCancelGroup').set_label(go('btnCancelUser').get_label())
go('btnAddAccount').set_label(go('btnUserAdd').get_label())
go('btnRemoveAccount').set_label(self.btnUserRemove.get_label())
# Init
self.ec = ExecCmd()
self.usr = User()
self.usersInfo = {}
self.user = {}
self.users = []
self.accounts = []
self.shells = self.usr.getShells()
self.groups = None
self.loggedinUser = self.usr.getLoggedinUser()
self.loggedinUserPrimaryGroup = self.usr.getUserPrimaryGroupName(self.loggedinUser)
self.selectedUser = None
self.selectedGroup = None
self.selectedGroupAccounts = None
self.userFace = None
self.radEnabled.set_active(True)
self.radValidUntilAlways.set_active(True)
self.setValidUntil(False)
self.setRequirePasswordAfter(False)
self.setEnforceMinimumPasswordAge(False)
self.txtUserID.set_editable(False)
self.txtUserID.set_can_focus(False)
self.txtLastChanged.set_editable(False)
self.txtLastChanged.set_can_focus(False)
self.txtGroupID.set_editable(False)
self.txtGroupID.set_can_focus(False)
self.passwordChanged = False
self.homeDirChanged = False
self.tempFace = "/tmp/face.png"
self.filterText(self.txtLoginName)
self.filterText(self.txtGroupName)
self.filterText(self.cmbPrimaryGroupEntry)
# Treeviews
self.tvHandlerUsersMain = TreeViewHandler(self.tvUsersMain)
self.tvHandlerUserGroupsMain = TreeViewHandler(self.tvUserGroupsMain)
self.tvHandlerUserGroups = TreeViewHandler(self.tvUserGroups)
self.tvHandlerAccounts = TreeViewHandler(self.tvAccounts)
self.tvHandlerSelectedAccounts = TreeViewHandler(self.tvSelectedAccounts)
self.tvHandlerGroupsMain = TreeViewHandler(self.tvGroupsMain)
self.tvHandlerGroupAccounts = TreeViewHandler(self.tvGroupAccounts)
# Comboboxes
self.cmbHandlerShells = ComboBoxHandler(self.cmbShells)
self.cmbHandlerPrimaryGroup = ComboBoxHandler(self.cmbPrimaryGroup)
self.cmbHandlerValidUntilMonth = ComboBoxHandler(self.cmbValidUntilMonth)
# Get data
self.refreshData()
self.cmbHandlerShells.fillComboBox(self.shells)
self.cmbHandlerValidUntilMonth.fillComboBox(functions.getMonthsList())
year = datetime.now().year
adj = Gtk.Adjustment(value=year, lower=year, upper=year + 10, step_incr=1, page_incr=0, page_size=0)
self.spbValidUntilYear.set_adjustment(adj)
# Connect the signals and show the window
self.builder.connect_signals(self)
self.window.show()
# ===============================================
# User functions
# ===============================================
def on_tvUsersMain_cursor_changed(self, widget, event=None):
self.showUserData()
def on_tvUsersMain_row_activated(self, widget, path, column):
self.on_btnUserEdit_clicked(None)
def on_btnUserAdd_clicked(self, widget):
self.imgFace.set_from_pixbuf(self.usr.getUserFacePixbuf())
self.tvHandlerUserGroups.fillTreeview(contentList=self.getUserGroupsComplete(), columnTypesList=['bool', 'str'])
self.cmbHandlerShells.selectValue('/bin/bash')
self.cmbHandlerPrimaryGroup.fillComboBox(self.groups)
self.txtLoginName.set_editable(True)
self.txtLoginName.set_can_focus(True)
self.txtLoginName.set_text("new_user")
self.txtUserID.set_text(str(self.usr.getNewUserID()))
self.updateNewLogin()
self.txtPassword.set_text("")
self.radValidUntilAlways.set_active(True)
self.windowUser.show()
def on_btnUserEdit_clicked(self, widget):
if self.selectedUser is not None:
if not self.userFace is None:
self.imgFace.set_from_pixbuf(self.userFace)
self.imgFace.show()
self.nbUser.get_nth_page(2).show()
else:
self.imgFace.hide()
# Hide passwords tab
self.nbUser.get_nth_page(2).hide()
self.txtLoginName.set_editable(False)
self.txtLoginName.set_can_focus(False)
self.txtLoginName.set_text(self.user['user'].pw_name)
if ",,," in self.user['user'].pw_gecos:
self.txtRealName.set_text(self.user['user'].pw_name)
else:
self.txtRealName.set_text(self.user['user'].pw_gecos)
self.txtUserID.set_text(str(self.user['user'].pw_uid))
self.txtHomeDirectory.set_text(self.user['user'].pw_dir)
self.tvHandlerUserGroups.fillTreeview(contentList=self.getUserGroupsComplete(self.user['groups']), columnTypesList=['bool', 'str'])
self.cmbHandlerShells.selectValue(self.user['user'].pw_shell)
self.cmbHandlerPrimaryGroup.fillComboBox(self.groups, self.user['prgrp'])
self.txtPassword.set_text("")
self.txtLastChanged.set_text(self.usr.intToDate(nr_days=self.user['pwd'].sp_lstchg, format_string='%d %B %Y', start_date=datetime(1970, 1, 1)))
daysMin = self.user['pwd'].sp_min
print((">>> daysMin = %d" % daysMin))
if daysMin > 0:
self.radValidUntilDate.set_active(True)
dt = self.usr.intToDate(nr_days=daysMin)
self.spbValidUntilDay.set_value(dt.day)
self.cmbValidUntilMonth.set_active(dt.month - 1)
self.spbValidUntilYear.set_value(dt.year)
else:
self.radValidUntilAlways.set_active(True)
daysMax = self.user['pwd'].sp_max
daysWarn = self.user['pwd'].sp_warn
daysInact = self.user['pwd'].sp_inact
print((">>> daysMax = %d" % daysMax))
print((">>> daysWarn = %d" % daysWarn))
print((">>> daysInact = %d" % daysInact))
if daysMax < 1000 and daysWarn >= 0 and daysInact >= 0:
self.chkRequirePasswordAfter.set_active(True)
self.spbRequirePasswordAfter.set_value(daysMax)
self.spbWarnBeforePasswordExpires.set_value(daysWarn)
self.spbDisableAccountWhenPasswordExpires.set_value(daysInact)
else:
self.chkRequirePasswordAfter.set_active(False)
daysExpire = self.user['pwd'].sp_expire
print((">>> daysExpire = %d" % daysExpire))
if daysExpire >= 0:
self.chkEnforceMinimumPasswordAge.set_active(True)
self.spbEnforceMinimumPasswordAge.set_value(daysExpire)
else:
self.chkEnforceMinimumPasswordAge.set_active(False)
self.windowUser.show()
def on_btnUserRemove_clicked(self, widget):
# Remove user
if self.selectedUser is not None:
title = _("Remove user")
if self.selectedUser != self.loggedinUser:
qd = QuestionDialog(title, _("Are you sure you want to remove the following user:\n\n'%(user)s'") % { "user": self.selectedUser }, self.window)
answer = qd.show()
if answer:
# TODO
ret = self.usr.deleteUser(self.selectedUser)
if ret == "":
self.showInfo(title, _("User successfully removed: %(user)s") % {"user": self.selectedUser}, self.window)
self.refreshData()
else:
retMsg = "\n\n%s" % ret
self.showError(title, _("Could not remove user: %(user)s %(retmsg)s") % {"user": self.selectedUser, "retmsg": retMsg}, self.window)
else:
self.showError(title, _("You cannot remove the currently logged in user"), self.window)
def on_chkShowSystemUsers_toggled(self, widget):
self.refreshData()
def on_btnBrowse_clicked(self, widget):
directory = SelectDirectoryDialog(_('Select user directory'), self.txtHomeDirectory.get_text(), self.windowUser).show()
if directory is not None:
self.user['user'].pw_dir = directory
self.txtHomeDirectory.set_text(self.user['user'].pw_dir)
def on_radEnabled_toggled(self, widget):
if widget.get_active():
print(">>> Enable user")
def on_radDisabled_toggled(self, widget):
if widget.get_active():
print(">>> Disable user")
def on_radValidUntilAlways_toggled(self, widget):
if widget.get_active():
self.setValidUntil(False)
def on_radValidUntilDate_toggled(self, widget):
if widget.get_active():
self.setValidUntil(True)
def setValidUntil(self, boolean=True):
self.spbValidUntilDay.set_sensitive(boolean)
self.cmbValidUntilMonth.set_sensitive(boolean)
self.spbValidUntilYear.set_sensitive(boolean)
if not boolean:
self.spbValidUntilDay.set_value(0)
self.cmbValidUntilMonth.set_active(-1)
self.spbValidUntilYear.set_value(0)
def on_chkRequirePasswordAfter_toggled(self, widget):
if widget.get_active():
self.setRequirePasswordAfter(True)
else:
self.setRequirePasswordAfter(False)
def setRequirePasswordAfter(self, boolean=True):
self.spbRequirePasswordAfter.set_sensitive(boolean)
self.spbWarnBeforePasswordExpires.set_sensitive(boolean)
self.spbDisableAccountWhenPasswordExpires.set_sensitive(boolean)
if not boolean:
self.spbRequirePasswordAfter.set_value(0)
self.spbWarnBeforePasswordExpires.set_value(0)
self.spbDisableAccountWhenPasswordExpires.set_value(0)
def on_chkEnforceMinimumPasswordAge_toggled(self, widget):
if widget.get_active():
self.setEnforceMinimumPasswordAge(True)
else:
self.setEnforceMinimumPasswordAge(False)
def setEnforceMinimumPasswordAge(self, boolean=True):
self.spbEnforceMinimumPasswordAge.set_sensitive(boolean)
if not boolean:
self.spbEnforceMinimumPasswordAge.set_value(0)
def on_txtLoginName_changed(self, widget):
self.updateNewLogin()
def on_btnSaveUser_clicked(self, widget):
errMsgs = []
changed = False
name = self.txtLoginName.get_text().strip()
if name != "":
userExists = self.usr.doesUserExist(name)
realName = self.txtRealName.get_text().strip()
if realName != "":
if self.user['user'].pw_gecos == realName:
realName = ""
else:
print(">>> realName changed")
changed = True
prGroup = self.cmbHandlerPrimaryGroup.getValue()
if prGroup != "":
if self.user['prgrp'] == prGroup:
prGroup = ""
else:
print(">>> prGroup changed")
changed = True
elif not userExists:
errMsgs.append(_("You need to provide a primary group for a new user"))
home = self.txtHomeDirectory.get_text().strip()
facePath = ""
if home != "":
if exists(self.tempFace):
facePath = join(home, ".face")
if self.user['user'].pw_dir == home:
home = ""
else:
print(">>> home changed")
changed = True
groups = self.tvHandlerUserGroups.getToggledValues()
if functions.areListsEqual(groups, self.user['groups']):
groups = []
else:
print(">>> groups changed")
changed = True
password = self.txtPassword.get_text().strip()
if password != "":
encPwd = self.usr.encryptPassword(password)
if self.user['pwd'].sp_pwd == encPwd:
password = ""
else:
print(">>> password changed")
changed = True
elif not userExists:
errMsgs.append(_("You need to provide a password for a new user"))
else:
errMsgs.append(_("You need to provide a name for a new user"))
shell = self.cmbHandlerShells.getValue()
# TODO
#if self.radValidUntilDate.get_active():
#vuDay = self.spbValidUntilDay.get_value_as_int()
#vuMonth = self.cmbValidUntilMonth.get_active() + 1
#vuYear = self.spbValidUntilYear.get_value_as_int()
#(datetime(vuYear, vuMonth, vuDay, 0, 0, 0) - datetime.now()).days
#manageUser(self, user, primary_group="", group_list=[], shell="", home_dir="", full_name="", password="", expire_date="", inactive_days=""):
title = _("Save user settings")
if errMsgs:
msg = "\n".join(errMsgs)
self.showError(title, msg, self.windowUser)
elif changed or facePath != "":
err = 0
if changed:
err = self.usr.manageUser(user=name, full_name=realName, primary_group=prGroup, home_dir=home, group_list=groups, shell=shell, password=password)
if err == 0:
if exists(self.tempFace):
move(self.tempFace, facePath)
self.showInfo(title, _("User saved: %(user)s") % {"user": name}, self.windowUser)
self.refreshData()
else:
# TODO
pass
if exists(self.tempFace):
os.remove(self.tempFace)
self.windowUser.hide()
else:
self.showInfo(title, _("Nothing was changed on user: %(user)s") % {"user": name}, self.windowUser)
self.windowUser.hide()
if exists(self.tempFace):
os.remove(self.tempFace)
def on_cmbValidUntilMonth_changed(self, widget):
monthDays = functions.getDaysInMonth(widget.get_active() + 1)
adj = Gtk.Adjustment(value=1, lower=1, upper=monthDays, step_incr=1, page_incr=0, page_size=0)
self.spbValidUntilDay.set_adjustment(adj)
self.spbValidUntilDay.set_value(1)
def updateNewLogin(self):
txt = self.txtLoginName.get_text()
self.txtRealName.set_text(txt)
self.txtHomeDirectory.set_text("/home/%s" % txt)
self.cmbHandlerPrimaryGroup.setValue(txt)
def getUsers(self):
self.users = []
self.usersInfo = self.usr.getAllUsersInfoDict(self.chkShowSystemUsers.get_active())
for ui in self.usersInfo:
self.users.append([ui['face'], ui['user'].pw_name])
def showUserData(self):
# Show user groups in tvUserGroupsMain
self.selectedUser = self.tvHandlerUsersMain.getSelectedValue(1)
print((">>> selectedUser = %s" % self.selectedUser))
for ui in self.usersInfo:
if ui['user'].pw_name == self.selectedUser:
self.user = ui
break
self.tvHandlerUserGroupsMain.fillTreeview(contentList=self.user['groups'], columnTypesList=['str'])
self.userFace = self.usr.getUserFacePixbuf(self.user['user'].pw_name, None, 32)
def getUserGroupsComplete(self, userGroups=None):
ugc = []
for group in self.groups:
if group != self.user['prgrp']:
isUg = False
if userGroups is not None:
for ug in userGroups:
if ug == group:
isUg = True
break
ugc.append([isUg, group])
return ugc
def on_btnCancelUser_clicked(self, widget):
# Close add share window without saving
if exists(self.tempFace):
os.remove(self.tempFace)
self.windowUser.hide()
def on_ebFace_button_release_event(self, widget, data=None):
imagePath = SelectImageDialog(_('Select user image'), self.user['user'].pw_dir, self.windowUser).show()
if imagePath is not None:
ih = ImageHandler(imagePath)
ih.makeFaceImage(self.tempFace)
if exists(self.tempFace):
self.imgFace.set_from_pixbuf(ih.pixbuf)
def on_usermanagerUserWindow_delete_event(self, widget, data=None):
if exists(self.tempFace):
os.remove(self.tempFace)
self.windowUser.hide()
return True
def fillTreeViewUsers(self):
self.tvHandlerUsersMain.fillTreeview(contentList=self.users, columnTypesList=['GdkPixbuf.Pixbuf', 'str'], fixedImgHeight=48)
def on_ebFace_enter_notify_event(self, widget, data=None):
self.windowUser.get_window().set_cursor(Gdk.Cursor(Gdk.CursorType.HAND2))
def on_ebFace_leave_notify_event(self, widget, data=None):
self.windowUser.get_window().set_cursor(None)
# ===============================================
# Group functions
# ===============================================
def on_tvGroupsMain_cursor_changed(self, widget, event=None):
self.selectedGroup = self.tvHandlerGroupsMain.getSelectedValue()
self.selectedGroupAccounts = self.usr.getGroupAccounts(self.selectedGroup)
print((">>> selectedGroup 1 = %s" % str(self.selectedGroup)))
print((">>> selectedGroupAccounts 1 = %s" % str(self.selectedGroupAccounts)))
self.tvHandlerGroupAccounts.fillTreeview(self.selectedGroupAccounts, ['str'])
def on_tvGroupsMain_row_activated(self, widget, path, column):
self.on_btnGroupEdit_clicked(None)
def on_btnGroupAdd_clicked(self, widget):
self.selectedGroup = None
self.selectedGroupAccounts = None
print((">>> selectedGroup 2 = %s" % str(self.selectedGroup)))
print((">>> selectedGroupAccounts 2 = %s" % str(self.selectedGroupAccounts)))
self.txtGroupName.set_editable(True)
self.txtGroupName.set_can_focus(True)
self.txtGroupName.set_text("")
self.txtGroupID.set_text(str(self.usr.getNewGroupID()))
self.fillTreeViewAccounts(self.users)
self.windowGroup.show()
def on_btnGroupEdit_clicked(self, widget):
if self.selectedGroup is not None:
availableAccounts = []
groupAccounts = []
for acc in self.users:
if acc[1] in self.selectedGroupAccounts:
groupAccounts.append(acc)
else:
availableAccounts.append(acc)
self.fillTreeViewAccounts(availableAccounts, groupAccounts)
self.txtGroupName.set_text(self.selectedGroup)
self.txtGroupName.set_editable(False)
self.txtGroupName.set_can_focus(False)
self.txtGroupID.set_text(str(self.usr.getGroupID(self.selectedGroup)))
self.tvAccounts.grab_focus()
self.windowGroup.show()
def on_btnGroupRemove_clicked(self, widget):
# Remove group
if self.selectedGroup is not None:
title = _("Remove group")
if self.selectedGroup != self.loggedinUserPrimaryGroup:
qd = QuestionDialog(title, _("Are you sure you want to remove the following group:\n\n'%(group)s'") % { "group": self.selectedGroup }, self.windowGroup)
answer = qd.show()
if answer:
ret = self.usr.deleteGroup(self.selectedGroup)
if ret == "":
self.showInfo(title, _("Group successfully removed: %(group)s") % {"group": self.selectedGroup}, self.windowGroup)
self.refreshData()
else:
retMsg = "\n\n%s" % ret
self.showError(title, _("Could not remove group: %(group)s %(retmsg)s") % {"group": self.selectedGroup, "retmsg": retMsg}, self.windowGroup)
else:
self.showError(title, _("You cannot remove the currently logged in user's primary group"), self.windowGroup)
def on_btnAddAccount_clicked(self, widget):
rows = self.tvHandlerAccounts.getSelectedRows()
for row in rows:
self.tvHandlerAccounts.delRow(row[0])
self.tvHandlerSelectedAccounts.addRow(row[1])
def on_btnRemoveAccount_clicked(self, widget):
# TODO: check if you want to remove an account from its own primary group
rows = self.tvHandlerSelectedAccounts.getSelectedRows()
for row in rows:
acc = row[1][1]
pg = self.usr.getUserPrimaryGroupName(acc)
if pg != self.selectedGroup:
self.tvHandlerSelectedAccounts.delRow(row[0])
self.tvHandlerAccounts.addRow(row[1])
else:
self.showError(_("Remove account"), _("You cannot remove an account from its own primary group"), self.windowGroup)
def on_btnOkGroup_clicked(self, widget):
group = self.txtGroupName.get_text()
if group != "":
newSelAccs = self.tvHandlerSelectedAccounts.getColumnValues(1)
if self.selectedGroupAccounts is not None:
# New accounts in group
for a in newSelAccs:
if a not in self.selectedGroupAccounts:
# Add group to account
self.usr.addGroupToAccount(a, group)
# Removed accounts in group
for a in self.selectedGroupAccounts:
if a not in newSelAccs:
# Remove group from account
self.usr.removeGroupFromAccount(a, group)
else:
# Create new group
self.usr.createGroup(group)
# Add group to each selected account
for a in newSelAccs:
self.usr.addGroupToAccount(a, group)
# Close the user window
self.windowGroup.hide()
self.refreshData()
def on_btnCancelGroup_clicked(self, widget):
self.windowGroup.hide()
def on_usermanagerGroupWindow_delete_event(self, widget, data=None):
self.windowGroup.hide()
return True
def on_tvAccounts_row_activated(self, widget, path, column):
self.on_btnAddAccount_clicked(None)
def on_tvSelectedAccounts_row_activated(self, widget, path, column):
self.on_btnRemoveAccount_clicked(None)
def fillTreeViewAccounts(self, availableAccounts, selectedAccounts=None):
if (availableAccounts is None or not availableAccounts) and selectedAccounts:
# Dirty hack to prepare the available accounts treeview
self.tvHandlerAccounts.fillTreeview(contentList=selectedAccounts, columnTypesList=['GdkPixbuf.Pixbuf', 'str'], fixedImgHeight=24)
self.tvHandlerAccounts.clearTreeView()
else:
self.tvHandlerAccounts.fillTreeview(contentList=availableAccounts, columnTypesList=['GdkPixbuf.Pixbuf', 'str'], fixedImgHeight=24)
if (selectedAccounts is None or not selectedAccounts) and availableAccounts:
# Dirty hack to prepare the selected accounts treeview
self.tvHandlerSelectedAccounts.fillTreeview(contentList=availableAccounts, columnTypesList=['GdkPixbuf.Pixbuf', 'str'], fixedImgHeight=24)
self.tvHandlerSelectedAccounts.clearTreeView()
else:
self.tvHandlerSelectedAccounts.fillTreeview(contentList=selectedAccounts, columnTypesList=['GdkPixbuf.Pixbuf', 'str'], fixedImgHeight=24)
def fillTreeViewGroups(self):
self.tvHandlerGroupsMain.fillTreeview(contentList=self.groups, columnTypesList=['str'])
self.on_tvGroupsMain_cursor_changed(None)
# ===============================================
# General functions
# ===============================================
def refreshData(self):
self.getUsers()
self.groups = self.usr.getGroups()
self.fillTreeViewUsers()
self.fillTreeViewGroups()
self.showUserData()
def showInfo(self, title, message, parent):
MessageDialogSave(title, message, Gtk.MessageType.INFO, parent).show()
def showError(self, title, message, parent):
MessageDialogSave(title, message, Gtk.MessageType.ERROR, parent).show()
def filterText(self, widget):
def filter(entry, *args):
text = entry.get_text().strip().lower()
entry.set_text(''.join([i for i in text if i in '0123456789abcdefghijklmnopqrstuvwxyz-._']))
widget.connect('changed', filter)
# Close the gui
def on_usermanagerWindow_destroy(self, widget):
# Close the app
Gtk.main_quit()
class PartitionDialog(object):
def __init__(self, partition, mount_as, format_as):
self.partition = partition
# Load window and widgets
self.scriptName = basename(__file__)
self.scriptDir = abspath(dirname(__file__))
self.mediaDir = join(self.scriptDir, '../../share/live-installer-3')
self.builder = Gtk.Builder()
self.builder.add_from_file(join(self.mediaDir, 'live-installer-3-dialog.glade'))
# Main window objects
self.go = self.builder.get_object
self.window = self.go("dialog")
self.window.set_transient_for(installer.window)
self.window.set_destroy_with_parent(True)
self.window.set_modal(True)
self.window.set_title(_("Edit partition"))
self.loading = True
self.txt_label = self.go("txt_label")
self.cmb_mount_point = self.go("combobox_mount_point")
self.cmb_mount_point_handler = ComboBoxHandler(self.cmb_mount_point)
self.cmb_use_as = self.go("combobox_use_as")
self.cmb_use_as_handler = ComboBoxHandler(self.cmb_use_as)
# Translations
self.go("label_partition").set_markup("<b>%s</b>" % _("Device"))
self.go("label_use_as").set_markup(_("Format as"))
self.go("label_mount_point").set_markup(_("Mount point"))
self.go("label_label").set_markup(_("Label (optional)"))
self.go("chk_encryption").set_label(_("Encrypt partition"))
self.go("label_encryption_pwd").set_label(_("Password"))
# Show the selected partition path
self.go("label_partition_value").set_label(self.partition.path)
# Encryption
self.go("chk_encryption").set_active(partition.encrypt)
if partition.encrypt:
self.go("frm_partition_encryption").set_sensitive(True)
self.go("entry_encpass1").set_text(partition.enc_passphrase)
self.go("entry_encpass2").set_text(partition.enc_passphrase)
else:
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text('')
self.go("entry_encpass2").set_text('')
# Label
label_len = 16
if "fat" in partition.type or "fat" in partition.mount_as:
label_len = 11
self.txt_label.set_max_length(label_len)
self.txt_label.set_text(partition.label)
# Build list of pre-provided mountpoints
mounts = ",/,/home,/boot,/boot/efi,/srv,/tmp,/var,swap".split(',')
self.cmb_mount_point_handler.fillComboBox(mounts)
# Build supported filesystems list
filesystems = sorted(['', 'swap'] +
[fs[11:] for fs in getoutput('echo /sbin/mkfs.*').split()],
key=lambda x: 0 if x in ('', 'ext4') else 1 if x == 'swap' else 2)
self.cmb_use_as_handler.fillComboBox(filesystems)
self.cmb_mount_point_handler.selectValue(mount_as)
self.cmb_use_as_handler.selectValue(format_as)
# Connect builder signals and show window
self.builder.connect_signals(self)
self.window.show_all()
self.loading = False
def on_combobox_mount_point_changed(self, widget):
mount_as = self.cmb_mount_point_handler.getValue()
if "boot" in mount_as:
# Cannot encrypt
self.go("chk_encryption").set_active(False)
self.go("chk_encryption").set_sensitive(False)
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text('')
self.go("entry_encpass2").set_text('')
else:
# Can encrypt
self.loading = True
self.go("chk_encryption").set_sensitive(True)
self.loading = False
# Set label for OS
if self.txt_label.get_text() == '':
if mount_as == '/':
self.txt_label.set_text(get_release_name())
# swap
if mount_as == 'swap' and self.partition.type != 'swap':
self.cmb_use_as_handler.selectValue('swap')
# efi
if 'efi' in mount_as and 'fat' not in self.partition.type:
self.cmb_use_as_handler.selectValue('vfat')
def on_combobox_use_as_changed(self, widget):
format_as = self.cmb_use_as_handler.getValue()
mount_as = self.cmb_mount_point_handler.getValue()
if format_as == 'swap' and mount_as != 'swap':
self.cmb_mount_point_handler.selectValue('swap')
def on_button_cancel_clicked(self, widget):
# Close window without saving
self.window.hide()
def on_chk_encryption_toggled(self, widget):
if self.loading: return
if widget.get_active():
# Show warning message
mount_as = self.cmb_mount_point_handler.getValue()
if mount_as == '/':
encrypt = QuestionDialog(_("Encryption"),
_("You chose to encrypt the root partition.\n\n"
"You will need to mount /boot on a separate non-encrypted partition (500 MB).\n"
"Without a non-encrypted /boot partition your system will be unbootable.\n\n"
"Encryption will erase all data from {}\n\n"
"Are you sure you want to continue?").format(self.partition.path))
else:
encrypt = QuestionDialog(_("Encryption"),
_("Encryption will erase all data from {}\n\n"
"Are you sure you want to continue?").format(self.partition.path))
if encrypt:
format_as = self.cmb_use_as_handler.getValue()
if not format_as:
self.cmb_use_as_handler.selectValue('ext4')
self.go("frm_partition_encryption").set_sensitive(True)
self.go("entry_encpass1").set_text(self.partition.enc_passphrase)
self.go("entry_encpass2").set_text(self.partition.enc_passphrase)
self.go("entry_encpass1").grab_focus()
else:
widget.set_active(False)
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text("")
self.go("entry_encpass2").set_text("")
else:
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text("")
self.go("entry_encpass2").set_text("")
def on_entry_encpass1_changed(self, widget):
self.assign_enc_password()
def on_entry_encpass2_changed(self, widget):
self.assign_enc_password()
def assign_enc_password(self):
encryption_pwd1 = self.go("entry_encpass1").get_text()
encryption_pwd2 = self.go("entry_encpass2").get_text()
if(encryption_pwd1 == "" and encryption_pwd2 == ""):
self.go("image_enc_mismatch").hide()
else:
self.go("image_enc_mismatch").show()
if(encryption_pwd1 != encryption_pwd2):
self.go("image_enc_mismatch").set_from_stock(Gtk.STOCK_NO, Gtk.IconSize.BUTTON)
else:
self.go("image_enc_mismatch").set_from_stock(Gtk.STOCK_OK, Gtk.IconSize.BUTTON)
def on_button_ok_clicked(self, widget):
# Collect data
format_as = self.cmb_use_as_handler.getValue()
mount_as = self.cmb_mount_point_handler.getValue()
encrypt = self.go("chk_encryption").get_active()
enc_passphrase1 = self.go("entry_encpass1").get_text().strip()
enc_passphrase2 = self.go("entry_encpass2").get_text().strip()
label = self.txt_label.get_text().strip()
# Check user input
if encrypt:
errorFound = False
if enc_passphrase1 == "":
errorFound = True
errorMessage = _("Please provide an encryption password.")
elif enc_passphrase1 != enc_passphrase2:
errorFound = True
errorMessage = _("Your encryption passwords do not match.")
elif not format_as:
errorFound = True
errorMessage = "{} {}".format(_("You need to choose a format type\n"
"for your encrypted partition (default: ext4):"), self.partition.path)
self.cmb_use_as_handler.selectValue('ext4')
if not mount_as:
errorFound = True
errorMessage = "{} {}".format(_("You need to choose a mount point for partition:"), self.partition.path)
if errorFound:
ErrorDialog(_("Encryption"), errorMessage)
return True
else:
# For good measure
enc_passphrase1 = ''
# Save the settings and close the window
assign_mount_point(self.partition, mount_as, format_as, encrypt, enc_passphrase1, label)
self.window.hide()
def on_dialog_delete_event(self, widget, data=None):
self.window.hide()
return True
class PartitionDialog(object):
def __init__(self, partition, mount_as, format_as):
self.partition = partition
# Load window and widgets
self.scriptName = basename(__file__)
self.scriptDir = abspath(dirname(__file__))
self.mediaDir = join(self.scriptDir, '../../share/live-installer-3')
self.builder = Gtk.Builder()
self.builder.add_from_file(
join(self.mediaDir, 'live-installer-3-dialog.glade'))
# Main window objects
self.go = self.builder.get_object
self.window = self.go("dialog")
self.window.set_transient_for(installer.window)
self.window.set_destroy_with_parent(True)
self.window.set_modal(True)
self.window.set_title(_("Edit partition"))
self.loading = True
self.txt_label = self.go("txt_label")
self.cmb_mount_point = self.go("combobox_mount_point")
self.cmb_mount_point_handler = ComboBoxHandler(self.cmb_mount_point)
self.cmb_use_as = self.go("combobox_use_as")
self.cmb_use_as_handler = ComboBoxHandler(self.cmb_use_as)
# Translations
self.go("label_partition").set_markup("<b>%s</b>" % _("Device"))
self.go("label_use_as").set_markup(_("Format as"))
self.go("label_mount_point").set_markup(_("Mount point"))
self.go("label_label").set_markup(_("Label (optional)"))
self.go("chk_encryption").set_label(_("Encrypt partition"))
self.go("label_encryption_pwd").set_label(_("Password"))
# Show the selected partition path
self.go("label_partition_value").set_label(self.partition.path)
# Encryption
self.go("chk_encryption").set_active(partition.encrypt)
if partition.encrypt:
self.go("frm_partition_encryption").set_sensitive(True)
self.go("entry_encpass1").set_text(partition.enc_passphrase)
self.go("entry_encpass2").set_text(partition.enc_passphrase)
else:
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text('')
self.go("entry_encpass2").set_text('')
# Label
label_len = 16
if "fat" in partition.type or "fat" in partition.mount_as:
label_len = 11
self.txt_label.set_max_length(label_len)
self.txt_label.set_text(partition.label)
# Build list of pre-provided mountpoints
mounts = ",/,/home,/boot,/boot/efi,/srv,/tmp,/var,swap".split(',')
self.cmb_mount_point_handler.fillComboBox(mounts)
# Build supported filesystems list
filesystems = sorted(
['', 'swap'] +
[fs[11:] for fs in getoutput('echo /sbin/mkfs.*').split()],
key=lambda x: 0 if x in ('', 'ext4') else 1 if x == 'swap' else 2)
self.cmb_use_as_handler.fillComboBox(filesystems)
self.cmb_mount_point_handler.selectValue(mount_as)
self.cmb_use_as_handler.selectValue(format_as)
# Connect builder signals and show window
self.builder.connect_signals(self)
self.window.show_all()
self.loading = False
def on_combobox_mount_point_changed(self, widget):
mount_as = self.cmb_mount_point_handler.getValue()
if "boot" in mount_as:
# Cannot encrypt
self.go("chk_encryption").set_active(False)
self.go("chk_encryption").set_sensitive(False)
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text('')
self.go("entry_encpass2").set_text('')
else:
# Can encrypt
self.loading = True
self.go("chk_encryption").set_sensitive(True)
self.loading = False
# Set label for OS
if self.txt_label.get_text() == '':
if mount_as == '/':
self.txt_label.set_text(get_release_name())
# swap
if mount_as == 'swap' and self.partition.type != 'swap':
self.cmb_use_as_handler.selectValue('swap')
# efi
if 'efi' in mount_as and 'fat' not in self.partition.type:
self.cmb_use_as_handler.selectValue('vfat')
def on_combobox_use_as_changed(self, widget):
format_as = self.cmb_use_as_handler.getValue()
mount_as = self.cmb_mount_point_handler.getValue()
if format_as == 'swap' and mount_as != 'swap':
self.cmb_mount_point_handler.selectValue('swap')
def on_button_cancel_clicked(self, widget):
# Close window without saving
self.window.hide()
def on_chk_encryption_toggled(self, widget):
if self.loading: return
if widget.get_active():
# Show warning message
mount_as = self.cmb_mount_point_handler.getValue()
if mount_as == '/':
encrypt = QuestionDialog(
_("Encryption"),
_("You chose to encrypt the root partition.\n\n"
"You will need to mount /boot on a separate non-encrypted partition (500 MB).\n"
"Without a non-encrypted /boot partition your system will be unbootable.\n\n"
"Encryption will erase all data from {}\n\n"
"Are you sure you want to continue?").format(
self.partition.path))
else:
encrypt = QuestionDialog(
_("Encryption"),
_("Encryption will erase all data from {}\n\n"
"Are you sure you want to continue?").format(
self.partition.path))
if encrypt:
format_as = self.cmb_use_as_handler.getValue()
if not format_as:
self.cmb_use_as_handler.selectValue('ext4')
self.go("frm_partition_encryption").set_sensitive(True)
self.go("entry_encpass1").set_text(
self.partition.enc_passphrase)
self.go("entry_encpass2").set_text(
self.partition.enc_passphrase)
self.go("entry_encpass1").grab_focus()
else:
widget.set_active(False)
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text("")
self.go("entry_encpass2").set_text("")
else:
self.go("frm_partition_encryption").set_sensitive(False)
self.go("entry_encpass1").set_text("")
self.go("entry_encpass2").set_text("")
def on_entry_encpass1_changed(self, widget):
self.assign_enc_password()
def on_entry_encpass2_changed(self, widget):
self.assign_enc_password()
def assign_enc_password(self):
encryption_pwd1 = self.go("entry_encpass1").get_text()
encryption_pwd2 = self.go("entry_encpass2").get_text()
if (encryption_pwd1 == "" and encryption_pwd2 == ""):
self.go("image_enc_mismatch").hide()
else:
self.go("image_enc_mismatch").show()
if (encryption_pwd1 != encryption_pwd2):
self.go("image_enc_mismatch").set_from_stock(
Gtk.STOCK_NO, Gtk.IconSize.BUTTON)
else:
self.go("image_enc_mismatch").set_from_stock(
Gtk.STOCK_OK, Gtk.IconSize.BUTTON)
def on_button_ok_clicked(self, widget):
# Collect data
format_as = self.cmb_use_as_handler.getValue()
mount_as = self.cmb_mount_point_handler.getValue()
encrypt = self.go("chk_encryption").get_active()
enc_passphrase1 = self.go("entry_encpass1").get_text().strip()
enc_passphrase2 = self.go("entry_encpass2").get_text().strip()
label = self.txt_label.get_text().strip()
# Check user input
if encrypt:
errorFound = False
if enc_passphrase1 == "":
errorFound = True
errorMessage = _("Please provide an encryption password.")
elif enc_passphrase1 != enc_passphrase2:
errorFound = True
errorMessage = _("Your encryption passwords do not match.")
elif not format_as:
errorFound = True
errorMessage = "{} {}".format(
_("You need to choose a format type\n"
"for your encrypted partition (default: ext4):"),
self.partition.path)
self.cmb_use_as_handler.selectValue('ext4')
if not mount_as:
errorFound = True
errorMessage = "{} {}".format(
_("You need to choose a mount point for partition:"),
self.partition.path)
if errorFound:
ErrorDialog(_("Encryption"), errorMessage)
return True
else:
# For good measure
enc_passphrase1 = ''
# Save the settings and close the window
assign_mount_point(self.partition, mount_as, format_as, encrypt,
enc_passphrase1, label)
self.window.hide()
def on_dialog_delete_event(self, widget, data=None):
self.window.hide()
return True
