def run(self):
""" """
options, remainder = self.parser.parse_args()
if options.source_device is not None:
if options.output_folder is None:
self.parser.print_help()
sys.exit(1)
ic = ImageCreator(options.image_name, options.source_device,
options.output_folder, options.compressor_level)
ic.create_image()
elif options.target_device:
if options.image_folder is None:
self.parser.print_help()
sys.exit(1)
ir = ImageRestorer(options.image_folder, options.target_device)
ir.restore_image()
def run(self):
""" """
opt, remainder = self.parser.parse_args()
if opt.source_device is not None:
if opt.output_folder is None:
self.parser.print_help()
sys.exit(1)
ic = ImageCreator(opt.source_device, opt.output_folder, opt.image_name, \
opt.compressor_level, opt.raw, opt.split_size)
ic.create_image()
elif opt.target_device:
if opt.image_folder is None:
self.parser.print_help()
sys.exit(1)
ir = ImageRestorer(opt.image_folder, opt.target_device)
ir.restore_image()
class Cli:
def __init__(self):
self.parser = optparse.OptionParser(usage=self.usage(),
version=get_version())
self.lock = Lock()
self._set_options()
self._register_signals()
self.operation = None
def _register_signals(self):
signal.signal(signal.SIGINT, self.cancel_operation)
signal.signal(signal.SIGUSR1, self.cancel_operation)
def cancel_operation(self, signal=None, frame=None):
if self.operation is not None:
self.operation.cancel()
def usage(self):
usage = "usage: %prog [options]\n"
usage += "\nCreating: \n"
usage += "\t%prog -s /dev/sda -o /media/external_hd\n"
usage += "\t%prog -s /dev/sda1 -o /media/external_hd -r -z -c 0\n"
usage += "\t%prog -s /dev/sda -o /media/external_hd -p 4G -m\n"
usage += "\nRestoring: \n"
usage += "\t%prog -t /dev/sda -i /media/external_hd\n"
usage += "\t%prog -t /dev/sda -i /media/external_hd -x 1,3 -e\n"
return usage
def _set_options(self):
""" """
create_group = optparse.OptionGroup(
self.parser,
"Creating Image Options",
)
restore_group = optparse.OptionGroup(
self.parser,
"Restoring Image Options",
)
information_group = optparse.OptionGroup(
self.parser,
"Image Information Options",
)
create_group.add_option(
"-s",
"--source-device",
dest="source_device",
)
create_group.add_option(
"-o",
"--output-folder",
dest="output_folder",
)
create_group.add_option(
"-n",
"--image-name",
dest="image_name",
default="image",
help="[default: %default]",
)
create_group.add_option(
"-c",
"--compressor-level",
dest="compressor_level",
type="int",
default=6,
help="An integer from 0 to 9 controlling "
"the level of compression; 0 for no "
"compression, 1 is fastest and produces "
"the least compression, 9 is slowest and "
"produces the most. "
"[default: %default]",
)
create_group.add_option(
"-r",
"--raw",
dest="raw",
action="store_true",
default=False,
help="Create raw images.",
)
create_group.add_option(
"-m",
"--iso",
dest="iso",
action="store_true",
default=False,
help="Create iso(s) from file(s).",
)
create_group.add_option(
"-z",
"--fill-with-zeros",
dest="fill_with_zeros",
action="store_true",
default=False,
help="Fill the filesystem with zeros. "
"Because Raw image copies every block of "
"the filesystem this will aid compression "
"of the image (If compression is enabled).",
)
create_group.add_option(
"-p",
"--split",
dest="split_size",
type="string",
help="Split the image file into smaller "
"chunks of required size.",
)
restore_group.add_option(
"-t",
"--target-device",
dest="target_device",
)
restore_group.add_option(
"-i",
"--image-folder",
dest="image_folder",
)
restore_group.add_option(
"-x",
"--image-partitions",
dest="partition_numbers",
help="Restore only the given "
"partition number(s) (Comma separated). "
"ONLY FOR DISK RECOVERIES. "
"[Ex: 1,2,3]",
)
restore_group.add_option(
"-e",
"--expand",
dest="expand",
action="store_true",
default=False,
help="If possible resize the last partition "
"until it fills the whole disk. "
"ONLY FOR DISK RECOVERIES.",
)
information_group.add_option(
"-q",
"--image-information",
dest="image_folder",
help="Show the information about the image.",
)
self.parser.add_option_group(create_group)
self.parser.add_option_group(restore_group)
self.parser.add_option_group(information_group)
def status(self, action, dict={}):
""" """
self.lock.acquire()
response = None
if action == "progress":
sys.stdout.write("%d%%\r" % dict["percent"])
elif action == "finish":
sys.stdout.write("Finished.\r\n")
elif action == "checking_filesystem":
sys.stdout.write("Checking filesystem of %s...\n" % dict["device"])
elif action == "expand":
sys.stdout.write("Expanding filesystem of %s...\n" %
dict["device"])
elif action == "filling_with_zeros":
sys.stdout.write("Zeroing filesystem of %s...\n" % dict["device"])
elif action == "iso":
sys.stdout.write("Creating ISO %d of %d...\n" %
(dict["volume"], dict["total"]))
elif action == "canceled":
sys.stdout.write("\r%s operation canceled!\n" % dict["operation"])
elif action == "base_files_not_found":
sys.stdout.write("\nCarbono files cannt be found in %s.\n" \
% dict["device"])
response = raw_input("Please type another device " + \
"(or leave blank to cancel): ")
elif action == "file_not_found":
sys.stdout.write("\nThe file %s cannot be found at %s.\n" \
% (dict["file"], dict["path"]))
response = raw_input("Type the file location " + \
"(or leave blank to cancel): ")
sys.stdout.flush()
self.lock.release()
return response
def run(self):
""" """
opt, remainder = self.parser.parse_args()
if opt.source_device is not None:
if opt.output_folder is None:
self.parser.print_help()
sys.exit(1)
if not check_if_root():
sys.stderr.write("You need to run this option as root.\n")
sys.exit(1)
split_size = 0
if opt.split_size:
l = opt.split_size[-1]
if l.isdigit():
split_size = float(opt.split_size)
else:
try:
split_size = float(opt.split_size[:-1])
except ValueError:
raise Exception("Invalid split size")
l = l.upper()
if l == 'M':
split_size = int(split_size * 1024**2)
elif l == 'G':
split_size = int(split_size * 1024**3)
else:
raise Exception("Cannot determine split size")
self.operation = ImageCreator(opt.source_device, opt.output_folder,
self.status, opt.image_name,
opt.compressor_level, opt.raw,
split_size, opt.iso,
opt.fill_with_zeros)
self.operation.create_image()
elif opt.target_device is not None:
if opt.image_folder is None:
self.parser.print_help()
sys.exit(1)
if not check_if_root():
sys.stderr.write("You need to run this option as root.\n")
sys.exit(1)
partitions = None
if opt.partition_numbers is not None:
if not opt.partition_numbers.find(','):
partitions = list()
partitions.append(int(opt.partition_numbers))
else:
plist = opt.partition_numbers.split(',')
partitions = map(lambda x: int(x), plist)
self.operation = ImageRestorer(opt.image_folder, opt.target_device,
self.status, partitions, opt.expand)
try:
self.operation.restore_image()
except OSError, e:
if e.errno == errno.ECHILD:
pass
elif opt.image_folder is not None:
image_path = adjust_path(opt.image_folder)
inf = Information(image_path)
inf.load()
sys.stdout.write("Name:\t\t\t%s\n" % inf.get_image_name())
sys.stdout.write("Is disk:\t\t%s\n" % inf.get_image_is_disk())
sys.stdout.write("Compressor level:\t%d\n\n" %
inf.get_image_compressor_level())
sys.stdout.write("Partitions:\n")
sys.stdout.write("%-10s %-15s %-15s %-16s %-36s\n" %
("Number", "Type", "Size", "Label", "UUID"))
for part in inf.get_partitions():
uuid = ''
label = ''
if hasattr(part, "uuid"):
uuid = part.uuid
if hasattr(part, "label"):
label = part.label
sys.stdout.write(
"%-10d %-15s %-15d %-16s %-36s\n" %
(part.number, part.type, part.size, label, uuid))
sys.stdout.flush()
class Cli:
def __init__(self):
self.parser = optparse.OptionParser(usage=self.usage(),
version=get_version())
self.lock = Lock()
self._set_options()
self._register_signals()
self.operation = None
def _register_signals(self):
signal.signal(signal.SIGINT, self.cancel_operation)
signal.signal(signal.SIGUSR1, self.cancel_operation)
def cancel_operation(self, signal=None, frame=None):
if self.operation is not None:
self.operation.cancel()
def usage(self):
usage = "usage: %prog [options]\n"
usage += "\nCreating: \n"
usage += "\t%prog -s /dev/sda -o /media/external_hd\n"
usage += "\t%prog -s /dev/sda1 -o /media/external_hd -r -z -c 0\n"
usage += "\t%prog -s /dev/sda -o /media/external_hd -p 4G -m\n"
usage += "\nRestoring: \n"
usage += "\t%prog -t /dev/sda -i /media/external_hd\n"
usage += "\t%prog -t /dev/sda -i /media/external_hd -x 1,3 -e\n"
return usage
def _set_options(self):
""" """
create_group = optparse.OptionGroup(self.parser,
"Creating Image Options",)
restore_group = optparse.OptionGroup(self.parser,
"Restoring Image Options",)
information_group = optparse.OptionGroup(self.parser,
"Image Information Options",)
create_group.add_option("-s", "--source-device",
dest="source_device",)
create_group.add_option("-o", "--output-folder",
dest="output_folder",)
create_group.add_option("-n", "--image-name",
dest="image_name",
default="image",
help="[default: %default]",)
create_group.add_option("-c", "--compressor-level",
dest="compressor_level",
type="int",
default=6,
help="An integer from 0 to 9 controlling "
"the level of compression; 0 for no "
"compression, 1 is fastest and produces "
"the least compression, 9 is slowest and "
"produces the most. "
"[default: %default]",)
create_group.add_option("-r", "--raw",
dest="raw",
action="store_true",
default=False,
help="Create raw images.",)
create_group.add_option("-m", "--iso",
dest="iso",
action="store_true",
default=False,
help="Create iso(s) from file(s).",)
create_group.add_option("-z", "--fill-with-zeros",
dest="fill_with_zeros",
action="store_true",
default=False,
help="Fill the filesystem with zeros. "
"Because Raw image copies every block of "
"the filesystem this will aid compression "
"of the image (If compression is enabled).",)
create_group.add_option("-p", "--split",
dest="split_size",
type="string",
help="Split the image file into smaller "
"chunks of required size.",)
restore_group.add_option("-t", "--target-device",
dest="target_device",)
restore_group.add_option("-i", "--image-folder",
dest="image_folder",)
restore_group.add_option("-x", "--image-partitions",
dest="partition_numbers",
help="Restore only the given "
"partition number(s) (Comma separated). "
"ONLY FOR DISK RECOVERIES. "
"[Ex: 1,2,3]",)
restore_group.add_option("-e", "--expand",
dest="expand",
action="store_true",
default=False,
help="If possible resize the last partition "
"until it fills the whole disk. "
"ONLY FOR DISK RECOVERIES.",)
information_group.add_option("-q", "--image-information",
dest="image_folder",
help="Show the information about the image.",)
self.parser.add_option_group(create_group)
self.parser.add_option_group(restore_group)
self.parser.add_option_group(information_group)
def status(self, action, dict={}):
""" """
self.lock.acquire()
response = None
if action == "progress":
sys.stdout.write("%d%%\r" % dict["percent"])
elif action == "finish":
sys.stdout.write("Finished.\r\n")
elif action == "checking_filesystem":
sys.stdout.write("Checking filesystem of %s...\n" %
dict["device"])
elif action == "expand":
sys.stdout.write("Expanding filesystem of %s...\n" %
dict["device"])
elif action == "filling_with_zeros":
sys.stdout.write("Zeroing filesystem of %s...\n" %
dict["device"])
elif action == "iso":
sys.stdout.write("Creating ISO %d of %d...\n" %
(dict["volume"], dict["total"]))
elif action == "canceled":
sys.stdout.write("\r%s operation canceled!\n" %
dict["operation"])
elif action == "base_files_not_found":
sys.stdout.write("\nCarbono files cannt be found in %s.\n" \
% dict["device"])
response = raw_input("Please type another device " + \
"(or leave blank to cancel): ")
elif action == "file_not_found":
sys.stdout.write("\nThe file %s cannot be found at %s.\n" \
% (dict["file"], dict["path"]))
response = raw_input("Type the file location " + \
"(or leave blank to cancel): ")
sys.stdout.flush()
self.lock.release()
return response
def run(self):
""" """
opt, remainder = self.parser.parse_args()
if opt.source_device is not None:
if opt.output_folder is None:
self.parser.print_help()
sys.exit(1)
if not check_if_root():
sys.stderr.write("You need to run this option as root.\n")
sys.exit(1)
split_size = 0
if opt.split_size:
l = opt.split_size[-1]
if l.isdigit():
split_size = float(opt.split_size)
else:
try:
split_size = float(opt.split_size[:-1])
except ValueError:
raise Exception("Invalid split size")
l = l.upper()
if l == 'M':
split_size = int(split_size *1024**2)
elif l == 'G':
split_size = int(split_size *1024**3)
else:
raise Exception("Cannot determine split size")
self.operation = ImageCreator(opt.source_device, opt.output_folder,
self.status, opt.image_name,
opt.compressor_level, opt.raw, split_size,
opt.iso, opt.fill_with_zeros)
self.operation.create_image()
elif opt.target_device is not None:
if opt.image_folder is None:
self.parser.print_help()
sys.exit(1)
if not check_if_root():
sys.stderr.write("You need to run this option as root.\n")
sys.exit(1)
partitions = None
if opt.partition_numbers is not None:
if not opt.partition_numbers.find(','):
partitions = list()
partitions.append(int(opt.partition_numbers))
else:
plist = opt.partition_numbers.split(',')
partitions = map(lambda x: int(x), plist)
self.operation = ImageRestorer(opt.image_folder, opt.target_device,
self.status, partitions, opt.expand)
try:
self.operation.restore_image()
except OSError, e:
if e.errno == errno.ECHILD:
pass
elif opt.image_folder is not None:
image_path = adjust_path(opt.image_folder)
inf = Information(image_path)
inf.load()
sys.stdout.write("Name:\t\t\t%s\n" % inf.get_image_name())
sys.stdout.write("Is disk:\t\t%s\n" % inf.get_image_is_disk())
sys.stdout.write("Compressor level:\t%d\n\n" %
inf.get_image_compressor_level())
sys.stdout.write("Partitions:\n")
sys.stdout.write("%-10s %-15s %-15s %-16s %-36s\n" %
("Number", "Type", "Size", "Label", "UUID"))
for part in inf.get_partitions():
uuid = ''
label = ''
if hasattr(part, "uuid"):
uuid = part.uuid
if hasattr(part, "label"):
label = part.label
sys.stdout.write("%-10d %-15s %-15d %-16s %-36s\n" %
(part.number, part.type, part.size,
label, uuid))
sys.stdout.flush()