-
Notifications
You must be signed in to change notification settings - Fork 3
/
__init__.py
271 lines (263 loc) · 15.7 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
import argparse
import logging
import time
import traceback
from .data_manager import UnZipper, Compiler
from .exceptions import (LinuxSystemNotFound,
RootPrivilegesNotGiven,
raiserModuleNotFound,
NotEnoughFreeSpaceAvailable
)
from .net.DependenciesInstaller import Dependencies
from .net.Downloader import Downloader
from .net.KernelVersions import KernelVersions
from .net.PageInfo import Connection
from .utils import (isRunningLinux,
isUserAdmin,
getLinuxVersion,
getFreeSpaceAvailable,
cleanupSpace,
isNewVersionAvailable,
cleanupOldLogs,
printKernelTable
)
from .utils.anim import Animation
from .utils.colors import OutputColors as Colors
from .utils.logger import setup_logging
from .values.Constants import OP_REPO_URL, EXU_USAGE, OP_VERSION, LOG_FILE_PATH, LOG_FILENAME, LOG_COMPILER_FILENAME
__program_name = """Kernel Upgrader for Linux"""
__program_description = """Download, compile and install the latest stable kernel for your Linux system. Automate
this tool for upgrading your kernel periodically"""
__program_version = "Current running version: " + OP_VERSION + " - " + OP_REPO_URL
def application(arg):
usage = arg.usage
show_version = arg.version
check_kernel_updates = arg.check
cores = arg.cores if arg.cores != 0 else None
interactive = arg.interactive
if usage:
print(EXU_USAGE)
elif show_version:
print(__program_version)
elif check_kernel_updates:
if not isRunningLinux():
raise LinuxSystemNotFound(Colors.FAIL +
"Your OS is not running under a Linux installation. It is not possible"
" to update the kernel" + Colors.ENDC)
else:
from packaging import version
info = Connection()
current_version = getLinuxVersion()
new_version = info.getLatestVersionCode()
if version.parse(current_version) >= version.parse(new_version):
print(Colors.FAIL + "You already have the latest version: " + current_version + Colors.ENDC)
exit(1)
else:
print(Colors.OKGREEN + "There is a new version available: " + Colors.ENDC + Colors.BOLD + new_version +
Colors.ENDC + "\n\nRun this program without \"-c\" (\"--check\") option to upgrade")
exit(0)
else:
if isNewVersionAvailable():
print(Colors.HEADER + "New version available" + Colors.ENDC + Colors.OKBLUE + " | Download it with pip"
" or go to this URL: "
+ Colors.ENDC + Colors.UNDERLINE + OP_REPO_URL + Colors.ENDC + "\n")
time.sleep(8)
if not isUserAdmin():
raise RootPrivilegesNotGiven(Colors.FAIL + "This application needs root rights in order to work properly."
" Run with \"-u\" option to get more information" + Colors.ENDC)
else:
cleanupOldLogs()
setup_logging("kernel_logging", LOG_FILE_PATH + LOG_FILENAME)
setup_logging("compiler_logging", LOG_FILE_PATH + LOG_COMPILER_FILENAME)
__log = logging.getLogger("kernel_logging")
animator = Animation(0.1)
try:
if not isRunningLinux():
__log.error("OS is not under a Linux installation. Aborting kernel upgrade...")
raise LinuxSystemNotFound(Colors.FAIL +
"Your OS is not running under a Linux installation. It is not possible"
" to update the kernel" + Colors.ENDC)
else:
__log.info("Checking for free space available...")
free_space = float(getFreeSpaceAvailable())
if free_space < 20:
__log.error("There is not enough free space available. Current free space (in GB): "
+ str(free_space))
raise NotEnoughFreeSpaceAvailable(Colors.FAIL + "There is not enough free space available on "
"drive which mounts \"/home\" 20GB are needed "
"at least" + Colors.ENDC)
__log.info("There is enough free space available. Current free space: " + str(free_space) + " GB")
if cores is not None:
__log.info("Using predefined number of cores: " + str(cores))
else:
__log.info("Using default number of cores (system available cores)")
__log.info("Starting kernel compiling")
__log.debug("Checking versions")
current_version = getLinuxVersion()
__log.info("Current version detected: " + current_version)
from packaging import version
if not interactive:
__log.info("Running in non-interactive mode")
info = Connection()
new_version = info.getLatestVersionCode()
version_url = info.getLatestVersionURL()
else:
__log.info("Running in interactive mode")
interactive_mode = KernelVersions()
available_versions = interactive_mode.obtain_current_available_kernels()
__log.info("Available versions:\n" + str(available_versions))
supported_versions = []
for release in available_versions:
release_version = release.get("release_version", None)
release_version = release_version.replace("[EOL]", '')
if version.parse(current_version) < version.parse(release_version) \
and not version.parse(release_version).is_prerelease:
supported_versions.append(release)
if len(supported_versions) == 0:
__log.debug("The version installed is the same or greater than the available one. "
"Current version: " + current_version)
print(Colors.FAIL + "You already have the latest version" + Colors.ENDC)
exit(1)
__log.info("Supported versions (higher than the current one):\n" + str(supported_versions))
printKernelTable(['#', "Release type", "Version", "Date"], supported_versions)
is_index_correct = False
index = -1
while not is_index_correct:
index = int(input("Number of the version to install: "))
if index < len(supported_versions):
is_index_correct = True
else:
print(Colors.FAIL + "Please, select a number of the version to install. Those numbers "
"are the first ones that appears next to the text identifying "
"the kernel release type" + Colors.ENDC)
chosen_version = supported_versions[index]
new_version = chosen_version["release_version"]
version_url = chosen_version["release_url"]
if not interactive and version.parse(current_version) >= version.parse(new_version):
__log.debug("The version installed is the same or greater than the available one. "
"Current version: " + current_version + " | Available version: " + new_version)
print(Colors.FAIL + "You already have the latest version" + Colors.ENDC)
exit(1)
else:
__log.debug("There is a new version available - new kernel upgrade process started")
if not interactive:
print(Colors.OKGREEN + "There is a new version available." +
Colors.ENDC)
print(Colors.OKBLUE + "Downloading new version... " + Colors.ENDC + "| New version: " +
new_version)
__log.debug("Starting new version download... | New version: " + new_version)
__log.info("Downloading from: " + version_url)
downloader = Downloader(version_url, new_version)
download_path, current_date = downloader.startDownload()
__log.info("Downloaded to path: \"" + download_path + "\"")
__log.debug("Starting dependencies installation...")
print(Colors.OKBLUE + "Installing required dependencies... " + Colors.ENDC)
Dependencies.installRequiredDependencies()
__log.debug("Required dependencies installed/satisfied")
__log.debug("Starting kernel decompression")
animator.animate(Colors.OKBLUE + "Decompressing downloaded kernel..." + Colors.ENDC,
Colors.OKBLUE)
unzipper = UnZipper(download_path)
kernel_folder = unzipper.unzip()
animator.stop()
__log.debug("Finished kernel decompression")
time.sleep(1)
__log.debug("Starting kernel compilation...")
__log.debug("Cleaning up space of old kernels")
print(Colors.OKBLUE + "Copying old configuration & cleaning up space..." + Colors.ENDC)
compiler = Compiler(kernel_folder, new_version, current_date, cores)
__log.debug("Copying old kernel boot config")
if compiler.copy_latest_config():
__log.debug("Copied old kernel boot configuration")
__log.debug("Adapting latest config for the new kernel version")
animator.animate(Colors.OKBLUE + "Adapting old configuration to the new kernel..."
+ Colors.ENDC, Colors.OKBLUE)
compiler.adaptOldConfig()
animator.stop()
__log.debug("Adapted old kernel configuration to the newer version")
time.sleep(1)
__log.debug("Performing kernel compilation...")
print(Colors.OKBLUE + "Starting kernel compilation..." + Colors.ENDC)
print(Colors.WARNING + "This process will take a long time to finish. You can do it "
"in background by pressing \"Ctrl + Z\" and then, type \"bg\" at"
" your terminal. To resume, just type \"fg\"." + Colors.ENDC)
animator.animate(Colors.BOLD + "Performing kernel compilation..." + Colors.ENDC,
Colors.BOLD)
compiler.compileKernel()
animator.stop()
__log.debug("Kernel compilation finished")
time.sleep(1)
__log.debug("Starting kernel installation...")
animator.animate(Colors.OKBLUE + "Installing the new kernel..." + Colors.ENDC,
Colors.OKBLUE)
compiler.installKernel()
animator.stop()
__log.debug("Finished correctly kernel installation. New version installed: " + new_version)
time.sleep(1)
print(Colors.OKGREEN + "Kernel completely installed. Now you should reboot in order to "
"apply changes. New version: " + new_version + Colors.ENDC)
__log.debug("Cleaning-up space for downloaded & compiled files")
animator.animate(Colors.UNDERLINE + "Cleaning up used space..." + Colors.ENDC, None)
cleanupSpace()
animator.stop()
__log.debug("Space cleaned-up")
time.sleep(1)
exit(0)
else:
__log.error("Impossible to copy latest kernel configuration. Aborting...")
except ImportError as e:
raiserModuleNotFound(e)
except KeyboardInterrupt:
animator.force_stop()
print("\n")
print(Colors.FAIL + "User pressed Ctrl + C - stopping..." + Colors.ENDC)
__log.error("User pressed keyboard interrupt. Stopping program...")
exit(2)
except Exception as e:
print("Ooops!! There was an unexpected error :O\n\nPlease, refer to: "
">> https://github.com/Javinator9889/KernelUpgrader/issues << and submit the following red code:")
print(Colors.FAIL)
print(e)
traceback.print_exc()
print(Colors.ENDC)
print("Also you can include the logs that are found at: \n"
"\t/var/log/kernel_upgrader.log & \n"
"\t/var/log/kernel_upgrader.compiler.log")
animator.force_stop()
__log.error("Exception catch | " + str(e))
exit(3)
def main():
arguments = argparse.ArgumentParser(prog=__program_name,
description=__program_description,
epilog=__program_version)
arguments.add_argument("-u",
"--usage",
action="store_true",
help="show full usage of this program")
arguments.add_argument("-v",
"--version",
action="store_true",
help="show program version")
arguments.add_argument("-c",
"--check",
action="store_true",
help="only checks if there is any new version available")
arguments.add_argument("-j",
"--cores",
type=int,
default=0,
help="use a predefined number of cores for thread compilation. Default: 0 (uses the number"
"of cores available on your system)")
arguments.add_argument("--interactive",
action="store_true",
help="launches the KernelUpgrader tool with interactive mode for selecting the kernel "
"you want to install")
args = arguments.parse_args()
try:
application(args)
except Exception as e:
print(e)
traceback.print_exc()
exit(-1)
if __name__ == '__main__':
main()