Merge pull request #1 from Tracks12:dev

Dev

Author: Tracks12

core/__init__.py

@@ -14,7 +14,8 @@
 	"version": "0.1"
 })
 
-REGEX_ARGS = str("\\s(?=(?:[^\"'`]*[\"'`][^\"'`]*[\"'`])*[^\"'`]*$)")
+REGEX_ARGS	= str("\\s(?=(?:[^\"'`]*[\"'`][^\"'`]*[\"'`])*[^\"'`]*$)")
+UNITS		= tuple[str](("o", "ko", "Mo", "Go", "To"))
 
 def helper(commands: tuple) -> None:
 	colors = tuple[str]((Colors.cyan, Colors.yellow, Colors.red))

libs/wslbuilder/archlinux.ico

@@ -103,16 +103,20 @@ def config(cfg: Config) -> bool:
 				print(f"{Icons.warn}{e}")
 
 		elif(args[0] in commands[1][0]):
-			match(args[1]):
-				case "encode":
-					print(cfg.getEncoding())
+			try:
+				match(args[1]):
+					case "encode":
+						print(cfg.getEncoding())
 
-				case "splash":
-					print(cfg.getSplash())
+					case "splash":
+						print(cfg.getSplash())
+
+					case "all":
+						print(f"encode: {cfg.getEncoding()}")
+						print(f"splash: {cfg.getSplash()}")
 
-				case "all":
-					print(f"encode: {cfg.getEncoding()}")
-					print(f"splash: {cfg.getSplash()}")
+			except IndexError:
+				print(f"{Icons.warn}No value was entered !")
 
 		elif(args[0] in commands[-2][0]):
 			helper(commands)

libs/wslbuilder/debian.ico

@@ -2,7 +2,9 @@
 # -*- coding: utf-8 -*-
 
 from os import system as shell
+from traceback import format_exc
 
+from core.icons import Icons
 from core.tool import Tool
 
 class Shell(Tool):
@@ -24,5 +26,25 @@ def __init__(self, args: list[str]):
 
 		self._run(args)
 
+	def _run(self, args: list[str]) -> bool:
+		try:
+			for i, a in enumerate(self._args):
+				if(args[1] in a[0]):
+					self._execs[i](args[2: len(args)])
+					return(True)
+
+			self._exec(args[1: len(args)])
+
+		except IndexError:
+			print(' To see more of command type "-h" or "--help" on arguments')
+
+		except ValueError as e:
+			print(f"{Icons.warn}{e}")
+
+		except Exception:
+			print(f"{Icons.warn}{format_exc()}")
+
+		return(False)
+
 	def _exec(self, args: list[str]) -> None:
 		shell(" ".join(args))

libs/wslbuilder/fedora.ico

@@ -2,8 +2,9 @@
 # -*- coding: utf-8 -*-
 
 from os import listdir, mkdir, remove, rmdir, system as shell
-from os.path import abspath, dirname
+from os.path import abspath, dirname, getsize
 
+from core import UNITS
 from core.icons import Icons
 from core.tool import Tool
 
@@ -134,9 +135,19 @@ def _init(self) -> None:
 	def _list(self) -> None:
 		__distros = listdir(self.__path)
 
-		print(f"\n {' '*1}*  Name{' '*(18-len('Name'))}Path")
+		print(f"\n {' '*1}*  Name{' '*(18-len('Name'))}Size{' '*(12-len('Size'))}Path")
 		for i, distro in enumerate(__distros, start=1):
-			print(f" {' '*(2-len(str(i)))}{i}. {distro.replace('-', ':')}{' '*(18-len(distro))}{abspath(f'{self.__path}/{distro}')}")
+			size = [getsize(abspath(f"{self.__path}/{distro}/ext4.vhdx")), UNITS[0]]
+
+			for j in range(1, len(UNITS)):
+				size[0] /= 1000
+				size[1] = UNITS[j]
+				if(size[0] < 1024):
+					break
+
+			size = f"{round(size[0], 2)} {size[1]}"
+
+			print(f" {' '*(2-len(str(i)))}{i}. {distro.replace('-', ':')}{' '*(18-len(distro))}{size}{' '*(12-len(size))}{abspath(f'{self.__path}/{distro}')}")
 
 	def _start(self, args: list[str]) -> None:
 		__distros = listdir(self.__path)