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