o
    5h                     @   s   d dl mZ d dlmZ d dlZddlmZ d dlZddlm	Z	m
Z
mZ d dlZdd Zd	d
 Zdd Zdd Zdd Zdd Zdd ZdS )    )HttpResponse)peopleN   )FileSerializer)recognize_facesadd_userencode_known_facec                 C   s  | j dkrxzZt| jd|id}| rVtjj|d}dt|j }t	j
|s,t	| |jd }t|t|j}|rIddt|j|jd	d
}nddt|j|jd	d
}tt|ddW S  tjyw   ddd}tt|dd Y S w ddd}tt|ddS )NPOST	people_iddatacontextpk	training/image_1zSucces train200001idnamemessagecoder   Face not found400002application/jsoncontent_typePeople Not Found400001r   r   )methodr   FILESis_validr   objectsgetstrr   ospathexistsmakedirsvalidated_datar   r   r   jsondumpsDoesNotExist)requestr
   
serializerpeopleModelr(   filetrainr    r4   ,/var/www/html/face-recog_v2/peoples/views.pyr3   	   sD   



r3   c                 O   s  | j dkr}t }t| jd|id}| rdtjj|| j	d dd}dt
|j }tj|s4t| |jd }t||}|rNd	d
t
|j|jdd}	nddt
|j|jdd}	tt|	ddS t|j d d }
|
dd}	tt|	ddS tdddS )Nr	   r
   r   r   g333333?)r   r   	tolerancer   r   zSucces creater   r   r   r   r   r   r   r   400003r    Method not allowed  status)r!   uuiduuid4r   r"   r#   r   r$   creater	   r&   r   r'   r(   r)   r*   r+   r   r   r   r,   r-   listerrorsvalues)r/   argskwargsprimary_uuidr0   personr(   r2   addr   error_messager4   r4   r5   create_person7   s<   




rH   c                 C   s  | j dkrzjtjj|d}t| jd|id}| rV|jd }t||j	|j
}t|t|j	krGddt|j	|jdd	}tt|d
dW S ddd}tt|d
dW S t|j d d }|dd}tt|d
dW S  tjy   ddd}tt|d
d Y S w tdddS )Nr	   r   r
   r   r   zImage foundr   r   r   r   r   zImage not foundr   r    r   r7   r   r   r8   r9   r:   )r!   r   r$   r%   r   r"   r#   r+   r   r   r6   r&   r   r   r,   r-   r?   r@   rA   r.   )r/   r
   r1   r0   r2   tesr   rG   r4   r4   r5   	facerecogb   s>   

rJ   c                 C      t | d S Nprintr/   r
   r4   r4   r5   add_face_left      rP   c                 C   rK   rL   rM   rO   r4   r4   r5   add_face_right   rQ   rR   c                 C   rK   rL   rM   rO   r4   r4   r5   add_face_up   rQ   rS   c                 C   rK   rL   rM   rO   r4   r4   r5   add_face_down   rQ   rT   )django.httpr   peoples.modelsr   r,   serializersr   r'   utilitysr   r   r   r<   r3   rH   rJ   rP   rR   rS   rT   r4   r4   r4   r5   <module>   s    .++