Условные операторы с текстовым файлом

у меня есть задача, что мне нужно включить выбор файла в консоли по цифрам через операторы if,elif,else, Всего 2 файла .тхт у обоих одинаковый тип данных идет в строчку

1 ссылками, а 2 никнеймами, так вот как мне сделать чтобы он выбирался и я смог его реализовать в методе или просто принтануть, поправьте меня, пожалуйста

import sys

mode = int(input("Введите цифру какой у вас режим файла?\n@Username - 1\nСсылки - 2\n"))

if mode == 1:
    chats_username = open("chats_username.txt", mode="r", encoding="UTF-8", errors="ignore")
    file = chats_username.read()
    files = file.splitlines(keepends=False)
    chats_username.close()
elif mode == 2:
    link = open("chats_link.txt", mode="r", encoding="UTF-8", errors="ignore")
    url = link.readlines()
    for i in url:
        links = i.split("/")[3].strip("\n")
    link.close()
else:
    sys.exit("Извините вы не правильно ввели цифру при выборе мода")
print(mode)