From b95e688825bc0c553765f01e4f593a92f6a90ac7 Mon Sep 17 00:00:00 2001
From: norohind <60548839+norohind@users.noreply.github.com>
Date: Fri, 22 Aug 2025 21:25:14 +0000
Subject: [PATCH] Introduce mechanism to skip token use counter update

---
 capi/model.py | 6 ++++--
 config.py     | 2 ++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/capi/model.py b/capi/model.py
index ba17b9a..2db335a 100644
--- a/capi/model.py
+++ b/capi/model.py
@@ -1,6 +1,7 @@
 import sqlite3
 from typing import Union
 
+import config
 from . import sqlite_requests
 from EDMCLogging import get_main_logger
 
@@ -129,8 +130,9 @@ class Model:
     def get_token_for_user(self, state: str) -> dict:
         sql_params = {'state': state}
         token = self.db.execute(sqlite_requests.get_token_for_user, sql_params).fetchone()
-        with self.db:
-            self.db.execute(sqlite_requests.increment_usages, sql_params)
+        if not config.do_skip_token_usage_increment:
+            with self.db:
+                self.db.execute(sqlite_requests.increment_usages, sql_params)
 
         return token
 
diff --git a/config.py b/config.py
index 9480f59..2f82187 100644
--- a/config.py
+++ b/config.py
@@ -6,6 +6,8 @@ from os import getenv
 CLIENT_ID = getenv('client_id')
 assert CLIENT_ID, "No client_id in env"
 
+do_skip_token_usage_increment = os.getenv('SKIP_INCREMENT', 'false').lower() in ('true', 't', '1')
+
 log_level = os.getenv('LOG_LEVEL', 'DEBUG').upper()
 
 access_key = os.getenv('access_key')