a
    h                     @   sR   d dl mZ d dlmZ G dd dejZG dd dejZG dd dejZd	S )
    )models)Userc                   @   s   e Zd Zg dZejeejddZej	ddZ
ej	deddZejd	d
ZejddZej	ddZejd	d
ZejedZejd	dZejd	dZG dd dZdd ZdS )MedicalRecord))Z
lab_reportz
Lab Report)ZprescriptionZPrescription)ZxrayzX-Ray)ZmriZMRI)Zct_scanzCT Scan)Zdoctor_notezDoctor Note)otherOtherZmedical_records	on_deleterelated_name   )
max_length2   r   )r   choicesdefaultTblankzmedical_records/%Y/%m/%d/)	upload_to
   )r   auto_now_addauto_nowc                   @   s   e Zd ZdgZdS )zMedicalRecord.Metaz-created_atN__name__
__module____qualname__ordering r   r   1/home/qiu/health_record_app/backend/api/models.pyMeta   s   r   c                 C   s   | j  d| jj S )N - )titleuserusernameselfr   r   r   __str__   s    zMedicalRecord.__str__N)r   r   r   ZRECORD_TYPESr   
ForeignKeyr   CASCADEr!   	CharFieldr    Zrecord_type	TextFielddescription	FileFieldfileZ	file_typeZextracted_text	JSONFielddictmetadataDateTimeField
created_at
updated_atr   r%   r   r   r   r   r      s   
r   c                   @   sZ   e Zd ZejeejddZejdddZ	ej
ddZej
ddZG dd	 d	Zd
d ZdS )ChatSessionZchat_sessionsr   r
   T)r   r   r   r   c                   @   s   e Zd ZdgZdS )zChatSession.Metaz-updated_atNr   r   r   r   r   r   (   s   r   c                 C   s   d| j  d| jj S )NzChat r   )idr!   r"   r#   r   r   r   r%   +   s    zChatSession.__str__N)r   r   r   r   r&   r   r'   r!   r(   r    r0   r1   r2   r   r%   r   r   r   r   r3   "   s   r3   c                   @   sl   e Zd ZddgZejeejddZej	dedZ
e ZejeddZejdd	ZG d
d dZdd ZdS )ChatMessage)r!   r   )Z	assistantZ	Assistantmessagesr   r   )r   r   Tr   r   c                   @   s   e Zd ZdgZdS )zChatMessage.Metar1   Nr   r   r   r   r   r   ;   s   r   c                 C   s   | j  d| jd d  dS )Nz: r   z...)rolecontentr#   r   r   r   r%   >   s    zChatMessage.__str__N)r   r   r   ZROLE_CHOICESr   r&   r3   r'   sessionr(   r7   r)   r8   ManyToManyFieldr   Zreferenced_recordsr0   r1   r   r%   r   r   r   r   r5   /   s   r5   N)	django.dbr   django.contrib.auth.modelsr   Modelr   r3   r5   r   r   r   r   <module>   s   