Python program to shutdown and restart computer

import os;
check = input("Want to shutdown your computer ? (y/n): ");
if check == 'n':
    check = input("Want to restart your computer ? (y/n): ");
    if check == 'y':
        os.system("shutdown /r /t 1");
    else:
        exit();
else:
    os.system("shutdown /s /t 1");