1
0
Fork 0
mirror of https://codeberg.org/Mo8it/dotfiles.git synced 2024-10-17 20:52:40 +00:00
dotfiles/.scripts/fetch_installed_packages.py
2022-09-24 21:45:26 +02:00

131 lines
3.4 KiB
Python
Executable file

#!/usr/bin/env python3
import json
import re
import subprocess
from os import environ
from pathlib import Path
def run(command, **kwargs):
return subprocess.run(command, shell=True, check=True, **kwargs)
def cap_run(command, **kwargs):
return run(
command, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL, text=True, **kwargs
).stdout.strip()
def rpm_ostree_packages(output_dir):
out = cap_run("rpm-ostree status --json")
json_out = json.loads(out)
deployments = json_out["deployments"]
booted_deployment = deployments[0]
if not booted_deployment["booted"]:
for depl in deployments:
if depl["booted"]:
booted_deployment = depl
break
removals = booted_deployment["requested-base-removals"]
additions = booted_deployment["requested-packages"]
file = output_dir / "rpm-ostree.sh"
with file.open("w") as f:
f.write("rpm-ostree override remove")
for removal in removals:
f.write(" " + removal)
f.write("\nrpm-ostree install")
for addition in additions:
f.write(" " + addition)
def flatpak_packages(output_dir):
out = cap_run("flatpak list --app --columns=application")
apps = out.split()
file = output_dir / "flatpak.sh"
with file.open("w") as f:
f.write("flatpak install")
for app in apps:
f.write(" " + app)
def cargo_packages(output_dir):
out = cap_run("cargo install --list")
pattern = r"(.+) .+:.*"
crates = re.findall(pattern, out)
file = output_dir / "cargo.sh"
with file.open("w") as f:
f.write("cargo install")
for crate in crates:
f.write(" " + crate)
def npm_packages(output_dir):
out = cap_run("npm list --json --location=global")
json_out = json.loads(out)
deps = json_out["dependencies"].keys()
deps = [dep for dep in deps if dep != "npm"]
file = output_dir / "npm.sh"
with file.open("w") as f:
f.write("npm config set prefix '~/.npm-global'\n")
f.write("npm install --location=global npm\n")
f.write("~/.npm-global/bin/npm install --location=global")
for dep in deps:
f.write(" " + dep)
def pipx_packages(output_dir):
out = cap_run("pipx list --json")
json_out = json.loads(out)
venvs = json_out["venvs"]
packages = []
injections = {}
for ind, venv_key in enumerate(venvs.keys()):
metadata = venvs[venv_key]["metadata"]
package = metadata["main_package"]["package_or_url"]
packages.append(package)
package_injections = metadata["injected_packages"].keys()
if len(package_injections) > 0:
injections[package] = package_injections
file = output_dir / "pipx.sh"
with file.open("w") as f:
for package in packages:
f.write(f"pipx install {package}\n")
f.write("\n")
for package in injections.keys():
for injection in injections[package]:
f.write(f"pipx inject {package} {injection}\n")
def main():
output_dir = Path(f"{environ['HOME']}/bk/installed_packages_info")
output_dir.mkdir(parents=True, exist_ok=True)
rpm_ostree_packages(output_dir)
flatpak_packages(output_dir)
pipx_packages(output_dir)
cargo_packages(output_dir)
npm_packages(output_dir)
if __name__ == "__main__":
main()