From 127e3f6df0dd9e11cd566a3ceb7c2194ce2fbfa3 Mon Sep 17 00:00:00 2001 From: Gary Talent Date: Sat, 20 Feb 2021 20:03:54 -0600 Subject: [PATCH] [nostalgia/scripts/pybb] Fix directory removal --- scripts/pybb | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/scripts/pybb b/scripts/pybb index 853d24f6..25c5267f 100644 --- a/scripts/pybb +++ b/scripts/pybb @@ -1,6 +1,8 @@ #! /usr/bin/env python3 import os +import shutil +import stat import sys def mkdir(path): @@ -8,8 +10,12 @@ def mkdir(path): os.mkdir(path) def rm(path): - if os.path.exists(path) or os.path.islink(path): + if (os.path.exists(path) or os.path.islink(path)) and not os.path.isdir(path): + os.chmod(path, stat.S_IWRITE) os.remove(path) + elif os.path.isdir(path): + os.chmod(path, stat.S_IWRITE) + shutil.rmtree(path) if sys.argv[1] == 'mkdir': mkdir(sys.argv[2])