mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-11-28 22:54:25 -05:00
15f79d1586
* fix user creation when signups are supposed to be diabled * add user registration tests * run formatter * fix test filename --------- Co-authored-by: Michael Genson <71845777+michael-genson@users.noreply.github.com>
105 lines
4.2 KiB
Python
105 lines
4.2 KiB
Python
from logging import Logger
|
|
from uuid import uuid4
|
|
|
|
from fastapi import HTTPException, status
|
|
|
|
from mealie.core.security import hash_password
|
|
from mealie.lang.providers import Translator
|
|
from mealie.repos.repository_factory import AllRepositories
|
|
from mealie.schema.group.group_preferences import CreateGroupPreferences
|
|
from mealie.schema.user.registration import CreateUserRegistration
|
|
from mealie.schema.user.user import GroupBase, GroupInDB, PrivateUser, UserIn
|
|
from mealie.services.group_services.group_service import GroupService
|
|
from mealie.services.seeder.seeder_service import SeederService
|
|
|
|
|
|
class RegistrationService:
|
|
logger: Logger
|
|
repos: AllRepositories
|
|
|
|
def __init__(self, logger: Logger, db: AllRepositories, translator: Translator):
|
|
self.logger = logger
|
|
self.repos = db
|
|
self.t = translator.t
|
|
|
|
def _create_new_user(self, group: GroupInDB, new_group: bool) -> PrivateUser:
|
|
new_user = UserIn(
|
|
email=self.registration.email,
|
|
username=self.registration.username,
|
|
password=hash_password(self.registration.password),
|
|
full_name=self.registration.username,
|
|
advanced=self.registration.advanced,
|
|
group=group.name,
|
|
can_invite=new_group,
|
|
can_manage=new_group,
|
|
can_organize=new_group,
|
|
)
|
|
|
|
# TODO: problem with repository type, not type here
|
|
return self.repos.users.create(new_user) # type: ignore
|
|
|
|
def _register_new_group(self) -> GroupInDB:
|
|
group_data = GroupBase(name=self.registration.group)
|
|
|
|
group_preferences = CreateGroupPreferences(
|
|
group_id=uuid4(),
|
|
private_group=self.registration.private,
|
|
first_day_of_week=0,
|
|
recipe_public=not self.registration.private,
|
|
recipe_show_nutrition=self.registration.advanced,
|
|
recipe_show_assets=self.registration.advanced,
|
|
recipe_landscape_view=False,
|
|
recipe_disable_comments=self.registration.advanced,
|
|
recipe_disable_amount=self.registration.advanced,
|
|
)
|
|
|
|
return GroupService.create_group(self.repos, group_data, group_preferences)
|
|
|
|
def register_user(self, registration: CreateUserRegistration) -> PrivateUser:
|
|
self.registration = registration
|
|
|
|
if self.repos.users.get_by_username(registration.username):
|
|
raise HTTPException(status.HTTP_409_CONFLICT, {"message": self.t("exceptions.username-conflict-error")})
|
|
elif self.repos.users.get_one(registration.email, "email"):
|
|
raise HTTPException(status.HTTP_409_CONFLICT, {"message": self.t("exceptions.email-conflict-error")})
|
|
|
|
token_entry = None
|
|
new_group = False
|
|
|
|
if registration.group_token and registration.group_token != "":
|
|
token_entry = self.repos.group_invite_tokens.get_one(registration.group_token)
|
|
if not token_entry:
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST, {"message": "Invalid group token"})
|
|
|
|
maybe_none_group = self.repos.groups.get_one(token_entry.group_id)
|
|
|
|
if maybe_none_group is None:
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST, {"message": "Invalid group token"})
|
|
|
|
group = maybe_none_group
|
|
elif registration.group:
|
|
new_group = True
|
|
group = self._register_new_group()
|
|
else:
|
|
raise HTTPException(status.HTTP_400_BAD_REQUEST, {"message": "Missing group"})
|
|
|
|
self.logger.info(f"Registering user {registration.username}")
|
|
user = self._create_new_user(group, new_group)
|
|
|
|
if new_group and registration.seed_data:
|
|
seeder_service = SeederService(self.repos, user, group)
|
|
seeder_service.seed_foods(registration.locale)
|
|
seeder_service.seed_labels(registration.locale)
|
|
seeder_service.seed_units(registration.locale)
|
|
|
|
if token_entry and user:
|
|
token_entry.uses_left = token_entry.uses_left - 1
|
|
|
|
if token_entry.uses_left == 0:
|
|
self.repos.group_invite_tokens.delete(token_entry.token)
|
|
|
|
else:
|
|
self.repos.group_invite_tokens.update(token_entry.token, token_entry)
|
|
|
|
return user
|