secstate/kowalski
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow requesting dump of DB

Browse Source
This commit is contained in:
Colin Powell
2025-02-07 21:15:32 -05:00
parent 05d8578a4e
commit c7d47dbecc
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
envrc.example
View File

@ -1,4 +1,5 @@
export SLACK_BOT_TOKEN="xoxb-*" export SLACK_BOT_TOKEN="xoxb-*"
export SLACK_APP_TOKEN="xapp-*" export SLACK_APP_TOKEN="xapp-*"
export DB_NAME="" export DB_NAME=""
export DB_DUMP_URL=""
export BACKDOOR_USERS="username,username2" export BACKDOOR_USERS="username,username2"

14
kowalski.py
View File

@ -40,6 +40,7 @@ CONGRATS = [
INDICATOR = "$$" INDICATOR = "$$"
DB_NAME = os.environ.get("DB_NAME", "kowalski.db") DB_NAME = os.environ.get("DB_NAME", "kowalski.db")
BACKDOOR_USERS = os.environ.get("BACKDOOR_USERS", "").split(",") BACKDOOR_USERS = os.environ.get("BACKDOOR_USERS", "").split(",")
DB_DUMP_URL = os.environ.get("DB_DUMP_URL", "")
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
# Initialize SQLite database # Initialize SQLite database
@ -164,6 +165,19 @@ def handle_message_events(event, say):
random_congrats = random.choice(CONGRATS) random_congrats = random.choice(CONGRATS)
say(f"{display_name} has ${user_count}, {random_congrats}") say(f"{display_name} has ${user_count}, {random_congrats}")
@app.event("app_mention")
def handle_message_events(event, say):
sender_id = event.get("user")
text = event.get("text")
username, display = get_username(sender_id)
if "dump database" in text and DB_DUMP_URL:
say(f"Sure thing @{display}! You can download my DB from {DB_DUMP_URL}")
return
say("Sorry, I didn't understand that!")
# Start the bot # Start the bot
if __name__ == "__main__": if __name__ == "__main__":
handler = SocketModeHandler(app, os.environ.get("SLACK_APP_TOKEN")) handler = SocketModeHandler(app, os.environ.get("SLACK_APP_TOKEN"))