Create usage.py

services/usage.py

@@ -0,0 +1,71 @@
+# services/usage.py
+import json
+import os
+from datetime import datetime, timezone
+from pathlib import Path
+from collections import defaultdict
+
+class UsageTracker:
+    def __init__(self, daily_limit: int, data_dir: Path = None):
+        self.daily_limit = daily_limit
+        self.data_dir = data_dir or Path("./data")
+        self.data_dir.mkdir(exist_ok=True)
+        self.usage_file = self.data_dir / "usage.json"
+        self.usage = self._load()
+    
+    def _load(self) -> dict:
+        try:
+            if self.usage_file.exists():
+                with open(self.usage_file, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+        except:
+            pass
+        return {}
+    
+    def _save(self):
+        try:
+            with open(self.usage_file, 'w', encoding='utf-8') as f:
+                json.dump(self.usage, f, ensure_ascii=False, indent=2)
+        except:
+            pass
+    
+    def _today(self) -> str:
+        return datetime.now(timezone.utc).strftime('%Y-%m-%d')
+    
+    def get_count(self, user_id: int) -> int:
+        uid = str(user_id)
+        today = self._today()
+        if uid not in self.usage or self.usage[uid].get('date') != today:
+            self.usage[uid] = {'date': today, 'count': 0}
+            self._save()
+            return 0
+        return self.usage[uid].get('count', 0)
+    
+    def check_limit(self, user_id: int) -> bool:
+        used = self.get_count(user_id)
+        return used < self.daily_limit
+    
+    def increment(self, user_id: int) -> bool:
+        if not self.check_limit(user_id):
+            return False
+        uid = str(user_id)
+        today = self._today()
+        if uid not in self.usage or self.usage[uid].get('date') != today:
+            self.usage[uid] = {'date': today, 'count': 1}
+        else:
+            self.usage[uid]['count'] = self.usage[uid].get('count', 0) + 1
+        self._save()
+        return True
+    
+    def get_stats_text(self, user_id: int) -> str:
+        used = self.get_count(user_id)
+        remaining = self.daily_limit - used
+        today = self._today()
+        percent = (used / self.daily_limit) * 100 if self.daily_limit > 0 else 0
+        bar = "█" * int(20 * percent / 100) + "░" * (20 - int(20 * percent / 100))
+        return (
+            f"📊 **Статистика**\n"
+            f"📅 `{today}`\n"
+            f"📈 [{bar}] {percent:.0f}%\n"
+            f"✅ `{used}` / ❌ `{remaining}` из `{self.daily_limit}`"
+        )