1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
#!/usr/bin/python
from RepSys import Error, config
import getpass
import sys
import os
import re
import logging
#import commands
log = logging.getLogger("repsys")
# Our own version of commands' getstatusoutput(). We have a commands
# module directory, so we can't import Python's standard module
def commands_getstatusoutput(cmd):
"""Return (status, output) of executing cmd in a shell."""
import os
pipe = os.popen('{ ' + cmd + '; } 2>&1', 'r')
text = pipe.read()
sts = pipe.close()
if sts is None: sts = 0
if text[-1:] == '\n': text = text[:-1]
return sts, text
def execcmd(*cmd, **kwargs):
cmdstr = " ".join(cmd)
if kwargs.get("show"):
status = os.system(cmdstr)
output = ""
else:
status, output = commands_getstatusoutput("LANG=C "+cmdstr)
if status != 0 and not kwargs.get("noerror"):
raise Error, "command failed: %s\n%s\n" % (cmdstr, output)
if config.getbool("global", "verbose", 0):
print cmdstr
sys.stdout.write(output)
return status, output
def get_auth(username=None, password=None):
set_username = 1
set_password = 1
if not username:
username = config.get("auth", "username")
if not username:
username = raw_input("username: ")
else:
set_username = 0
if not password:
password = config.get("auth", "password")
if not password:
password = getpass.getpass("password: ")
else:
set_password = 0
if set_username:
config.set("auth", "username", username)
if set_password:
config.set("auth", "password", password)
return username, password
def mapurl(url):
"""Maps a url following the regexp provided by the option url-map in
repsys.conf
"""
urlmap = config.get("global", "url-map")
newurl = url
if urlmap:
try:
expr_, replace = urlmap.split()[:2]
except ValueError:
log.error("invalid url-map: %s", urlmap)
else:
try:
newurl = re.sub(expr_, replace, url)
except re.error, errmsg:
log.error("error in URL mapping regexp: %s", errmsg)
return newurl
# vim:et:ts=4:sw=4
|
