def link_clean(self):
for name in glob.iglob(os.path.join(self.home, '?:')):
try:
if os.path.islink(name):
os.remove(name)
except Exception as e:
Utils.exc("Unable to delete symbolic link {0}".format(name), e)
def install(self, work):
changes = self.cache.get_changes()
if len(changes) > 0:
work.set('install', 200)
log.write(log.DEBUG, "Installing apps: {0}".format(changes))
try:
self.cache.commit()
except Exception as e:
Utils.exc("Failed application installation", e)
self.refresh()
else:
work.set('install', 0)
work.set('appscripts', len(self.app_scripts))
self.process_scripts(work, self.app_scripts)
work.set('icons', len(self.icons))
self.icons_mkdir()
for location in apps_icons:
self.icons_store(apps_icons[location], location)
for icon in self.icons:
idx = work.update("Configurant icones de '{0}'".format(icon[0]))
self.icon_install(icon)
work.complete(idx)
work.set('scripts', len(self.scripts))
self.process_scripts(work, self.scripts)
def icons_mkdir(self, *path):
try:
path = os.path.join(self.base, *path)
if not os.path.exists(path):
os.mkdir(path)
return path
except Exception as e:
Utils.exc("Unable to create directory '{0}'".format(path), e)
return None
def main():
log.write(log.DEBUG, "Args: {0}".format(sys.argv))
data = pwd.getpwnam(smb.user)
progress = Progress(data.pw_uid)
progress.start()
printers = Printers(smb.user)
drives = Drives(smb.domain, data.pw_uid, data.pw_gid, data.pw_dir)
apps = Appliations(data.pw_dir)
dc_count = len(dns.resolve(dns.domain, 'A'))
is_login = (os.getenv('PAM_TYPE') != 'close_session') and (dc_count > 0)
if is_login:
for item in clean_files:
try:
log.write(log.DEBUG, "Removing file '{0}'".format(item))
file_to_remove = os.path.join(data.pw_dir, item)
if os.path.exists(file_to_remove):
os.remove(file_to_remove)
except Exception as e:
Utils.exc("Unable to remove file '{0}'".format(item), e)
if is_requested('apps') and is_login:
for location in user_icons:
apps.icons_store(
None, os.path.join(data.pw_dir, user_icons[location][0]),
user_icons[location][1])
work = progress.add('Apps', 500, "Configurant aplicacions")
subtask = core.call_delayed(None, apps.task_install, work)
core.call_cleanup(subtask, work_finish, work)
if is_requested('printers'):
work = progress.add('PrintersDel', 100, "Eliminant impressores")
subtask = core.call_delayed(None, printers.task_remove, work)
core.call_cleanup(subtask, work_finish, work)
if is_login:
work = progress.add('PrintersAdd', 200, "Creant impressores")
subtask = core.call_delayed(subtask, printers.task_create, work)
core.call_cleanup(subtask, work_finish, work)
if is_requested('drives'):
work = progress.add('DrivesDel', 100, "Eliminant unitats")
subtask = core.call_delayed(None, drives.task_umount, work)
core.call_cleanup(subtask, work_finish, work)
if is_login:
core.sync_point_add()
work = progress.add('DrivesAdd', 100, "Creant unitats")
subtask = core.call_delayed(subtask, drives.task_mount, work)
core.call_cleanup(subtask, work_finish, work, True)
if is_login:
core.call(None, user_init)
core.call(None, ws_init)
def icons_store(self, iid, path, prefix=''):
try:
log.write(log.DEBUG, "Store icons: {0} ({1}, {2})".format(path, prefix, iid))
if iid:
if self.icons_prepare(iid) is None:
return False
path = os.path.abspath(path)
for root, subdirs, files in os.walk(path):
for item in files:
if item[-8:] != '.desktop':
continue
NoDisplay_true = False
NoDisplay_false = False
src = os.path.join(root, item)
with open(src, 'r') as f:
if self.re_NoDisplay_true.search(f.read()) is not None:
NoDisplay_true = True
log.write(log.DEBUG, "Icona {0} visible en {1}".format(item, path))
elif self.re_NoDisplay_false.search(f.read()) is not None:
NoDisplay_false = True
log.write(log.DEBUG, "Icona {0} oculta en {1}".format(item, path))
else:
log.write(log.DEBUG, "Icona {0} visible en {1} per omissió del paràmetre NoDisplay".format(item, path))
if NoDisplay_true:
continue
elif NoDisplay_false:
log.write(log.DEBUG, "Amagant icona {0} ...".format(src))
src_tmp = src + ".tmp"
with open(src, 'r') as f_source:
with open(src_tmp, 'w+') as f_temp:
for line in f_source:
f_temp.write(line.replace("NoDisplay=false", "NoDisplay=true"))
os.remove(src)
os.rename(src_tmp, src)
else:
log.write(log.DEBUG, "Marcant icona {0} com a oculta ...".format(src))
src_tmp = src + ".tmp"
NoDisplay_trobat = False
with open(src, 'r') as f_source:
with open(src_tmp, 'w+') as f_temp:
for line in f_source:
f_temp.write(line)
if "[Desktop Entry]" in line:
f_temp.write("NoDisplay=true\n")
NoDisplay_trobat = True
if NoDisplay_trobat == False:
f_temp.write("NoDisplay=true\n")
os.remove(src)
os.rename(src_tmp, src)
except Exception as e:
Utils.exc("Unable to store icons: {0} ({1}, {2})".format(path, prefix, iid), e)
return None
def refresh(self):
os.environ['PATH'] = '/usr/sbin:/usr/bin:/sbin:/bin'
os.environ['DEBIAN_FRONTEND'] = 'noninteractive'
try:
self.cache = apt.Cache()
self.cache.update()
except Exception as e:
Utils.exc("Unable to update repositories", e)
self.cache = apt.Cache()
self.cache.open()
def mod_desktop_file(self, filename, visible):
try:
config = ConfigParser.RawConfigParser()
config.optionxform = str
config.read(filename)
if not config.has_section('Desktop Entry'):
config.add_section('Desktop Entry')
config.set("Desktop Entry", "NoDisplay",
"{0}".format(not visible).lower())
with open(filename, 'wb') as desktopfile:
config.write(desktopfile)
with open(filename, 'r') as desktopfile:
filedata = desktopfile.read()
filedata = filedata.replace(' = ', '=')
with open(filename, 'w') as desktopfile:
desktopfile.write(filedata)
except Exception as e:
Utils.exc("Unable mod desktop file '{0}'".format(filename), e)
return None
def dir_clean(self, path, create=False):
log.write(log.DEBUG, "Cleaning directory {0}".format(path))
if os.path.ismount(path):
log.write(log.DEBUG, "Directory {0} is a mount point".format(path))
return self.umount(path)
if not create:
return True
if os.path.isdir(path):
log.write(log.DEBUG, "Directory {0} already exists".format(path))
return True
if os.path.exists(path):
log.write(log.ERR, "Entry {0} already exists and it's not a directory".format(path))
return False
try:
os.mkdir(path)
except Exception as e:
Utils.exc("Unable to create directory {0}".format(path), e)
return False
log.write(log.INFO, "Directory {0} created".format(path))
return True
def icons_store(self, iid, path, prefix = ''):
log.write(log.DEBUG, "Store icons: {0} ({1}, {2})".format(path, prefix,
iid))
if iid:
if self.icons_prepare(iid) is None:
return False
path = os.path.abspath(path)
for root, subdirs, files in os.walk(path):
for item in files:
if item[-8:] != '.desktop':
continue
src = os.path.join(root, item)
with open(src, 'r') as f:
if self.re_disp.search(f.read()) is not None:
log.write(log.DEBUG,
"Skipped hidden icon '{0}'".format(src))
continue
if iid:
tgt = self.icons_prepare(iid, os.path.relpath(root, path))
if tgt is None:
continue
dst = os.path.join(tgt, item)
log.write(log.DEBUG, "Moving icon {0}".format(src))
try:
shutil.move(src, dst)
except Exception as e:
Utils.exc(("Failed to move icon '{0}' "
"to '{1}'").format(src, dst), e)
elif item.startswith(prefix):
log.write(log.DEBUG, "Deleting icon {0}".format(src))
try:
os.remove(src)
except Exception as e:
Utils.exc("Failed to delete icon '{0}'".format(src), e)
def create_link(self, path, link):
log.write(log.DEBUG, "Creating link {0} -> {1}".format(link, path))
dirname, filename = link.rsplit('/', 1)
if os.path.islink(link):
src = os.path.join(dirname, path)
try:
dst = os.path.join(dirname, os.readlink(link))
except Exception as e:
Utils.exc("Failed to read symbolic link {0}".format(link), e)
return False
if src == dst:
log.write(log.DEBUG, "Link {0} already exists".format(link))
return True
log.write(log.WARN, "Link {0} points to {1}".format(link, dst))
try:
os.remove(link)
except Exception as e:
Utils.exc("Unable to remove link {0}".format(link), e)
return False
log.write(log.INFO, "Link {0} removed".format(link))
if os.path.exists(link):
log.write(log.WARN,
"Entry {0} exists but it's not a link".format(link))
now = datetime.now()
tgt = ('{0}_{1:04d}{2:02d}{3:02d}_'
'{4:02d}{5:02d}{6:02d}{7:06d}').format(filename, now.year,
now.month, now.day,
now.hour, now.minute,
now.second,
now.microsecond)
try:
os.rename(link, os.path.join(dirname, tgt))
except Exception as e:
Utils.exc("Unable to rename directory {0} to {1}".format(link,
tgt),
e)
return False
try:
os.symlink(path, link)
except Exception as e:
Utils.exc("Unable to create link {0}".format(link), e)
return False
log.write(log.INFO, "Link {0} created".format(link))
return True
def icons_store(self, iid, path, prefix=''):
try:
log.write(log.DEBUG,
"Store icons: {0} ({1}, {2})".format(path, prefix, iid))
if iid:
if self.icons_prepare(iid) is None:
return False
path = os.path.abspath(path)
for root, subdirs, files in os.walk(path):
for item in files:
if item[-8:] != '.desktop':
continue
src = os.path.join(root, item)
with open(src, 'r') as f:
if self.re_disp.search(f.read()) is not None:
log.write(log.DEBUG,
"Skipped hidden icon '{0}'".format(src))
continue
if iid:
tgt = self.icons_prepare(iid,
os.path.relpath(root, path))
if tgt is None:
continue
dst = os.path.join(tgt, item)
log.write(log.DEBUG, "Copying icon {0}".format(src))
try:
shutil.copy(src, dst)
os.chmod(dst, 0755)
except Exception as e:
Utils.exc(
"Failed to copy a icon '{0}' to '{1}'".format(
src, dst), e)
log.write(log.DEBUG,
"Hidding icon {0} after copying".format(src))
self.mod_desktop_file(src, False)
elif item.startswith(prefix):
log.write(
log.DEBUG,
"Hidding custom (_imi_) icon {0}".format(src))
try:
self.mod_desktop_file(src, False)
except Exception as e:
Utils.exc("Failed to hide icon '{0}'".format(src),
e)
except Exception as e:
Utils.exc("Failed in 'icons_store' '{0}'".format(path), e)
def icon_install(self, icon):
if len(icon) < 2:
log.write(log.ERR, "Insufficient arguments for an 'icon' action")
return
app = icon[0]
log.write(log.DEBUG, "Adding icons for '{0}'".format(app))
if (app not in self.cache) or not self.cache[app].is_installed:
log.write(log.WARN, "Application not available")
return
avail = []
for name in self.cache[app].installed_files:
if not name.endswith('.desktop'):
continue
for location in apps_icons:
if name.startswith(location):
avail.append('[{0}]{1}'.format(
apps_icons[location],
os.path.relpath(name[:-8], location)))
tmp = avail
log.write(log.DEBUG, "Available icons: {0}".format(avail))
log.write(log.DEBUG, "Icon filter: {0}".format(icon[2:]))
if len(icon) > 2:
tmp = []
for name in icon[2:]:
match = self.re_icon.match(name)
if match is None:
log.write(log.ERR,
"Invalid icon reference: '{0}'".format(name))
continue
if match.group(1) is None:
base = 'common'
else:
base = match.group(1)
key = '[{0}]{1}'.format(base, match.group(2))
if key in avail:
tmp.append(key)
else:
log.write(log.WARN, (
"Icon '{0}' not available for application {1}").format(
key, app))
names = []
for name in tmp:
match = self.re_icon.match(name)
names.append([match.group(1), match.group(2)])
log.write(log.DEBUG, "Filtered list of icons: {0}".format(names))
place = icon[1].lower()
paths = []
for location in user_icons:
if (place == 'all') or (place == location):
paths.append([
os.path.join(self.home, user_icons[location][0]),
user_icons[location][1]
])
log.write(log.DEBUG, "Installation paths: {0}".format(paths))
for path in paths:
for data in names:
base = data[0]
name = data[1] + '.desktop'
tgt = os.path.join(path[0], os.path.dirname(name),
path[1] + os.path.basename(name))
log.write(log.DEBUG,
"Installing icon '{0}' to {1}".format(name, tgt))
try:
shutil.copy(os.path.join(self.base, base, name), tgt)
os.chmod(tgt, 0755)
except Exception as e:
Utils.exc("Unable to install icon '{0}'".format(name), e)