dschandra commited on
Commit
998b660
·
verified ·
1 Parent(s): d077798

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -13,13 +13,27 @@ SF_DOMAIN = "login" # Use "test" if using a sandbox
13
  # Connect to Salesforce
14
  sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN, domain=SF_DOMAIN)
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  @app.route('/')
17
  def home():
18
  return render_template('Menu.html')
19
 
20
  @app.route('/get_menu', methods=['GET'])
21
  def get_menu():
22
- return jsonify({"message": "Menu endpoint is working!"})
 
23
 
24
  if __name__ == "__main__":
25
- app.run(host="0.0.0.0", port=7860, debug=True)
 
13
  # Connect to Salesforce
14
  sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN, domain=SF_DOMAIN)
15
 
16
+ def fetch_menu_items():
17
+ query = "SELECT Name, Image1__c, Image2__c, Ingredients__c, Price__c, Category__c, Section__c FROM Menu_Item__c"
18
+ headers = {
19
+ "Authorization": f"Bearer {ACCESS_TOKEN}",
20
+ "Content-Type": "application/json"
21
+ }
22
+ response = requests.get(f"{SALESFORCE_URL}?q={query}", headers=headers)
23
+
24
+ if response.status_code == 200:
25
+ return response.json()["records"]
26
+ else:
27
+ return []
28
+
29
  @app.route('/')
30
  def home():
31
  return render_template('Menu.html')
32
 
33
  @app.route('/get_menu', methods=['GET'])
34
  def get_menu():
35
+ menu_items = fetch_menu_items()
36
+ return jsonify(menu_items)
37
 
38
  if __name__ == "__main__":
39
+ app.run(host="0.0.0.0", port=7860, debug=True, use_reloader=False)