diff --git a/app/__init__.py b/app/__init__.py index ebe2384..13aa835 100644 --- a/app/__init__.py +++ b/app/__init__.py @@ -27,15 +27,3 @@ def create_app(): return app - -def download(): - res = requests.get('http://resource-base.herokuapp.com/download/data.db') - with open('./app/db/data.db', 'wb') as fp: - fp.write(res.content) - - -if __name__ != '__main__': - if not os.path.exists('./app/db/data.db'): - print('数据库文件未找到, 正在下载中') - download() - print('下载完成') diff --git a/manage.py b/manage.py index 4ca9213..74068cb 100644 --- a/manage.py +++ b/manage.py @@ -6,12 +6,25 @@ # @File Name: manage.py +import os +import requests from gevent import pywsgi from app import create_app app = create_app() +def download(): + res = requests.get('http://resource-base.herokuapp.com/download/data.db') + with open('./app/db/data.db', 'wb') as fp: + fp.write(res.content) + + +if not os.path.exists('./app/db/data.db'): + print('数据库文件未找到, 正在下载中') + download() + print('下载完成') + if __name__ == '__main__': server = pywsgi.WSGIServer(('0.0.0.0', 5000), app) server.serve_forever()