401 or 403? in Python

The shop now hands out keys to its staff too, for other jobs later. Adding products stays the owner's job. So a refused request can mean two different things, and HTTP has a status for each.

401 Unauthorized means "I don't know who you are": the key is missing, or it isn't one of the shop's keys. 403 Forbidden means "I know who you are, and you may not do this": the key is real, but its role isn't allowed. After a 401 a client should send a valid key; after a 403, trying again won't help.

A dictionary can map each valid key to its role. in looks at the dictionary's keys, and square brackets give the value stored for a key. Here with made-up keys for a music app:

PLANS = {"free-key-1": "free", "pro-key-2": "pro"}
print("pro-key-2" in PLANS)   # True
print(None in PLANS)          # False
print(PLANS["pro-key-2"])     # pro

These are the requests your code must answer:

POST /products  {"name": "Vase", "price": 18}  (no key)                    →  401  {"error": "Missing or unknown key"}
POST /products  {"name": "Vase", "price": 18}  X-Admin-Key: let-me-in      →  401  {"error": "Missing or unknown key"}
POST /products  {"name": "Vase", "price": 18}  X-Admin-Key: staff-key-456  →  403  {"error": "Admins only"}
POST /products  {"name": "Vase", "price": 18}  X-Admin-Key: admin-key-123  →  201  {"id": 9, "name": "Vase", "price": 18}
GET /products/10  →  404  {"error": "Product not found"}

Your task: replace ADMIN_KEY with a dictionary API_KEYS that maps "admin-key-123" to "admin" and "staff-key-456" to "staff". In add_product, answer 401 when the key is not in API_KEYS, then 403 with {"error": "Admins only"} when its role isn't "admin".

Assignment
Follow the Coding Tutorial to tell an unknown caller (401) from a known caller who isn't allowed (403).

Hint
Keep the order: 401 first, then 403. API_KEYS[key] fails for a key that isn't in the dictionary, but after the 401 check every key that's left is in it.