How to get account given user and using base models?

[simkimsia]

i am using base models. not abstract.

class CustomUser(AbstractUser, LifecycleModel):
    pass
    # add additional fields in here

    def __str__(self):
        return self.username

    @hook(AFTER_CREATE)
    def create_account(self):
        account = Account(name=self.username)
        account.save()
        au = AccountUser(organization=account, user=self)
        au.save()
        ao = AccountOwner(organization=account, organization_user=au)
        ao.save()
        
class AccountUser(OrganizationUserBase):
    class USER_TYPE(models.TextChoices):

        # first value is db value
        # second value is human readable label, you need this in case you have multiple words
        OWNER = "OWNER", "Owner"
        MEMBER = "MEMBER", "Member"

    user_type = models.CharField(
        max_length=20, choices=USER_TYPE.choices, default=USER_TYPE.OWNER
    )

I want to have a filter queryset to get the Account given the CustomUser object. What's the easiest way?

[simkimsia]

Both models above are in the app called core in Django 3.2

So what I did is I have a function in core/services.py

def get_account_by_user(user: CustomUser) -> Account:
    """
    given CustomUser, get the associated Account via core_accountuser relation
    """
    return user.core_accountuser.first().organization

see #233 for an application of this method