fake-slack-app/main.py

98 lines
4 KiB
Python
Raw Normal View History

2024-06-03 23:24:56 +00:00
import os
from slack_bolt import App
from slack_bolt.adapter.socket_mode import SocketModeHandler
from dotenv import load_dotenv
import firepup650 as fp
input = fp.replitInput
fp.replitCursor = fp.bcolors.REPLIT + ">>>" + fp.bcolors.RESET # Totally not hijacking one of my functions to use ;P
2024-06-03 23:24:56 +00:00
load_dotenv()
for requiredVar in ["SLACK_BOT_TOKEN", "SLACK_APP_TOKEN"]:
if not os.environ.get(requiredVar):
raise ValueError(f'Missing required environment variable "{requiredVar}". Please create a .env file in the same directory as this script and define it.')
2024-06-03 23:24:56 +00:00
print("Establishing a connection to slack...")
2024-06-03 23:24:56 +00:00
app = App(token=os.environ.get("SLACK_BOT_TOKEN"))
client = app.client
print("Building a list of users, please stand by.")
users_list = []
cursor = "N/A"
pages = 0
while cursor:
data = ""
if cursor != "N/A":
data = client.users_list(cursor=cursor, limit=1000)
else:
data = client.users_list(limit=1000)
cursor = data["response_metadata"]["next_cursor"]
users_list.extend(data["members"])
pages += 1
print(f"Pages of users loaded: {pages}")
print("All pages loaded, generating user mappings now.")
del pages
user_mappings = {}
for user in users_list:
2024-06-04 22:18:40 +00:00
user_mappings[f"<@{user['id']}>"] = f"<@{user['profile']['display_name']}>" if user["profile"]["display_name"] else f"<@{user['id']}>"
print("User mappings loaded. User count:", len(user_mappings))
2024-06-03 23:24:56 +00:00
if __name__ == "__main__":
print("^D at any time to terminate program")
2024-06-03 23:24:56 +00:00
while 1:
chan = input("Channel ID")
2024-06-03 23:24:56 +00:00
try:
print("^C to change channel")
while 1:
thread = input("Reply to a thread? (y|N)").lower().startswith("y")
2024-06-03 23:24:56 +00:00
ts = None
if thread:
hasID = input("Do you have the TS ID? (y|N))").lower().startswith("y")
2024-06-03 23:24:56 +00:00
if not hasID:
try:
print("Getting the last 50 messages for threading options...")
2024-06-03 23:24:56 +00:00
res = client.conversations_history(
channel=chan, inclusive=True, limit=50
2024-06-03 23:24:56 +00:00
)
messages = res["messages"]
texts = {}
print("Building messages, this might take a little bit...")
for i in range(len(messages)):
label = f'{messages[i]["text"]} ({messages[i]["ts"]})'
for user in user_mappings:
label = label.replace(user, user_mappings[user])
texts[label] = i
found = messages[fp.menu(texts, "Please select the message to reply to as a thread")]
2024-06-03 23:24:56 +00:00
ts = found["ts"]
except Exception as E:
print(f"Exception: {E}")
break
2024-06-03 23:24:56 +00:00
else:
ts = input("TS ID")
print("^C to change/exit thread (^C twice if you want to change channel)")
try:
while 1:
msg = input("[THREAD] Message (Raw text, not blocks)")
2024-06-03 23:24:56 +00:00
try:
print(
client.chat_postMessage(
channel=chan, text=msg, thread_ts=ts
)
)
except Exception as E:
print(f"Exception: {E}")
except KeyboardInterrupt:
print()
if ts:
continue
msg = input("Message (Raw text, not blocks)")
2024-06-03 23:24:56 +00:00
try:
print(client.chat_postMessage(channel=chan, text=msg))
except Exception as E:
print(f"Exception: {E}")
except KeyboardInterrupt:
print()