|
|
|
@ -3,243 +3,127 @@ import apt |
|
|
|
import apt_pkg |
|
|
|
import sys |
|
|
|
from ee.core.logging import Log |
|
|
|
from sh import apt_get |
|
|
|
|
|
|
|
|
|
|
|
class EEAptGet(): |
|
|
|
"""Generic apt-get intialisation""" |
|
|
|
|
|
|
|
def update(self): |
|
|
|
"""Similar to apt-get update""" |
|
|
|
|
|
|
|
# app.log.debug("Update cache") |
|
|
|
cache = apt.Cache() |
|
|
|
fprogress = apt.progress.text.AcquireProgress() |
|
|
|
iprogress = apt.progress.base.InstallProgress() |
|
|
|
cache.update() |
|
|
|
|
|
|
|
def upgrade(self, packages): |
|
|
|
"""Similar to apt-get update""" |
|
|
|
cache = apt.Cache() |
|
|
|
fprogress = apt.progress.text.AcquireProgress() |
|
|
|
iprogress = apt.progress.base.InstallProgress() |
|
|
|
my_selected_packages = [] |
|
|
|
# Cache Initialization |
|
|
|
if not cache: |
|
|
|
cache = apt.Cache() |
|
|
|
# Cache Read |
|
|
|
cache.open() |
|
|
|
for package in packages: |
|
|
|
pkg = cache[package] |
|
|
|
# Check Package Installed |
|
|
|
if pkg.is_installed: |
|
|
|
# Check Package is Upgradeble |
|
|
|
if pkg.is_upgradable: |
|
|
|
with cache.actiongroup(): |
|
|
|
# Mark Package for Upgrade |
|
|
|
pkg.mark_upgrade() |
|
|
|
my_selected_packages.append(pkg.installed) |
|
|
|
else: |
|
|
|
print("\'{package_name}-{package_ver.version}\'" |
|
|
|
"already the newest version" |
|
|
|
.format(package_name=pkg.name, |
|
|
|
package_ver=pkg.installed)) |
|
|
|
|
|
|
|
# Check if packages available for install. |
|
|
|
if len(my_selected_packages) > 0: |
|
|
|
print("The following packages will be upgraded:" |
|
|
|
"\n {pkg_name}" |
|
|
|
.format(pkg_name=my_selected_packages)) |
|
|
|
print("{pkg_install_count} newly installed." |
|
|
|
.format(pkg_upgrade_count=len(my_selected_packages))) |
|
|
|
print("Need to get {req_download} bytes of archives" |
|
|
|
.format(req_download=cache.required_download)) |
|
|
|
print("After this operation, {space} bytes of" |
|
|
|
"additional disk space will be used." |
|
|
|
.format(space=cache.required_space)) |
|
|
|
try: |
|
|
|
# Commit changes in cache (actually install) |
|
|
|
cache.commit(fprogress, iprogres) |
|
|
|
except Exception as e: |
|
|
|
print("package installation failed. [{err}]" |
|
|
|
.format(err=str(e))) |
|
|
|
return(False) |
|
|
|
return(True) |
|
|
|
try: |
|
|
|
apt_cache = apt.cache.Cache() |
|
|
|
apt_cache.update() |
|
|
|
success = (apt_cache.commit( |
|
|
|
apt.progress.text.AcquireProgress(), |
|
|
|
apt.progress.base.InstallProgress())) |
|
|
|
#apt_cache.close() |
|
|
|
return success |
|
|
|
except AttributeError as e: |
|
|
|
Log.error(self, 'AttributeError: ' + str(e)) |
|
|
|
|
|
|
|
def dist_upgrade(): |
|
|
|
apt_cache = apt.cache.Cache() |
|
|
|
apt_cache.update() |
|
|
|
apt_cache.open(None) |
|
|
|
apt_cache.upgrade(True) |
|
|
|
success = (apt_cache.commit( |
|
|
|
apt.progress.text.AcquireProgress(), |
|
|
|
apt.progress.base.InstallProgress())) |
|
|
|
#apt_cache.close() |
|
|
|
return success |
|
|
|
|
|
|
|
def install(self, packages): |
|
|
|
"""Installation of packages""" |
|
|
|
cache = apt.Cache() |
|
|
|
fprogress = apt.progress.text.AcquireProgress() |
|
|
|
iprogress = apt.progress.base.InstallProgress() |
|
|
|
my_selected_packages = [] |
|
|
|
# Cache Initialization |
|
|
|
if not cache: |
|
|
|
cache = apt.Cache() |
|
|
|
# Cache Read |
|
|
|
cache.open() |
|
|
|
|
|
|
|
for package in packages: |
|
|
|
try: |
|
|
|
pkg = cache[package] |
|
|
|
except KeyError as e: |
|
|
|
continue |
|
|
|
# Check Package Installed |
|
|
|
if pkg.is_installed or pkg.marked_install: |
|
|
|
# Check Package is Upgradeble |
|
|
|
if pkg.is_upgradable: |
|
|
|
print("latest version of \'{package_name}\' available." |
|
|
|
.format(package_name=pkg.installed)) |
|
|
|
def install_package(self, package_name): |
|
|
|
apt_pkg.init() |
|
|
|
# #apt_pkg.PkgSystemLock() |
|
|
|
apt_cache = apt.cache.Cache() |
|
|
|
pkg = apt_cache[package_name.strip()] |
|
|
|
if package_name.strip() in apt_cache: |
|
|
|
if pkg.is_installed: |
|
|
|
#apt_pkg.PkgSystemUnLock() |
|
|
|
Log.info(self, 'Trying to install a package that ' |
|
|
|
'is already installed (' + |
|
|
|
package_name.strip() + ')') |
|
|
|
#apt_cache.close() |
|
|
|
return False |
|
|
|
else: |
|
|
|
# Check if package already marked for install |
|
|
|
if not pkg.marked_install: |
|
|
|
print("\'{package_name}-{package_ver}\'" |
|
|
|
"already the newest version" |
|
|
|
.format(package_name=pkg.shortname, |
|
|
|
package_ver=pkg.installed)) |
|
|
|
else: |
|
|
|
with cache.actiongroup(): |
|
|
|
# Mark Package for Installation |
|
|
|
pkg.mark_install() |
|
|
|
my_selected_packages.append(pkg.name) |
|
|
|
|
|
|
|
# Check if packages available for install. |
|
|
|
if cache.install_count > 0: |
|
|
|
print("The following NEW packages will be installed:" |
|
|
|
"\n {pkg_name}" |
|
|
|
.format(pkg_name=my_selected_packages)) |
|
|
|
print("{pkg_install_count} newly installed." |
|
|
|
.format(pkg_install_count=cache.install_count)) |
|
|
|
print("Need to get {req_download} bytes of archives" |
|
|
|
.format(req_download=cache.required_download)) |
|
|
|
print("After this operation, {space:.2f} MB of" |
|
|
|
" additional disk space will be used." |
|
|
|
.format(space=cache.required_space/1e6)) |
|
|
|
try: |
|
|
|
# Commit changes in cache (actually install) |
|
|
|
cache.commit() |
|
|
|
except Exception as e: |
|
|
|
print("package installation failed. [{err}]" |
|
|
|
.format(err=str(e))) |
|
|
|
return(False) |
|
|
|
return(True) |
|
|
|
|
|
|
|
def remove(self, packages, auto=False, purge=False): |
|
|
|
def __dependencies_loop(cache, deplist, pkg, onelevel=True): |
|
|
|
""" Loops through pkg's dependencies. |
|
|
|
Returns a list with every package found. """ |
|
|
|
if onelevel: |
|
|
|
onelevellist = [] |
|
|
|
if not pkg.is_installed: |
|
|
|
return |
|
|
|
for depf in pkg.installed.dependencies: |
|
|
|
for dep in depf: |
|
|
|
# if (dep.name in cache and not cache[dep.name] |
|
|
|
# in deplist): |
|
|
|
# deplist.append(cache[dep.name]) |
|
|
|
# __dependencies_loop(cache, deplist, cache[dep.name]) |
|
|
|
# if onelevel: |
|
|
|
if dep.name in cache: |
|
|
|
if (cache[dep.name].is_installed and |
|
|
|
cache[dep.name].is_auto_installed): |
|
|
|
onelevellist.append(cache[dep.name]) |
|
|
|
# if onelevel: |
|
|
|
return onelevellist |
|
|
|
|
|
|
|
cache = apt.Cache() |
|
|
|
fprogress = apt.progress.text.AcquireProgress() |
|
|
|
iprogress = apt.progress.base.InstallProgress() |
|
|
|
|
|
|
|
onelevel = [] |
|
|
|
try: |
|
|
|
#apt_pkg.PkgSystemUnLock() |
|
|
|
result = apt_cache.commit() |
|
|
|
#apt_cache.close() |
|
|
|
return result |
|
|
|
except SystemError as e: |
|
|
|
Log.error(self, 'SystemError: ' + str(e)) |
|
|
|
#apt_cache.close() |
|
|
|
else: |
|
|
|
#apt_cache.close() |
|
|
|
Log.error(self, 'Unknown package selected (' + |
|
|
|
package_name.strip() + ')') |
|
|
|
|
|
|
|
my_selected_packages = [] |
|
|
|
# Cache Initialization |
|
|
|
if not cache: |
|
|
|
cache = apt.Cache() |
|
|
|
# Cache Read |
|
|
|
cache.open() |
|
|
|
for package in packages: |
|
|
|
print("processing", package) |
|
|
|
try: |
|
|
|
pkg = cache[package] |
|
|
|
except KeyError as e: |
|
|
|
Log.debug(self, "{0}".format(e)) |
|
|
|
continue |
|
|
|
if not pkg.is_installed: |
|
|
|
Log.info(self, "Package '{package_name}' is not installed," |
|
|
|
" so not removed." |
|
|
|
.format(package_name=pkg.name)) |
|
|
|
if not install_package(self, package): |
|
|
|
continue |
|
|
|
my_selected_packages.append(pkg.name) |
|
|
|
# How logic works: |
|
|
|
# 1) We loop trough dependencies's dependencies and add them to |
|
|
|
# the list. |
|
|
|
# 2) We sequentially remove every package in list |
|
|
|
# - via is_auto_installed we check if we can safely remove it |
|
|
|
deplist = [] |
|
|
|
onelevel = onelevel + __dependencies_loop(cache, deplist, pkg, |
|
|
|
onelevel=True) |
|
|
|
# Mark for deletion the first package, to fire up |
|
|
|
# auto_removable Purge? |
|
|
|
|
|
|
|
try: |
|
|
|
if purge: |
|
|
|
pkg.mark_delete(purge=True) |
|
|
|
else: |
|
|
|
pkg.mark_delete(purge=False) |
|
|
|
except SystemError as e: |
|
|
|
Log.debug(self, "{0}".format(e)) |
|
|
|
apt.ProblemResolver(cache).remove(pkg) |
|
|
|
# print(pkg.inst_state) |
|
|
|
# Log.error(self, "Unable to purge packages.") |
|
|
|
|
|
|
|
for dep in onelevel: |
|
|
|
my_selected_packages.append(dep.name) |
|
|
|
try: |
|
|
|
if purge: |
|
|
|
dep.mark_delete(purge=True) |
|
|
|
def remove(self, packages, auto=False, purge=False): |
|
|
|
def remove_package(self, package_name, purge=False): |
|
|
|
apt_pkg.init() |
|
|
|
#apt_pkg.PkgSystemLock() |
|
|
|
apt_cache = apt.cache.Cache() |
|
|
|
pkg = apt_cache[package_name.strip()] |
|
|
|
if package_name.strip() in apt_cache: |
|
|
|
if not pkg.is_installed: |
|
|
|
#apt_pkg.PkgSystemUnLock() |
|
|
|
Log.info(self, 'Trying to uninstall a package ' |
|
|
|
'that is not installed (' + |
|
|
|
package_name.strip() + ')') |
|
|
|
return False |
|
|
|
else: |
|
|
|
dep.mark_delete(purge=False) |
|
|
|
except SystemError as e: |
|
|
|
Log.debug(self, "{0}".format(e)) |
|
|
|
Log.error(self, "Unable to purge depedencies.") |
|
|
|
try: |
|
|
|
pkg.mark_delete(purge) |
|
|
|
except SystemError as e: |
|
|
|
Log.debug(self, 'SystemError: ' + str(e)) |
|
|
|
return False |
|
|
|
try: |
|
|
|
#apt_pkg.PkgSystemUnLock() |
|
|
|
result = apt_cache.commit() |
|
|
|
#apt_cache.close() |
|
|
|
return result |
|
|
|
except SystemError as e: |
|
|
|
Log.debug(self, 'SystemError: ' + str(e)) |
|
|
|
return False |
|
|
|
#apt_cache.close() |
|
|
|
else: |
|
|
|
#apt_cache.close() |
|
|
|
Log.error(self, 'Unknown package selected (' + |
|
|
|
package_name.strip() + ')') |
|
|
|
|
|
|
|
# Check if packages available for remove/update. |
|
|
|
if cache.delete_count > 0: |
|
|
|
# app.log.debug('packages will be REMOVED ') |
|
|
|
print("The following packages will be REMOVED:" |
|
|
|
"\n {pkg_name}" |
|
|
|
.format(pkg_name=my_selected_packages)) |
|
|
|
print("{pkg_remove_count} to remove." |
|
|
|
.format(pkg_remove_count=cache.delete_count)) |
|
|
|
# app.log.debug('bytes disk space will be freed') |
|
|
|
print("After this operation, {space:.2f} MB disk space " |
|
|
|
"will be freed.".format(space=cache.required_space/1e6)) |
|
|
|
try: |
|
|
|
cache.commit(fprogress, iprogress) |
|
|
|
except Exception as e: |
|
|
|
# app.log.error('Sorry, package installation failed ') |
|
|
|
print("Sorry, package installation failed [{err}]" |
|
|
|
.format(err=str(e))) |
|
|
|
return(False) |
|
|
|
return(True) |
|
|
|
for package in packages: |
|
|
|
if not remove_package(self, package, purge=purge): |
|
|
|
continue |
|
|
|
|
|
|
|
def is_installed(self, package): |
|
|
|
cache = apt.Cache() |
|
|
|
fprogress = apt.progress.text.AcquireProgress() |
|
|
|
iprogress = apt.progress.base.InstallProgress() |
|
|
|
def auto_clean(self): |
|
|
|
try: |
|
|
|
apt_get.autoclean("-y") |
|
|
|
except ErrorReturnCode as e: |
|
|
|
Log.debug(self, "{0}".format(e)) |
|
|
|
Log.error(self, "Unable to apt-get autoclean") |
|
|
|
|
|
|
|
# Cache Initialization |
|
|
|
if not cache: |
|
|
|
cache = apt.Cache() |
|
|
|
# Cache Read |
|
|
|
cache.open() |
|
|
|
def auto_remove(self): |
|
|
|
try: |
|
|
|
pkg = cache[package] |
|
|
|
# Check Package Installed |
|
|
|
if pkg.is_installed: |
|
|
|
return True |
|
|
|
else: |
|
|
|
return False |
|
|
|
except KeyError as e: |
|
|
|
Log.debug(self, "Running apt-get autoremove") |
|
|
|
apt_get.autoremove("-y") |
|
|
|
except ErrorReturnCode as e: |
|
|
|
Log.debug(self, "{0}".format(e)) |
|
|
|
except Exception as e: |
|
|
|
return False |
|
|
|
Log.error(self, "Unable to apt-get autoremove") |
|
|
|
|
|
|
|
def is_installed(self, package_name): |
|
|
|
apt_cache = apt.cache.Cache() |
|
|
|
apt_cache.open() |
|
|
|
if (package_name.strip() in apt_cache and |
|
|
|
apt_cache[package_name.strip()].is_installed): |
|
|
|
apt_cache.close() |
|
|
|
return True |
|
|
|
#apt_cache.close() |
|
|
|
return False |
|
|
|
|
harshadyeola
10 years ago