[Python] FastAPI: Using a common module app.py for socketify tests

Author: remittor

frameworks/Python/fastapi/app-socketify-asgi.py

@@ -1,7 +1,12 @@
 import multiprocessing
 from contextlib import asynccontextmanager
 
-import asyncpg
+try:
+    import asyncpg
+    _DB_ALLOWED = True
+except ImportError:
+    _DB_ALLOWED = False
+
 import os
 from fastapi import FastAPI, Request
 from fastapi.responses import PlainTextResponse
@@ -34,8 +39,6 @@ def get_num_queries(queries):
     return query_count
 
 
-connection_pool = None
-
 templates = Jinja2Templates(directory="templates")
 
 
@@ -60,7 +63,7 @@ async def lifespan(app: FastAPI):
     await app.state.connection_pool.close()
 
 
-app = FastAPI(lifespan=lifespan)
+app = FastAPI(lifespan=lifespan) if _DB_ALLOWED else FastAPI()
 
 
 @app.get("/json")

frameworks/Python/fastapi/app.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+import os
+import sys
+import fastapi
+
+from app import app
+
+if __name__ == "__main__":
+    import optparse
+    import multiprocessing
+    import logging
+
+    parser = optparse.OptionParser("usage: %prog [options]", add_help_option=False)
+    parser.add_option("-h", "--host", dest="host", default='0.0.0.0', type="string")
+    parser.add_option("-p", "--port", dest="port", default=8080, type="int")
+    parser.add_option("-s", "--server", dest="server", default="", type="string")
+    parser.add_option("-w", "--workers", dest="workers", default=0, type="int")
+    parser.add_option("-k", "--keepalive", dest="keepalive", default=60, type="int")
+    parser.add_option("-v", "--verbose", dest="verbose", default=0, type="int")
+    (opt, args) = parser.parse_args() 
+
+    _is_travis = os.environ.get('TRAVIS') == 'true'
+
+    workers = opt.workers
+    if workers <= 0:
+        workers = int(multiprocessing.cpu_count())
+
+    if _is_travis:
+        workers = 2
+
+    def run_app():
+        if opt.server in [ 'si', 'socketify' ]:
+            import socketify
+            siapp = socketify.ASGI(app)
+            siapp.listen(opt.port, lambda config: logging.info(f"Listening on port http://localhost:{opt.port} now\n"))
+            siapp.run()
+            return
+        
+        raise Exception(f'Unknown server name = "{opt.server}"')
+
+    def create_fork():
+        n = os.fork()
+        # n greater than 0 means parent process
+        if not n > 0:
+            run_app()
+
+    # fork limiting the cpu count - 1
+    for i in range(1, workers):
+        create_fork()
+
+    run_app()  # run app on the main process too :) 

frameworks/Python/fastapi/app_server.py

@@ -13,4 +13,4 @@ RUN pip3 install -r requirements-socketify-pypy.txt
 
 EXPOSE 8080
 
-CMD python ./app-socketify-asgi.py
+CMD python app_server.py -s socketify

frameworks/Python/fastapi/fastapi-socketify-asgi-pypy.dockerfile

@@ -16,4 +16,4 @@ EXPOSE 8080
 
 ENV USE_ORJSON=1
 
-CMD python ./app-socketify-asgi.py
+CMD python app_server.py -s socketify