Skip to content

Commit

Permalink
(PC-32194)[API] feat: add user and user_offerer for offerers without …
Browse files Browse the repository at this point in the history
…one in industrial sandbox
  • Loading branch information
jcicurel-pass committed Oct 10, 2024
1 parent 8e6c5a6 commit 35bcdeb
Show file tree
Hide file tree
Showing 2 changed files with 35 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@
)
from pcapi.sandboxes.scripts.creators.industrial.create_offers_with_status import create_offers_with_status
from pcapi.sandboxes.scripts.creators.industrial.create_special_events import create_special_events
from pcapi.sandboxes.scripts.creators.industrial.create_user_offerers import create_user_offerers


def save_industrial_sandbox() -> None:
Expand Down Expand Up @@ -154,3 +155,6 @@ def save_industrial_sandbox() -> None:
create_industrial_commercial_gestures()

create_special_events()

# run this last as we fill out missing user offerers
create_user_offerers()
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
import logging

from pcapi.core.offerers import factories as offerers_factories
from pcapi.core.offerers import models as offerers_models
from pcapi.core.users import factories as users_factories


logger = logging.getLogger(__name__)


def create_user_offerers() -> None:
# Add a user and user_offerer for each offerer without one, except if the offerer has a provider
logger.info("create_user_offerers")
user = None
count = 0

query = (
offerers_models.Offerer.query.outerjoin(offerers_models.Offerer.UserOfferers)
.outerjoin(offerers_models.Offerer.offererProviders)
.filter(offerers_models.UserOfferer.id.is_(None), offerers_models.OffererProvider.id.is_(None))
)
for offerer in query:
if user is None:
user = users_factories.ProFactory(
email="[email protected]", firstName="Compte pro", lastName="Rataché"
)

offerers_factories.UserOffererFactory(offerer=offerer, user=user)
count += 1

logger.info("created %d user offerers", count)

0 comments on commit 35bcdeb

Please sign in to comment.