问答机器人从0到1 03

问答机器人从0到1 03

import re,sys#导入模块
from time import strftime,gmtime
import difflib
f=open('main.txt','r',encoding='utf-8')#以读取的方式打开txt数据库
main=f.read()#读取txt文件到缓存并赋值给main
dim_nub=0#定义模糊搜索开关
def welcome():
    return_word('先生，输入$可以关闭我嚛,模糊搜索已开启，输入&关闭')
    time_return()#时间问候
    wait_enter()
    #while True:
def wait_enter():
    question=input('...')#等待用户输入问题
    if dim_nub==0:
        open_function(question)
    else:
        find_return(question)
def open_function(function):
    global dim_nub
    if function=='$':
        return_word('后会有期！')
        sys.exit()
    elif function=='&':
        return_word('模糊搜索已关闭')
        dim_nub=1
        wait_enter()
    else:
        if dim_nub==0:
            dim(function)
def dim(word):#定义模糊搜索关键字
    keywords=re.findall('\n(.*?)@',main,re.S)#问题列表
    answers=re.findall('@(.*?)\n',main,re.S)#回答列表
    value=keywords+answers#合并问题和回答也就是正向和逆向搜索
    return_words=difflib.get_close_matches(word,value,10, cutoff=0.6)#模糊匹配
    nub=int(len(return_words))#判断有多少给合适条件
    if nub==1:#判断是否有多个回答
        for return_word in return_words:
            find_return(return_word)#得到模糊问题
    elif nub==0:
        return_word('worry,数据库模糊匹配失败....')
    else:
        dim_more(return_words)#跳转到多个匹配结果处理


def dim_more(key):#模糊搜索多个匹配结果处理
    try:
        a=key[0]#获取每一个数据
        b=key[1]
        c=key[2]
        key_nub=input('你可能想问：\n1.{}\n2.{}\n3.{}\n'.format(a,b,c)+'..:输入序列号查看结果')
        try:#尝试把key_nub转换为整型，并执行匹配操作
            key_nub=int(key_nub)
            if key_nub==1:#根据输入序号传递问题
                find_return(a)
            elif key_nub==2:
                find_return(b)
            elif key_nub==3:
                find_return(c)
                
            else:#输入其他数值后跳转
                return_word('worry,请输入正确数值，正在跳转....')#输入错误跳转
        except:
            return_word('worry,输入非法字符,正在跳转....')
    except:
        d=key[0]
        find_return(d)
def time_return():
    data=strftime("%Y年%m月%d日 %H:%M")#获取当前时间
    return_word('当前日期为{}'.format(data))#打印当前时间
    hour=int(strftime("%H"))#获取当前小时
    if 4<hour<12:#不同的时间段不同回答
        return_word('上午好，先生')
    elif 11<hour<17:
        return_word('中午好，先生')
    elif 16<hour<24:
        return_word('晚上好')
    else:
        return_word('这么晚还不睡，和我的粑粑一起变秃头小宝贝叭~~~')
def find_return(question):
    try:
        #通过正则查找方便快捷
        try:#正向查找
            
            answer=re.findall('{}@(.*?)\n'.format(question),main,re.S)[0]#通过问题寻找答案
            
            return_word(answer)#打印回答
            wait_enter()#跳转到回答
        except:#反向查找
            answer=re.findall('\n(.*?)@{}\n'.format(question),main)[0]#通过答案寻找问题
            return_word(answer)
            wait_enter()
    except:
        turn='没有找寻到答案，请换个说法试试:-)正在跳转....'
        return_word(turn)
        wait_enter()
def return_word(word):
    print('>>{}'.format(word))
welcome()

新增功能：

模糊搜索以及其开关，单独定义了一个输出打印功能方便以后拓展

下篇将会增加根据用户输入特定关键词动态回答用户

对了，附图一张