From e9ce95e511b23870677a00f222e4c7da2970324f Mon Sep 17 00:00:00 2001
From: neingeist <neingeist@bl0rg.net>
Date: Sun, 15 Nov 2015 18:46:58 +0100
Subject: [PATCH] Rewrite in Python

---
 check_dir_empty | 34 ++++++++++++++++++++--------------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/check_dir_empty b/check_dir_empty
index 6113b2a..efaaed2 100755
--- a/check_dir_empty
+++ b/check_dir_empty
@@ -1,18 +1,24 @@
-#!/bin/sh
+#!/usr/bin/python
+from __future__ import division, print_function
+import argparse
+import os
+import sys
 
-dir=$1
 
-if [ "$dir" = "" ]; then
-  echo "Usage: $0 <dir>" >&2
-  exit 1
-fi
+parser = argparse.ArgumentParser(
+    description='Check that the directories given are empty')
+parser.add_argument(
+    'directories', metavar='dir', nargs='+', type=str,
+    help='directory to be checked')
+args = parser.parse_args()
 
-count=`find "$dir" -mindepth 1 -maxdepth 1 | wc -l`
 
-if [ "$count" = "0" ]; then
-  echo "OK: '$dir' is empty."
-  exit 0
-else
-  echo "WARNING: '$dir' is NOT empty ($count entries)."
-  exit 1
-fi
+for directory in args.directories:
+    listdir = os.listdir(directory)
+    if len(listdir) != 0:
+        print('WARNING: Directory {} is not empty ({} entries)'
+              .format(directory, len(listdir)))
+        sys.exit(1)
+
+print('OK: All given directories are empty')
+sys.exit(0)