add simple character stats generator

This commit is contained in:
Greg Gauthier 2024-06-22 14:22:39 +01:00
parent 366a340b1d
commit 796aac4d80
2 changed files with 117 additions and 11 deletions

107
app.py
View File

@ -1,7 +1,7 @@
from flask import Flask
from flask_restx import Api, Resource, fields
from flask_restx import Api, Resource
from marshmallow import Schema, fields, validate
from models import dice_model, ability_model, hp_model
from models import dice_model, ability_model, hp_model, character_model
import random
app = Flask(__name__)
@ -12,10 +12,12 @@ api = Api(app, version='1.0', title='Gamma World Dice', description='Rolled Dice
dice = api.namespace('dice', description='Dice operations')
ability = api.namespace('ability', description='Ability operations')
hp = api.namespace('hp', description='HP operations')
character = api.namespace('character', description='Character operations')
dice_model = dice.model('Dice', dice_model)
ability_model = ability.model('Ability', ability_model)
hp_model = hp.model('HP', hp_model)
character_model = character.model('Character', character_model)
class DiceSchema(Schema):
@ -27,6 +29,22 @@ class DiceSchema(Schema):
dice_schema = DiceSchema()
class CharacterSchema(Schema):
chartype = fields.String(
required=True,
validate=validate.OneOf(["human", "mutant", "android", "robot"]),
description='The characters type of being'
)
emphasis = fields.String(
required=True,
validate=validate.OneOf(["physical", "mental", "random"]),
description='Valid inputs: physical, mental, random'
)
character_schema = CharacterSchema()
@api.route('/roll/dice', methods=['POST'])
class RollDice(Resource):
@dice.expect(dice_model)
@ -72,16 +90,48 @@ class RollHP(Resource):
return roll_dices(conscore, geometry, False), 200
@api.route('/roll/tohit', methods=['GET'])
class RollToHit(Resource):
def get(self):
return roll_dices(1, 20, False), 200
@api.route('/roll/chance', methods=['GET'])
class RollChance(Resource):
def get(self):
return roll_dices(1, 100, False), 200
@api.route('/roll/tohit', methods=['GET'])
class RollToHit(Resource):
@api.route('/coinflip', methods=['GET'])
class RollCoinflip(Resource):
def get(self):
return roll_dices(1, 20, False), 200
return random.choice(['Heads', 'Tails']), 200
@api.route('/character/generate')
class GenerateCharacter(Resource):
@character.expect(character_model)
def post(self):
data = api.payload
errors = character_schema.validate(data)
if errors:
return errors, 400
chartype = data.get('chartype')
char_emphasis = data.get('emphasis')
character_sheet = {}
if chartype == 'human':
geometry = 8
else:
geometry = 6
scores = roll_ability_scores(chartype)
assigned_abilities = assign_ability_scores(scores, char_emphasis)
character_sheet['abilities'] = assigned_abilities
character_sheet['hp'] = roll_hp(chartype, assigned_abilities['constitution'])
character_sheet['gold'] = roll_dices(4, 4, False).get('result')
character_sheet['domar'] = roll_dices(2, 4, False).get('result')
return character_sheet, 200
def roll_dices(dice_count, dice_sides, discard_lowest):
@ -115,5 +165,52 @@ def roll_dices(dice_count, dice_sides, discard_lowest):
return result
def roll_ability_scores(chartype):
if chartype == 'human':
geometry = 8
else:
geometry = 6
scores = []
for _ in range(6):
scores.append(roll_dices(4, geometry, True)['result'])
return scores
def assign_ability_scores(scores, char_emphasis):
ability_scores = {}
if char_emphasis == 'physical':
ability_list = ['p-strength', 'constitution', 'dexterity',
'charisma', 'intelligence', 'm-strength']
for ab in ability_list:
max_score = max(scores)
ability_scores[ab] = max_score
scores.remove(max_score)
elif char_emphasis == 'mental':
ability_list = ['m-strength', 'intelligence', 'charisma',
'constitution', 'dexterity', 'p-strength']
for ab in ability_list:
max_score = max(scores)
ability_scores[ab] = max_score
scores.remove(max_score)
else:
ability_list = ['p-strength', 'constitution', 'dexterity',
'charisma', 'intelligence', 'm-strength']
random.shuffle(scores)
for ab, score in zip(ability_list, scores):
ability_scores[ab] = score
return ability_scores
def roll_hp(chartype, conscore):
if chartype == 'human':
geometry = 8
else:
geometry = 6
return roll_dices(conscore, geometry, False).get('result')
if __name__ == '__main__':
app.run()

View File

@ -1,8 +1,9 @@
from flask_restx import fields
# Common fields
chartype_field = fields.String(required=False, default="human", description='Character type. Allowed values: "human", '
'"mutant", "android", "robot"')
chartype_field = fields.String(
required=False, default="human",
description='Character type. Allowed values: "human", "mutant", "android", "robot"')
# Dice model
dice_model = {
@ -21,3 +22,11 @@ hp_model = {
'chartype': chartype_field,
'conscore': fields.Integer(required=True, description='Conscore')
}
character_model = {
'chartype': chartype_field,
'emphasis': fields.String(
required=False,
default="random",
description='The attribute emphasis of your character. Choices: "physical", "mental", "random"')
}