98 lines
3.8 KiB
Python
98 lines
3.8 KiB
Python
import flask
|
|
from flask import request
|
|
|
|
import channel_rewards.channelRewards_loader as rewards_loader
|
|
from channel_rewards.channelRewards_base import AbstractChannelRewards
|
|
|
|
api = flask.Flask(__name__)
|
|
# enable/disable this to get web pages of crashes returned
|
|
api.config["DEBUG"] = True
|
|
|
|
loadedRewards = {}
|
|
|
|
def init():
|
|
# todo load entire reward library and cache it here
|
|
print("init stuff")
|
|
loadedRewards[AbstractChannelRewards.ChannelRewardsType.channelPoints] = rewards_loader.load_rewards(AbstractChannelRewards.ChannelRewardsType.channelPoints)
|
|
loadedRewards[AbstractChannelRewards.ChannelRewardsType.twitch_bits] = rewards_loader.load_rewards(AbstractChannelRewards.ChannelRewardsType.twitch_bits)
|
|
loadedRewards[AbstractChannelRewards.ChannelRewardsType.twitch_subs] = rewards_loader.load_rewards(AbstractChannelRewards.ChannelRewardsType.twitch_subs)
|
|
|
|
|
|
def is_reward(reward_name, reward_type) -> bool:
|
|
#global loadedRewards
|
|
tempType = reward_type.replace('ChannelRewardsType.', '')
|
|
realTempType = AbstractChannelRewards.ChannelRewardsType.__dict__[tempType]
|
|
#print(loadedRewards[realTempType])
|
|
|
|
for reward in loadedRewards[realTempType]:
|
|
print("found: ", reward, "type: ", type(reward))
|
|
if reward_name == reward:
|
|
print("Equal")
|
|
return True
|
|
|
|
|
|
if reward_name == "!echo":
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
def handle_reward(source, username, reward_name, reward_type, rewardPrompt, userInput, bonusData):
|
|
#reward:AbstractChannelRewards = loadedRewards[reward_name]
|
|
try:
|
|
tempType = reward_type.replace('ChannelRewardsType.', '')
|
|
realTempType = AbstractChannelRewards.ChannelRewardsType.__dict__[tempType]
|
|
reward:AbstractChannelRewards = loadedRewards[realTempType][reward_name]
|
|
if reward is not None:
|
|
reward_response = reward.do_ChannelReward(source, username, reward_name, rewardPrompt, userInput, bonusData)
|
|
return flask.make_response("{\"message\":\"%s\"}" % reward_response, 200, {"Content-Type": "application/json"})
|
|
except:
|
|
return "None"
|
|
#print("Doing a reward")
|
|
|
|
|
|
@api.route('/api/v1/reward', methods=['GET'])
|
|
def reward_check():
|
|
if 'reward_name' in request.args and 'reward_type' in request.args:
|
|
print("reward_name:", request.args['reward_name'],"reward_type:", request.args['reward_type'])
|
|
if is_reward(request.args['reward_name'], request.args['reward_type']):
|
|
print("about to send")
|
|
return flask.make_response('', 200)
|
|
else:
|
|
return flask.make_response('', 404)
|
|
|
|
|
|
@api.route('/api/v1/exec_reward', methods=['GET'])
|
|
def exec_reward():
|
|
if 'reward_name' not in request.args:
|
|
return flask.make_response('{\"text\":"Argument \'reward_name\' not in request"}', 400)
|
|
if 'reward_type' not in request.args:
|
|
return flask.make_response('{\"text\":"Argument \'reward_name\' not in request"}', 400)
|
|
if 'reward_prompt' not in request.args:
|
|
return flask.make_response('{\"text\":"Argument \'reward_prompt\' not in request"}', 400)
|
|
if 'user_input' not in request.args:
|
|
return flask.make_response('{\"text\":"Argument \'user_input\' not in request"}', 400)
|
|
|
|
|
|
if 'reward_source' not in request.args:
|
|
return flask.make_response('{\"text\":"Argument \'reward_source\' not in request"}', 400)
|
|
|
|
if 'user_name' not in request.args:
|
|
username = "User"
|
|
else:
|
|
username = request.args['user_name']
|
|
|
|
return handle_reward(
|
|
request.args['reward_source'],
|
|
username,
|
|
request.args['reward_name'],
|
|
request.args['reward_type'],
|
|
request.args['reward_prompt'],
|
|
request.args['user_input'],
|
|
request.args['bonus_data'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
init()
|
|
api.run(host='0.0.0.0', port=6969)
|