129 lines
4.1 KiB
Python
Executable File
129 lines
4.1 KiB
Python
Executable File
# Create your views here.
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth import authenticate, login, logout
|
|
from ox.django.shortcuts import render_to_json_response, json_response
|
|
from django import forms
|
|
import json
|
|
|
|
def register(request):
|
|
'''
|
|
param data {
|
|
username: 'username',
|
|
password: 'password',
|
|
email: 'emailaddress'
|
|
}
|
|
|
|
return {
|
|
status: {'code': int, 'text': string}
|
|
data: {
|
|
errors: {
|
|
username: 'Unknown Username',
|
|
password: 'Incorrect Password'
|
|
}
|
|
user: {
|
|
...
|
|
}
|
|
}
|
|
}
|
|
'''
|
|
data = json.loads(request.POST['data'])
|
|
form = RegisterForm(data, request.FILES)
|
|
if form.is_valid():
|
|
if models.User.objects.filter(username=form.data['username']).count() > 0:
|
|
response = json_response({
|
|
'errors': {
|
|
'username': 'Username already exists'
|
|
}
|
|
})
|
|
elif models.User.objects.filter(email=form.data['email']).count() > 0:
|
|
response = json_response({
|
|
'errors': {
|
|
'email': 'Email address already exits'
|
|
}
|
|
})
|
|
elif not form.data['password']:
|
|
response = json_response({
|
|
'errors': {
|
|
'password': 'Password can not be empty'
|
|
}
|
|
})
|
|
else:
|
|
first_user = models.User.objects.count() == 0
|
|
user = models.User(username=form.data['username'], email=form.data['email'])
|
|
user.set_password(form.data['password'])
|
|
#make first user admin
|
|
user.is_superuser = first_user
|
|
user.is_staff = first_user
|
|
user.save()
|
|
user = authenticate(username=form.data['username'],
|
|
password=form.data['password'])
|
|
login(request, user)
|
|
user_json = models.get_user_json(user)
|
|
response = json_response({
|
|
'user': user_json
|
|
}, text='account created')
|
|
else:
|
|
response = json_response(status=400, text='invalid data')
|
|
return render_to_json_response(response)
|
|
|
|
|
|
class LoginForm(forms.Form):
|
|
username = forms.TextInput()
|
|
password = forms.TextInput()
|
|
|
|
def api_login(request):
|
|
'''
|
|
param data {
|
|
username: 'username',
|
|
password: 'password'
|
|
}
|
|
|
|
return {
|
|
status: {'code': 200, 'text': 'ok'}
|
|
data: {
|
|
errors: {
|
|
username: 'Unknown Username',
|
|
password: 'Incorrect Password'
|
|
}
|
|
user: {
|
|
...
|
|
}
|
|
}
|
|
}
|
|
'''
|
|
data = json.loads(request.POST['data'])
|
|
form = LoginForm(data, request.FILES)
|
|
if form.is_valid():
|
|
if models.User.objects.filter(username=form.data['username']).count() == 0:
|
|
response = json_response({
|
|
'errors': {
|
|
'username': 'Unknown Username'
|
|
}
|
|
})
|
|
else:
|
|
user = authenticate(username=data['username'], password=data['password'])
|
|
if user is not None:
|
|
if user.is_active:
|
|
login(request, user)
|
|
user_json = models.get_user_json(user)
|
|
response = json_response({
|
|
'user': user_json
|
|
})
|
|
else:
|
|
response = json_response({
|
|
'errors': {
|
|
'username': 'User Disabled'
|
|
}
|
|
})
|
|
else:
|
|
response = json_response({
|
|
'errors': {
|
|
'password': 'Incorrect Password'
|
|
}
|
|
})
|
|
else:
|
|
response = json_response(status=400, text='invalid data')
|
|
return render_to_json_response(response)
|
|
actions.register(api_login, 'login')
|
|
|