
    di
                         d dl mZ d dlmZmZmZmZmZmZ d dl	m
Z
mZmZmZ d dlmZ d dlmZ  G d de      Z G d d	e      Zy
)    )Field)DOUBLEINTEGERFloat
ForeignKeyIntegerString)MappedSessionmapped_columnrelationship)Base)CustomModelc                       e Zd ZU  ed      Zedz  ed<    ed      Zedz  ed<    ed      Zedz  ed<    ed      Z	e
dz  ed<    ed      Zedz  ed<    ed      Zedz  ed<   y)	CatchmentPotentialBaseN)defaultcatch_idISEC_Segmentnumber_of_householdspercentage_of_segmentpotential_number_of_householdsgroup_id)__name__
__module____qualname__r   r   int__annotations__r   r   r   floatr   r        J/var/www/html/retail-simulation-api/app/models/main/catchment_potential.pyr   r      sy     .HcDj.$T2L#*2',T':#*:*/*=54<=383F"C4KF "$/HsTz/r    r   c            	          e Zd ZU dZ edd      Zee   ed<    ede	d      Z
ee   ed<    eed      Zee   ed<    eed      Zee   ed	<    eed      Zee   ed
<    ede	 ed      d      Zee   ed<    edd      Zedededd fd       Zedededdfd       Zededededdfd       Zedededd fd       Zy)TblCatchmentPotentialtbl_catchment_potentialT)primary_keyautoincrementr   r   F)nullabler   r   r   r   ztbl_group.group_idTblGroupcatchmentPotential)back_populatesdatadbreturnc                 x    |j                         } | di |}|j                  |       |j                          |S )Nr   )
model_dumpaddflush)clsr+   r,   	data_dictnew_datas        r!   createzTblCatchmentPotential.create   s5    OO%	##
x

r    zTblCatchmentPotential | Nonec                 x    |j                  |       j                  | j                  |k(        j                         S N)queryfilterr   all)r2   r   r,   s      r!   	get_by_idzTblCatchmentPotential.get_by_id'   s-    xx}##CLLH$<=AACCr    c                 4   |j                  |       j                  | j                  |k(        j                         }|sy |j	                         }|j                         D ]  \  }}|	t        |||        |j                          |j                  |       |S r7   )	r8   r9   r   firstr/   itemssetattrcommitrefresh)r2   r   r+   r,   get_datar3   keyvalues           r!   updatezTblCatchmentPotential.update+   s    88C=''(@AGGIOO%	#//+JC #u- , 			


8r    c                     |j                  |       j                  | j                  |k(        j                         }|sy|j	                  |       |j                          |S )NF)r8   r9   r   r=   deleter@   )r2   r   r,   objs       r!   rG   zTblCatchmentPotential.delete8   sK    hhsm""3<<8#;<BBD
		#
		
r    N)r   r   r   __tablename__r   r   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   groupclassmethodr   r   r5   r;   rE   rG   r   r    r!   r#   r#      sZ   -M)d$OHfSkO -ngPU VL&+V(5fu(M&+M+8+O6%=O2?QV2W"F3KW **gzJ^?_jnoHfSko4HIE0 g BY   D D' D6T D D 
c 
)? 
W 
Qo 
 
 c w 3J  r    r#   N)pydanticr   
sqlalchemyr   r   r   r   r   r	   sqlalchemy.ormr
   r   r   r   app.models.mainr   app.utils.schemas_utilsr   r   r#   r   r    r!   <module>rQ      s2     J J G G   /0[ 0.D .r    