116 lines
3.5 KiB
Plaintext
116 lines
3.5 KiB
Plaintext
|
#!/usr/bin/env python
|
||
|
|
#
|
||
|
|
# Original author: Richard Bateman <taxilian@gmail.com>
|
||
|
|
#
|
||
|
|
# Any questions or concerns about how this works should be addressed to
|
||
|
|
# me, not to sitaram. Please note that neither I nor sitaram make any
|
||
|
|
# guarantees about the security or usefulness of this script. It may
|
||
|
|
# be used without warantee or any guarantee of any kind.
|
||
|
|
#
|
||
|
|
# That said, it works fine for me.
|
||
|
|
#
|
||
|
|
# This script is licensed under the New BSD license
|
||
|
|
# Copyright 2011 Richard Bateman
|
||
|
|
#
|
||
|
|
|
||
|
|
import sys, os
|
||
|
|
from pygitolite import *
|
||
|
|
|
||
|
|
def list(gl, user, repo, filter_var = ""):
|
||
|
|
perms = gl.get_perms(repo, user)
|
||
|
|
for var, ppl in perms.iteritems():
|
||
|
|
if filter_var == "" or filter_var == var:
|
||
|
|
print "%s:" % var
|
||
|
|
for item in ppl:
|
||
|
|
print " %s" % item
|
||
|
|
|
||
|
|
def clear(gl, user, repo, filter_var = ""):
|
||
|
|
try:
|
||
|
|
os.system(r"echo Are you sure? Type YES \(all caps\) to continue: ")
|
||
|
|
bval = raw_input()
|
||
|
|
if bval != "YES":
|
||
|
|
print "Canceling..."
|
||
|
|
|
||
|
|
if filter_var == "":
|
||
|
|
gl.set_perms(repo, user, {})
|
||
|
|
else:
|
||
|
|
perms = gl.get_perms(repo, user)
|
||
|
|
if filter_var in perms:
|
||
|
|
del perms[filter_var]
|
||
|
|
gl.set_perms(repo, user, perms)
|
||
|
|
print "Perms after clear:"
|
||
|
|
list(gl, user, repo)
|
||
|
|
except:
|
||
|
|
print "An error occured"
|
||
|
|
|
||
|
|
def add(gl, user, repo, var, *users):
|
||
|
|
perms = gl.get_perms(repo, user)
|
||
|
|
if var not in perms:
|
||
|
|
perms[var] = []
|
||
|
|
if len(users) == 0:
|
||
|
|
print "Usage: perms add %s %s <username>" % (repo, var)
|
||
|
|
return
|
||
|
|
for cur in users:
|
||
|
|
if cur not in perms[var]:
|
||
|
|
perms[var].append(cur)
|
||
|
|
gl.set_perms(repo, user, perms)
|
||
|
|
list(gl, user, repo, var)
|
||
|
|
|
||
|
|
def set(gl, user, repo, var, *users):
|
||
|
|
perms = gl.get_perms(repo, user)
|
||
|
|
perms[var] = []
|
||
|
|
if len(users) == 0:
|
||
|
|
print "Usage: perms set %s %s <username>" % (repo, var)
|
||
|
|
return
|
||
|
|
for cur in users:
|
||
|
|
if cur not in perms[var]:
|
||
|
|
perms[var].append(cur)
|
||
|
|
gl.set_perms(repo, user, perms)
|
||
|
|
list(gl, user, repo, var)
|
||
|
|
|
||
|
|
def remove(gl, user, repo, var, *users):
|
||
|
|
perms = gl.get_perms(repo, user)
|
||
|
|
if var not in perms:
|
||
|
|
print "%s isn't a valid type" % var
|
||
|
|
return
|
||
|
|
if len(users) == 0:
|
||
|
|
print "No users specified to remove; perhaps you want clear?"
|
||
|
|
return
|
||
|
|
for cur in users:
|
||
|
|
if cur in perms[var]:
|
||
|
|
perms[var].remove(cur)
|
||
|
|
gl.set_perms(repo, user, perms)
|
||
|
|
list(gl, user, repo, var)
|
||
|
|
|
||
|
|
commands = {
|
||
|
|
"list": list,
|
||
|
|
"clear": clear,
|
||
|
|
"add": add,
|
||
|
|
"set": set,
|
||
|
|
"remove": remove,
|
||
|
|
}
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if "GL_USER" not in os.environ:
|
||
|
|
raise "No user!"
|
||
|
|
user = os.environ["GL_USER"]
|
||
|
|
command = sys.argv[1] if len(sys.argv) > 2 else ""
|
||
|
|
if len(sys.argv) < 3 or command not in commands:
|
||
|
|
print "Usage: perms <command> <repository> <args>"
|
||
|
|
print " list <repository> [TYPE]"
|
||
|
|
print " clear <repository>"
|
||
|
|
print " add <repository> <TYPE> [user and group list]"
|
||
|
|
print " set <repository> <TYPE> [user and group list]"
|
||
|
|
print " remove <repository> <TYPE> [user and group list]"
|
||
|
|
sys.exit(1)
|
||
|
|
repo = sys.argv[2]
|
||
|
|
|
||
|
|
gl = gitolite()
|
||
|
|
rights, owner = gl.get_rights_and_owner(repo, user)
|
||
|
|
|
||
|
|
if owner != user:
|
||
|
|
print "Either %s does not exist or you are not the owner." % repo
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
commands[command](gl, user, repo, *sys.argv[3:])
|