Merge 243:1f764e5dd1cc

1 files changed, 8 insertions, 7 deletions

diff --git a/cobbler/action_import.py b/cobbler/action_import.py
index 69bfa97..486842e 100644
--- a/cobbler/action_import.py
+++ b/cobbler/action_import.py
@@ -85,18 +85,19 @@ class Importer:
        if self.path and not os.path.isdir(self.path):
            raise cexceptions.CobblerException("import_failed","bad path")
        if self.mirror is not None:
-           if not self.mirror.startswith("rsync://"):
-               raise cexceptions.CobblerException("import_failed","expecting rsync:// url")
            if self.mirror_name is None:
                raise cexceptions.CobblerException("import_failed","must specify --mirror-name")
            print "This will take a while..."
            self.path = "/var/www/cobbler/localmirror/%s" % self.mirror_name
            try:
-               os.mkdir(self.path)
+               os.makedirs(self.path)
            except:
                if not os.path.exists(self.path):
                    raise cexceptions.CobblerException("couldn't create: %s" % (self.path))
-           cmd = "rsync -az %s /var/www/cobbler/localmirror/%s --progress" % (self.mirror, self.mirror_name)
+           spacer = ""
+           if not self.mirror.startswith("rsync://"):
+               spacer = ' -e "ssh" '
+           cmd = "rsync -az %s %s /var/www/cobbler/localmirror/%s --progress" % (spacer, self.mirror, self.mirror_name)
            sub_process.call(cmd,shell=True)
            update_file = open(os.path.join(self.path,"update.sh"),"w+")
            update_file.write("#!/bin/sh")
@@ -186,7 +187,7 @@ class Importer:
                if tentative == filter_out:
                    fnames.remove(tentative)
        print "%s" % dirname
-       if not self.is_pxe_or_xen_dir(dirname):
+       if not self.is_pxe_or_virt_dir(dirname):
            return
        for x in fnames:
            if x.startswith("initrd"):
@@ -235,8 +236,8 @@ class Importer:
           return "x86"
        return "x86"
 
-   def is_pxe_or_xen_dir(self,dirname):
-       if dirname.find("pxe") != -1 or dirname.find("xen") != -1:
+   def is_pxe_or_virt_dir(self,dirname):
+       if dirname.find("pxe") != -1 or dirname.find("xen") != -1 or dirname.find("virt") != -1:
            return True
        return False