Fixed the conditional statements for image tests and cleaned up

the hostname handling.

1 files changed, 15 insertions, 5 deletions

diff --git a/cobbler/item_image.py b/cobbler/item_image.py
index 481951d8..b11dc9c0 100644
--- a/cobbler/item_image.py
+++ b/cobbler/item_image.py
@@ -119,6 +119,7 @@ class Image(item.Item):
         * hostname:/path/to/the/filename.ext
         * /path/to/the/filename.ext
         """
+        print "STARTING WITH FILENAME: %s" % filename
         uri = ""
         auth = hostname = path = ""
         # we'll discard the protocol if it's supplied, for legacy support
@@ -128,13 +129,22 @@ class Image(item.Item):
         else:
             uri = filename
 
-        if filename.find("@") != -1: auth, filename = filename.split("@")
-        if filename.find(":") != -1: hostname, filename = filename.split(":")
+        if filename.find("@") != -1:
+            auth, filename = filename.split("@")
+        if filename.find(":") != -1:
+            hostname, filename = filename.split(":")
+        elif filename[0] != '/':
+            index = filename.find("/")
+            hostname = filename[:index]
+            filename = filename[index:]
         # raise an exception if we don't have a valid path
-        if filename[0] != '/': raise CX(_("file contains an invalid path"))
-        if filename.find("/") != -1: path, filename = filename.rsplit("/", 1)
+        if filename[0] != '/':
+            raise CX(_("file contains an invalid path: %s" % filename))
+        if filename.find("/") != -1:
+            path, filename = filename.rsplit("/", 1)
 
-        if len(filename) == 0:  raise CX(_("missing filename"))
+        if len(filename) == 0:
+            raise CX(_("missing filename"))
         if len(auth) > 0 and len(hostname) == 0:
             raise CX(_("a hostname must be specified with authentication details"))