Merge pull request #68 from 0b01001001/dev

fix starlette static file bug

examples/starlette_demo.py

@@ -60,7 +60,7 @@ def get(self, request):
         Route('/ping', Ping),
         Mount('/api', routes=[
             Route('/predict/{luck:int}', predict, methods=['POST'])
-        ])
+        ]),
     ])
     api.register(app)
 

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name='spectree',
-    version='0.3.8',
+    version='0.3.9',
     author='Keming Yang',
     author_email='kemingy94@gmail.com',
     description=('generate OpenAPI document and validate request&response '

spectree/plugins/starlette_plugin.py

@@ -101,6 +101,9 @@ def find_routes(self):
         routes = []
 
         def parse_route(app, prefix=''):
+            # :class:`starlette.staticfiles.StaticFiles` doesn't have routes
+            if not app.routes:
+                return
             for route in app.routes:
                 if route.path.startswith(f'/{self.config.PATH}'):
                     continue

tests/test_plugin_starlette.py

@@ -4,6 +4,7 @@
 from starlette.endpoints import HTTPEndpoint
 from starlette.responses import JSONResponse
 from starlette.routing import Mount, Route
+from starlette.staticfiles import StaticFiles
 from starlette.testclient import TestClient
 
 from spectree import SpecTree, Response
@@ -60,7 +61,8 @@ async def user_score(request):
         Mount('/user', routes=[
             Route('/{name}', user_score, methods=['POST']),
         ])
-    ])
+    ]),
+    Mount('/static', app=StaticFiles(directory='docs'), name='static'),
 ])
 api.register(app)