1
0
mirror of https://github.com/ycd/manage-fastapi.git synced 2021-11-08 01:34:39 +03:00
Files
manage-fastapi/manage_fastapi/schemas.py
Marcelo Trylesinski d20bba6b44 Add database support
2020-12-07 14:42:54 -03:00

47 lines
1.2 KiB
Python

import subprocess
from datetime import datetime
from typing import Optional
from pydantic import BaseModel, EmailStr, root_validator
from manage_fastapi.config import FASTAPI_VERSION
from manage_fastapi.constants import Database, License, PackageManager, PythonVersion
class Context(BaseModel):
name: str
folder_name: str
packaging: PackageManager
username: Optional[str] = None
email: Optional[EmailStr] = None
python: PythonVersion
fastapi: str = FASTAPI_VERSION
license: License
year: int
pre_commit: bool
docker: bool
database: Database
@root_validator(pre=True)
def git_info(cls, values: dict):
try:
values["username"] = subprocess.check_output(
["git", "config", "--get", "user.name"]
)
values["email"] = subprocess.check_output(
["git", "config", "--get", "user.email"]
)
except subprocess.CalledProcessError:
...
values["folder_name"] = values["name"].lower().replace(" ", "-").strip()
values["year"] = datetime.today().year
return values
class Config:
use_enum_values = True