diff --git a/servermanager.py b/servermanager.py index b7fb84a..5e48ab5 100644 --- a/servermanager.py +++ b/servermanager.py @@ -221,6 +221,26 @@ def server_command_builder( command.append(" --vwrap") if not hwrap: command.append(" --nohwrap") + if not ai_slots == []: + for ai in ai_slots: + if ai[0] not in available_nations[era]: + return "ERROR_INVALID_NATION" + match ai[1]: + case 1: + command.append(" --easyai " + str(ai[0])) + case 2: + command.append(" --normai " + str(ai[0])) + case 3: + command.append(" --diffai " + str(ai[0])) + case 4: + command.append(" --mightyai " + str(ai[0])) + case 5: + command.append(" --masterai " + str(ai[0])) + case 6: + command.append(" --impai" + str(ai[0])) + case _: + return "EROR_INVALID_AI" + return "".join(command)