klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 05:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Write generated server id to env file when env file does not contain it

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2023-05-17 19:12:05 +05:30
parent dc0626856e
commit d02415edcc
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/khoj/utils/helpers.py
View File

@@ -138,6 +138,8 @@ def get_server_id():
"""Get, Generate Persistent, Random ID per server install. """Get, Generate Persistent, Random ID per server install.
Helps count distinct khoj servers deployed. Helps count distinct khoj servers deployed.
Maintains anonymity by using non-PII random id.""" Maintains anonymity by using non-PII random id."""
# Initialize server_id to None
server_id = None
# Expand path to the khoj env file. It contains persistent internal app data # Expand path to the khoj env file. It contains persistent internal app data
app_env_filename = path.expanduser(constants.app_env_filepath) app_env_filename = path.expanduser(constants.app_env_filepath)
@@ -153,15 +155,19 @@ def get_server_id():
if key.strip() == "server_id": if key.strip() == "server_id":
server_id = value.strip() server_id = value.strip()
break break
else:
# If server_id is not found, generate and write to env file
if server_id is None:
# If server_id is not found, generate a new one # If server_id is not found, generate a new one
server_id = str(uuid.uuid4()) server_id = str(uuid.uuid4())
with open(app_env_filename, "a") as f:
f.write("server_id=" + server_id + "\n")
else: else:
# Generate a new server id # If server_id is not found, generate a new one
server_id = str(uuid.uuid4()) server_id = str(uuid.uuid4())
# Write the server_id to the file # Write the server_id to the env file
with open(app_env_filename, "w") as f: with open(app_env_filename, "w") as f:
f.write("server_id=" + server_id + "\n") f.write("server_id=" + server_id + "\n")