hassoudi commited on
Commit
e1ec338
·
verified ·
1 Parent(s): b1c2af3

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -19,7 +19,9 @@ limiter = Limiter(key_func=get_remote_address)
19
  app.state.limiter = limiter
20
  app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
21
 
22
- API_KEY_FILE = "api_keys.json"
 
 
23
  token_header = APIKeyHeader(name="X-API-KEY")
24
 
25
  if os.path.exists(API_KEY_FILE):
@@ -27,6 +29,10 @@ if os.path.exists(API_KEY_FILE):
27
  API_KEYS_STORE = json.load(f)
28
  else:
29
  API_KEYS_STORE = {"users": {}}
 
 
 
 
30
 
31
  API_KEYS = API_KEYS_STORE.get("users", {})
32
  ADMIN_KEY = os.getenv("ADMIN_KEY")
 
19
  app.state.limiter = limiter
20
  app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
21
 
22
+ #API_KEY_FILE = "api_keys.json"
23
+ API_KEY_FILE = "/tmp/api_keys.json"
24
+
25
  token_header = APIKeyHeader(name="X-API-KEY")
26
 
27
  if os.path.exists(API_KEY_FILE):
 
29
  API_KEYS_STORE = json.load(f)
30
  else:
31
  API_KEYS_STORE = {"users": {}}
32
+ with open(API_KEY_FILE, "w") as f:
33
+ json.dump(API_KEYS_STORE, f)
34
+
35
+
36
 
37
  API_KEYS = API_KEYS_STORE.get("users", {})
38
  ADMIN_KEY = os.getenv("ADMIN_KEY")