def __init__(self, **kwargs): if (os.geteuid() != 0 and '--help' not in sys.argv and '-h' not in sys.argv): # Inform people with insufficient privileges before parsing args # so they don't have to wade through required arg messages and # whatnot first. raise Exception("must be superuser") BundleCreator.__init__(self, **kwargs)
def configure(self): BundleCreator.configure(self) # kernel/ramdisk image IDs if self.args.get('kernel') == 'true': self.args['image_type'] = 'kernel' if self.args.get('ramdisk') == 'true': self.args['image_type'] = 'ramdisk' if self.args['image_type'] == 'kernel': if self.args.get('kernel') and self.args['kernel'] != 'true': raise ArgumentError("argument --kernel: not compatible with " "image type 'kernel'") if self.args.get('ramdisk'): raise ArgumentError("argument --ramdisk: not compatible with " "image type 'kernel'") if self.args['image_type'] == 'ramdisk': if self.args.get('kernel'): raise ArgumentError("argument --kernel: not compatible with " "image type 'ramdisk'") if self.args.get('ramdisk') and self.args['ramdisk'] != 'true': raise ArgumentError("argument --ramdisk: not compatible with " "image type 'ramdisk'")
def configure(self): BundleCreator.configure(self) self.args['user'] = self.args.get('user').replace('-', '')