Compare commits
No commits in common. "d76a965e7508af5caeb7a1ac660751436384bc73" and "979e15f525eaba9052f4b78c43330ef0e66919a4" have entirely different histories.
d76a965e75
...
979e15f525
7 changed files with 17 additions and 61 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -182,4 +182,3 @@ cython_debug/
|
|||
|
||||
# Custom stuff
|
||||
gacha_game.db
|
||||
config.ini
|
||||
|
|
|
@ -1,5 +1,4 @@
|
|||
import time
|
||||
import traceback
|
||||
import misskey
|
||||
from parsing import parse_notification
|
||||
from db_utils import get_or_create_user, add_pull, get_config, set_config
|
||||
|
@ -69,7 +68,7 @@ def stream_notifications():
|
|||
time.sleep(5)
|
||||
|
||||
except Exception as e:
|
||||
print(f"An exception has occured: {e}\n{traceback.format_exc()}")
|
||||
print(f"Error: {e}")
|
||||
time.sleep(5)
|
||||
|
||||
|
||||
|
|
|
@ -1,19 +1,8 @@
|
|||
'''Essentials for the bot to function'''
|
||||
import configparser
|
||||
config = configparser.ConfigParser()
|
||||
config.read('config.ini')
|
||||
# Essential for the bot to function
|
||||
INSTANCE = "" # Bots Misskey instance's URL **with schema**
|
||||
KEY = "" # API key for the bot
|
||||
|
||||
# Username for the bot
|
||||
USER = config['application']['BotUser']
|
||||
|
||||
# API key for the bot
|
||||
KEY = config['application']['ApiKey']
|
||||
# Bot's Misskey instance URL
|
||||
INSTANCE = config['application']['InstanceUrl']
|
||||
|
||||
# Extra stuff for control of the bot
|
||||
|
||||
# TODO: move this to db
|
||||
# Fedi handles in the traditional 'user@domain.tld' style, allows these users
|
||||
# to use extra admin exclusive commands with the bot'''
|
||||
ADMINS = config['application']['DefaultAdmins']
|
||||
# Extra stuff for control of the bot
|
||||
ADMINS = [] # Fedi handles in the traditional 'user@domain.tld' style,
|
||||
# allows these users to use extra admin exclusive commands
|
||||
# with the bot
|
||||
|
|
|
@ -17,10 +17,10 @@ def get_or_create_user(username):
|
|||
conn.row_factory = sqlite3.Row
|
||||
cur = conn.cursor()
|
||||
cur.execute('SELECT id FROM users WHERE username = ?', (username,))
|
||||
user = cur.fetchone()
|
||||
user = cur.fetchone()[0]
|
||||
if user:
|
||||
conn.close()
|
||||
return user[0]
|
||||
return user
|
||||
|
||||
# New user starts with has_rolled = False
|
||||
cur.execute(
|
||||
|
|
|
@ -18,14 +18,6 @@ def get_character():
|
|||
|
||||
return chosen['id'], chosen['name'], chosen['file_id'], chosen['rarity']
|
||||
|
||||
def is_float(val):
|
||||
try:
|
||||
float(val)
|
||||
return True
|
||||
except ValueError:
|
||||
return False
|
||||
|
||||
|
||||
# TODO: See issue #3, separate command parsing from game logic.
|
||||
def gacha_response(command,full_user, arguments,note_obj):
|
||||
'''Parses a given command with arguments, processes the game state and
|
||||
|
@ -42,7 +34,7 @@ def gacha_response(command,full_user, arguments,note_obj):
|
|||
|
||||
add_pull(user_id,character_id)
|
||||
stars = '⭐️' * rarity
|
||||
return([f"@{full_user} 🎲 Congrats! You rolled {stars} **{character_name}**\nShe's all yours now~ 💖✨",[file_id]])
|
||||
return([f"@{full_user} 🎲 Congrats! You rolled **{character_name}** she's {stars} stars! She's all yours now~ 💖✨",[file_id]])
|
||||
|
||||
if command == "create":
|
||||
# Example call from bot logic
|
||||
|
@ -50,15 +42,6 @@ def gacha_response(command,full_user, arguments,note_obj):
|
|||
if not image_url:
|
||||
return "You need an image to create a character, dumbass."
|
||||
|
||||
if len(arguments) != 3:
|
||||
return "Please specify the following attributes in order: name, rarity, drop weighting"
|
||||
|
||||
if not (arguments[1].isnumeric() and 1 <= int(arguments[1]) <= 5):
|
||||
return f"Invalid rarity: '{arguments[1]}' must be a number between 1 and 5"
|
||||
|
||||
if not (is_float(arguments[2]) and 0.0 < float(arguments[2]) <= 1.0):
|
||||
return f"Invalid drop weight: '{arguments[2]}' must be a decimal value between 0.0 and 1.0"
|
||||
|
||||
character_id, file_id = add_character(
|
||||
name=arguments[0],
|
||||
rarity=int(arguments[1]),
|
||||
|
|
|
@ -1,5 +1,4 @@
|
|||
import random, re
|
||||
import config
|
||||
from gacha_response import gacha_response
|
||||
|
||||
def parse_notification(notification,client):
|
||||
|
@ -32,14 +31,15 @@ def parse_notification(notification,client):
|
|||
|
||||
note = note_text.strip().lower() if note_text else ""
|
||||
|
||||
# Check for both short and fully-qualified name mentions
|
||||
username_variants = [
|
||||
config.USER,
|
||||
f'@{config.USER.split('@')[1]}'
|
||||
# Will want to move bot name to a config file or find a way to derermine
|
||||
# AP ID at runtime
|
||||
waifumelon_variants = [
|
||||
"@waifumelon", # simple mention
|
||||
"@waifumelon@mai.waifuism.life", # fully qualified
|
||||
]
|
||||
|
||||
# Make sure the notification text explicitly mentions the bot
|
||||
if not any(variant in note for variant in username_variants):
|
||||
if not any(variant in note for variant in waifumelon_variants):
|
||||
return
|
||||
|
||||
# Find command and arguments after the mention
|
||||
|
|
|
@ -1,14 +0,0 @@
|
|||
; Rename me to config.ini and put your values in here
|
||||
[application]
|
||||
; Full fedi handle of the bot user
|
||||
BotUser = @bot@example.tld
|
||||
|
||||
; API key for the bot
|
||||
; Generate one by going to Settings > API > Generate access token
|
||||
ApiKey = abcdefghijklmnopqrstuvwxyz012345
|
||||
|
||||
; Fully qualified URL of the instance hosting the bot
|
||||
InstanceUrl = http://example.tld
|
||||
|
||||
; Comma separated list of fedi handles for any administrator users
|
||||
DefaultAdmins = ['admin@example.tld']
|
Loading…
Add table
Reference in a new issue