#! /usr/bin/env python3

import os

import shutil

from pathlib import Path

import yaml

DEFAULT_CONFIG = """

- ~/.adobe # Flash crap

- ~/.macromedia # Flash crap

- ~/.recently-used

- ~/.local/share/recently-used.xbel

- ~/.thumbnails

- ~/.gconfd

- ~/.gconf

- ~/.local/share/gegl-0.2

- ~/.FRD/log/app.log # FRD

- ~/.FRD/links.txt # FRD

- ~/.objectdb # FRD

- ~/.gstreamer-0.10

- ~/.pulse

- ~/.esd_auth

- ~/.config/enchant

- ~/.spicec # contains only log file; unconfigurable

- ~/.dropbox-dist

- ~/.parallel

- ~/.dbus

- ~/ca2 # WTF?

- ~/ca2~ # WTF?

- ~/.distlib/ # contains another empty dir, don't know which software creates it

- ~/.bazaar/ # bzr insists on creating files holding default values

- ~/.bzr.log

- ~/.nv/

- ~/.viminfo # configured to be moved to ~/.cache/vim/viminfo, but it is still sometimes created...

- ~/.npm/ # npm cache

- ~/.java/

- ~/.swt/

- ~/.oracle_jre_usage/

- ~/.openjfx/

- ~/.org.jabref.gui.JabRefMain/

- ~/.org.jabref.gui.MainApplication/

- ~/.jssc/

- ~/.tox/ # cache directory for tox

- ~/.pylint.d/

- ~/.qute_test/

- ~/.QtWebEngineProcess/

- ~/.qutebrowser/ # created empty, only with webengine backend

- ~/.asy/

- ~/.cmake/

- ~/.gnome/

- ~/unison.log

- ~/.texlive/

- ~/.w3m/

- ~/.subversion/

- ~/nvvp_workspace/ # created empty even when the path is set differently in nvvp

- ~/.ansible/

- ~/.fltk/

- ~/.vnc/

- ~/.local/share/Trash/ # VSCode puts deleted files here

"""

def get_size(path):

if Path(path).is_dir():

return sum(p.stat().st_size for p in Path(path).rglob("*"))

return Path(path).stat().st_size

def read_config():

"""

Reads the list of shitty files from a YAML config.

"""

config_dir = os.getenv("XDG_CONFIG_HOME", os.path.expanduser("~/.config/"))

config_path = Path(config_dir) / "rmshit.yaml"

# write default config if it does not exist

if not config_path.exists():

with open(config_path, "w") as f:

print(DEFAULT_CONFIG.strip(), file=f)

with open(config_path, "r") as f:

return yaml.safe_load(f)

def yesno(question, default="n"):

"""

Asks the user for YES or NO, always case insensitive.

Returns True for YES and False for NO.

"""

prompt = f"{question} (y/[n]) "

ans = input(prompt).strip().lower()

if not ans:

ans = default

if ans == "y":

return True

return False

def format_size(size_in_bytes):

"""Format file size in bytes to a human-readable string."""

if size_in_bytes <= 0:

return "0 bytes"

units = ["bytes", "KiB", "MiB", "GiB"]

size = float(size_in_bytes)

unit_index = min(int((size_in_bytes.bit_length() - 1) // 10), len(units) - 1)

size /= 1024**unit_index

return f"{size:.4g} {units[unit_index]}"

def rmshit():

shittyfiles = read_config()

print("Found shittyfiles:")

found = []

total_size = 0

for f in shittyfiles:

absf = os.path.expanduser(f)

if os.path.exists(absf):

found.append(absf)

size = get_size(absf)

total_size += size

print(f" {f} ({format_size(size)})")

if len(found) == 0:

print("No shitty files found :)")

return

if yesno("Remove all?", default="n"):

for f in found:

if os.path.isfile(f):

os.remove(f)

else:

shutil.rmtree(f)

print(f"All cleaned, {format_size(total_size)} freed.")

else:

print("No file removed")

if __name__ == "__main__":

rmshit()