U
    2\j&                     @   s   d Z ddlmZmZmZmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZmZmZ ddlmZmZmZ ed	d
gdZejdedee	feedddZejdedee	feedddZdS )u   注册 / 登录路由。    )	APIRouterDependsHTTPExceptionstatus)Session   )get_db)User)
RegisterInLoginInTokenOut)hash_passwordverify_passwordcreate_access_tokenz	/api/authauth)prefixtagsz	/register)response_model)bodydbc                 C   sp   | ttj| jk r(ttjdt| jt| j	d}|
| |  || t|j}t||jdS )Nu   用户名已存在)usernamepassword_hashaccess_tokenr   )queryr	   filterr   firstr   r   HTTP_409_CONFLICTr   passwordaddcommitrefreshr   idr   r   r   usertoken r&   &/root/pica-gateway/app/routers/auth.pyregister   s    


r(   z/loginc                 C   sR   | ttj| jk }|r.t| j|js:tt	j
dt|j}t||jdS )Nu   用户名或密码错误r   )r   r	   r   r   r   r   r   r   r   r   HTTP_401_UNAUTHORIZEDr   r"   r   r#   r&   r&   r'   login   s
    
r*   N)__doc__fastapir   r   r   r   sqlalchemy.ormr   databaser   modelsr	   schemasr
   r   r   securityr   r   r   routerpostr(   r*   r&   r&   r&   r'   <module>   s   