Criador de instalador USB Linux bootável com Python
Publicado por Marcos Fernandes (última atualização em 07/05/2020)
[ Hits: 1.615 ]
Mesmo com tantas opções, resolvi por questão de praticar com Python, construir uma aplicação que efetue o download de uma ISO e permita, logo após, gravar a mesma em um dispositivo USB.
Requisitos:
beautifulsoup4==4.9.0
requests==2.23.0
wget==3.2
https://github.com/markirius/ulmaker
#!/usr/bin/env python import re import signal import sys from pathlib import Path from subprocess import DEVNULL, STDOUT, call, check_output try: import wget import requests from bs4 import BeautifulSoup except Exception as err: sys.exit(f"[!] {err}, please use 'pip install -r requirements.txt'") URLS = [] HOME = Path.home() # when user exits with Ctrl-C, don't show error msg and remove wget's temp files signal.signal(signal.SIGINT, lambda x, y: clean_temp()) def download(dist): regex = re.compile(".*iso$") if dist == "ubuntu": url = "https://releases.ubuntu.com/20.04/" regex = re.compile(".*desktop.*.iso$") if dist == "archlinux": url = "https://mirror.rackspace.com/archlinux/iso/latest/" if dist == "fedora": url = "https://download.fedoraproject.org/pub/fedora/linux/releases/32/Workstation/x86_64/iso/" if dist == "mint": url = "https://mirrors.edge.kernel.org/linuxmint/stable/19.3/" regex = re.compile(".*cinnamon-64.*iso$") page = requests.get(url) soup = BeautifulSoup(page.content, "html.parser") for link in soup.find_all("a", attrs={"href": regex}): URLS.append(url+link.get("href")) iso_file = Path(f"{HOME}/Downloads/{link.get('href')}") if iso_file.exists(): print("[?] ISO file already downloaded.") return print(f"Downloading {link.get('href')} and saving on {HOME}/Downloads") wget.download(URLS[0], f"{HOME}/Downloads/{link.get('href')}") print() print("Download complete.") def create_usb(dist): dir = f"{HOME}/Downloads" try: iso = check_output(f"find {dir} -name {dist}*.iso", shell=True, stderr=STDOUT).decode('utf-8').split('\n')[:-1] print(f"ISO file {iso[0]} found!") except Exception as err: print(f"[!] {err}: {dir} is not a valid directory!") exit() if not len(iso): print("You don't have ISO files in this directory!") exit() # empty line print() drive_list = check_output("ls -l /dev/disk/by-id/", shell=True).decode("utf-8") if "usb" not in drive_list: print("You don't have any USB drive connected!") exit() usb_list = [d for d in drive_list.split("\n")[:-1] if not d[-1].isdigit() and "usb" in d] print("List of your USB drives:") labels = [] for i, usb in enumerate(usb_list): name = " ".join(usb.split("usb-")[1].split("_")[:-1]) label = usb.split("/")[-1] labels.append(label) size = int(check_output(f"cat /sys/class/block/{label}/size", shell=True)) / 2 / 1024 / 1024 print(f"\n {i} {name} - {size:.2f} GB") # drive selection drive = input("\nSelect your USB drive! (default: 0) ") if drive.isdigit() and int(drive) < len(labels): drive = "/dev/" + labels[int(drive)] elif drive == "": drive = "/dev/" + labels[0] else: print("[!] This is not a valid number drive.\n") exit() confirm = True if input(f"\nThis will erease {drive} Are you sure? (y/n) ") == 'y' else False if confirm: is_mounted = True if call(f"mount | grep {drive}", shell=True, stdout=DEVNULL, stderr=STDOUT) == 0 else False if is_mounted: mount_part = check_output(f"mount | grep {drive}", shell=True).decode("utf-8").split(" on")[0] call(f"umount {mount_part}", shell=True) print("drive unmounted!") call(f"sudo dd if={iso[0]} of={drive} status=progress bs=4M", shell=True) call(f"sync && sudo eject {drive}", shell=True) print("\nDone! Now you can remove your USB drive!") else: exit() def info(): print(""" All systems and software are downloaded in x64 only... sorry. Supported distros and software at moment: ubuntu = Ubuntu 20.04 (LTS Always) fedora = Fedora 32 archlinux = Archlinux (latest) mint = Mint Linux (19.3) """) def clean_temp(): try: temp_files = check_output( f"find {HOME}/Downloads -name *.tmp", shell=True, stderr=STDOUT ).decode('utf-8').split('\n')[:-1] if temp_files: call(f"rm {HOME}/Downloads/*.tmp", shell=True) exit() except Exception as err: print(f"[!] {err}: {dir} is not a valid directory!") exit() def main(argv): try: if argv: download(argv) create_usb(argv) except Exception: info() if __name__ == "__main__": try: dist = sys.argv[1].lower() main(dist) except Exception: info()
Aprenda a Gerenciar Permissões de Arquivos no Linux
Como transformar um áudio em vídeo com efeito de forma de onda (wave form)
Como aprovar Pull Requests em seu repositório Github via linha de comando
Aplicativo simples para gravar tela
Quebra de linha na data e hora no Linux Mint
trocar linhas [RESOLVIDO] (11)
Firefox não abre em usuário não administradores (2)
Ubuntu com problemas no áudio (1)