Formatting

This commit is contained in:
Firepup Sixfifty 2024-06-05 13:01:23 -05:00
parent deb52bf348
commit 27fe210429
Signed by: Firepup650
GPG key ID: 7C92E2ABBBFAB9BA

19
main.py
View file

@ -29,13 +29,18 @@ try:
raise ImportError("User requested to skip cache")
print("Trying to load user mappings from cache...")
from cache import userMappings
print("Cache load OK. Reminder: If you need to regenerate the cache, call the script with `--no-cache`")
print(
"Cache load OK. Reminder: If you need to regenerate the cache, call the script with `--no-cache`"
)
except ImportError:
users_list = []
print("Cache load failed, falling back to full load from slack...")
cursor = "N/A"
pages = 0
while cursor: # If slack gives us a cursor, then we ain't done loading user data yet
while (
cursor
): # If slack gives us a cursor, then we ain't done loading user data yet
data = ""
if cursor != "N/A":
data = client.users_list(cursor=cursor, limit=1000)
@ -47,7 +52,11 @@ except ImportError:
print(f"Pages of users loaded: {pages}")
del pages
print("Building user mappings now, this shouldn't take long...")
for user in users_list: # Map user ID mentions to user name mentions, it's nicer when printing messages for thread selection.
for (
user
) in (
users_list
): # Map user ID mentions to user name mentions, it's nicer when printing messages for thread selection.
userMappings[f"<@{user['id']}>"] = (
f"<@{user['profile']['display_name_normalized']}>"
if user["profile"]["display_name_normalized"]
@ -58,7 +67,9 @@ except ImportError:
)
)
print("All mappings generated, writing cache file now...")
with open("cache.py", "w") as cacheFile: # It is many times faster to load from a local file instead of from slack
with open(
"cache.py", "w"
) as cacheFile: # It is many times faster to load from a local file instead of from slack
cacheFile.write(f"userMappings = {userMappings}")
print("Cache saved.")