if文において 平成 (何か数字) と含んでいるときした時と条件付けしたい

Question

前提・実現したいこと

変数commandに”平成 何か数字”　と入力した時に（例えば 平成 1990）西暦に変換し、それ以外はおうむ返しするbotを作成しているのですが、

このままだと”平成”とだけ入力されるとsplit部でエラーが発生するので、
その条件付けをしたいのですが、どのように記述すればよいでしょうか？

該当のソースコードぶ

if "平成" in command: #平成だけ入力されたらエラー吐きますねこれ
        heisei, year_str = command.split()
        year = int(year_str)

###思いついた解決法
平成だけ入力されたら西暦から計算するか尋ねて。数字入力を促す。　でも二度手間ですね。。。

Answer 1

re.match(r"平成 (\d+)", command)
とすれば、数字を取り出すことができます。

次のコードと、その実行例を研究してみてください。
z.py

import re

while True:
    command = input()
    m = re.match(r"平成 (\d+)", command)
    if m:
        year = int(m.group(1))
        if year > 1988:
            print("平成", year - 1988)
        elif 0 < year and year <= 31:
            print("西暦", year + 1988)
        else:
            print(command)
    else:
        print(command)

実行例

Comments

[OkomekO]

回答ありがとうございました！ぜひ実行して動きを見てみたいと思います

Answer 2

正規表現でチェックするのがお手軽でしょう。

import re

if re.match(r'平成 \d{4}', command):
    ...

Comments

[OkomekO]

正規表現ですか。とても便利ですね！勉強してきます回答ありがとうございました！