Sabbirr12 commited on
Commit
86121f6
·
verified ·
1 Parent(s): 18f31e3

Create quota_manager.py

Browse files
Files changed (1) hide show
  1. quota_manager.py +23 -0
quota_manager.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import datetime, json
2
+
3
+ QUOTA_FILE = "quota.json"
4
+
5
+ def check_quota():
6
+ today = datetime.date.today().isoformat()
7
+ try:
8
+ with open(QUOTA_FILE, "r") as f:
9
+ data = json.load(f)
10
+ except:
11
+ data = {}
12
+ return data.get(today, 0) < 7
13
+
14
+ def increment_quota():
15
+ today = datetime.date.today().isoformat()
16
+ try:
17
+ with open(QUOTA_FILE, "r") as f:
18
+ data = json.load(f)
19
+ except:
20
+ data = {}
21
+ data[today] = data.get(today, 0) + 1
22
+ with open(QUOTA_FILE, "w") as f:
23
+ json.dump(data, f)