tkfinder.py

# -*- coding: utf-8 -*-
import os
import json

dirname = os.path.dirname(__file__)

def get_character(chara_name: str) -> dict:
    '''Gets character details from character_misc.json, if character exists
    returns character details as dict if exists, else None
    '''
   
    filepath = dirname + '/newjsons/character_misc.json'
    with open(filepath) as chara_misc_file:
        contents = chara_misc_file.read()
    
    chara_misc_json = json.loads(contents) 
    chara_details = list(filter(lambda x: (x['name'] == chara_name), chara_misc_json))
    print(chara_name)
    if chara_details:
        return chara_details[0]
    else:
        return None

def get_move(character: dict, move_command: str, case_important: bool) -> dict:
    '''Gets move from local_json, if exists
    returns move if exists, else None
    '''

    move_file_name = '/newjsons/' + character.get('local_json')
    filepath = dirname + move_file_name
    with open(filepath) as move_file:
        move_file_contents = move_file.read()
    move_json = json.loads(move_file_contents)
    
    if case_important:
        move = list(filter(lambda x: (x['Command'] == move_command), move_json))
    else: 
        move = list(filter(lambda x: (move_simplifier(x['Command'].lower())
        == move_simplifier(move_command.lower())), move_json))
        if not move:
            move = list(filter(lambda x: move_simplifier(move_command.lower()) 
            in move_simplifier(x['Command'].lower()), move_json))
    
    print(move, "bab")
    if move:
        return move[0]
    else:
        return None

def get_by_move_type(character: dict, move_type: str) -> list:
    '''Gets a list of moves that match move_type from local_json
    returns a list of move Commands if finds match(es), else empty list'''

    move_file_name = '/newjsons/' + character.get('local_json')
    filepath = dirname + move_file_name
    with open(filepath) as move_file:
        move_file_contents = move_file.read()
    move_json = json.loads(move_file_contents)

    moves = list(filter(lambda x: (move_type.lower() in x['Notes'].lower()), move_json))
    
    if moves:
        move_list = []
        for move in moves:
            move_list.append(move['Command'])
        return list(set(move_list))
    else:
        return []

def move_simplifier(move_input):
    '''Removes bells and whistles from the move_input'''

    move_replacements = {
        'fff': 'f,f,f',
        'ff': 'f,f',
        'bf': 'b,f',
        'fb': 'f,b',
        'ddf': 'd,df',
        'cd': 'f,n,d,df',
        'wr': 'f,f,f',
        'ewgf': 'f,n,d,df+2'
    }

    # Don't apply the above replacements for any of the moves with the following notation
    replacements_blacklist = ["cds"]

    for move in move_replacements:
        if not any([mv in move_input for mv in replacements_blacklist]) and move in move_input:
            move_input = move_input.replace(move, move_replacements[move])

    move_input = move_input.replace(' ', '')
    move_input = move_input.replace('/', '')
    move_input = move_input.replace('+', '')

    return move_input