def register(request):
if request.user != None and request.user.is_authenticated():
return redirect('/')
if request.method == 'POST':
form = RegistrationForm(request.POST)
if form.is_valid():
#Create the user
owner_email = form.cleaned_data.get('owner_email')
passwd = form.cleaned_data.get('password1')
user = User(email=owner_email)
user.set_password(passwd)
user.save()
user.username = user.pk
user.save()
user.backend = 'user_backends.email_username.EmailOrUsernameModelBackend'
#create the enterprise
ep_name = form.cleaned_data.get('enterprise')
ep = Enterprise(name=ep_name)
ep.save()
#create the enterprise member
ep_member = EnterpriseMember(user=user, enterprise=ep)
ep_member.member_type = EnterpriseMember.MEMBER_TYPE.owner
ep_member.save()
#logs the new user
login(request, user)
return redirect('/')
else:
form = RegistrationForm()
return locals()
def register(request):
if request.user != None and request.user.is_authenticated():
return redirect('/')
if request.method == 'POST':
form = RegistrationForm(request.POST)
if form.is_valid():
#Create the user
owner_email = form.cleaned_data.get('owner_email')
passwd = form.cleaned_data.get('password1')
user = User(email=owner_email)
user.set_password(passwd)
user.save()
user.username = user.pk
user.save()
user.backend='user_backends.email_username.EmailOrUsernameModelBackend'
#create the enterprise
ep_name = form.cleaned_data.get('enterprise')
ep = Enterprise(name=ep_name)
ep.save()
#create the enterprise member
ep_member = EnterpriseMember(user=user,enterprise=ep)
ep_member.member_type = EnterpriseMember.MEMBER_TYPE.owner
ep_member.save()
#logs the new user
login(request,user)
return redirect('/')
else:
form = RegistrationForm()
return locals()
def addUser(request):
form = UserForm(request.POST)
if form.is_valid():
try:
with transaction.atomic():
enterprise = Enterprise()
enterprise.save()
request.session['idEnterprise'] = enterprise.id;
location= Location(enterprise=enterprise, lat=0, lng=0, name='Main Office')
location.save()
user = User(location = location, email=form.cleaned_data['email'],
password=form.cleaned_data['password'])
user.save()
request.session['idUser'] = user.id;
profile = Profile(user = user, role="Administrator")
profile.save()
return render(request, 'users/dashboard.html')
except Exception as e:
print(e)
messages.error(request, 'Sorry, Internal Error')
else:
messages.error(request, 'Please fill the form')
return HttpResponseRedirect('/signup')
def addUser(request):
form = UserForm(request.POST)
if form.is_valid():
try:
with transaction.atomic():
enterprise = Enterprise()
enterprise.save()
request.session['idEnterprise'] = enterprise.id
location = Location(enterprise=enterprise,
lat=0,
lng=0,
name='Main Office')
location.save()
user = User(location=location,
email=form.cleaned_data['email'],
password=form.cleaned_data['password'])
user.save()
request.session['idUser'] = user.id
profile = Profile(user=user, role="Administrator")
profile.save()
return render(request, 'users/dashboard.html')
except Exception as e:
print(e)
messages.error(request, 'Sorry, Internal Error')
else:
messages.error(request, 'Please fill the form')
return HttpResponseRedirect('/signup')
def add_member(request):
"""
Add a new member to a given enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
if request.method == 'POST':
form = NewEnterpriseMemberForm(request.POST)
if form.is_valid():
#Create the user
owner_email = form.cleaned_data.get('member_email')
passwd = form.cleaned_data.get('password1')
user = User(email=owner_email)
user.set_password(passwd)
user.save()
user.username = user.pk
user.backend = 'user_backends.email_username.EmailOrUsernameModelBackend'
user.save()
#create the enterprise member
ep_member = form.save(commit=False) #(user=user,enterprise=ep)
ep_member.user = user
ep_member.enterprise = enterprise
ep_member.save()
return redirect('/')
else:
form = NewEnterpriseMemberForm()
return locals()
def add_member(request):
"""
Add a new member to a given enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
if request.method == 'POST':
form = NewEnterpriseMemberForm(request.POST)
if form.is_valid():
#Create the user
owner_email = form.cleaned_data.get('member_email')
passwd = form.cleaned_data.get('password1')
user = User(email=owner_email)
user.set_password(passwd)
user.save()
user.username = user.pk
user.backend='user_backends.email_username.EmailOrUsernameModelBackend'
user.save()
#create the enterprise member
ep_member = form.save(commit=False)#(user=user,enterprise=ep)
ep_member.user = user
ep_member.enterprise = enterprise
ep_member.save()
return redirect('/')
else:
form = NewEnterpriseMemberForm()
return locals()
def manage_members(request):
"""
Show other options to manage the enterprise, like:
Add members and change their type
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
return locals()
def manage_projects(request):
"""
List the projects for this enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
projects = enterprise.projects.all()
return locals()
def manage_members(request):
"""
Show other options to manage the enterprise, like:
Add members and change their type
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
return locals()
def manage_projects(request):
"""
List the projects for this enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
projects = enterprise.projects.all()
return locals()
def wrapper(request, *args, **kwargs):
enterprise = Enterprise.get_from_user_or_404(request.user)
get_kwargs = {
'enterprise': enterprise,
'user': request.user,
}
if owner:
get_kwargs['member_type'] = EnterpriseMember.MEMBER_TYPE.owner
member = get_object_or_404(EnterpriseMember, **get_kwargs)
return function(request, *args, **kwargs)
def wrapper(request, *args, **kwargs):
enterprise = Enterprise.get_from_user_or_404(request.user)
get_kwargs = {
'enterprise': enterprise,
'user':request.user,
}
if owner:
get_kwargs['member_type'] = EnterpriseMember.MEMBER_TYPE.owner
member = get_object_or_404(EnterpriseMember, **get_kwargs)
return function(request, *args, **kwargs)
def add_project(request):
"""
Add a new project to a given enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
if request.method == 'POST':
form = NewProjectForm(enterprise, request.POST)
if form.is_valid():
new_project = form.save(commit=False)
new_project.enterprise = enterprise
new_project.save()
form.save_m2m()
return redirect('/')
else:
form = NewProjectForm(enterprise)
return locals()
def add_project(request):
"""
Add a new project to a given enterprise
"""
enterprise = Enterprise.get_from_user_or_404(request.user)
if request.method == 'POST':
form = NewProjectForm(enterprise,request.POST)
if form.is_valid():
new_project = form.save(commit=False)
new_project.enterprise = enterprise
new_project.save()
form.save_m2m()
return redirect('/')
else:
form = NewProjectForm(enterprise)
return locals()
