From 3e2bad2c1ae93b2ab6758df0727a6a0238ebc41a Mon Sep 17 00:00:00 2001 From: Holger Levsen Date: Tue, 26 Feb 2013 19:16:05 +0100 Subject: [PATCH] count existing images and issue warning if number is greater then 10 --- cleanup_liveboot.py | 17 ++++++++++++++++- debian_repos2db.sh | 7 +++++++ 2 files changed, 23 insertions(+), 1 deletion(-) diff --git a/cleanup_liveboot.py b/cleanup_liveboot.py index 4fff2ef..928c239 100755 --- a/cleanup_liveboot.py +++ b/cleanup_liveboot.py @@ -27,6 +27,18 @@ def liveboot_to_keep(name): if cur.fetchone(): return True +def count_liveboots(): + con = db_connect() + cur = con.cursor() + try: + cur.execute("SELECT COUNT(keep_image) FROM liveboot WHERE keep_image = true") + except psycopg2.DatabaseError as e: + logger.debug("SELECT COUNT(keep_image) FROM liveboot WHERE keep_image = true") + return False + print cur.fetchone()[0] + return True + + def set_liveboot_to_delete(name): con = db_connect() cur = con.cursor() @@ -38,7 +50,10 @@ def set_liveboot_to_delete(name): return False if __name__ == '__main__': - if len(sys.argv) != 3 or sys.argv[1] not in ('keep', 'delete'): + if len(sys.argv) == 1: + count_liveboots() + sys.exit(0) + elif len(sys.argv) != 3 or sys.argv[1] not in ('keep', 'delete'): print("usage: %s keep $liveboot_name") print(" will return true or false, depending if 'keep_image' is set for this image") print("usage: %s delete $liveboot_name") diff --git a/debian_repos2db.sh b/debian_repos2db.sh index fd26667..5aae8cf 100755 --- a/debian_repos2db.sh +++ b/debian_repos2db.sh @@ -31,3 +31,10 @@ for REPO in $(seq 0 $REPOS) ; do done done done + +EXISTING_IMAGES=$(./cleanup_liveboot.py) + +if [ "$EXISTING_IMAGES" -gt 10 ] ; then + figlet WARNING + echo "$EXISTING_IMAGES liveboot images exist, you probably need to delete some..." +fi -- 2.39.5