view.py: double blank lines at toplevel
[invirt/packages/invirt-web.git] / code / xen-ips
1 #!/usr/bin/python
2 import random
3 from invirt import database
4 import sqlalchemy.exceptions
5 import sys
6
7 # stolen directly from xend/server/netif.py
8 def randomMAC():
9     """Generate a random MAC address.
10
11     Uses OUI (Organizationally Unique Identifier) 00-16-3E, allocated to
12     Xensource, Inc. The OUI list is available at
13     http://standards.ieee.org/regauth/oui/oui.txt.
14
15     The remaining 3 fields are random, with the first bit of the first
16     random field set 0.
17
18     @return: MAC address string
19     """
20     mac = [ 0x00, 0x16, 0x3e,
21             random.randint(0x00, 0x7f),
22             random.randint(0x00, 0xff),
23             random.randint(0x00, 0xff) ]
24     return ':'.join(map(lambda x: "%02x" % x, mac))
25
26 # ... and stolen from xend/uuid.py
27 def randomUUID():
28     """Generate a random UUID."""
29
30     return [ random.randint(0, 255) for _ in range(0, 16) ]
31
32 def uuidToString(u):
33     return "-".join(["%02x" * 4, "%02x" * 2, "%02x" * 2, "%02x" * 2,
34                      "%02x" * 6]) % tuple(u)
35
36
37 def usage():
38     print >> sys.stderr, "USAGE: " + sys.argv[0] + " <ip>"
39
40 def addip(ip):
41     try:
42         n = database.NIC(machine=None, mac_addr=randomMAC(), ip=ip, hostname=None)
43         database.session.save(n)
44         database.session.flush()
45     except sqlalchemy.exceptions.IntegrityError:
46         pass
47
48
49 if __name__ == '__main__':
50     if len(sys.argv) == 2:
51         ip = sys.argv[1]
52     else:
53         usage()
54         sys.exit(1)
55     database.connect()
56     addip(ip)