fix: Make the URL check more robust (fixes #32)

skorokithakis · Jun 15, 2017 · bd2f38e · bd2f38e
1 parent e60207b
commit bd2f38e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/shortuuid/main.py b/shortuuid/main.py
@@ -79,10 +79,10 @@ def uuid(self, name=None, pad_length=None):
         # If no name is given, generate a random UUID.
         if name is None:
             uuid = _uu.uuid4()
-        elif "http" not in name.lower():
-            uuid = _uu.uuid5(_uu.NAMESPACE_DNS, name)
-        else:
+        elif name.lower().startswith(("http://", "https://")):
             uuid = _uu.uuid5(_uu.NAMESPACE_URL, name)
+        else:
+            uuid = _uu.uuid5(_uu.NAMESPACE_DNS, name)
         return self.encode(uuid, pad_length)
 
     def random(self, length=None):