简单对话机器人

对话机器人

import random
# 定语从句语法
grammar = '''
战斗 => 施法  ， 结果 。
施法 => 主语 动作 技能 
结果 => 主语 获得 效果
主语 => 张飞 | 关羽 | 赵云 | 典韦 | 许褚 | 刘备 | 黄忠 | 曹操 | 鲁班七号 | 貂蝉
动作 => 施放 | 使用 | 召唤 
技能 => 一骑当千 | 单刀赴会 | 青龙偃月 | 刀锋铁骑 | 黑暗潜能 | 画地为牢 | 守护机关 | 狂兽血性 | 龙鸣 | 惊雷之龙 | 破云之龙 | 天翔之龙
获得 => 损失 | 获得 
效果 => 数值 状态
数值 => 1 | 1000 |5000 | 100 
状态 => 法力 | 生命
'''

def getGrammarDict(gram , linesplit = '\n',gramsplit = '=>'):
    result={}

    for line in gram.split(linesplit):
        if not line.strip():
            continue
        expr,statement = line.split(gramsplit)
        result[expr.strip()]=[i.split()for i in statement.split("|")]
        #print(result)
    return result


def generate(gramdict , target , isEng = False ):
    if target not in gramdict:
        return target
    find = random.choice(gramdict[target])
    print(find)
    blank=''
    if isEng:
        blank = ' '
    #blank.join() 不断添加字符，字符中间添加指定符号，此处为添加空格。
    return blank.join(generate(gramdict,t,isEng)for t in find)

gramdict = getGrammarDict(grammar)
print(gramdict)

print(generate(gramdict,"战斗"))
print(generate(gramdict,"战斗",True))

spilt() 函数 按照指定方式换行（默认为\n）
strip() 函数除去首尾的字符