在python中自定义一个终端命令
这里我们想要将一个csv文件中的数据导入到数据库中,就可以定义一个终端命令,直接一行命令就可以将我们文件中的数据导入到数据库中,特别的简单
首先,我们先创建一个py文件
management/commands/import_candidate.py
接下来,我们就可以在这个py文件中定义终端命令了
import csv
from django.core.management import BaseCommand
from interview.models import Candidate
class Command(BaseCommand):
help = '从一个CSV文件的内容中导入候选人列表,导入到数据库中'
def add_argument(self, parser):
parser.add_argument('--path', type=str)
def handle(self, *args, **kwargs):
path = kwargs['path']
with open(path, 'rt', encoding='gbk') as f:
reader = csv.reader(f, dialect='excel', delimiter=';')
for row in reader:
candidate = Candidate.object.create(
username=row[0],
city=row[1],
phone=row[2],
bachelor_school=row[3],
major=row[4],
degree=row[5],
test_score_of_general_ability=row[6],
paper_score=row[7]
)
print(candidate)
然后, 我们就可以在终端执行命令了
python manage.py import_candidate --path ~/recruitment/应聘信息.csv
终于,文件中的数据就导入到数据库中了
手机扫一扫
移动阅读更方便
你可能感兴趣的文章