소스 검색

add entrypoint

david 7 년 전
부모
커밋
25e667a66b
1개의 변경된 파일7개의 추가작업 그리고 4개의 파일을 삭제
  1. 7 4
      bookworm/web.py

+ 7 - 4
bookworm/web.py

@@ -6,6 +6,9 @@ import redis
 
 s3client = s3.client()
 app = Flask(__name__)
+pool = pg_simple.config_pool(dsn='dbname=david user=david')
+db = pg_simple.PgSimple(pool)
+r = redis.StrictRedis(host='localhost', port=6379, decode_responses=True)
 
 @app.route('/books/status', methods=['GET'])
 def status_books():
@@ -52,8 +55,8 @@ def fetch_books():
     return ''
 
 
-if __name__ == '__main__':
-    pool = pg_simple.config_pool(dsn='dbname=david user=david')
-    db = pg_simple.PgSimple(pool)
-    r = redis.StrictRedis(host='localhost', port=6379, decode_responses=True)
+def main():
     app.run(host='0.0.0.0', debug=True)
+
+if __name__ == '__main__':
+    main()