def upload_to_sfdc(request):
gmail_service = Google.get_gmail_service(request.session['user_id'])
attachment_id = request.GET['att_id']
message_id = request.GET['msg_id']
file_name = request.GET['file_name']
attachment = gmail_service.users().messages().attachments().get(userId='me',
messageId=message_id, id=attachment_id).execute()
file_data = attachment['data']
if file_data:
sf_auth = Salesforce.objects.get(user__id=request.session['user_id'])
Sfdc.insert_document(sf_auth, file_name, file_data)
return HttpResponse('success')
def sfdc_oauth_callback(request):
credentials = Sfdc.exchange_code(request.GET['code'])
try:
sfdc_auth = Salesforce.objects.get(user__id=request.session['user_id'])
except ObjectDoesNotExist:
sfdc_auth = None
if sfdc_auth is None:
user = MyUser.objects.get(id=request.session['user_id'])
sfdc_auth = Salesforce(user=user, access_token=credentials['access_token'],
refresh_token=credentials['refresh_token'],
sf_used_id=credentials['user_id'])
sfdc_auth.save()
else:
sfdc_auth.access_token = credentials['access_token']
sfdc_auth.refresh_token = credentials['refresh_token']
sfdc_auth.sf_used_id = credentials['user_id']
sfdc_auth.save()
return redirect(reverse('close_window'))
def sfdc_oauth(request):
return redirect(Sfdc.get_auth_url())