fix bugs in get_string_num

tomzhang · Feb 1, 2018 · 0e5071b · 0e5071b
1 parent 83b931b
commit 0e5071b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/spider/utilities/util_parse.py b/spider/utilities/util_parse.py
@@ -15,12 +15,12 @@
 ]
 
 
-def get_string_num(string):
+def get_string_num(string, ignore_sign=False):
     """
     get a float number from a string
     """
-    string_re = re.search(r"(?P<num>\d+(\.\d+)?)", get_string_strip(string.replace(",", ""), replace_char=""), flags=re.IGNORECASE)
-    return float(string_re.group("num")) if string_re else 0.0
+    string_re = re.search(r"(?P<sign>-*?)(?P<num>\d+(\.\d+)?)", get_string_strip(string.replace(",", ""), replace_char=""), flags=re.IGNORECASE)
+    return float((string_re.group("sign") if not ignore_sign else "") + string_re.group("num")) if string_re else 0.0
 
 
 def get_string_strip(string, replace_char=" "):