B JDb@spddlmZddlmZddlmZddlmZddlm Z GdddeZ Gdd d eZ e Z e Zd S) )Response)JSONWebTokenAPIView)ObtainAuthToken) StaffToken) serializersc@seZdZejZddZdS)StaffObtainAuthCompanyTokencOsJ|j|jd|id}|jdd|jd}tjj|d\}}td|jiS)Nrequest)datacontextT)Zraise_exceptionuser)r token) serializer_classr Zis_validZvalidated_datarZobjectsZ get_or_createrkey)selfrargskwargsZ serializerr r ZcreatedrGG:\easytimepro\master/mysite/staff\api\staff_authtoken_company_views.pyposts    z StaffObtainAuthCompanyToken.postN)__name__ __module__ __qualname__rZAuthTokenCompanySerializerr rrrrrr src@seZdZdZejZdS)StaffObtainJSONWebCompanyTokenz API View that receives a POST with a user's username and password. Returns a JSON Web Token that can be used for authenticated requests. N)rrr__doc__rZJSONWebTokenCompanySerializerr rrrrrsrN)Zrest_framework.responserZrest_framework_jwt.viewsrZrest_framework.authtoken.viewsrZmysite.staff.models.staff_tokenrZmysite.staff.apirrrZas_viewZobtain_auth_company_tokenZobtain_jwt_company_tokenrrrrs