Firepup650/fake-slack-app
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add caching support

Browse source
This commit is contained in:
Firepup Sixfifty 2024-06-05 12:39:00 -05:00
parent eeef9c607e
commit c8cb828b70
Signed by: Firepup650
GPG key ID: 7C92E2ABBBFAB9BA
1 changed files with 38 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
main.py
View file

@ -1,4 +1,4 @@
import os import os, sys
from slack_bolt import App from slack_bolt import App
from slack_bolt.adapter.socket_mode import SocketModeHandler from slack_bolt.adapter.socket_mode import SocketModeHandler
from dotenv import load_dotenv from dotenv import load_dotenv
@ -22,11 +22,18 @@ print("Establishing a connection to slack...")
app = App(token=os.environ.get("SLACK_BOT_TOKEN")) app = App(token=os.environ.get("SLACK_BOT_TOKEN"))
client = app.client client = app.client
print("Building a list of users, please stand by.") userMappings = {}
users_list = [] try:
cursor = "N/A" if "--no-cache" in sys.argv:
pages = 0 raise ImportError("User requested to skip cache")
while cursor: print("Trying to load user mappings from cache...")
from cache import userMappings
except ImportError:
users_list = []
print("Cache load failed, falling back to full load from slack...")
cursor = "N/A"
pages = 0
while cursor:
data = "" data = ""
if cursor != "N/A": if cursor != "N/A":
data = client.users_list(cursor=cursor, limit=1000) data = client.users_list(cursor=cursor, limit=1000)
@ -36,17 +43,21 @@ while cursor:
users_list.extend(data["members"]) users_list.extend(data["members"])
pages += 1 pages += 1
print(f"Pages of users loaded: {pages}") print(f"Pages of users loaded: {pages}")
print("All pages loaded, generating user mappings now.") del pages
del pages print("Building user mappings now, this shouldn't take long...")
user_mappings = {} userMappings = {}
for user in users_list: for user in users_list:
user_mappings[f"<@{user['id']}>"] = ( userMappings[f"<@{user['id']}>"] = (
f"<@{user['profile']['display_name']}>" f"<@{user['profile']['display_name']}>"
if user["profile"]["display_name"] if user["profile"]["display_name"]
else f"<@{user['id']}>" else f"<@{user['id']}>"
) )
print("All mappings generated, writing cache file now...")
with open("cache.py", "w") as cacheFile:
cacheFile.write(f"userMappings = {userMappings}")
print("Cache saved.")
print("User mappings loaded. User count:", len(user_mappings)) print("User mappings loaded. User count:", len(userMappings))
if __name__ == "__main__": if __name__ == "__main__":
print("^D at any time to terminate program") print("^D at any time to terminate program")
@ -74,8 +85,8 @@ if __name__ == "__main__":
print("Building messages, this might take a little bit...") print("Building messages, this might take a little bit...")
for i in range(len(messages)): for i in range(len(messages)):
label = f'{messages[i]["text"]} ({messages[i]["ts"]})' label = f'{messages[i]["text"]} ({messages[i]["ts"]})'
for user in user_mappings: for user in userMappings:
label = label.replace(user, user_mappings[user]) label = label.replace(user, userMappings[user])
texts[label] = i texts[label] = i
found = messages[ found = messages[
fp.menu( fp.menu(