Compare commits

..

No commits in common. "900131fae843202f8c1cccf30b2bf3aa68c87fad" and "4959f1987cd40d64ddf2c2f12d7a3a6dfae92ad7" have entirely different histories.

2 changed files with 12 additions and 32 deletions

View file

@ -32,38 +32,15 @@ class DBType(Enum):
class DBConfig(BaseModel): class DBConfig(BaseModel):
type: DBType = DBType.sqlite db_type: DBType = DBType.sqlite
user: str | None = None
password: str | None = None
host: str | None = None
database: str | None = "./tmp/vpn.db"
mysql_driver: str = "pymysql"
mysql_args: list[str] = ["charset=utf8mb4"]
@property @property
async def db_engine(self) -> Engine: async def db_engine(self) -> Engine:
if self.type is DBType.sqlite:
# SQLite backend
return create_engine( return create_engine(
f"sqlite:///{self.database}", "sqlite:///./tmp/vpn.db",
connect_args={"check_same_thread": False}, connect_args={"check_same_thread": False},
) )
elif self.type is DBType.mysql:
# MySQL backend
if self.mysql_args:
args_str = "?" + "&".join(self.mysql_args)
else:
args_str = ""
return create_engine(
f"mysql+{self.mysql_driver}://"
f"{self.user}:{self.password}@{self.host}"
f"/{self.database}{args_str}",
pool_recycle=3600,
)
class JWTConfig(BaseModel): class JWTConfig(BaseModel):
secret: str | None = None secret: str | None = None

View file

@ -48,17 +48,20 @@ async def set_config(
}, },
) )
async def add_user( async def add_user(
user: schemas.UserCreate, user_name: str,
user_password: str,
current_config: Config | None = Depends(Config.load), current_config: Config | None = Depends(Config.load),
db: Session | None = Depends(Connection.get), db: Session | None = Depends(Connection.get),
): ):
if current_config is None: if current_config is None:
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST) raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST)
user.capabilities.append("admin")
schemas.User.create( schemas.User.create(
db=db, db=db,
user=user, user=schemas.UserCreate(
name=user_name,
password=user_password,
capabilities=["admin"],
),
crypt_context=await current_config.crypto.crypt_context, crypt_context=await current_config.crypto.crypt_context,
) )