RasadDam_Backend/common/helpers.py

import random
import string
import typing
from apps.authorization.models import UserRelations


def detect_file_extension(file_name: str) -> typing.AnyStr:
    """ detect extension of a file like: jpg, png, pdf """
    extended = file_name.split('.')
    return extended[1]


def get_organization_by_user(user: object = None) -> typing.Any:
    """ get organization object by request user """
    organization = UserRelations.objects.select_related('organization').get(user=user).organization
    return organization


def generate_code(length=6):
    """ generate 6 digit code """
    return ''.join(random.choices(string.digits, k=length))