parent
741ed68428
commit
3d878cc7dc
3 changed files with 37 additions and 3 deletions
@ -0,0 +1,35 @@ |
||||
import os, sys, django, csv |
||||
|
||||
from django.db import IntegrityError |
||||
|
||||
sys.path.append("../") |
||||
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "lms.settings") |
||||
django.setup() |
||||
|
||||
from library.models import Tags, ArticleSection, Article |
||||
|
||||
if __name__ == '__main__': |
||||
with open('./article/tags.csv') as article_csv: |
||||
article_reader = csv.DictReader(article_csv) |
||||
for row in article_reader: |
||||
Tags.objects.get_or_create(**row) |
||||
|
||||
with open('./article/section.csv') as article_csv: |
||||
article_reader = csv.DictReader(article_csv) |
||||
for row in article_reader: |
||||
ArticleSection.objects.get_or_create(**row) |
||||
|
||||
with open('./article/articles.csv') as article_csv: |
||||
article_reader = csv.DictReader(article_csv) |
||||
for row in article_reader: |
||||
row = dict(row) |
||||
tags = row.pop('tags', None).split("[")[1].split("]")[0].split(", ") |
||||
article, _is_create = Article.objects.get_or_create(**row) |
||||
|
||||
try: |
||||
for tag in tags: |
||||
if tags: |
||||
article.teachers.add(tags) |
||||
|
||||
except IntegrityError: |
||||
pass |
||||
Loading…
Reference in new issue