File tree Expand file tree Collapse file tree 1 file changed +5
-15
lines changed Expand file tree Collapse file tree 1 file changed +5
-15
lines changed Original file line number Diff line number Diff line change 1
- from flask import Flask , request , abort
1
+ from flask import Flask
2
2
from azure .identity import DefaultAzureCredential
3
3
from azure .keyvault .secrets import SecretClient
4
4
import os
5
5
6
6
app = Flask (__name__ )
7
7
8
- # Your Key Vault URL (replace with yours)
9
8
KEY_VAULT_URL = "https://yashkeyvaultverysafe.vault.azure.net/"
10
-
11
- # Initialize Key Vault client with managed identity
12
9
credential = DefaultAzureCredential ()
13
10
client = SecretClient (vault_url = KEY_VAULT_URL , credential = credential )
14
11
15
12
@app .route ("/" )
16
13
def home ():
17
14
try :
18
- # Get secret from Key Vault (executed at request-time, not startup)
19
- retrieved_secret = client .get_secret ("app-auth-secret" ).value
20
-
21
- # Get API key from incoming request
22
- api_key = request .headers .get ('x-api-key' )
23
-
24
- # If API key is missing or incorrect, deny access
25
- if api_key != retrieved_secret :
26
- abort (403 , "Invalid API key" )
15
+ # Try to access a secret from Key Vault
16
+ secret_value = client .get_secret ("app-auth-secret" ).value
27
17
28
- return "✅ Authorized! Access granted. "
18
+ return f "✅ Welcome! Secret loaded from Key Vault: { secret_value } "
29
19
30
20
except Exception as e :
31
- return f"❌ Error: { str (e )} " , 500
21
+ return f"❌ Access denied or error reading from Key Vault:<br><br> { str (e )} " , 500
32
22
33
23
@app .route ("/ping" )
34
24
def ping ():
You can’t perform that action at this time.
0 commit comments