提交

PixivSearch/dao/bangumi.py

@@ -14,6 +14,7 @@ django.setup()
 from PixivSearch.model.config import mediaInfo, stat
 
 current_mediaInfo = mediaInfo(id=0, chn_name='null')
+
 flag = True
 
 
@@ -21,6 +22,7 @@ class bangumi(threading.Thread):
     begin = 0
     end = 0
     id = 0
+    flag = True
 
     def __init__(self, begin, end):
         threading.Thread.__init__(self)
@@ -47,11 +49,18 @@ class bangumi(threading.Thread):
         self.go(self.begin, self.end)
 
     def get(self):
+        global current_mediaInfo
         return current_mediaInfo
 
+    def stop(self):
+        global flag
+        flag = False
+
     def go(self, start, end):
+        global flag
+        flag = True
         for num in range(start, end):
-            if (flag):
+            if flag:
                 time.sleep(1)
                 logger.info("爬虫进度：%d" % num)
                 self.id = num

PixivSearch/urls.py

@@ -14,7 +14,7 @@ Including another URLconf
     2. Add a URL to urlpatterns:  url(r'^blog/', include('blog.urls'))
 """
 from django.conf.urls import url
-from PixivSearch.view import search, index, download, saveConfig, save, get, start
+from PixivSearch.view import search, index, download, saveConfig, save, get, start, stop
 
 urlpatterns = [
     url(r'^$', index),
@@ -24,4 +24,5 @@ urlpatterns = [
     url(r'^bangumi/save', save),
     url(r'^bangumi/get', get),
     url(r'^bangumi/start', start),
+    url(r'^bangumi/stop', stop)
 ]

PixivSearch/view.py

@@ -89,7 +89,7 @@ def save(request):
 
 
 def get(request):
-    return HttpResponse(str(bangumi.get(None).__str__()))
+    return HttpResponse(str(bangumi.get(bangumi).__str__()))
 
 
 def start(request):
@@ -98,3 +98,7 @@ def start(request):
     bangumi(begin=begin, end=end).start()
 
     return HttpResponse("start success")
+
+def stop(request):
+    bangumi.stop(bangumi)
+    return HttpResponse("stop success")