fix burst not work bug

binux · Jan 20, 2015 · 0f43efc · 0f43efc
1 parent 6b02ed5
commit 0f43efc
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyspider/database/local/projectdb.py b/pyspider/database/local/projectdb.py
@@ -24,20 +24,20 @@ def __init__(self, files):
                 continue
             self.projects[project['name']] = project
 
-    rate_re = re.compile(r'^\s*#\s*rate.*(\d+(\.\d+)?)', re.I)
-    burst_re = re.compile(r'^\s*#\s*burst.*(\d+(\.\d+)?)', re.I)
+    rate_re = re.compile(r'^\s*#\s*rate.*?(\d+(\.\d+)?)', re.I | re.M)
+    burst_re = re.compile(r'^\s*#\s*burst.*?(\d+(\.\d+)?)', re.I | re.M)
 
     def _build_project(self, filename):
         try:
             with open(filename) as fp:
                 script = fp.read()
-            m = self.rate_re.match(script)
+            m = self.rate_re.search(script)
             if m:
                 rate = float(m.group(1))
             else:
                 rate = 1
 
-            m = self.burst_re.match(script)
+            m = self.burst_re.search(script)
             if m:
                 burst = float(m.group(1))
             else: