🔧 Add configuration on generated projects

manage_fastapi/constants.py

@@ -14,7 +14,7 @@ class BaseMetadataEnum(EnumMeta):
             return True
 
 
-class BaseEnum(Enum, metaclass=BaseMetadataEnum):
+class BaseEnum(str, Enum, metaclass=BaseMetadataEnum):
     """Base enum class."""
 
 

manage_fastapi/schemas.py

@@ -1,13 +1,19 @@
 import subprocess
 from datetime import datetime
+from enum import Enum
 from typing import Optional
 
 from pydantic import BaseModel, EmailStr, root_validator
+from pydantic.class_validators import validator
 
 from manage_fastapi.config import FASTAPI_VERSION
 from manage_fastapi.constants import PackageManager, PythonVersion
 
 
+def enum_to_str(enum: Enum) -> str:
+    return enum.value
+
+
 class Context(BaseModel):
     name: str
     folder_name: str
@@ -36,3 +42,5 @@ class Context(BaseModel):
         values["folder_name"] = values["name"].lower().replace(" ", "-").strip()
         values["year"] = datetime.today().year
         return values
+
+    _python = validator("python", allow_reuse=True)(enum_to_str)

manage_fastapi/templates/project/{{ cookiecutter.folder_name }}/app/config.py

@@ -0,0 +1,11 @@
+from pydantic import BaseSettings
+
+
+class Settings(BaseSettings):
+    PROJECT_NAME: str
+
+    class Config:
+        env_file = ".env"
+
+
+settings = Settings()

manage_fastapi/templates/project/{{ cookiecutter.folder_name }}/app/main.py

@@ -1,3 +1,5 @@
 from fastapi import FastAPI
 
-app = FastAPI()
+from app.config import settings
+
+app = FastAPI(title=settings.PROJECT_NAME)

setup.cfg

@@ -1,6 +1,7 @@
 [isort]
 profile = black
-known_first_party = manage_fastapi
+known_first_party = manage_fastapi,app
+skip = '{{ cookiecutter.folder_name }}/*'
 
 [flake8]
 max-complexity = 7