|
105
|
1 #!/usr/bin/env python2 |
|
|
2 # this script add aliases to /etc/hosts |
|
|
3 import sys, re |
|
|
4 |
|
|
5 if len(sys.argv) < 2 or len(sys.argv) > 3: |
|
|
6 sys.exit(1) |
|
|
7 |
|
|
8 host = sys.argv[1] |
|
|
9 alias = sys.argv[2] if len(sys.argv) == 3 else "localhost" |
|
|
10 |
|
|
11 if host == "localhost" or host == alias: |
|
|
12 sys.exit(0) |
|
|
13 |
|
|
14 print "Adding host {} as an alias of {}".format(host, alias) |
|
|
15 with open("/etc/hosts", "r+") as f: |
|
|
16 buf = re.sub(r"\\b{}\\b".format(alias), "{}\\t{}".format(alias, host), f.read(), 1) |
|
|
17 f.seek(0) |
|
|
18 f.write(buf) |
