Spaces:
Sleeping
Sleeping
File size: 4,059 Bytes
a7abf85 d3245ed 36d2eb6 b12f5e4 f4feb03 b12f5e4 444fe60 9a31124 e4e2d12 9a31124 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | from flask import Flask, render_template, request, jsonify, redirect, url_for, session
from flask_session import Session # Import the Session class
from flask.sessions import SecureCookieSessionInterface # Import the class
from salesforce import get_salesforce_connection
from flask_cors import CORS
import os
from flask import Flask, redirect, request, jsonify, session
import requests
import os
app = Flask(__name__)
app.secret_key = os.urandom(24) # Random secret key for session management
# Salesforce OAuth URLs and credentials
SALESFORCE_OAUTH_URL = "https://login.salesforce.com/services/oauth2/authorize"
SALESFORCE_TOKEN_URL = "https://login.salesforce.com/services/oauth2/token"
CLIENT_ID = '3MVG9WVXk15qiz1JbtW1tT9a7WojFUbAfMVyVXfvI4PISHAKAxmZ8RLS1lBHqpnaDPQPZOOInuVdcQpi7smWc' # Your Consumer Key
CLIENT_SECRET = '36C463CD713C420BA2ED78F853359EACCE1ECCE2954C9810FFD7F946564CB0E8' # Your Consumer Secret
REDIRECT_URI = 'https://huggingface.co/spaces/nagasurendra/BiryaniHubflask21' # Your Hugging Face redirect URI
INSTANCE_URL = 'https://biryanihub-dev-ed.develop.my.site.com/s/welcomePage' # Your Salesforce instance URL
# OAuth flow to redirect to Salesforce login
@app.route('/login')
def login():
oauth_url = f"{SALESFORCE_OAUTH_URL}?response_type=code&client_id={CLIENT_ID}&redirect_uri={REDIRECT_URI}"
return redirect(oauth_url)
# Handle the OAuth callback from Salesforce and exchange the authorization code for an access token
@app.route('/oauth/callback')
def oauth_callback():
auth_code = request.args.get('code') # The code returned from Salesforce
if not auth_code:
return jsonify({"error": "No authorization code provided"}), 400
# Exchange the authorization code for an access token
token_data = {
'grant_type': 'authorization_code',
'code': auth_code,
'client_id': CLIENT_ID,
'client_secret': CLIENT_SECRET,
'redirect_uri': REDIRECT_URI
}
token_response = requests.post(SALESFORCE_TOKEN_URL, data=token_data)
if token_response.status_code == 200:
token_info = token_response.json()
access_token = token_info['sSSjyhInIsUohKpG8sHzty2q']
instance_url = token_info['https://biryanihub-dev-ed.develop.my.site.com/s/welcomePage'] # The Salesforce instance URL
# Store access token in session for future API requests
session['access_token'] = access_token
session['instance_url'] = instance_url
return redirect('/dashboard') # Redirect user to the Hugging Face dashboard or home page
else:
return jsonify({"error": "Failed to authenticate with Salesforce"}), 400
# Example protected route that uses the Salesforce access token
@app.route('/dashboard')
def dashboard():
if 'access_token' not in session:
return redirect('/login') # If the user is not logged in, redirect to login
access_token = session['access_token']
instance_url = session['instance_url']
# Use the access token to make API calls to Salesforce (example: fetch user info)
headers = {'Authorization': f'Bearer {access_token}'}
user_info_url = f"{instance_url}/services/oauth2/userinfo"
user_info_response = requests.get(user_info_url, headers=headers)
if user_info_response.status_code == 200:
user_info = user_info_response.json()
return jsonify(user_info) # Display user info from Salesforce
else:
return jsonify({"error": "Failed to fetch user info from Salesforce"}), 400
if token_response.status_code == 200:
token_info = token_response.json()
access_token = token_info['access_token'] # Correctly fetch the access token
instance_url = token_info['instance_url'] # Correctly fetch the instance URL
# Store access token in session for future API requests
session['access_token'] = access_token
session['instance_url'] = instance_url
return redirect('/dashboard') # Redirect user to the Hugging Face dashboard or home page
if __name__ == '__main__':
app.run(debug=True)
|