Update fancy.py

Procfile

@@ -1 +1 @@
-web: gunicorn fancy:app
+web: gunicorn fancy:app -t 3000

fancy.py

@@ -1,7 +1,7 @@
 import urllib.request
 import http.cookiejar
 from bs4 import BeautifulSoup
-import re
+import re, os
 from flask import Flask, render_template
 from flask_bootstrap import Bootstrap
 from pymongo import *
@@ -76,25 +76,32 @@ def get_info(links):
             print(item, ' processed')
         i+=1
         print(i, 'done,',len(links) - i ,'remaining')
+        if i >= 20:
+            return itemlist
+    print('信息收集完毕')
     return itemlist
 
 @app.before_first_request
 def update():
-    # client = MongoClient('mongodb://uPvyEVzH4jIWGhcn:[email protected]:27017/')
-    client = MongoClient()
-    db = client.test
+    client = MongoClient(os.environ.get('DATABASE_URL'))
+    db = client.fanci
     item_collection = db.item
-    itemlist = get_info(get_links())
-    item_collection.insert_many(itemlist)
+    if not item_collection.find_one():
+        itemlist = get_info(get_links())
+        item_collection.insert_many(itemlist)
+        print('信息已写入数据库')
+    else:
+        print('数据库已存在信息')
 
 @app.route('/')
 def index():
-    # client = MongoClient('mongodb://uPvyEVzH4jIWGhcn:[email protected]:27017/')
-    client = MongoClient()
-    db = client.test
+    client = MongoClient(os.environ.get('DATABASE_URL'))
+    # client = MongoClient()
+    db = client.fanci
     item_collection = db.item
     items = item_collection.find()
+    print('正从数据库取出信息')
     return render_template('index.html', items = items)
 
 if __name__ == '__main__':
-    app.run(debug=True,host='0.0.0.0', port = 3000)
+    app.run(debug=True,host='0.0.0.0', port = 3000)

requirements.txt

@@ -9,3 +9,4 @@ itsdangerous==0.24
 pymongo==3.2.2
 redis==2.10.5
 visitor==0.1.2
+gunicorn==19.6.0