rairo commited on
Commit
0fb2a5c
·
verified ·
1 Parent(s): 1345ea3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -2
main.py CHANGED
@@ -249,7 +249,7 @@ def get_spending_overview():
249
  # Modified verify_admin function (now checks database is_admin flag)
250
  # ========================================
251
  def verify_admin(auth_header):
252
- if not auth_header.startswith('Bearer '):
253
  raise ValueError('Invalid token')
254
 
255
  token = auth_header.split(' ')[1]
@@ -414,7 +414,11 @@ def view_receipt(transaction_id):
414
  verify_admin(request.headers.get('Authorization', ''))
415
  transaction_ref = db.reference(f'transactions/{transaction_id}')
416
  transaction_data = transaction_ref.get()
417
- return jsonify({'error': str(e)+f'transaction data:{transaction_data}'}), 500
 
 
 
 
418
 
419
  @app.route('/api/admin/receipt/<string:transaction_id>/download', methods=['GET'])
420
  def download_receipt(transaction_id):
 
249
  # Modified verify_admin function (now checks database is_admin flag)
250
  # ========================================
251
  def verify_admin(auth_header):
252
+ if not auth_header or not auth_header.startswith('Bearer '):
253
  raise ValueError('Invalid token')
254
 
255
  token = auth_header.split(' ')[1]
 
414
  verify_admin(request.headers.get('Authorization', ''))
415
  transaction_ref = db.reference(f'transactions/{transaction_id}')
416
  transaction_data = transaction_ref.get()
417
+ if not transaction_data:
418
+ return jsonify({'error': 'Transaction not found'}), 404
419
+
420
+ image_url = transaction_data.get('image_url')
421
+ return jsonify({'error': str(e)+f"image: {image_url}"}), 500
422
 
423
  @app.route('/api/admin/receipt/<string:transaction_id>/download', methods=['GET'])
424
  def download_receipt(transaction_id):