mirror of
https://github.com/librenms/librenms.git
synced 2024-10-07 16:52:45 +00:00
56 lines
2.2 KiB
Python
Executable File
56 lines
2.2 KiB
Python
Executable File
#! /usr/bin/env python
|
|
import argparse, datetime, sys
|
|
from subprocess import call, check_output
|
|
from os.path import dirname, abspath
|
|
|
|
raw_input = getattr(__builtins__, 'raw_input', input)
|
|
|
|
def confirm(question):
|
|
valid = {"yes":True, "y":True, "ye":True, "no":False, "n":False}
|
|
while 1:
|
|
sys.stdout.write(question + " [y/N] ")
|
|
choice = raw_input().lower()
|
|
if choice == '':
|
|
return False
|
|
elif choice in valid.keys():
|
|
return valid[choice]
|
|
else:
|
|
sys.stdout.write("Please respond with 'yes' or 'no' (or 'y' or 'n').\n")
|
|
|
|
|
|
librenms_dir = dirname(dirname(abspath(__file__)))
|
|
|
|
parser = argparse.ArgumentParser()
|
|
group = parser.add_mutually_exclusive_group(required=True)
|
|
group.add_argument("-d", "--discard", action="store_true", help="Discard all changes clean extra files")
|
|
group.add_argument("-s", "--save", action="store_true", help="Save and remove changes by stashing them. (git stash)")
|
|
group.add_argument("-r", "--restore", action="store_true", help="Attempt to restore saved changes (git stash pop)")
|
|
parser.add_argument("-v", "--vendor", action="store_true", help="Also discard changes to ./vendor (requires --discard)")
|
|
args = parser.parse_args()
|
|
|
|
|
|
if args.discard:
|
|
if confirm("Are you sure you want to delete all modified and untracked files?"):
|
|
dirs = ["contrib/", "doc/", "html/", "includes/", "lib/", "LibreNMS/",
|
|
"licenses/", "mibs/", "misc/", "scripts/", "sql-schema/", "tests/"]
|
|
|
|
call(["git", "reset", "-q"], cwd=librenms_dir)
|
|
call(["git", "checkout", "."], cwd=librenms_dir)
|
|
call(["git", "clean", "-d", "-f"] + dirs, cwd=librenms_dir)
|
|
|
|
if args.vendor:
|
|
call(["git", "clean", "-x", "-d", "-f", "vendor/"], cwd=librenms_dir)
|
|
|
|
elif args.save:
|
|
msg = "github-remove saved on "+str(datetime.datetime.now())
|
|
call(["git", "stash", "save", msg], cwd=librenms_dir)
|
|
|
|
elif args.restore:
|
|
list = check_output(["git", "stash", "list"])
|
|
last = list.decode("utf-8") .split('\n', 1)[0]
|
|
|
|
if "github-remove" in last:
|
|
call(["git", "stash", "pop"], cwd=librenms_dir)
|
|
else:
|
|
print("Last stash was not created by " + __file__ + ". Aborting.")
|