Fix get_model

This commit is contained in:
long2ice
2021-06-03 21:56:29 +08:00
parent 6f95f66393
commit 42c7351914
2 changed files with 106 additions and 71 deletions

View File

@ -14,7 +14,8 @@ def get_model(resource: Optional[str] = Path(...)):
if not resource:
return
for app, models in Tortoise.apps.items():
model = models.get(resource.title())
models = {key.lower(): val for key, val in models.items()}
model = models.get(resource)
if model:
return model