Как показать сообщение об ошибке csv файл не существует
если я добавляю имя файла csv неправильно, я хочу показать сообщение об ошибке введенный файл csv не существует, пожалуйста, напишите правильное имя файла csv. как это сделать?
class Command(BaseCommand):
help = 'Load a company csv file into the database'
def add_arguments(self, parser):
parser.add_argument('filename.csv', type=str)
def handle(self, *args, **options ):
if Company.objects.exists():
print("Data already loaded!!")
with open('companies.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
create= Company.objects.get_or_create(
name=row[0],
hr_name=row[1],
hr_email=row[2],
hr_verified=row[3],
user_id=row[4],
primary_phone=row[5],
comments=row[6],
)
Это скорее вопрос по Python, чем по Django. Вы можете сделать это с помощью try/catch, выбрасывающего FileNotFoundError исключение, когда указан неправильный файл.
def handle(self, *args, **options ):
if Company.objects.exists():
print("Data already loaded!!")
try:
with open('companies.csv', 'r') as file:
reader = csv.reader(file)
....
except FileNotFoundError:
# Yoour error message goes here
- Вы можете использовать попытку except:
try:
with open(filename, "r") as file:
# ...
except FileNotFoundError as e:
print("Please write the correct cvs file name")
- Вы можете проверить, существует ли файл:
import os
if not os.path.exists(filename):
print("Please write the correct csv file name")