code.grnet.gr Git - archipelago/blob - xseg/tools/vlmc-filed.py

   1 #!/usr/bin/env python2.7
   2 #
   3 # vlmc tool for filed
   4
   5 from vlmc_shared import *
   6 import os, sys, subprocess, argparse
   7
   8 def vlmc_map(args):
   9     xsegbd_loaded()
  10     name = args.name[0]
  11     prev = 0
  12     try:
  13         result = [int(open(XSEGBD_SYSFS + "devices/" + f + "/srcport").read().strip()) for f in os.listdir(XSEGBD_SYSFS + "devices/")]
  14         result.sort()
  15
  16         for p in result:
  17             if p - prev > 1:
  18                break
  19             else:
  20                prev = p
  21
  22         port = prev + 1
  23         fd = os.open(XSEGBD_SYSFS + "add", os.O_WRONLY)
  24         os.write(fd, "%s %d:%d:%d" % (name, port, FILED_PORT, REQS))
  25         os.close(fd)
  26     except Exception, reason:
  27         print >> sys.stderr, reason
  28         sys.exit(-1)
  29
  30 def vlmc_unmap(args):
  31     xsegbd_loaded()
  32     device = args.name[0]
  33     try:
  34         for f in os.listdir(XSEGBD_SYSFS + "devices/"):
  35             d_id = open(XSEGBD_SYSFS + "devices/" + f + "/id").read().strip()
  36             name = open(XSEGBD_SYSFS + "devices/"+ f + "/target").read().strip()
  37             if device == DEVICE_PREFIX + d_id:
  38                 fd = os.open(XSEGBD_SYSFS + "remove", os.O_WRONLY)
  39                 os.write(fd, d_id)
  40                 os.close(fd)
  41
  42                 sys.exit(0)
  43         print >> sys.stderr, "Device %s doesn't exist" % device
  44         sys.exit(-1)
  45     except Exception, reason:
  46         print >> sys.stderr, reason
  47         sys.exit(-1)
  48
  49 def vlmc_showmapped(args):
  50     xsegbd_loaded()
  51     print "id\tpool\timage\tsnap\tdevice"
  52     try:
  53         for f in os.listdir(XSEGBD_SYSFS + "devices/"):
  54             d_id = open(XSEGBD_SYSFS + "devices/" + f + "/id").read().strip()
  55             target = open(XSEGBD_SYSFS + "devices/"+ f + "/target").read().strip()
  56
  57             print "%s\t%s\t%s\t%s\t%s" % (d_id, '-', target, '-', DEVICE_PREFIX +
  58             d_id)
  59     except Exception, reason:
  60         print >> sys.stderr, reason
  61         sys.exit(-1)
  62
  63 if __name__ == "__main__":
  64     # parse arguments and discpatch to the correct func
  65     parser = argparse.ArgumentParser(description='vlmc tool')
  66     parser.add_argument('-c', '--config', type=str, nargs='?', help='config file')
  67     subparsers = parser.add_subparsers()
  68
  69     create_parser = subparsers.add_parser('create', help='Create volume')
  70     #group = create_parser.add_mutually_exclusive_group(required=True)
  71     create_parser.add_argument('-s', '--size', type=int, nargs='?', help='requested size in MB for create')
  72     create_parser.add_argument('--snap', type=str, nargs='?', help='create from snapshot')
  73     create_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
  74     create_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
  75     create_parser.set_defaults(func=vlmc_create)
  76
  77     remove_parser = subparsers.add_parser('remove', help='Delete volume')
  78     remove_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
  79     remove_parser.set_defaults(func=vlmc_remove)
  80     remove_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
  81
  82     rm_parser = subparsers.add_parser('rm', help='Delete volume')
  83     rm_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
  84     rm_parser.set_defaults(func=vlmc_remove)
  85     rm_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
  86
  87     map_parser = subparsers.add_parser('map', help='Map volume')
  88     map_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
  89     map_parser.set_defaults(func=vlmc_map)
  90     map_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
  91
  92     unmap_parser = subparsers.add_parser('unmap', help='Unmap volume')
  93     unmap_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
  94     unmap_parser.set_defaults(func=vlmc_unmap)
  95     unmap_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
  96
  97     showmapped_parser = subparsers.add_parser('showmapped', help='Show mapped volumes')
  98     showmapped_parser.set_defaults(func=vlmc_showmapped)
  99     showmapped_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
 100
 101     list_parser = subparsers.add_parser('list', help='List volumes')
 102     list_parser.set_defaults(func=vlmc_list)
 103     list_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
 104
 105     ls_parser = subparsers.add_parser('ls', help='List volumes')
 106     ls_parser.set_defaults(func=vlmc_list)
 107     ls_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
 108
 109     resize_parser = subparsers.add_parser('resize', help='Resize volume')
 110     resize_parser.add_argument('-s', '--size', type=int, nargs=1, help='requested size in MB for resize')
 111     resize_parser.add_argument('name', type=str, nargs=1, help='volume/device name')
 112     resize_parser.set_defaults(func=vlmc_resize)
 113     resize_parser.add_argument('-p', '--pool', type=str, nargs='?', help='for backwards compatiblity with rbd')
 114
 115     args = parser.parse_args()
 116     loadrc(args.config)
 117     args.func(args)