BLANK_SCHEME -> NO_SCHEME, option for custom value (#21)

* BLANK_SCHEME -> NO_SCHEME, option for custom value

Also use constant in test.

* Add docstring for param

domain_utils/domain_utils.py

@@ -3,7 +3,7 @@
 from tldextract import TLDExtract
 from urllib.parse import urlparse
 
-BLANK_SCHEME = 'blank'
+NO_SCHEME = 'no_scheme'
 
 
 def _load_and_update_extractor(function):
@@ -240,15 +240,17 @@ def get_stripped_url(url, scheme=False, drop_non_http=False, use_netloc=True, ex
     )
 
 
-def get_scheme(url):
-
+def get_scheme(url, no_scheme=NO_SCHEME):
     """
-    Given an url extract from it the scheme
+    Given an url, extract from it the scheme.
 
     Parameters
     ----------
     url: string
         The URL from where we want to get the scheme
+    no_scheme: any
+        The value to use if no scheme is detected.
+        Default is ``no_scheme``
 
     Returns
     ----------
@@ -261,4 +263,4 @@ def get_scheme(url):
     if scheme:
         return scheme
     else:
-        return BLANK_SCHEME
+        return no_scheme

tests/test_get_scheme.py

@@ -1,41 +1,45 @@
-from domain_utils import get_scheme
+from domain_utils import get_scheme, NO_SCHEME
 
 
-def test_get_scheme_no_scheme():
+def test_no_scheme():
     result = get_scheme("domain.net")
-    assert result == 'blank'
+    assert result == NO_SCHEME
 
 
-def test_get_scheme_file():
+def test_file():
     result = get_scheme("file:///home/user/index.html")
     assert result == 'file'
 
 
-def test_get_scheme_https():
+def test_https():
     result = get_scheme("https://domain.net")
     assert result == 'https'
 
 
-def test_get_scheme_http():
+def test_http():
     result = get_scheme("http://domain.net")
     assert result == 'http'
 
 
-def test_get_scheme_about():
+def test_about():
     result = get_scheme("about:config")
     assert result == 'about'
 
 
-def test_get_scheme_webpack():
+def test_webpack():
     result = get_scheme("webpack://index.js")
     assert result == 'webpack'
 
 
-def test_get_scheme_empty():
+def test_empty():
     result = get_scheme("")
-    assert result == 'blank'
+    assert result == NO_SCHEME
 
 
-def test_get_scheme_ws():
+def test_ws():
     result = get_scheme("ws://socket")
     assert result == 'ws'
+
+
+def test_pass_custom_empty_value():
+    assert get_scheme('10.0.0.1/path/to/index.html', no_scheme=None) is None