def index(request):
if request.method == 'POST':
login_form = LoginForm(request.POST)
if login_form.is_valid():
environment = login_form.cleaned_data['environment']
oauth_url = 'https://login.salesforce.com/services/oauth2/authorize'
if environment == 'Sandbox':
oauth_url = 'https://test.salesforce.com/services/oauth2/authorize'
oauth_url = oauth_url + '?response_type=code&client_id=' + settings.SALESFORCE_CONSUMER_KEY + '&redirect_uri=' + settings.SALESFORCE_REDIRECT_URI + '&state='+ environment
return HttpResponseRedirect(oauth_url)
else:
login_form = LoginForm()
return render_to_response('index.html', RequestContext(request,{'login_form': login_form}))
def oauth_response(request):
error_exists = False
error_message = ''
username = ''
org_name = ''
if request.GET:
oauth_code = request.GET.get('code')
environment = request.GET.get('state')
access_token = ''
instance_url = ''
org_id = ''
if 'Production' in environment:
login_url = 'https://login.salesforce.com'
else:
login_url = 'https://test.salesforce.com'
r = requests.post(login_url + '/services/oauth2/token', headers={ 'content-type':'application/x-www-form-urlencoded'}, data={'grant_type':'authorization_code','client_id': settings.SALESFORCE_CONSUMER_KEY,'client_secret':settings.SALESFORCE_CONSUMER_SECRET,'redirect_uri': settings.SALESFORCE_REDIRECT_URI,'code': oauth_code})
auth_response = json.loads(r.text)
if 'error_description' in auth_response:
error_exists = True
error_message = auth_response['error_description']
else:
access_token = auth_response['access_token']
instance_url = auth_response['instance_url']
user_id = auth_response['id'][-18:]
org_id = auth_response['id'][:-19]
org_id = org_id[-18:]
# get username of the authenticated user
r = requests.get(instance_url + '/services/data/v' + str(settings.SALESFORCE_API_VERSION) + '.0/sobjects/User/' + user_id + '?fields=Username', headers={'Authorization': 'OAuth ' + access_token})
query_response = json.loads(r.text)
username = query_response['Username']
# get the org name of the authenticated user
r = requests.get(instance_url + '/services/data/v' + str(settings.SALESFORCE_API_VERSION) + '.0/sobjects/Organization/' + org_id + '?fields=Name', headers={'Authorization': 'OAuth ' + access_token})
if 'Name' in json.loads(r.text):
org_name = json.loads(r.text)['Name']
else:
org_name = ''
login_form = LoginForm(initial={'environment': environment, 'access_token': access_token, 'instance_url': instance_url, 'org_id': org_id, 'package_option': 'all'})
if request.POST:
login_form = LoginForm(request.POST)
if login_form.is_valid():
environment = login_form.cleaned_data['environment']
access_token = login_form.cleaned_data['access_token']
instance_url = login_form.cleaned_data['instance_url']
org_id = login_form.cleaned_data['org_id']
package_option = login_form.cleaned_data['package_option']
if 'logout' in request.POST:
r = requests.post(instance_url + '/services/oauth2/revoke', headers={'content-type':'application/x-www-form-urlencoded'}, data={'token': access_token})
instance = instance_url.replace('https://','').replace('.salesforce.com','')
return HttpResponseRedirect('/logout?instance=' + instance)
if 'get_components' in request.POST:
# create the package record to store results
package = Package()
package.random_id = uuid.uuid4()
package.created_date = datetime.datetime.now()
package.username = org_id
package.api_version = str(settings.SALESFORCE_API_VERSION) + '.0'
package.access_token = access_token
package.instance_url = instance_url
package.component_option = package_option
package.status = 'Running'
package.save()
# Queue job to run async
try:
query_components_from_org.delay(package)
except:
# If fail above, wait 5 seconds and try again. Not ideal but should work for now
sleep(5)
try:
query_components_from_org.delay(package)
except:
# Sleep another 5
sleep(5)
query_components_from_org.delay(package)
return HttpResponseRedirect('/loading/' + str(package.random_id))
return render_to_response('oauth_response.html', RequestContext(request,{'error': error_exists, 'error_message': error_message, 'username': username, 'org_name': org_name, 'login_form': login_form}))
