【Python】コンソール上でしりとり♪

こんにちは！
はるさらと申します。

今回はPythonの練習がてら
「コンソール上で行うしりとり」を
作成してみました。

使用する方がいるかは分かりませんが・・・
どなたかの参考になれば幸いです！

開発環境

PyCharm を使用しております。
コピペしていただければ
同じ挙動が出来るように作成しております。

他のPythonの記事はこちらから参照できます♪

今回の目標

しりとりです。
単語を繋げて行って、
最後に「ん」がついたら
プログラム終了となります。

入力された単語はDBへ登録していき
最後に履歴として表示されるように
実装していきます。

コーディング

import sqlite3
count = 0
start ='しりとり'
print("「しりとり」の「り」からスタート")

# テーブルの作成
con = sqlite3.connect(":memory:")
c = con.cursor()

create_table = '''create table shiritori (id int, word varchar(64))'''
c.execute(create_table)

# データの追加
def insert(no,word):
    sql = 'insert into shiritori (id, word) values (?,?)'
    user = (no, word)
    c.execute(sql, user)

while True:
    word = input("ひらがなの単語を入力して下さい: ")
    if start[-1:] != word[:1]:
        print('前の単語の最後の文字から始めてください。')
    elif word[-1:] == 'ん':
        print('最後に「ん」が付きました。ゲーム終了')
        insert(count + 1,word)
        for a in c.execute("select * from shiritori"):
            print(a)
        break
    else:
        count += 1
        start = word
        insert(count,word)
        print('「' + word + '」→')

con.close()