import datetime, pytz from typing import Dict, List, Optional, Set, Union from pydantic import BaseModel, EmailStr, Field, Json, PrivateAttr, ValidationError, validator from app.db_sql import redis_lookup_id_random from app.lib_general import log, logging from app.models.common_field_schema import base_fields, default_num_bytes # from app.models.account_models import Account_Base from app.models.contact_models import Contact_Base # from app.models.person_models import Person_Base # from app.models.user_models import User_Base # ### BEGIN ### API Organization Models ### Organization_Base() ### class Organization_Base(BaseModel): log.setLevel(logging.WARNING) # DEBUG, INFO, WARNING, ERROR, EXCEPTION, CRITICAL log.debug(locals()) id_random: Optional[str] = Field( **base_fields['organization_id_random'], alias = 'organization_id_random', ) id: Optional[int] = Field( alias = 'organization_id' ) account_id_random: Optional[str] account_id: Optional[int] contact_id_random: Optional[str] contact_id: Optional[int] person_id_random: Optional[str] person_id: Optional[int] user_id_random: Optional[str] user_id: Optional[int] name: Optional[str] tagline: Optional[str] description: Optional[str] company: Optional[bool] nonprofit: Optional[bool] industry: Optional[int] start_date: Optional[datetime.datetime] = None end_date: Optional[datetime.datetime] = None logo_path: Optional[str] logo_bg_color: Optional[str] thumbnail_path: Optional[str] thumbnail_bg_color: Optional[str] enable: Optional[bool] hide: Optional[bool] priority: Optional[int] sort: Optional[int] group: Optional[str] notes: Optional[str] created_on: Optional[datetime.datetime] = None updated_on: Optional[datetime.datetime] = None # Including other related objects #account: Optional[Account_Base] contact: Optional[Contact_Base] #person: Optional[Person_Base] #user: Optional[User_Base] _processed_at: datetime.datetime = PrivateAttr(default_factory=datetime.datetime.now) #@validator('organization_id_random', always=True) def organization_id_random_copy(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if values['id_random']: return values['id_random'] return None @validator('id', always=True) def organization_id_lookup(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if values['id_random']: log.debug(values['id_random']) return redis_lookup_id_random(record_id_random=values['id_random'], table_name='organization') return None @validator('account_id', always=True) def account_id_lookup(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if isinstance(v, int) and v > 0: return v elif id_random := values.get('account_id_random'): return redis_lookup_id_random(record_id_random=id_random, table_name='account') return None @validator('account_id_random', always=True) def account_id_random_lookup(cls, v, values, **kwargs): if isinstance(v, str) and len(v) >= 11: return v elif account_id := values.get('account_id'): return get_id_random(record_id=account_id, table_name='account') return None @validator('contact_id', always=True) def contact_id_lookup(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if values['contact_id_random']: return redis_lookup_id_random(record_id_random=values['contact_id_random'], table_name='contact') return None @validator('person_id', always=True) def person_id_lookup(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if values['person_id_random']: return redis_lookup_id_random(record_id_random=values['person_id_random'], table_name='person') return None @validator('user_id', always=True) def user_id_lookup(cls, v, values, **kwargs): log.setLevel(logging.WARNING) log.debug(locals()) if values['user_id_random']: return redis_lookup_id_random(record_id_random=values['user_id_random'], table_name='user') return None class Config: underscore_attrs_are_private = True allow_population_by_field_name = True fields = base_fields # ### END ### API Organization Models ### Organization_Base() ###